3party/mongoose
0
0
Fork 0
mirror of https://github.com/cesanta/mongoose.git synced 2025-01-15 10:18:11 +08:00
Code Issues Projects Releases Wiki Activity

Added -w option

Browse Source
This commit is contained in:
valenok 2012-01-23 22:03:36 +00:00
parent aa90972a31
commit 01c5ee0973
2 changed files with 60 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
mongoose.c
View File

@ -405,7 +405,7 @@ enum {
GLOBAL_PASSWORDS_FILE, INDEX_FILES, GLOBAL_PASSWORDS_FILE, INDEX_FILES,
ENABLE_KEEP_ALIVE, ACCESS_CONTROL_LIST, MAX_REQUEST_SIZE, ENABLE_KEEP_ALIVE, ACCESS_CONTROL_LIST, MAX_REQUEST_SIZE,
EXTRA_MIME_TYPES, LISTENING_PORTS, EXTRA_MIME_TYPES, LISTENING_PORTS,
DOCUMENT_ROOT, SSL_CERTIFICATE, NUM_THREADS, RUN_AS_USER, DOCUMENT_ROOT, SSL_CERTIFICATE, NUM_THREADS, RUN_AS_USER, REWRITE,
NUM_OPTIONS NUM_OPTIONS
}; };
@ -432,6 +432,7 @@ static const char *config_options[] = {
"s", "ssl_certificate", NULL, "s", "ssl_certificate", NULL,
"t", "num_threads", "10", "t", "num_threads", "10",
"u", "run_as_user", NULL, "u", "run_as_user", NULL,
"w", "rewrite", NULL,
NULL NULL
}; };
#define ENTRIES_PER_CONFIG_OPTION 3 #define ENTRIES_PER_CONFIG_OPTION 3
@ -1544,15 +1545,59 @@ static int get_document_root(const struct mg_connection *conn,
return len_of_matched_uri; return len_of_matched_uri;
} }
static int match_prefix(const char *pattern, int pattern_len, const char *str) {
const char *or_str;
int i, j, len, res;
if ((or_str = memchr(pattern, '|', pattern_len)) != NULL) {
res = match_prefix(or_str + 1, (pattern + pattern_len) - (or_str + 1), str);
return res > 0 ? res : match_prefix(pattern, or_str - pattern, str);
}
i = j = res = 0;
for (; i < pattern_len; i++, j++) {
if (pattern[i] == '?' && str[j] != '\0') {
continue;
} else if (pattern[i] == '*') {
i++;
if (pattern[i] == '*') {
i++;
len = strlen(str + j);
} else {
len = strcspn(str + j, "/");
}
if (i == pattern_len) {
return j + len;
}
do {
res = match_prefix(pattern + i, pattern_len - i, str + j + len);
} while (res == 0 && len-- > 0);
return res == 0 ? 0 : j + res + len;
} else if (pattern[i] != str[j]) {
return 0;
}
}
return j;
}
static void convert_uri_to_file_name(struct mg_connection *conn, static void convert_uri_to_file_name(struct mg_connection *conn,
const char *uri, char *buf, const char *uri, char *buf,
size_t buf_len) { size_t buf_len) {
struct vec vec; struct vec vec, a, b;
const char *rewrite;
int match_len; int match_len;
match_len = get_document_root(conn, &vec); match_len = get_document_root(conn, &vec);
mg_snprintf(conn, buf, buf_len, "%.*s%s", vec.len, vec.ptr, uri + match_len); mg_snprintf(conn, buf, buf_len, "%.*s%s", vec.len, vec.ptr, uri + match_len);
rewrite = conn->ctx->config[REWRITE];
while ((rewrite = next_option(rewrite, &a, &b)) != NULL) {
if ((match_len = match_prefix(a.ptr, a.len, uri)) > 0) {
mg_snprintf(conn, buf, buf_len, "%.*s%s", b.len, b.ptr, uri + match_len);
break;
}
}
#if defined(_WIN32) && !defined(__SYMBIAN32__) #if defined(_WIN32) && !defined(__SYMBIAN32__)
change_slashes_to_backslashes(buf); change_slashes_to_backslashes(buf);
#endif // _WIN32 #endif // _WIN32

13
test/unit_test.c Normal file
View File

@ -0,0 +1,13 @@
#include "mongoose.c"
int main(void) {
assert(match_prefix("/a/", 3, "/a/b/c") == 3);
assert(match_prefix("/a/", 3, "/ab/c") == 0);
assert(match_prefix("/*/", 3, "/ab/c") == 4);
assert(match_prefix("**", 2, "/a/b/c") == 6);
assert(match_prefix("/*", 2, "/a/b/c") == 2);
assert(match_prefix("*/*", 3, "/a/b/c") == 2);
assert(match_prefix("**/", 3, "/a/b/c") == 5);
return 0;
}