walkero
/
nghttp2
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

tiny-nghttpd: Add token lookup for slight optimization

This commit is contained in:
Tatsuhiro Tsujikawa 2015-02-20 21:50:02 +09:00
parent b39aa43537
commit fe84ec5e8b
1 changed files with 60 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
examples/tiny-nghttpd.c
View File

@ -179,13 +179,57 @@ static char *io_buf_add_str(io_buf *buf, const void *src, size_t len) {
return (char *)start; return (char *)start;
} }
static int memseq(const uint8_t *a, size_t alen, const char *b) { static int memeq(const void *a, const void *b, size_t n) {
const uint8_t *last = a + alen; return memcmp(a, b, n) == 0;
}
for (; a != last && *b && *a == *b; ++a, ++b) #define streq(A, B, N) ((sizeof((A)) - 1) == (N) && memeq((A), (B), (N)))
;
return a == last && *b == 0; typedef enum {
NGHTTP2_TOKEN__AUTHORITY,
NGHTTP2_TOKEN__METHOD,
NGHTTP2_TOKEN__PATH,
NGHTTP2_TOKEN__SCHEME,
NGHTTP2_TOKEN_HOST,
} nghttp2_token;
/* Inspired by h2o header lookup. https://github.com/h2o/h2o */
static int lookup_token(const uint8_t *name, size_t namelen) {
switch (namelen) {
case 5:
switch (name[namelen - 1]) {
case 'h':
if (streq(":pat", name, 4)) {
return NGHTTP2_TOKEN__PATH;
}
break;
}
break;
case 7:
switch (name[namelen - 1]) {
case 'd':
if (streq(":metho", name, 6)) {
return NGHTTP2_TOKEN__METHOD;
}
break;
case 'e':
if (streq(":schem", name, 6)) {
return NGHTTP2_TOKEN__SCHEME;
}
break;
}
break;
case 10:
switch (name[namelen - 1]) {
case 'y':
if (streq(":authorit", name, 9)) {
return NGHTTP2_TOKEN__AUTHORITY;
}
break;
}
break;
}
return -1;
} }
static char *cpydig(char *buf, int n, size_t len) { static char *cpydig(char *buf, int n, size_t len) {
@ -939,48 +983,42 @@ static int on_header_callback(nghttp2_session *session,
return 0; return 0;
} }
if (memseq(name, namelen, ":method")) { switch (lookup_token(name, namelen)) {
case NGHTTP2_TOKEN__METHOD:
strm->method = io_buf_add_str(&strm->scrbuf, value, valuelen); strm->method = io_buf_add_str(&strm->scrbuf, value, valuelen);
if (!strm->method) { if (!strm->method) {
return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
} }
strm->methodlen = valuelen; strm->methodlen = valuelen;
return 0; break;
} case NGHTTP2_TOKEN__SCHEME:
if (memseq(name, namelen, ":scheme")) {
strm->scheme = io_buf_add_str(&strm->scrbuf, value, valuelen); strm->scheme = io_buf_add_str(&strm->scrbuf, value, valuelen);
if (!strm->scheme) { if (!strm->scheme) {
return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
} }
strm->schemelen = valuelen; strm->schemelen = valuelen;
return 0; break;
} case NGHTTP2_TOKEN__AUTHORITY:
if (memseq(name, namelen, ":authority")) {
strm->authority = io_buf_add_str(&strm->scrbuf, value, valuelen); strm->authority = io_buf_add_str(&strm->scrbuf, value, valuelen);
if (!strm->authority) { if (!strm->authority) {
return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
} }
strm->authoritylen = valuelen; strm->authoritylen = valuelen;
return 0; break;
} case NGHTTP2_TOKEN__PATH:
if (memseq(name, namelen, ":path")) {
strm->path = io_buf_add_str(&strm->scrbuf, value, valuelen); strm->path = io_buf_add_str(&strm->scrbuf, value, valuelen);
if (!strm->path) { if (!strm->path) {
return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
} }
strm->pathlen = valuelen; strm->pathlen = valuelen;
return 0; break;
} case NGHTTP2_TOKEN_HOST:
if (memseq(name, namelen, "host") && !strm->host) {
strm->host = io_buf_add_str(&strm->scrbuf, value, valuelen); strm->host = io_buf_add_str(&strm->scrbuf, value, valuelen);
if (!strm->host) { if (!strm->host) {
return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
} }
strm->hostlen = valuelen; strm->hostlen = valuelen;
break;
} }
return 0; return 0;