mirror/mongoose
1
0
Fork 0
mirror of https://github.com/cesanta/mongoose.git synced 2025-06-13 05:12:51 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Added test for rewrites

Browse Source
This commit is contained in:
Sergey Lyubka 2014-01-27 11:03:44 +00:00
parent 83a9a726d3
commit e31f3058fa
2 changed files with 36 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mongoose.c
View File

@ -1542,10 +1542,10 @@ static int must_hide_file(struct connection *conn, const char *path) {
static int convert_uri_to_file_name(struct connection *conn, char *buf, static int convert_uri_to_file_name(struct connection *conn, char *buf,
size_t buf_len, file_stat_t *st) { size_t buf_len, file_stat_t *st) {
struct vec a, b; struct vec a, b;
const char *rewrites = conn->server->config_options[URL_REWRITES], const char *rewrites = conn->server->config_options[URL_REWRITES];
*root = conn->server->config_options[DOCUMENT_ROOT], const char *root = conn->server->config_options[DOCUMENT_ROOT];
*cgi_pat = conn->server->config_options[CGI_PATTERN], const char *cgi_pat = conn->server->config_options[CGI_PATTERN];
*uri = conn->mg_conn.uri; const char *uri = conn->mg_conn.uri;
char *p; char *p;
int match_len; int match_len;
@ -3639,6 +3639,7 @@ int mg_connect(struct mg_server *server, const char *host, int port,
return 0; return 0;
} }
conn->server = server;
conn->client_sock = sock; conn->client_sock = sock;
conn->endpoint_type = EP_CLIENT; conn->endpoint_type = EP_CLIENT;
conn->handler = handler; conn->handler = handler;

41
unit_test.c
View File

@ -538,31 +538,32 @@ static const char *test_parse_multipart(void) {
return NULL; return NULL;
} }
#ifdef MONGOOSE_USE_SSL
static int us1(struct mg_connection *conn) { static int us1(struct mg_connection *conn) {
static const char *file_name = "mongoose.h"; static const char *file_name = "mongoose.h";
char *cp = (char *) conn->connection_param;
int file_size; int file_size;
if (conn->status_code == MG_CONNECT_SUCCESS) { if (conn->status_code == MG_CONNECT_SUCCESS) {
mg_printf(conn, "GET /%s HTTP/1.0\r\n\r\n", file_name); mg_printf(conn, "GET /%s HTTP/1.0\r\n\r\n", cp);
return 0; return 0;
} else if (conn->status_code == MG_DOWNLOAD_SUCCESS) { } else if (conn->status_code == MG_DOWNLOAD_SUCCESS) {
char *file_data = read_file(file_name, &file_size); char *file_data = read_file(file_name, &file_size);
sprintf((char *) conn->connection_param, "%d", sprintf(cp, "%d %s", (size_t) file_size == conn->content_len &&
(size_t) file_size == conn->content_len && memcmp(file_data, conn->content, file_size) == 0 ? 1 : 0,
memcmp(file_data, conn->content, file_size) == 0 ? 1 : 0); conn->query_string == NULL ? "?" : conn->query_string);
free(file_data); free(file_data);
} else if (conn->status_code == MG_CONNECT_FAILURE) { } else if (conn->status_code == MG_CONNECT_FAILURE) {
sprintf((char *) conn->connection_param, "%s", "cf"); sprintf(cp, "%s", "cf");
} else { } else {
sprintf((char *) conn->connection_param, "%s", "df"); sprintf(cp, "%s", "df");
} }
return 1; return 1;
} }
#ifdef MONGOOSE_USE_SSL
static const char *test_ssl(void) { static const char *test_ssl(void) {
static const char *ssl_cert = "examples/ssl_cert.pem"; static const char *ssl_cert = "examples/ssl_cert.pem";
char buf1[100] = ""; char buf1[100] = "mongoose.h";
struct mg_server *server = mg_create_server(NULL); struct mg_server *server = mg_create_server(NULL);
ASSERT(mg_set_option(server, "listening_port", LISTENING_ADDR) == NULL); ASSERT(mg_set_option(server, "listening_port", LISTENING_ADDR) == NULL);
@ -571,8 +572,8 @@ static const char *test_ssl(void) {
ASSERT(mg_connect(server, "127.0.0.1", atoi(HTTP_PORT), 1, us1, buf1) == 1); ASSERT(mg_connect(server, "127.0.0.1", atoi(HTTP_PORT), 1, us1, buf1) == 1);
{ int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1000); } { int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1); }
ASSERT(strcmp(buf1, "1") == 0); ASSERT(strcmp(buf1, "1 ?") == 0);
mg_destroy_server(&server); mg_destroy_server(&server);
return NULL; return NULL;
} }
@ -586,6 +587,25 @@ static const char *test_mg_set_option(void) {
return NULL; return NULL;
} }
static const char *test_rewrites(void) {
char buf1[100] = "xx";
struct mg_server *server = mg_create_server(NULL);
ASSERT(mg_set_option(server, "listening_port", "0") == NULL);
ASSERT(mg_set_option(server, "document_root", ".") == NULL);
ASSERT(mg_set_option(server, "url_rewrites", "/xx=./mongoose.h") == NULL);
ASSERT(mg_connect(server, "127.0.0.1",
atoi(mg_get_option(server, "listening_port")),
0, us1, buf1) == 1);
{ int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1); }
//printf("[%s]\n", buf1);
ASSERT(strcmp(buf1, "1 ?") == 0);
mg_destroy_server(&server);
return NULL;
}
static const char *run_all_tests(void) { static const char *run_all_tests(void) {
RUN_TEST(test_should_keep_alive); RUN_TEST(test_should_keep_alive);
RUN_TEST(test_match_prefix); RUN_TEST(test_match_prefix);
@ -602,6 +622,7 @@ static const char *run_all_tests(void) {
RUN_TEST(test_server); RUN_TEST(test_server);
RUN_TEST(test_mg_connect); RUN_TEST(test_mg_connect);
RUN_TEST(test_mg_set_option); RUN_TEST(test_mg_set_option);
RUN_TEST(test_rewrites);
#ifdef MONGOOSE_USE_SSL #ifdef MONGOOSE_USE_SSL
RUN_TEST(test_ssl); RUN_TEST(test_ssl);
#endif #endif