#include "test_framework.h"
#include "../src/types.h"
#include "../src/http.h"
#include "../src/config.h"
#include <stdio.h>
#include <unistd.h>
static const char *TEST_CONFIG_FILE = "/tmp/test_routing_config.json";
static void create_routing_config(void) {
FILE *f = fopen(TEST_CONFIG_FILE, "w");
if (f) {
fprintf(f,
"{\n"
" \"port\": 8080,\n"
" \"reverse_proxy\": [\n"
" {\n"
" \"hostname\": \"api.local\",\n"
" \"upstream_host\": \"backend-api\",\n"
" \"upstream_port\": 3000,\n"
" \"use_ssl\": false,\n"
" \"rewrite_host\": true\n"
" },\n"
" {\n"
" \"hostname\": \"secure.local\",\n"
" \"upstream_host\": \"backend-secure\",\n"
" \"upstream_port\": 443,\n"
" \"use_ssl\": true,\n"
" \"rewrite_host\": true\n"
" },\n"
" {\n"
" \"hostname\": \"passthrough.local\",\n"
" \"upstream_host\": \"backend-pass\",\n"
" \"upstream_port\": 8000,\n"
" \"use_ssl\": false,\n"
" \"rewrite_host\": false\n"
" }\n"
" ]\n"
"}\n");
fclose(f);
}
}
static void cleanup_routing_config(void) {
unlink(TEST_CONFIG_FILE);
}
void test_routing_host_rewrite(void) {
TEST_SUITE_BEGIN("Routing Host Rewrite Logic");
create_routing_config();
config_load(TEST_CONFIG_FILE);
route_config_t *route = config_find_route("api.local");
TEST_ASSERT(route != NULL, "Route for api.local found");
if (route) {
TEST_ASSERT_EQ(1, route->rewrite_host, "Host rewrite enabled for api.local");
TEST_ASSERT_STR_EQ("backend-api", route->upstream_host, "Upstream host is backend-api");
TEST_ASSERT_EQ(3000, route->upstream_port, "Upstream port is 3000");
}
route = config_find_route("passthrough.local");
TEST_ASSERT(route != NULL, "Route for passthrough.local found");
if (route) {
TEST_ASSERT_EQ(0, route->rewrite_host, "Host rewrite disabled for passthrough.local");
}
config_free();
cleanup_routing_config();
TEST_SUITE_END();
}
void test_routing_ssl_upstream(void) {
TEST_SUITE_BEGIN("Routing SSL Upstream");
create_routing_config();
config_load(TEST_CONFIG_FILE);
route_config_t *route = config_find_route("secure.local");
TEST_ASSERT(route != NULL, "Route for secure.local found");
if (route) {
TEST_ASSERT_EQ(1, route->use_ssl, "SSL enabled for secure.local");
TEST_ASSERT_EQ(443, route->upstream_port, "SSL port is 443");
}
route = config_find_route("api.local");
TEST_ASSERT(route != NULL, "Route for api.local found");
if (route) {
TEST_ASSERT_EQ(0, route->use_ssl, "SSL disabled for api.local");
}
config_free();
cleanup_routing_config();
TEST_SUITE_END();
}
void test_routing_dashboard_detection(void) {
TEST_SUITE_BEGIN("Routing Dashboard Detection");
http_request_t req;
const char *dashboard_req = "GET /rproxy/dashboard HTTP/1.1\r\nHost: anyhost.com\r\n\r\n";
http_parse_request(dashboard_req, strlen(dashboard_req), &req);
int is_dashboard = (strncmp(req.uri, "/rproxy/dashboard", 17) == 0);
TEST_ASSERT_EQ(1, is_dashboard, "Dashboard URI detected");
const char *api_req = "GET /rproxy/api/stats HTTP/1.1\r\nHost: anyhost.com\r\n\r\n";
http_parse_request(api_req, strlen(api_req), &req);
int is_api = (strncmp(req.uri, "/rproxy/api/stats", 17) == 0);
TEST_ASSERT_EQ(1, is_api, "API stats URI detected");
const char *regular_req = "GET /some/other/path HTTP/1.1\r\nHost: api.local\r\n\r\n";
http_parse_request(regular_req, strlen(regular_req), &req);
int is_regular = (strncmp(req.uri, "/rproxy/", 8) != 0);
TEST_ASSERT_EQ(1, is_regular, "Regular path not matched as internal");
TEST_SUITE_END();
}
void test_routing_keep_alive_handling(void) {
TEST_SUITE_BEGIN("Routing Keep-Alive Handling");
http_request_t req;
const char *ka_req = "GET /api HTTP/1.1\r\nHost: api.local\r\nConnection: keep-alive\r\n\r\n";
http_parse_request(ka_req, strlen(ka_req), &req);
TEST_ASSERT_EQ(1, req.keep_alive, "Keep-alive flag set when Connection: keep-alive");
const char *close_req = "GET /api HTTP/1.1\r\nHost: api.local\r\nConnection: close\r\n\r\n";
http_parse_request(close_req, strlen(close_req), &req);
TEST_ASSERT_EQ(0, req.keep_alive, "Keep-alive flag not set when Connection: close");
const char *default_req = "GET /api HTTP/1.1\r\nHost: api.local\r\n\r\n";
http_parse_request(default_req, strlen(default_req), &req);
TEST_ASSERT_EQ(1, req.keep_alive, "Keep-alive default for HTTP/1.1");
const char *http10_req = "GET /api HTTP/1.0\r\nHost: api.local\r\n\r\n";
http_parse_request(http10_req, strlen(http10_req), &req);
TEST_ASSERT_EQ(0, req.keep_alive, "Keep-alive default off for HTTP/1.0");
TEST_SUITE_END();
}
void test_routing_all_methods_accepted(void) {
TEST_SUITE_BEGIN("Routing All HTTP Methods Accepted");
http_request_t req;
int result;
const char *methods[] = {
"GET", "POST", "PUT", "DELETE", "HEAD", "OPTIONS", "PATCH", "TRACE", "CONNECT",
"PROPFIND", "PROPPATCH", "MKCOL", "MOVE", "COPY", "LOCK", "UNLOCK",
"SEARCH", "REPORT", "MKACTIVITY", "CHECKOUT", "MERGE",
"NOTIFY", "SUBSCRIBE", "UNSUBSCRIBE",
"CUSTOMMETHOD", "FOOBAR", "MYPROTO"
};
int num_methods = sizeof(methods) / sizeof(methods[0]);
for (int i = 0; i < num_methods; i++) {
char request[256];
snprintf(request, sizeof(request), "%s /path HTTP/1.1\r\nHost: test.com\r\n\r\n", methods[i]);
result = http_parse_request(request, strlen(request), &req);
char msg[128];
snprintf(msg, sizeof(msg), "Method %s accepted", methods[i]);
TEST_ASSERT_EQ(1, result, msg);
TEST_ASSERT_STR_EQ(methods[i], req.method, msg);
}
TEST_SUITE_END();
}
void test_routing_pipelined_requests(void) {
TEST_SUITE_BEGIN("Routing Pipelined Request Detection");
http_request_t req1, req2;
const char *first_req = "GET /first HTTP/1.1\r\nHost: api.local\r\nConnection: keep-alive\r\n\r\n";
int result1 = http_parse_request(first_req, strlen(first_req), &req1);
TEST_ASSERT_EQ(1, result1, "First request parsed");
TEST_ASSERT_STR_EQ("/first", req1.uri, "First request URI");
TEST_ASSERT_EQ(1, req1.keep_alive, "First request keep-alive");
const char *second_req = "GET /second HTTP/1.1\r\nHost: api.local\r\n\r\n";
int result2 = http_parse_request(second_req, strlen(second_req), &req2);
TEST_ASSERT_EQ(1, result2, "Second request parsed");
TEST_ASSERT_STR_EQ("/second", req2.uri, "Second request URI");
int is_new_request = http_is_request_start(second_req, strlen(second_req));
TEST_ASSERT_EQ(1, is_new_request, "New request detected for pipelining");
TEST_SUITE_END();
}
void test_routing_websocket_upgrade(void) {
TEST_SUITE_BEGIN("Routing WebSocket Upgrade");
http_request_t req;
const char *ws_req =
"GET /ws/chat HTTP/1.1\r\n"
"Host: api.local\r\n"
"Upgrade: websocket\r\n"
"Connection: Upgrade\r\n"
"Sec-WebSocket-Key: dGhlIHNhbXBsZSBub25jZQ==\r\n"
"Sec-WebSocket-Version: 13\r\n"
"\r\n";
int result = http_parse_request(ws_req, strlen(ws_req), &req);
TEST_ASSERT_EQ(1, result, "WebSocket upgrade request parsed");
TEST_ASSERT_EQ(1, req.is_websocket, "WebSocket flag set");
TEST_ASSERT_STR_EQ("/ws/chat", req.uri, "WebSocket URI parsed");
TEST_SUITE_END();
}
void test_routing_chunked_transfer(void) {
TEST_SUITE_BEGIN("Routing Chunked Transfer Encoding");
http_request_t req;
const char *chunked_req =
"POST /upload HTTP/1.1\r\n"
"Host: api.local\r\n"
"Transfer-Encoding: chunked\r\n"
"\r\n";
int result = http_parse_request(chunked_req, strlen(chunked_req), &req);
TEST_ASSERT_EQ(1, result, "Chunked request parsed");
TEST_ASSERT_EQ(1, req.is_chunked, "Chunked flag set");
const char *non_chunked_req =
"POST /upload HTTP/1.1\r\n"
"Host: api.local\r\n"
"Content-Length: 100\r\n"
"\r\n";
result = http_parse_request(non_chunked_req, strlen(non_chunked_req), &req);
TEST_ASSERT_EQ(1, result, "Non-chunked request parsed");
TEST_ASSERT_EQ(0, req.is_chunked, "Chunked flag not set");
TEST_ASSERT_EQ(100, req.content_length, "Content-Length parsed");
TEST_SUITE_END();
}
void run_routing_tests(void) {
test_routing_host_rewrite();
test_routing_ssl_upstream();
test_routing_dashboard_detection();
test_routing_keep_alive_handling();
test_routing_all_methods_accepted();
test_routing_pipelined_requests();
test_routing_websocket_upgrade();
test_routing_chunked_transfer();
}
Reference in New Issue View Git Blame Copy Permalink