mongoose/test/unit_test.c

3401 lines
119 KiB
C
Raw Normal View History

#include "mongoose.h"
2022-06-10 16:38:36 +08:00
#include "float.h" // For DBL_EPSILON and HUGE_VAL
#include "math.h"
2020-12-05 19:26:32 +08:00
static int s_num_tests = 0;
2020-12-05 19:26:32 +08:00
#define ASSERT(expr) \
do { \
s_num_tests++; \
if (!(expr)) { \
printf("FAILURE %s:%d: %s\n", __FILE__, __LINE__, #expr); \
2022-06-12 19:43:01 +08:00
abort(); \
2020-12-05 19:26:32 +08:00
} \
} while (0)
2020-12-05 19:26:32 +08:00
#define FETCH_BUF_SIZE (256 * 1024)
2023-07-25 20:07:28 +08:00
// Self-signed CA, CERT, KEY
const char *s_tls_ca =
2023-07-25 20:07:28 +08:00
"-----BEGIN CERTIFICATE-----\n"
"MIIBqjCCAU+gAwIBAgIUESoOPGqMhf9uarzblVFwzrQweMcwCgYIKoZIzj0EAwIw\n"
"RDELMAkGA1UEBhMCSUUxDzANBgNVBAcMBkR1YmxpbjEQMA4GA1UECgwHQ2VzYW50\n"
"YTESMBAGA1UEAwwJVGVzdCBSb290MCAXDTIwMDUwOTIxNTE0NFoYDzIwNTAwNTA5\n"
"MjE1MTQ0WjBEMQswCQYDVQQGEwJJRTEPMA0GA1UEBwwGRHVibGluMRAwDgYDVQQK\n"
"DAdDZXNhbnRhMRIwEAYDVQQDDAlUZXN0IFJvb3QwWTATBgcqhkjOPQIBBggqhkjO\n"
"PQMBBwNCAAQsq9ECZiSW1xI+CVBP8VDuUehVA166sR2YsnJ5J6gbMQ1dUCH/QvLa\n"
"dBdeU7JlQcH8hN5KEbmM9BnZxMor6ussox0wGzAMBgNVHRMEBTADAQH/MAsGA1Ud\n"
"DwQEAwIBrjAKBggqhkjOPQQDAgNJADBGAiEAnHFsAIwGQQyRL81B04dH6d86Iq0l\n"
"fL8OKzndegxOaB0CIQCPwSIwEGFdURDqCC0CY2dnMrUGY5ZXu3hHCojZGS7zvg==\n"
"-----END CERTIFICATE-----\n";
const char *s_tls_cert =
2023-07-25 20:07:28 +08:00
"-----BEGIN CERTIFICATE-----\n"
"MIIBhzCCASygAwIBAgIUbnMoVd8TtWH1T09dANkK2LU6IUswCgYIKoZIzj0EAwIw\n"
"RDELMAkGA1UEBhMCSUUxDzANBgNVBAcMBkR1YmxpbjEQMA4GA1UECgwHQ2VzYW50\n"
"YTESMBAGA1UEAwwJVGVzdCBSb290MB4XDTIwMDUwOTIxNTE0OVoXDTMwMDUwOTIx\n"
"NTE0OVowETEPMA0GA1UEAwwGc2VydmVyMFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcD\n"
"QgAEkuBGnInDN6l06zVVQ1VcrOvH5FDu9MC6FwJc2e201P8hEpq0Q/SJS2nkbSuW\n"
"H/wBTTBaeXN2uhlBzMUWK790KKMvMC0wCQYDVR0TBAIwADALBgNVHQ8EBAMCA6gw\n"
"EwYDVR0lBAwwCgYIKwYBBQUHAwEwCgYIKoZIzj0EAwIDSQAwRgIhAPo6xx7LjCdZ\n"
"QY133XvLjAgVFrlucOZHONFVQuDXZsjwAiEAzHBNligA08c5U3SySYcnkhurGg50\n"
"BllCI0eYQ9ggp/o=\n"
"-----END CERTIFICATE-----\n";
const char *s_tls_key =
2023-07-25 20:07:28 +08:00
"-----BEGIN PRIVATE KEY-----\n"
"MIGHAgEAMBMGByqGSM49AgEGCCqGSM49AwEHBG0wawIBAQQglNni0t9Dg9icgG8w\n"
"kbfxWSS+TuNgbtNybIQXcm3NHpmhRANCAASS4EacicM3qXTrNVVDVVys68fkUO70\n"
"wLoXAlzZ7bTU/yESmrRD9IlLaeRtK5Yf/AFNMFp5c3a6GUHMxRYrv3Qo\n"
"-----END PRIVATE KEY-----\n";
2020-12-05 19:26:32 +08:00
// Important: we use different port numbers for the Windows bug workaround. See
// https://support.microsoft.com/en-ae/help/3039044/error-10013-wsaeacces-is-returned-when-a-second-bind-to-a-excluded-por
static void test_globmatch(void) {
ASSERT(mg_globmatch("", 0, "", 0) == 1);
ASSERT(mg_globmatch("*", 1, "a", 1) == 1);
ASSERT(mg_globmatch("*", 1, "ab", 2) == 1);
ASSERT(mg_globmatch("", 0, "a", 1) == 0);
ASSERT(mg_globmatch("/", 1, "/foo", 4) == 0);
ASSERT(mg_globmatch("/*/foo", 6, "/x/bar", 6) == 0);
ASSERT(mg_globmatch("/*/foo", 6, "/x/foo", 6) == 1);
ASSERT(mg_globmatch("/*/foo", 6, "/x/foox", 7) == 0);
ASSERT(mg_globmatch("/*/foo*", 7, "/x/foox", 7) == 1);
ASSERT(mg_globmatch("/*", 2, "/abc", 4) == 1);
ASSERT(mg_globmatch("/*", 2, "/ab/", 4) == 0);
ASSERT(mg_globmatch("/*", 2, "/", 1) == 1);
ASSERT(mg_globmatch("/x/*", 4, "/x/2", 4) == 1);
ASSERT(mg_globmatch("/x/*", 4, "/x/2/foo", 8) == 0);
ASSERT(mg_globmatch("/x/*/*", 6, "/x/2/foo", 8) == 1);
ASSERT(mg_globmatch("#", 1, "///", 3) == 1);
ASSERT(mg_globmatch("/api/*", 6, "/api/foo", 8) == 1);
ASSERT(mg_globmatch("/api/*", 6, "/api/log/static", 15) == 0);
ASSERT(mg_globmatch("/api/#", 6, "/api/log/static", 15) == 1);
2021-09-15 14:43:48 +08:00
ASSERT(mg_globmatch("#.shtml", 7, "/ssi/index.shtml", 16) == 1);
ASSERT(mg_globmatch("#.c", 3, ".c", 2) == 1);
ASSERT(mg_globmatch("abc", 3, "ab", 2) == 0);
ASSERT(mg_globmatch("#.c", 3, "a.c", 3) == 1);
ASSERT(mg_globmatch("#.c", 3, "..c", 3) == 1);
ASSERT(mg_globmatch("#.c", 3, "/.c", 3) == 1);
ASSERT(mg_globmatch("#.c", 3, "//a.c", 5) == 1);
ASSERT(mg_globmatch("#.c", 3, "x/a.c", 5) == 1);
ASSERT(mg_globmatch("#.c", 3, "./a.c", 5) == 1);
ASSERT(mg_globmatch("#.shtml", 7, "./ssi/index.shtml", 17) == 1);
ASSERT(mg_globmatch("#aa#bb#", 7, "caabba", 6) == 1);
ASSERT(mg_globmatch("#aa#bb#", 7, "caabxa", 6) == 0);
ASSERT(mg_globmatch("a*b*c", 5, "a__b_c", 6) == 1);
{
struct mg_str caps[3];
ASSERT(mg_match(mg_str("//a.c"), mg_str("#.c"), NULL) == true);
ASSERT(mg_match(mg_str("a"), mg_str("#"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("a")) == 0);
ASSERT(mg_match(mg_str("//a.c"), mg_str("#.c"), caps) == true);
ASSERT(mg_match(mg_str("a_b_c_"), mg_str("a*b*c"), caps) == false);
ASSERT(mg_match(mg_str("a__b_c"), mg_str("a*b*c"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("__")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("_")) == 0);
ASSERT(mg_match(mg_str("a_b_c__c"), mg_str("a*b*c"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("_")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("_c__")) == 0);
ASSERT(mg_match(mg_str("a_xb_.c__c"), mg_str("a*b*c"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("_x")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("_.c__")) == 0);
ASSERT(mg_match(mg_str("a"), mg_str("#a"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("")) == 0);
ASSERT(mg_match(mg_str(".aa..b...b"), mg_str("#a#b"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str(".")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("a..b...")) == 0);
ASSERT(mg_strcmp(caps[2], mg_str("")) == 0);
ASSERT(mg_match(mg_str("/foo/bar"), mg_str("/*/*"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("foo")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("bar")) == 0);
ASSERT(mg_strcmp(caps[2], mg_str("")) == 0);
ASSERT(mg_match(mg_str("/foo/"), mg_str("/*/*"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("foo")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("")) == 0);
ASSERT(mg_strcmp(caps[2], mg_str("")) == 0);
ASSERT(mg_match(mg_str("abc"), mg_str("?#"), caps) == true);
ASSERT(mg_strcmp(caps[0], mg_str("a")) == 0);
ASSERT(mg_strcmp(caps[1], mg_str("bc")) == 0);
ASSERT(mg_strcmp(caps[2], mg_str("")) == 0);
}
2020-12-05 19:26:32 +08:00
}
static void test_http_get_var(void) {
char buf[256];
struct mg_str body;
body = mg_str("key1=value1&key2=value2&key3=value%203&key4=value+4");
ASSERT(mg_http_get_var(&body, "key1", buf, sizeof(buf)) == 6);
ASSERT(strcmp(buf, "value1") == 0);
ASSERT(mg_http_get_var(&body, "KEY1", buf, sizeof(buf)) == 6);
ASSERT(strcmp(buf, "value1") == 0);
ASSERT(mg_http_get_var(&body, "key2", buf, sizeof(buf)) == 6);
ASSERT(strcmp(buf, "value2") == 0);
ASSERT(mg_http_get_var(&body, "key3", buf, sizeof(buf)) == 7);
ASSERT(strcmp(buf, "value 3") == 0);
ASSERT(mg_http_get_var(&body, "key4", buf, sizeof(buf)) == 7);
ASSERT(strcmp(buf, "value 4") == 0);
ASSERT(mg_http_get_var(&body, "key", buf, sizeof(buf)) == -4);
ASSERT(mg_http_get_var(&body, "key1", NULL, sizeof(buf)) == -2);
ASSERT(mg_http_get_var(&body, "key1", buf, 0) == -2);
ASSERT(mg_http_get_var(&body, NULL, buf, sizeof(buf)) == -1);
ASSERT(mg_http_get_var(&body, "key1", buf, 1) == -3);
body = mg_str("key=broken%2");
ASSERT(mg_http_get_var(&body, "key", buf, sizeof(buf)) == -3);
body = mg_str("key=broken%2x");
ASSERT(mg_http_get_var(&body, "key", buf, sizeof(buf)) == -3);
ASSERT(mg_http_get_var(&body, "inexistent", buf, sizeof(buf)) == -4);
2020-12-11 17:35:50 +08:00
body = mg_str("key=%");
ASSERT(mg_http_get_var(&body, "key", buf, sizeof(buf)) == -3);
2020-12-12 18:03:19 +08:00
body = mg_str("&&&kEy=%");
ASSERT(mg_http_get_var(&body, "kEy", buf, sizeof(buf)) == -3);
2020-12-05 19:26:32 +08:00
}
static int vcmp(struct mg_str s1, const char *s2) {
// MG_INFO(("->%.*s<->%s<- %d %d %d", (int) s1.len, s1.ptr, s2,
2020-12-05 19:26:32 +08:00
//(int) s1.len, strncmp(s1.ptr, s2, s1.len), mg_vcmp(&s1, s2)));
return mg_vcmp(&s1, s2) == 0;
}
static void test_url(void) {
// Host
ASSERT(vcmp(mg_url_host("foo"), "foo"));
ASSERT(vcmp(mg_url_host("//foo"), "foo"));
ASSERT(vcmp(mg_url_host("foo:1234"), "foo"));
2021-11-29 21:57:33 +08:00
ASSERT(vcmp(mg_url_host(":1234"), ""));
2020-12-05 19:26:32 +08:00
ASSERT(vcmp(mg_url_host("//foo:1234"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo/"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo/x"), "foo"));
2021-03-24 23:49:35 +08:00
ASSERT(vcmp(mg_url_host("p://foo/x/"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo/x//"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo//x"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo///x"), "foo"));
ASSERT(vcmp(mg_url_host("p://foo///x//"), "foo"));
2020-12-05 19:26:32 +08:00
ASSERT(vcmp(mg_url_host("p://bar:1234"), "bar"));
ASSERT(vcmp(mg_url_host("p://bar:1234/"), "bar"));
ASSERT(vcmp(mg_url_host("p://bar:1234/a"), "bar"));
ASSERT(vcmp(mg_url_host("p://u@bar:1234/a"), "bar"));
ASSERT(vcmp(mg_url_host("p://u:p@bar:1234/a"), "bar"));
ASSERT(vcmp(mg_url_host("p://u:p@[::1]:1234/a"), "[::1]"));
ASSERT(vcmp(mg_url_host("p://u:p@[1:2::3]:1234/a"), "[1:2::3]"));
2021-07-31 01:34:55 +08:00
ASSERT(vcmp(mg_url_host("p://foo/x:y/z"), "foo"));
2020-12-05 19:26:32 +08:00
// Port
ASSERT(mg_url_port("foo:1234") == 1234);
2021-11-29 21:57:33 +08:00
ASSERT(mg_url_port(":1234") == 1234);
2020-12-05 19:26:32 +08:00
ASSERT(mg_url_port("x://foo:1234") == 1234);
ASSERT(mg_url_port("x://foo:1234/") == 1234);
ASSERT(mg_url_port("x://foo:1234/xx") == 1234);
ASSERT(mg_url_port("x://foo:1234") == 1234);
ASSERT(mg_url_port("p://bar:1234/a") == 1234);
2021-03-24 23:49:35 +08:00
ASSERT(mg_url_port("p://bar:1234/a:b") == 1234);
2020-12-05 19:26:32 +08:00
ASSERT(mg_url_port("http://bar") == 80);
ASSERT(mg_url_port("http://localhost:1234") == 1234);
ASSERT(mg_url_port("https://bar") == 443);
ASSERT(mg_url_port("wss://bar") == 443);
ASSERT(mg_url_port("wss://u:p@bar") == 443);
ASSERT(mg_url_port("wss://u:p@bar:123") == 123);
ASSERT(mg_url_port("wss://u:p@bar:123/") == 123);
ASSERT(mg_url_port("wss://u:p@bar:123/abc") == 123);
2020-12-22 17:44:59 +08:00
ASSERT(mg_url_port("http://u:p@[::1]/abc") == 80);
ASSERT(mg_url_port("http://u:p@[::1]:2121/abc") == 2121);
2021-03-24 23:49:35 +08:00
ASSERT(mg_url_port("http://u:p@[::1]:2121/abc/cd:ef") == 2121);
2020-12-05 19:26:32 +08:00
// User / pass
ASSERT(vcmp(mg_url_user("p://foo"), ""));
ASSERT(vcmp(mg_url_pass("p://foo"), ""));
ASSERT(vcmp(mg_url_user("p://:@foo"), ""));
ASSERT(vcmp(mg_url_pass("p://:@foo"), ""));
ASSERT(vcmp(mg_url_user("p://u@foo"), "u"));
ASSERT(vcmp(mg_url_pass("p://u@foo"), ""));
ASSERT(vcmp(mg_url_user("p://u:@foo"), "u"));
ASSERT(vcmp(mg_url_pass("p://u:@foo"), ""));
ASSERT(vcmp(mg_url_user("p://:p@foo"), ""));
ASSERT(vcmp(mg_url_pass("p://:p@foo"), "p"));
ASSERT(vcmp(mg_url_user("p://u:p@foo"), "u"));
ASSERT(vcmp(mg_url_pass("p://u:p@foo"), "p"));
2021-03-24 23:49:35 +08:00
ASSERT(vcmp(mg_url_pass("p://u:p@foo//a@b"), "p"));
2022-07-10 22:41:11 +08:00
ASSERT(vcmp(mg_url_user("p://foo/q?mail=a@b.c"), ""));
ASSERT(vcmp(mg_url_pass("p://foo/q?mail=a@b.c"), ""));
2020-12-05 19:26:32 +08:00
// URI
ASSERT(strcmp(mg_url_uri("p://foo"), "/") == 0);
ASSERT(strcmp(mg_url_uri("p://foo/"), "/") == 0);
ASSERT(strcmp(mg_url_uri("p://foo:12/"), "/") == 0);
ASSERT(strcmp(mg_url_uri("p://foo:12/abc"), "/abc") == 0);
ASSERT(strcmp(mg_url_uri("p://foo:12/a/b/c"), "/a/b/c") == 0);
2020-12-22 17:44:59 +08:00
ASSERT(strcmp(mg_url_uri("p://[::1]:12/a/b/c"), "/a/b/c") == 0);
ASSERT(strcmp(mg_url_uri("p://[ab::1]:12/a/b/c"), "/a/b/c") == 0);
2022-07-10 22:41:11 +08:00
ASSERT(strcmp(mg_url_uri("p://foo/q?mail=a@b.c"), "/q?mail=a@b.c") == 0);
2020-12-05 19:26:32 +08:00
}
static void test_base64(void) {
char buf[128];
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "", 0, buf, sizeof(buf)) == 0);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "x", 1, buf, sizeof(buf)) == 4);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "eA==") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "xyz", 3, buf, sizeof(buf)) == 4);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "eHl6") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "abcdef", 6, buf, sizeof(buf)) == 8);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "YWJjZGVm") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "ы", 2, buf, sizeof(buf)) == 4);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "0Ys=") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "xy", 3, buf, sizeof(buf)) == 4);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "eHkA") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "test", 4, buf, sizeof(buf)) == 8);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "dGVzdA==") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "abcde", 5, buf, sizeof(buf)) == 8);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "YWJjZGU=") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_encode((uint8_t *) "a", 1, buf, 0) == 0);
ASSERT(mg_base64_encode((uint8_t *) "a", 1, buf, 1) == 0);
ASSERT(mg_base64_encode((uint8_t *) "a", 1, buf, 2) == 0);
ASSERT(mg_base64_encode((uint8_t *) "a", 1, buf, 3) == 0);
ASSERT(mg_base64_encode((uint8_t *) "a", 1, buf, 4) == 0);
ASSERT(mg_base64_encode((uint8_t *) "a", 1, buf, 5) == 4);
ASSERT(mg_base64_decode("кю", 4, buf, sizeof(buf)) == 0);
ASSERT(mg_base64_decode("A", 1, buf, sizeof(buf)) == 0);
ASSERT(mg_base64_decode("A=", 2, buf, sizeof(buf)) == 0);
ASSERT(mg_base64_decode("AA=", 3, buf, sizeof(buf)) == 0);
ASSERT(mg_base64_decode("AAA=", 4, buf, sizeof(buf)) == 2);
ASSERT(mg_base64_decode("AAAA====", 8, buf, sizeof(buf)) == 0);
ASSERT(mg_base64_decode("AAAA----", 8, buf, sizeof(buf)) == 0);
ASSERT(mg_base64_decode("Q2VzYW50YQ==", 12, buf, sizeof(buf)) == 7);
2020-12-05 19:26:32 +08:00
ASSERT(strcmp(buf, "Cesanta") == 0);
2023-08-22 18:50:19 +08:00
ASSERT(mg_base64_decode("AAA=", 4, buf, 0) == 0);
ASSERT(mg_base64_decode("AAA=", 4, buf, 1) == 0);
ASSERT(mg_base64_decode("AAA=", 4, buf, 2) == 0);
ASSERT(mg_base64_decode("AAA=", 4, buf, 3) == 0);
ASSERT(mg_base64_decode("AAA=", 4, buf, 4) == 2);
2020-12-05 19:26:32 +08:00
}
static void test_iobuf(void) {
2022-08-01 19:53:25 +08:00
struct mg_iobuf io = {0, 0, 0, 10};
2020-12-05 19:26:32 +08:00
ASSERT(io.buf == NULL && io.size == 0 && io.len == 0);
mg_iobuf_resize(&io, 1);
2022-08-01 19:53:25 +08:00
ASSERT(io.buf != NULL && io.size == 10 && io.len == 0);
2021-08-28 15:08:54 +08:00
ASSERT(memcmp(io.buf, "\x00", 1) == 0);
2022-08-01 19:53:25 +08:00
mg_iobuf_add(&io, 3, "hi", 2);
2021-08-28 15:08:54 +08:00
ASSERT(io.buf != NULL && io.size == 10 && io.len == 5);
ASSERT(memcmp(io.buf, "\x00\x00\x00hi", 5) == 0);
2022-08-01 19:53:25 +08:00
mg_iobuf_add(&io, io.len, "!", 1);
2021-08-28 15:08:54 +08:00
ASSERT(io.buf != NULL && io.size == 10 && io.len == 6);
ASSERT(memcmp(io.buf, "\x00\x00\x00hi!", 6) == 0);
2022-08-01 19:53:25 +08:00
mg_iobuf_add(&io, 0, "x", 1);
2021-08-28 15:08:54 +08:00
ASSERT(memcmp(io.buf, "x\x00\x00\x00hi!", 7) == 0);
ASSERT(io.buf != NULL && io.size == 10 && io.len == 7);
mg_iobuf_del(&io, 1, 3);
ASSERT(io.buf != NULL && io.size == 10 && io.len == 4);
ASSERT(memcmp(io.buf, "xhi!", 3) == 0);
mg_iobuf_del(&io, 10, 100);
ASSERT(io.buf != NULL && io.size == 10 && io.len == 4);
2023-08-22 17:13:04 +08:00
ASSERT(memcmp(io.buf, "xhi!", io.len) == 0);
mg_iobuf_add(&io, io.len, "123456", 6);
ASSERT(io.buf != NULL && io.size == 10 && io.len == 10);
mg_iobuf_add(&io, io.len, "a", 1);
ASSERT(io.buf != NULL && io.size == 20 && io.len == 11);
ASSERT(memcmp(io.buf, "xhi!123456a", io.len) == 0);
mg_iobuf_free(&io);
2020-12-05 19:26:32 +08:00
}
static void sntp_cb(struct mg_connection *c, int ev, void *ev_data) {
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_SNTP_TIME) {
int64_t received = *(int64_t *) ev_data;
*(int64_t *) c->fn_data = received;
MG_DEBUG(("got time: %lld", received));
#if MG_ARCH == MG_ARCH_UNIX
struct timeval tv = {0, 0};
gettimeofday(&tv, 0);
int64_t ms = (int64_t) tv.tv_sec * 1000 + tv.tv_usec / 1000;
int64_t diff = ms > received ? ms - received : received - ms;
MG_DEBUG(("diff: %lld", diff));
// ASSERT(diff < 100);
#endif
2022-05-28 02:55:39 +08:00
} else if (ev == MG_EV_OPEN) {
c->is_hexdumping = 1;
2020-12-05 19:26:32 +08:00
}
(void) c;
}
2022-05-26 23:22:44 +08:00
static void test_sntp_server(const char *url) {
2021-12-22 05:50:18 +08:00
int64_t ms = 0;
2020-12-05 19:26:32 +08:00
struct mg_mgr mgr;
struct mg_connection *c = NULL;
2022-05-28 05:18:14 +08:00
int i;
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
2022-05-26 23:22:44 +08:00
c = mg_sntp_connect(&mgr, url, sntp_cb, &ms);
2020-12-05 19:26:32 +08:00
ASSERT(c != NULL);
ASSERT(c->is_udp == 1);
2022-05-28 05:18:14 +08:00
for (i = 0; i < 60 && ms == 0; i++) mg_mgr_poll(&mgr, 50);
2022-05-26 23:22:44 +08:00
MG_DEBUG(("server: %s, ms: %lld", url ? url : "(default)", ms));
#if !defined(NO_SNTP_CHECK)
2021-12-22 05:50:18 +08:00
ASSERT(ms > 0);
#endif
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
2022-05-26 23:22:44 +08:00
}
static void test_sntp(void) {
const unsigned char bad[] =
"\x55\x02\x00\xeb\x00\x00\x00\x1e\x00\x00\x07\xb6\x3e\xc9\xd6\xa2"
"\xdb\xde\xea\x30\x91\x86\xb7\x10\xdb\xde\xed\x98\x00\x00\x00\xde"
"\xdb\xde\xed\x99\x0a\xe2\xc7\x96\xdb\xde\xed\x99\x0a\xe4\x6b\xda";
ASSERT(mg_sntp_parse(bad, sizeof(bad)) < 0);
ASSERT(mg_sntp_parse(NULL, 0) == -1);
2022-08-09 03:56:10 +08:00
// NOTE(cpq): temporarily disabled until Github Actions fix their NTP
// port blockage issue, https://github.com/actions/runner-images/issues/5615
2022-11-30 22:47:37 +08:00
// test_sntp_server("udp://time.apple.com:123");
test_sntp_server("udp://time.windows.com:123");
test_sntp_server(NULL);
2020-12-05 19:26:32 +08:00
}
struct mqtt_data {
2023-06-08 02:38:03 +08:00
char *topic;
char *msg;
size_t topicsize;
size_t msgsize;
2023-06-01 08:23:26 +08:00
int flags;
};
2023-06-01 08:23:26 +08:00
#define flags_subscribed (1 << 0)
#define flags_published (1 << 1)
#define flags_received (1 << 2)
#define flags_released (1 << 3)
#define flags_completed (1 << 4)
static void mqtt_cb(struct mg_connection *c, int ev, void *ev_data) {
struct mqtt_data *test_data = (struct mqtt_data *) c->fn_data;
2023-06-08 02:38:03 +08:00
char *buf = test_data->msg;
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_MQTT_OPEN) {
buf[0] = *(int *) ev_data == 0 ? 'X' : 'Y';
2023-06-08 02:38:03 +08:00
} else if (ev == MG_EV_CLOSE) {
buf[0] = 0;
} else if (ev == MG_EV_MQTT_CMD) {
struct mg_mqtt_message *mm = (struct mg_mqtt_message *) ev_data;
if (mm->cmd == MQTT_CMD_SUBACK) {
2023-06-01 08:23:26 +08:00
test_data->flags = flags_subscribed;
} else if (mm->cmd == MQTT_CMD_PUBACK) { // here we assume the broker
test_data->flags = flags_published; // reported no errors,
} else if (mm->cmd == MQTT_CMD_PUBREC) { // either no var header or
test_data->flags |= flags_received; // reason code 0x00
} else if (mm->cmd == MQTT_CMD_PUBREL) {
test_data->flags |= flags_released;
} else if (mm->cmd == MQTT_CMD_PUBCOMP) {
test_data->flags |= flags_completed;
}
2020-12-05 19:26:32 +08:00
} else if (ev == MG_EV_MQTT_MSG) {
struct mg_mqtt_message *mm = (struct mg_mqtt_message *) ev_data;
2023-06-08 02:38:03 +08:00
snprintf(test_data->topic, test_data->topicsize, "%.*s",
(int) mm->topic.len, mm->topic.ptr);
snprintf(buf + 1, test_data->msgsize - 2, "%.*s", (int) mm->data.len,
mm->data.ptr);
if (mm->cmd == MQTT_CMD_PUBLISH && c->is_mqtt5) {
size_t pos = 0;
struct mg_mqtt_prop prop;
ASSERT((pos = mg_mqtt_next_prop(mm, &prop, pos)) > 0);
ASSERT(prop.iv == 10 && prop.id == MQTT_PROP_MESSAGE_EXPIRY_INTERVAL);
2023-06-07 18:47:11 +08:00
ASSERT((pos = mg_mqtt_next_prop(mm, &prop, pos)) > 0);
ASSERT(prop.id == MQTT_PROP_PAYLOAD_FORMAT_INDICATOR);
ASSERT((pos = mg_mqtt_next_prop(mm, &prop, pos)) > 0);
ASSERT(prop.id == MQTT_PROP_CONTENT_TYPE);
ASSERT(strncmp(prop.val.ptr, "test_content_val_2", prop.val.len) == 0 &&
prop.val.len == strlen("test_content_val_2"));
ASSERT((pos = mg_mqtt_next_prop(mm, &prop, pos)) > 0);
ASSERT(prop.id == MQTT_PROP_USER_PROPERTY);
ASSERT(strncmp(prop.key.ptr, "test_key_1", prop.key.len) == 0 &&
prop.key.len == strlen("test_key_1"));
ASSERT(strncmp(prop.val.ptr, "test_value_1", prop.val.len) == 0 &&
prop.val.len == strlen("test_value_1"));
ASSERT((pos = mg_mqtt_next_prop(mm, &prop, pos)) > 0);
ASSERT(prop.id == MQTT_PROP_USER_PROPERTY);
ASSERT(strncmp(prop.key.ptr, "test_key_2", prop.key.len) == 0 &&
prop.key.len == strlen("test_key_2"));
ASSERT(strncmp(prop.val.ptr, "test_value_2", prop.val.len) == 0 &&
prop.val.len == strlen("test_value_2"));
ASSERT((pos = mg_mqtt_next_prop(mm, &prop, pos)) == 0);
}
2020-12-05 19:26:32 +08:00
}
(void) c;
}
static void construct_props(struct mg_mqtt_prop *props) {
props[0].id = MQTT_PROP_MESSAGE_EXPIRY_INTERVAL;
props[0].iv = 10;
props[1].id = MQTT_PROP_USER_PROPERTY;
props[1].key = mg_str("test_key_1");
props[1].val = mg_str("test_value_1");
props[2].id = MQTT_PROP_USER_PROPERTY;
props[2].key = mg_str("test_key_2");
props[2].val = mg_str("test_value_2");
props[3].id = MQTT_PROP_CONTENT_TYPE;
props[3].val = mg_str("test_content_val_2");
2023-06-07 18:47:11 +08:00
props[4].id = MQTT_PROP_PAYLOAD_FORMAT_INDICATOR;
props[4].iv = 1;
}
static void test_mqtt_base(void);
2023-04-01 04:00:05 +08:00
static void test_mqtt_base(void) {
2023-06-08 02:38:03 +08:00
char buf[10] = {0}; // we won't use it
struct mqtt_data test_data = {buf, buf, 10, 10, 0};
struct mg_mgr mgr;
struct mg_connection *c;
const char *url = "mqtt://broker.hivemq.com:1883";
int i;
mg_mgr_init(&mgr);
// Ping the client
c = mg_mqtt_connect(&mgr, url, NULL, mqtt_cb, &test_data);
mg_mqtt_ping(c);
for (i = 0; i < 300 && !(c->is_client && !c->is_connecting); i++)
mg_mgr_poll(&mgr, 10);
ASSERT(c->is_client && !c->is_connecting);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2023-06-08 02:38:03 +08:00
static void check_mqtt_message(struct mg_mqtt_opts *opts,
struct mqtt_data *data, bool enforce) {
if (opts->topic.len != strlen(data->topic) ||
strcmp(opts->topic.ptr, data->topic)) {
MG_INFO(("TOPIC[%s]", data->topic));
if (enforce) ASSERT(0);
}
if (*data->msg != 'X' || opts->message.len != (strlen(&data->msg[1])) ||
strcmp(opts->message.ptr, &data->msg[1])) {
MG_INFO(("MSG[%s]", data->msg));
if (enforce) ASSERT(0);
}
}
// generate a random string ending in three digits taken from current time
static struct mg_str genstring(char *t, unsigned int sz) {
mg_random_str(t, sz - 3);
snprintf(t + sz - 4, 4, "%03u", (unsigned int) mg_millis() % 1000);
return mg_str(t);
}
2023-10-31 03:20:55 +08:00
static void test_mqtt_basic(void) {
char tbuf[16], mbuf[50] = {0}, topic[16];
2023-06-08 02:38:03 +08:00
struct mqtt_data test_data = {tbuf, mbuf, 16, 50, 0};
2020-12-05 19:26:32 +08:00
struct mg_mgr mgr;
struct mg_connection *c;
struct mg_mqtt_opts opts;
2020-12-11 00:36:24 +08:00
const char *url = "mqtt://broker.hivemq.com:1883";
2023-06-01 08:23:26 +08:00
int i, retries;
2020-12-05 19:26:32 +08:00
2023-10-31 03:20:55 +08:00
// Connect with empty client ID, no options, ergo MQTT = 3.1.1
2023-09-22 15:12:09 +08:00
mg_mgr_init(&mgr);
c = mg_mqtt_connect(&mgr, url, NULL, mqtt_cb, &test_data);
2023-06-08 02:38:03 +08:00
for (i = 0; i < 300 && mbuf[0] == 0; i++) mg_mgr_poll(&mgr, 10);
if (mbuf[0] != 'X') MG_INFO(("[%s]", mbuf));
ASSERT(mbuf[0] == 'X');
2023-06-01 08:23:26 +08:00
ASSERT(test_data.flags == 0);
2023-06-01 08:23:26 +08:00
// Subscribe with QoS1
opts.topic = genstring(topic, sizeof(topic));
2023-06-08 02:38:03 +08:00
opts.qos = 1;
mg_mqtt_sub(c, &opts);
2023-06-01 08:23:26 +08:00
for (i = 0; i < 500 && test_data.flags == 0; i++) mg_mgr_poll(&mgr, 10);
ASSERT(test_data.flags == flags_subscribed);
test_data.flags = 0;
2023-06-01 08:23:26 +08:00
// Publish with QoS0 to subscribed topic and check reception
2023-06-08 02:38:03 +08:00
// keep former opts.topic
opts.message = mg_str("hi0"), opts.qos = 0, opts.retain = false;
mg_mqtt_pub(c, &opts);
2023-06-08 02:38:03 +08:00
for (i = 0; i < 500 && mbuf[1] == 0; i++) mg_mgr_poll(&mgr, 10);
ASSERT(!(test_data.flags & flags_published)); // No PUBACK for QoS0
check_mqtt_message(&opts, &test_data, false); // We may not get the msg
memset(mbuf + 1, 0, sizeof(mbuf) - 1);
2023-06-01 08:23:26 +08:00
test_data.flags = 0;
// Publish with QoS1 to subscribed topic and check reception
2023-06-08 02:38:03 +08:00
// keep former opts.topic
opts.message = mg_str("hi1"), opts.qos = 1, opts.retain = false;
2023-06-06 21:36:58 +08:00
retries = 0; // don't do retries for test speed
2023-06-08 02:38:03 +08:00
do { // retry on failure after an expected timeout
2023-06-01 08:23:26 +08:00
mg_mqtt_pub(c, &opts);
for (i = 0; i < 500 && test_data.flags == 0; i++) mg_mgr_poll(&mgr, 10);
} while (test_data.flags == 0 && retries--);
ASSERT(test_data.flags == flags_published);
2023-06-08 02:38:03 +08:00
for (i = 0; i < 500 && mbuf[1] == 0; i++) mg_mgr_poll(&mgr, 10);
check_mqtt_message(&opts, &test_data, true);
memset(mbuf + 1, 0, sizeof(mbuf) - 1);
2023-06-01 08:23:26 +08:00
test_data.flags = 0;
2023-10-31 03:20:55 +08:00
// Clean Disconnect !
2023-06-08 02:38:03 +08:00
mg_mqtt_disconnect(c, NULL);
for (i = 0; i < 10 && mbuf[0] != 0; i++) mg_mgr_poll(&mgr, 10);
2023-06-01 08:23:26 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
2023-10-31 03:20:55 +08:00
}
static void test_mqtt_ver(uint8_t mqtt_version) {
char tbuf[16], mbuf[50] = {0}, client_id[16], topic[16];
struct mqtt_data test_data = {tbuf, mbuf, 16, 50, 0};
struct mg_mgr mgr;
struct mg_connection *c;
struct mg_mqtt_opts opts;
struct mg_mqtt_prop properties[5];
const char *url = "mqtt://broker.hivemq.com:1883";
int i, retries;
2020-12-05 19:26:32 +08:00
2023-10-31 03:20:55 +08:00
// Connect with options: version, clean session, last will, keepalive
2023-06-08 02:38:03 +08:00
// time. Don't set retain, some runners are not random
2023-06-01 08:23:26 +08:00
test_data.flags = 0;
2023-06-08 02:38:03 +08:00
memset(mbuf, 0, sizeof(mbuf));
2020-12-05 19:26:32 +08:00
memset(&opts, 0, sizeof(opts));
2023-06-01 08:23:26 +08:00
mg_mgr_init(&mgr);
2023-06-08 02:38:03 +08:00
opts.clean = true, opts.qos = 1, opts.retain = false, opts.keepalive = 20;
2022-06-28 18:31:13 +08:00
opts.version = mqtt_version;
opts.topic = genstring(topic, sizeof(topic));
opts.message = mg_str("mg_will_messsage");
opts.client_id = genstring(client_id, sizeof(client_id));
c = mg_mqtt_connect(&mgr, url, &opts, mqtt_cb, &test_data);
2023-06-08 02:38:03 +08:00
for (i = 0; i < 300 && mbuf[0] == 0; i++) mg_mgr_poll(&mgr, 10);
if (mbuf[0] != 'X') MG_INFO(("[%s]", mbuf));
ASSERT(mbuf[0] == 'X');
2023-06-01 08:23:26 +08:00
ASSERT(test_data.flags == 0);
2023-06-01 08:23:26 +08:00
// Subscribe with QoS2 (reception downgrades to published QoS)
opts.topic = genstring(topic, sizeof(topic));
2023-06-08 02:38:03 +08:00
opts.qos = 2;
mg_mqtt_sub(c, &opts);
2023-06-01 08:23:26 +08:00
for (i = 0; i < 500 && test_data.flags == 0; i++) mg_mgr_poll(&mgr, 10);
ASSERT(test_data.flags == flags_subscribed);
test_data.flags = 0;
2023-06-01 08:23:26 +08:00
// Publish with QoS1 to subscribed topic and check reception
2023-06-08 02:38:03 +08:00
// keep former opts.topic
opts.message = mg_str("hi1"), opts.qos = 1, opts.retain = false;
if (mqtt_version == 5) {
opts.props = properties;
2023-06-07 18:47:11 +08:00
opts.num_props = 5;
construct_props(properties);
}
2023-06-06 21:36:58 +08:00
retries = 0; // don't do retries for test speed
2023-06-08 02:38:03 +08:00
do { // retry on failure after an expected timeout
2023-06-01 08:23:26 +08:00
mg_mqtt_pub(c, &opts);
for (i = 0; i < 500 && test_data.flags == 0; i++) mg_mgr_poll(&mgr, 10);
} while (test_data.flags == 0 && retries--);
ASSERT(test_data.flags == flags_published);
2023-06-08 02:38:03 +08:00
for (i = 0; i < 500 && mbuf[1] == 0; i++) mg_mgr_poll(&mgr, 10);
check_mqtt_message(&opts, &test_data, true);
memset(mbuf + 1, 0, sizeof(mbuf) - 1);
2023-06-01 08:23:26 +08:00
test_data.flags = 0;
2023-06-01 08:23:26 +08:00
// Publish with QoS2 to subscribed topic and check (simultaneous) reception
2023-06-08 02:38:03 +08:00
// keep former opts.topic
opts.message = mg_str("hi2"), opts.qos = 2, opts.retain = false;
2023-06-01 08:23:26 +08:00
if (mqtt_version == 5) {
opts.props = properties;
2023-06-07 18:47:11 +08:00
opts.num_props = 5;
2023-06-01 08:23:26 +08:00
construct_props(properties);
}
2023-06-06 21:36:58 +08:00
retries = 0; // don't do retries for test speed
2023-06-08 02:38:03 +08:00
do { // retry on failure after an expected timeout
2023-06-01 08:23:26 +08:00
mg_mqtt_pub(c, &opts);
for (i = 0; i < 500 && !(test_data.flags & flags_received); i++)
mg_mgr_poll(&mgr, 10);
} while (!(test_data.flags & flags_received) && retries--);
ASSERT(test_data.flags & flags_received);
test_data.flags &= ~flags_received;
// Mongoose sent PUBREL, wait for PUBCOMP
for (i = 0; i < 500 && !(test_data.flags & flags_completed); i++)
mg_mgr_poll(&mgr, 10);
// TODO(): retry sending PUBREL on failure after an expected timeout
2023-06-03 07:32:21 +08:00
// or broker sends PUBREC again
2023-06-01 08:23:26 +08:00
ASSERT(test_data.flags & flags_completed);
2023-06-08 02:38:03 +08:00
for (i = 0; i < 500 && mbuf[1] == 0; i++) mg_mgr_poll(&mgr, 10);
check_mqtt_message(&opts, &test_data, true);
2023-06-01 08:23:26 +08:00
for (i = 0; i < 500 && !(test_data.flags & flags_released); i++)
mg_mgr_poll(&mgr, 10);
ASSERT(test_data.flags & flags_released); // Mongoose sent PUBCOMP
2023-06-08 02:38:03 +08:00
memset(mbuf + 1, 0, sizeof(mbuf) - 1);
2023-06-01 08:23:26 +08:00
test_data.flags = 0;
2023-06-08 02:38:03 +08:00
// dirty disconnect
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2022-06-28 18:31:13 +08:00
static void test_mqtt(void) {
test_mqtt_base();
2023-10-31 03:20:55 +08:00
test_mqtt_basic();
2022-06-28 18:31:13 +08:00
test_mqtt_ver(4);
test_mqtt_ver(5);
2022-06-28 18:31:13 +08:00
}
static void eh1(struct mg_connection *c, int ev, void *ev_data) {
struct mg_tls_opts *topts = (struct mg_tls_opts *) c->fn_data;
if (ev == MG_EV_ACCEPT && topts != NULL) mg_tls_init(c, topts);
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
MG_INFO(("[%.*s %.*s] message len %d", (int) hm->method.len, hm->method.ptr,
(int) hm->uri.len, hm->uri.ptr, (int) hm->message.len));
2020-12-05 19:26:32 +08:00
if (mg_http_match_uri(hm, "/foo/*")) {
2020-12-18 06:45:22 +08:00
mg_http_reply(c, 200, "", "uri: %.*s", hm->uri.len - 5, hm->uri.ptr + 5);
2020-12-05 19:26:32 +08:00
} else if (mg_http_match_uri(hm, "/ws")) {
mg_ws_upgrade(c, hm, NULL);
2020-12-05 19:26:32 +08:00
} else if (mg_http_match_uri(hm, "/body")) {
2020-12-18 06:45:22 +08:00
mg_http_reply(c, 200, "", "%.*s", (int) hm->body.len, hm->body.ptr);
2020-12-05 19:26:32 +08:00
} else if (mg_http_match_uri(hm, "/bar")) {
2020-12-18 06:45:22 +08:00
mg_http_reply(c, 404, "", "not found");
} else if (mg_http_match_uri(hm, "/no_reason")) {
mg_printf(c, "%s", "HTTP/1.0 200\r\nContent-Length: 2\r\n\r\nok");
2020-12-05 19:26:32 +08:00
} else if (mg_http_match_uri(hm, "/badroot")) {
struct mg_http_serve_opts sopts;
memset(&sopts, 0, sizeof(sopts));
sopts.root_dir = "/BAAADDD!";
mg_http_serve_dir(c, hm, &sopts);
2020-12-13 04:38:09 +08:00
} else if (mg_http_match_uri(hm, "/creds")) {
char user[100], pass[100];
mg_http_creds(hm, user, sizeof(user), pass, sizeof(pass));
2020-12-18 06:45:22 +08:00
mg_http_reply(c, 200, "", "[%s]:[%s]", user, pass);
2020-12-13 04:38:09 +08:00
} else if (mg_http_match_uri(hm, "/upload")) {
mg_http_upload(c, hm, &mg_fs_posix, ".", 99999);
c->is_hexdumping = 1;
2020-12-13 04:38:09 +08:00
} else if (mg_http_match_uri(hm, "/test/")) {
struct mg_http_serve_opts sopts;
memset(&sopts, 0, sizeof(sopts));
sopts.root_dir = ".";
sopts.extra_headers = "A: B\r\nC: D\r\n";
mg_http_serve_dir(c, hm, &sopts);
2021-07-29 04:11:07 +08:00
} else if (mg_http_match_uri(hm, "/servefile")) {
struct mg_http_serve_opts sopts;
memset(&sopts, 0, sizeof(sopts));
sopts.mime_types = "foo=a/b,txt=c/d";
mg_http_serve_file(c, hm, "test/data/a.txt", &sopts);
2020-12-05 19:26:32 +08:00
} else {
struct mg_http_serve_opts sopts;
memset(&sopts, 0, sizeof(sopts));
sopts.root_dir = "./test/data";
sopts.ssi_pattern = "#.shtml";
sopts.extra_headers = "C: D\r\n";
mg_http_serve_dir(c, hm, &sopts);
}
} else if (ev == MG_EV_WS_OPEN) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
ASSERT(mg_strcmp(hm->uri, mg_str("/ws")) == 0);
mg_ws_send(c, "opened", 6, WEBSOCKET_OP_BINARY);
2020-12-13 04:38:09 +08:00
} else if (ev == MG_EV_WS_MSG) {
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
mg_ws_send(c, wm->data.ptr, wm->data.len, WEBSOCKET_OP_BINARY);
}
}
2020-12-05 19:26:32 +08:00
struct fetch_data {
char *buf;
int code, closed;
};
static void fcb(struct mg_connection *c, int ev, void *ev_data) {
struct fetch_data *fd = (struct fetch_data *) c->fn_data;
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
snprintf(fd->buf, FETCH_BUF_SIZE, "%.*s", (int) hm->message.len,
hm->message.ptr);
fd->code = atoi(hm->uri.ptr);
fd->closed = 1;
c->is_closing = 1;
(void) c;
2022-08-03 22:07:16 +08:00
} else if (ev == MG_EV_CLOSE) {
fd->closed = 1;
}
}
2020-12-05 19:26:32 +08:00
static int fetch(struct mg_mgr *mgr, char *buf, const char *url,
const char *fmt, ...) {
struct fetch_data fd = {buf, 0, 0};
int i;
2023-07-25 20:07:28 +08:00
struct mg_connection *c = NULL;
2020-12-05 19:26:32 +08:00
va_list ap;
c = mg_http_connect(mgr, url, fcb, &fd);
ASSERT(c != NULL);
if (c != NULL && mg_url_is_ssl(url)) {
2020-12-05 19:26:32 +08:00
struct mg_tls_opts opts;
2023-09-12 03:33:43 +08:00
memset(&opts, 0, sizeof(opts)); // read CA from packed_fs
opts.ca = mg_unpacked("/test/data/ca.pem");
2020-12-05 19:26:32 +08:00
if (strstr(url, "127.0.0.1") != NULL) {
// Local connection, use self-signed certificates
opts.ca = mg_str(s_tls_ca);
// opts.cert = mg_str(s_tls_cert);
// opts.key = mg_str(s_tls_key);
2020-12-05 19:26:32 +08:00
}
mg_tls_init(c, &opts);
}
2022-08-03 22:07:16 +08:00
// c->is_hexdumping = 1;
2020-12-05 19:26:32 +08:00
va_start(ap, fmt);
2022-09-23 15:59:02 +08:00
mg_vprintf(c, fmt, &ap);
2020-12-05 19:26:32 +08:00
va_end(ap);
buf[0] = '\0';
2022-08-03 22:07:16 +08:00
for (i = 0; i < 50 && buf[0] == '\0'; i++) mg_mgr_poll(mgr, 1);
2020-12-05 19:26:32 +08:00
if (!fd.closed) c->is_closing = 1;
mg_mgr_poll(mgr, 1);
return fd.code;
}
static struct mg_http_message gethm(const char *buf) {
2020-12-05 19:26:32 +08:00
struct mg_http_message hm;
memset(&hm, 0, sizeof(hm));
mg_http_parse(buf, strlen(buf), &hm);
return hm;
}
static int cmpbody(const char *buf, const char *str) {
struct mg_str s = mg_str(str);
struct mg_http_message hm = gethm(buf);
size_t len = strlen(buf);
// mg_http_parse(buf, len, &hm);
if (hm.body.len > len) hm.body.len = len - (size_t) (hm.body.ptr - buf);
return mg_strcmp(hm.body, s);
}
static bool cmpheader(const char *buf, const char *name, const char *value) {
struct mg_http_message hm = gethm(buf);
struct mg_str *h = mg_http_get_header(&hm, name);
return h != NULL && mg_strcmp(*h, mg_str(value)) == 0;
}
static void wcb(struct mg_connection *c, int ev, void *ev_data) {
int *p = (int *) c->fn_data;
if (ev == MG_EV_WS_OPEN) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
struct mg_str *wsproto = mg_http_get_header(hm, "Sec-WebSocket-Protocol");
ASSERT(wsproto != NULL);
mg_ws_printf(c, WEBSOCKET_OP_BINARY, "%.3s", "boo!!!!");
mg_ws_printf(c, WEBSOCKET_OP_BINARY, "%s", "foobar");
mg_ws_send(c, "", 0, WEBSOCKET_OP_PING);
2022-04-13 18:14:57 +08:00
p[0] += 100;
} else if (ev == MG_EV_WS_MSG) {
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
2022-04-13 18:14:57 +08:00
if (mg_strstr(wm->data, mg_str("foobar")))
mg_ws_send(c, "", 0, WEBSOCKET_OP_CLOSE);
2022-04-13 18:14:57 +08:00
if (mg_strstr(wm->data, mg_str("boo"))) p[0] += 2;
if (mg_strstr(wm->data, mg_str("foobar"))) p[0] += 3;
} else if (ev == MG_EV_CLOSE) {
2022-04-13 18:14:57 +08:00
p[0] += 10;
}
}
static void ew2(struct mg_connection *c, int ev, void *ev_data) {
size_t size = 65 * 1024 + 737;
if (ev == MG_EV_WS_OPEN) {
2022-08-13 21:57:15 +08:00
char *msg = (char *) calloc(1, size + 1);
memset(msg, 'A', size);
mg_ws_printf(c, WEBSOCKET_OP_TEXT, "%s", msg);
free(msg);
} else if (ev == MG_EV_WS_MSG) {
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
if (wm->data.len == 6) {
// Ignore the "opened" message from server
} else {
size_t ok = 1, i;
ASSERT(wm->data.len == size);
for (i = 0; i < size; i++) {
if (wm->data.ptr[i] != 'A') ok = 0;
}
ASSERT(ok == 1);
*(int *) c->fn_data = 1;
}
}
}
static void test_ws(void) {
char buf[FETCH_BUF_SIZE];
const char *url = "ws://LOCALHOST:12343/ws";
struct mg_mgr mgr;
int i, done = 0;
mg_mgr_init(&mgr);
ASSERT(mg_http_listen(&mgr, url, eh1, NULL) != NULL);
mg_ws_connect(&mgr, url, wcb, &done, "%s", "Sec-WebSocket-Protocol: meh\r\n");
for (i = 0; i < 30; i++) mg_mgr_poll(&mgr, 1);
// MG_INFO(("--> %d", done));
2022-04-13 18:14:57 +08:00
ASSERT(done == 115);
// Test that non-WS requests fail
ASSERT(fetch(&mgr, buf, url, "GET /ws HTTP/1.0\r\n\n") == 426);
// Test large WS frames, over 64k
done = 0;
mg_ws_connect(&mgr, url, ew2, &done, NULL);
for (i = 0; i < 1000 && done == 0; i++) mg_mgr_poll(&mgr, 1);
ASSERT(done == 1);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
static void eh9(struct mg_connection *c, int ev, void *ev_data) {
2020-12-12 06:58:50 +08:00
if (ev == MG_EV_ERROR) {
ASSERT(!strcmp((char *) ev_data, "socket error"));
*(int *) c->fn_data = 7;
2020-12-12 06:58:50 +08:00
}
2020-12-12 01:35:58 +08:00
(void) c;
}
2020-12-05 19:26:32 +08:00
static void test_http_server(void) {
struct mg_mgr mgr;
const char *url = "http://127.0.0.1:12346";
char buf[FETCH_BUF_SIZE];
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
mg_http_listen(&mgr, url, eh1, NULL);
2020-12-05 19:26:32 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
// Invalid header: failure
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\nA B\n\n") == 0);
2021-01-22 01:14:08 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /%%61.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
// Responses with missing reason phrase must also work
ASSERT(fetch(&mgr, buf, url, "GET /no_reason HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "ok") == 0);
2021-07-23 03:16:50 +08:00
// Fetch file with unicode chars in filename
ASSERT(fetch(&mgr, buf, url, "GET /київ.txt HTTP/1.0\n\n") == 200);
MG_INFO(("%s", buf));
2021-07-23 03:16:50 +08:00
ASSERT(cmpbody(buf, "є\n") == 0);
2023-06-30 17:25:58 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /../fuzz.c HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url, "GET /.%%2e/fuzz.c HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url, "GET /.%%2e%%2ffuzz.c HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url, "GET /..%%2f%%20fuzz.c HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url, "GET /..%%2ffuzz.c%%20 HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url, "GET /dredir HTTP/1.0\n\n") == 301);
ASSERT(cmpheader(buf, "Location", "/dredir/"));
ASSERT(fetch(&mgr, buf, url, "GET /dredir/ HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hi\n") == 0);
ASSERT(fetch(&mgr, buf, url,
"GET /dredirgz/ HTTP/1.0\n"
"Accept-Encoding: gzip\n\n") == 200);
ASSERT(cmpheader(buf, "Content-Type", "text/html; charset=utf-8"));
ASSERT(cmpheader(buf, "Content-Encoding", "gzip"));
ASSERT(fetch(&mgr, buf, url, "GET /gzip.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hi\n") == 0);
ASSERT(gethm(buf).body.len == 3);
ASSERT(cmpheader(buf, "Content-Encoding", "gzip") == false);
ASSERT(fetch(&mgr, buf, url,
"GET /gzip.txt HTTP/1.0\n"
"Accept-Encoding: foo,gzip\n\n") == 200);
mg_hexdump(buf, strlen(buf));
ASSERT(cmpheader(buf, "Content-Encoding", "gzip") == true);
ASSERT(gethm(buf).body.len == 23);
2023-06-30 17:25:58 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /..ddot HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url, "GET /..ddot/ HTTP/1.0\n\n") == 400);
ASSERT(fetch(&mgr, buf, url,
"GET /a.txt HTTP/1.0\n"
"Content-Length: -123\n\n") == 0);
ASSERT(fetch(&mgr, buf, url,
"POST /a.txt HTTP/1.0\n"
"Content-Length: -123\n\n") == 0);
ASSERT(fetch(&mgr, buf, url,
"GET /a.txt HTTP/1.0\n"
"Content-Length: 19000000000000000000\n\n") == 0);
ASSERT(fetch(&mgr, buf, url,
"POST /a.txt HTTP/1.0\n"
"Content-Length: 19000000000000000000\n\n") == 0);
2023-06-28 22:01:59 +08:00
ASSERT(fetch(&mgr, buf, url,
"GET /a.txt HTTP/1.0\n"
":\n" // truncated header
"Content-Length: 1\n\n") == 0);
{
2021-07-29 21:21:20 +08:00
extern char *mg_http_etag(char *, size_t, size_t, time_t);
struct mg_http_message hm;
2020-12-05 19:26:32 +08:00
char etag[100];
2021-07-29 21:21:20 +08:00
size_t size = 0;
time_t mtime = 0;
ASSERT(mg_fs_posix.st("./test/data/a.txt", &size, &mtime) != 0);
2021-07-29 21:21:20 +08:00
ASSERT(mg_http_etag(etag, sizeof(etag), size, mtime) == etag);
2020-12-05 19:26:32 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\nIf-None-Match: %s\n\n",
etag) == 304);
ASSERT(mg_http_parse(buf, strlen(buf), &hm) > 0);
MG_INFO(("%s", buf));
ASSERT(mg_http_get_header(&hm, "C") != NULL);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "C"), mg_str("D")) == 0);
}
2021-07-29 04:11:07 +08:00
// Text mime type override
ASSERT(fetch(&mgr, buf, url, "GET /servefile HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
{
struct mg_http_message hm;
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_http_get_header(&hm, "Content-Type") != NULL);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "Content-Type"), mg_str("c/d")) ==
0);
}
2020-12-05 19:26:32 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /foo/1 HTTP/1.0\r\n\n") == 200);
// MG_INFO(("%d %.*s", (int) hm.len, (int) hm.len, hm.buf));
2020-12-05 19:26:32 +08:00
ASSERT(cmpbody(buf, "uri: 1") == 0);
2020-12-05 19:26:32 +08:00
ASSERT(fetch(&mgr, buf, url, "%s",
"POST /body HTTP/1.1\r\n"
"Content-Length: 4\r\n\r\nkuku") == 200);
ASSERT(cmpbody(buf, "kuku") == 0);
ASSERT(fetch(&mgr, buf, url, "GET /ssi HTTP/1.1\r\n\r\n") == 301);
ASSERT(fetch(&mgr, buf, url, "GET /ssi/ HTTP/1.1\r\n\r\n") == 200);
ASSERT(cmpbody(buf,
"this is index\n"
"this is nested\n\n"
"this is f1\n\n\n\n"
"recurse\n\n"
"recurse\n\n"
"recurse\n\n"
"recurse\n\n"
"recurse\n\n") == 0);
{
struct mg_http_message hm;
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_http_get_header(&hm, "Content-Length") != NULL);
ASSERT(mg_http_get_header(&hm, "Content-Type") != NULL);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "Content-Type"),
mg_str("text/html; charset=utf-8")) == 0);
}
2022-04-05 19:54:20 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /badroot HTTP/1.0\r\n\n") == 404);
// ASSERT(cmpbody(buf, "Invalid web root [/BAAADDD!]\n") == 0);
{
struct mg_str data = mg_file_read(&mg_fs_posix, "./test/data/ca.pem");
2020-12-05 19:26:32 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /ca.pem HTTP/1.0\r\n\n") == 200);
ASSERT(cmpbody(buf, data.ptr) == 0);
free((void *) data.ptr);
}
2020-12-15 19:58:30 +08:00
{
// Test mime type
struct mg_http_message hm;
ASSERT(fetch(&mgr, buf, url, "GET /empty.js HTTP/1.0\r\n\n") == 200);
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_http_get_header(&hm, "Content-Type") != NULL);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "Content-Type"),
2021-07-14 02:24:46 +08:00
mg_str("text/javascript; charset=utf-8")) == 0);
2020-12-15 19:58:30 +08:00
}
2020-12-12 01:35:58 +08:00
{
// Test connection refused
2020-12-12 06:58:50 +08:00
int i, errored = 0;
mg_connect(&mgr, "tcp://127.0.0.1:55117", eh9, &errored);
// Give it a couple of seconds, see #1605
for (i = 0; i < 200 && errored == 0; i++) mg_mgr_poll(&mgr, 10);
2022-06-30 02:19:19 +08:00
MG_INFO(("errored: %d, expected: 7", errored));
2020-12-12 06:58:50 +08:00
ASSERT(errored == 7);
2020-12-12 01:35:58 +08:00
}
2020-12-13 04:38:09 +08:00
// Directory listing
fetch(&mgr, buf, url, "GET /test/ HTTP/1.0\n\n");
ASSERT(fetch(&mgr, buf, url, "GET /test/ HTTP/1.0\n\n") == 200);
ASSERT(mg_strstr(mg_str(buf), mg_str(">Index of /test/<")) != NULL);
ASSERT(mg_strstr(mg_str(buf), mg_str(">fuzz.c<")) != NULL);
{
// Credentials
struct mg_http_message hm;
ASSERT(fetch(&mgr, buf, url, "%s",
"GET /creds?access_token=x HTTP/1.0\r\n\r\n") == 200);
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_strcmp(hm.body, mg_str("[]:[x]")) == 0);
ASSERT(fetch(&mgr, buf, url, "%s",
"GET /creds HTTP/1.0\r\n"
"Authorization: Bearer x\r\n\r\n") == 200);
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_strcmp(hm.body, mg_str("[]:[x]")) == 0);
ASSERT(fetch(&mgr, buf, url, "%s",
"GET /creds HTTP/1.0\r\n"
"Authorization: Basic Zm9vOmJhcg==\r\n\r\n") == 200);
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_strcmp(hm.body, mg_str("[foo]:[bar]")) == 0);
ASSERT(fetch(&mgr, buf, url, "%s",
"GET /creds HTTP/1.0\r\n"
"Cookie: blah; access_token=hello\r\n\r\n") == 200);
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_strcmp(hm.body, mg_str("[]:[hello]")) == 0);
}
{
// Test upload
struct mg_str s;
2020-12-13 04:38:09 +08:00
remove("uploaded.txt");
s = mg_file_read(&mg_fs_posix, "uploaded.txt");
ASSERT(s.ptr == NULL);
2020-12-13 04:38:09 +08:00
ASSERT(fetch(&mgr, buf, url,
"POST /upload HTTP/1.0\n"
"Content-Length: 1\n\nx") == 400);
ASSERT(fetch(&mgr, buf, url,
"POST /upload?file=uploaded.txt HTTP/1.0\r\n"
2020-12-13 04:38:09 +08:00
"Content-Length: 5\r\n"
"\r\nhello") == 200);
ASSERT(fetch(&mgr, buf, url,
"POST /upload?file=uploaded.txt&offset=5 HTTP/1.0\r\n"
"Content-Length: 6\r\n"
"\r\n\nworld") == 200);
s = mg_file_read(&mg_fs_posix, "uploaded.txt");
ASSERT(s.ptr != NULL);
ASSERT(strcmp(s.ptr, "hello\nworld") == 0);
free((void *) s.ptr);
2020-12-13 04:38:09 +08:00
remove("uploaded.txt");
}
{
// Test upload directory traversal
struct mg_str s;
remove("uploaded.txt");
s = mg_file_read(&mg_fs_posix, "uploaded.txt");
ASSERT(s.ptr == NULL);
ASSERT(fetch(&mgr, buf, url,
"POST /upload?file=../uploaded.txt HTTP/1.0\r\n"
"Content-Length: 5\r\n"
"\r\nhello") == 400);
s = mg_file_read(&mg_fs_posix, "uploaded.txt");
ASSERT(s.ptr == NULL);
}
2020-12-05 19:26:32 +08:00
// HEAD request
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\n\n") == 200);
ASSERT(fetch(&mgr, buf, url, "HEAD /a.txt HTTP/1.0\n\n") == 200);
// Pre-compressed files
{
struct mg_http_message hm;
ASSERT(fetch(&mgr, buf, url,
"HEAD /hello.txt HTTP/1.0\n"
"Accept-Encoding: gzip\n\n") == 200);
mg_http_parse(buf, strlen(buf), &hm);
ASSERT(mg_http_get_header(&hm, "Content-Encoding") != NULL);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "Content-Encoding"),
mg_str("gzip")) == 0);
}
2020-12-22 18:16:31 +08:00
#if MG_ENABLE_IPV6
{
const char *url6 = "http://[::1]:12366";
2020-12-22 18:16:31 +08:00
ASSERT(mg_http_listen(&mgr, url6, eh1, NULL) != NULL);
ASSERT(fetch(&mgr, buf, url6, "GET /a.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
}
#endif
// #2552, reject chunks with no length
ASSERT(fetch(&mgr, buf, url,
"POST / HTTP/1.1\r\n"
"Transfer-Encoding: chunked\r\n\r\n"
"1\r\n"
"Z\r\n"
"?\r\n"
"\r\n") == 0);
ASSERT(fetch(&mgr, buf, url,
"POST / HTTP/1.1\r\n"
"Transfer-Encoding: chunked\r\n\r\n"
"1\r\n"
"Z\r\n"
"\r\n"
"\r\n") == 0);
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
static void h4(struct mg_connection *c, int ev, void *ev_data) {
2022-06-01 06:44:03 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
MG_INFO(("[%.*s %.*s] message len %d", (int) hm->method.len, hm->method.ptr,
(int) hm->uri.len, hm->uri.ptr, (int) hm->message.len));
if (mg_http_match_uri(hm, "/a/#")) {
struct mg_http_serve_opts opts;
memset(&opts, 0, sizeof(opts));
opts.root_dir = "/a=./test/data";
opts.page404 = "./test/data/404.html"; // existing 404 page
mg_http_serve_dir(c, hm, &opts);
} else if (mg_http_match_uri(hm, "/b/#")) {
struct mg_http_serve_opts opts;
memset(&opts, 0, sizeof(opts));
opts.root_dir = "/b=./test/data";
opts.page404 = "./test/data/nooooo.html"; // non-existing 404 page
mg_http_serve_dir(c, hm, &opts);
} else { // null 404 page
struct mg_http_serve_opts opts;
memset(&opts, 0, sizeof(opts));
opts.root_dir = "./test/data";
mg_http_serve_dir(c, hm, &opts);
}
}
}
static void test_http_404(void) {
struct mg_mgr mgr;
const char *url = "http://127.0.0.1:22343";
char buf[FETCH_BUF_SIZE];
mg_mgr_init(&mgr);
mg_http_listen(&mgr, url, h4, NULL);
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
ASSERT(fetch(&mgr, buf, url, "GET /a/a.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
ASSERT(fetch(&mgr, buf, url, "GET /b/a.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
ASSERT(fetch(&mgr, buf, url, "GET /xx.txt HTTP/1.0\n\n") == 404);
ASSERT(cmpbody(buf, "Not found\n") == 0);
ASSERT(fetch(&mgr, buf, url, "GET /a/xx.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "boo\n") == 0);
2024-03-06 08:57:31 +08:00
{ // txt requested, existent 404 is html
struct mg_http_message hm;
struct mg_str *cl;
mg_http_parse(buf, strlen(buf), &hm);
cl = mg_http_get_header(&hm, "Content-Type");
ASSERT(cl != NULL);
2024-03-06 08:57:31 +08:00
if (cl->len > 9) cl->len = 9; // restrict to text/html len max
ASSERT(mg_strcmp(*cl, mg_str("text/html")) == 0);
}
2022-06-01 06:44:03 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /b/xx.txt HTTP/1.0\n\n") == 404);
ASSERT(cmpbody(buf, "Not found\n") == 0);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2020-12-05 19:26:32 +08:00
static void test_tls(void) {
2023-09-19 23:08:02 +08:00
// return;
2023-07-25 20:07:28 +08:00
#if MG_TLS
2020-12-05 19:26:32 +08:00
struct mg_mgr mgr;
struct mg_connection *c;
const char *url = "https://127.0.0.1:12347";
char buf[FETCH_BUF_SIZE];
struct mg_tls_opts opts;
struct mg_str data = mg_unpacked("/Makefile");
memset(&opts, 0, sizeof(opts));
// opts.ca = mg_str(s_tls_ca);
opts.cert = mg_str(s_tls_cert);
opts.key = mg_str(s_tls_key);
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
c = mg_http_listen(&mgr, url, eh1, &opts);
2020-12-05 19:26:32 +08:00
ASSERT(c != NULL);
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\n\n") == 200);
// MG_INFO(("%s", buf));
2020-12-05 19:26:32 +08:00
ASSERT(cmpbody(buf, "hello\n") == 0);
// POST a larger file, make sure we drain TLS buffers and read all, #2619
ASSERT(data.ptr != NULL && data.len > 0);
ASSERT(fetch(&mgr, buf, url,
"POST /foo/bar HTTP/1.0\n"
"Content-Length: %lu\n\n"
"%s",
data.len, data.ptr) == 200);
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
#endif
}
static void f3(struct mg_connection *c, int ev, void *ev_data) {
int *ok = (int *) c->fn_data;
// MG_INFO(("%d", ev));
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_CONNECT) {
// c->is_hexdumping = 1;
2022-04-12 16:38:25 +08:00
mg_printf(c, "GET /%s HTTP/1.0\r\nHost: %s\r\n\r\n",
c->rem.is_ip6 ? "" : "/robots.txt",
2022-02-23 06:00:55 +08:00
c->rem.is_ip6 ? "ipv6.google.com" : "cesanta.com");
2020-12-05 19:26:32 +08:00
} else if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
// MG_INFO(("-->[%.*s]", (int) hm->message.len, hm->message.ptr));
2020-12-05 19:26:32 +08:00
// ASSERT(mg_vcmp(&hm->method, "HTTP/1.1") == 0);
// ASSERT(mg_vcmp(&hm->uri, "301") == 0);
2022-02-20 00:19:51 +08:00
*ok = mg_http_status(hm);
2020-12-05 19:26:32 +08:00
} else if (ev == MG_EV_CLOSE) {
if (*ok == 0) *ok = 888;
} else if (ev == MG_EV_ERROR) {
if (*ok == 0) *ok = 777;
}
2020-12-05 19:26:32 +08:00
}
2020-12-05 19:26:32 +08:00
static void test_http_client(void) {
struct mg_mgr mgr;
2023-07-25 20:07:28 +08:00
struct mg_connection *c = NULL;
const char *url = "http://cesanta.com";
2020-12-05 19:26:32 +08:00
int i, ok = 0;
struct mg_tls_opts opts;
2023-07-25 22:55:01 +08:00
memset(&opts, 0, sizeof(opts));
opts.ca = mg_unpacked("/test/data/ca.pem");
opts.name = mg_url_host(url);
ASSERT(opts.ca.len > 0);
ASSERT(opts.name.len > 0);
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
c = mg_http_connect(&mgr, url, f3, &ok);
2020-12-05 19:26:32 +08:00
ASSERT(c != NULL);
for (i = 0; i < 500 && ok <= 0; i++) mg_mgr_poll(&mgr, 1);
MG_INFO(("%d", ok));
2020-12-05 19:26:32 +08:00
ASSERT(ok == 301);
c->is_closing = 1;
2024-02-14 20:32:26 +08:00
ASSERT((c->loc.ip[0] != 0)); // Make sure that c->loc address is populated
2020-12-05 19:26:32 +08:00
mg_mgr_poll(&mgr, 0);
ok = 0;
2023-07-25 20:07:28 +08:00
#if MG_TLS
2023-09-12 03:33:43 +08:00
c = mg_http_connect(&mgr, "https://cesanta.com", f3, &ok);
ASSERT(c != NULL);
mg_tls_init(c, &opts);
for (i = 0; i < 1500 && ok <= 0; i++) mg_mgr_poll(&mgr, 1);
2023-09-12 03:33:43 +08:00
ASSERT(ok == 200);
c->is_closing = 1;
mg_mgr_poll(&mgr, 1);
// Test failed host validation
c = mg_http_connect(&mgr, "https://cesanta.com", f3, &ok);
ASSERT(c != NULL);
opts.name = mg_str("dummy"); // Set some invalid hostname value
mg_tls_init(c, &opts);
ok = 0;
for (i = 0; i < 500 && ok <= 0; i++) mg_mgr_poll(&mgr, 10);
MG_INFO(("OK: %d", ok));
ASSERT(ok == 777);
mg_mgr_poll(&mgr, 1);
opts.name = mg_str("cesanta.com");
opts.ca = mg_str("");
c = mg_http_connect(&mgr, "https://cesanta.com", f3, &ok);
mg_tls_init(c, &opts);
ok = 0;
for (i = 0; i < 500 && ok <= 0; i++) mg_mgr_poll(&mgr, 10);
MG_INFO(("OK: %d", ok));
ASSERT(ok == 200);
mg_mgr_poll(&mgr, 1);
2020-12-05 19:26:32 +08:00
#endif
2020-12-21 00:55:33 +08:00
#if MG_ENABLE_IPV6
ok = 0;
2020-12-22 17:44:59 +08:00
// ipv6.google.com does not have IPv4 address, only IPv6, therefore
// it is guaranteed to hit IPv6 resolution path.
2020-12-21 00:55:33 +08:00
c = mg_http_connect(&mgr, "http://ipv6.google.com", f3, &ok);
for (i = 0; i < 500 && ok <= 0; i++) mg_mgr_poll(&mgr, 10);
2020-12-22 17:44:59 +08:00
ASSERT(ok == 200);
2020-12-21 00:55:33 +08:00
#endif
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2023-09-12 03:33:43 +08:00
// Test host validation only (no CA, no cert)
static void test_host_validation(void) {
#if MG_TLS
const char *url = "https://cesanta.com";
struct mg_tls_opts opts;
struct mg_mgr mgr;
struct mg_connection *c = NULL;
int i, ok = 0;
memset(&opts, 0, sizeof(opts));
mg_mgr_init(&mgr);
ok = 0;
c = mg_http_connect(&mgr, url, f3, &ok);
ASSERT(c != NULL);
opts.ca = mg_unpacked("/test/data/ca.pem");
mg_tls_init(c, &opts);
2023-09-12 03:33:43 +08:00
for (i = 0; i < 1500 && ok <= 0; i++) mg_mgr_poll(&mgr, 10);
ASSERT(ok == 200);
c->is_closing = 1;
mg_mgr_poll(&mgr, 1);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
#endif
}
static void f4(struct mg_connection *c, int ev, void *ev_data) {
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
mg_printf(c, "HTTP/1.0 200 OK\n\n%.*s/%s", (int) hm->uri.len, hm->uri.ptr,
2022-04-21 09:50:40 +08:00
"abcdef");
strcat((char *) c->fn_data, "m");
2020-12-05 19:26:32 +08:00
c->is_draining = 1;
2022-04-21 09:50:40 +08:00
} else if (ev == MG_EV_CLOSE) {
strcat((char *) c->fn_data, "c");
}
2020-12-05 19:26:32 +08:00
}
static void f4c(struct mg_connection *c, int ev, void *ev_data) {
2022-04-21 10:13:31 +08:00
if (ev == MG_EV_CONNECT) {
mg_printf(c, "GET /foo/bar HTTP/1.0\n\n");
} else if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
ASSERT(mg_strcmp(hm->body, mg_str("/foo/bar/abcdef")) == 0);
strcat((char *) c->fn_data, "m");
2022-04-21 10:13:31 +08:00
} else if (ev == MG_EV_CLOSE) {
strcat((char *) c->fn_data, "c");
2022-04-21 10:13:31 +08:00
}
}
static void f41(struct mg_connection *c, int ev, void *ev_data) {
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
mg_printf(c, "HTTP/1.0 200 OK\n\n%.*s/%s", (int) hm->uri.len, hm->uri.ptr,
"abcdef");
}
}
2020-12-05 19:26:32 +08:00
static void test_http_no_content_length(void) {
2022-04-21 10:13:31 +08:00
char buf1[10] = {0}, buf2[10] = {0};
char buf[100];
2020-12-05 19:26:32 +08:00
struct mg_mgr mgr;
const char *url = "http://127.0.0.1:12348";
2022-04-21 10:13:31 +08:00
int i;
2024-03-06 08:57:31 +08:00
const char *post_req =
"POST / HTTP/1.1\r\nContent-Type:"
"b/a\r\nContent-Length: 15\r\n\r\n"
"{\"key\": \"value\"}";
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
2022-04-21 10:13:31 +08:00
mg_http_listen(&mgr, url, f4, (void *) buf1);
mg_http_connect(&mgr, url, f4c, (void *) buf2);
2022-07-02 19:45:44 +08:00
for (i = 0; i < 1000 && strchr(buf2, 'c') == NULL; i++) mg_mgr_poll(&mgr, 10);
MG_INFO(("[%s] [%s]", buf1, buf2));
2023-09-27 02:59:42 +08:00
ASSERT(strcmp(buf1, "mc") == 0);
ASSERT(strcmp(buf2, "mc") == 0);
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
mg_mgr_init(&mgr);
mg_http_listen(&mgr, url, f41, (void *) NULL);
ASSERT(fetch(&mgr, buf, url, "POST / HTTP/1.1\r\n\r\n") == 411);
ASSERT(fetch(&mgr, buf, url, "HTTP/1.1 200\r\n\r\n") == 411);
ASSERT(fetch(&mgr, buf, url, "HTTP/1.1 100\r\n\r\n") != 411);
ASSERT(fetch(&mgr, buf, url, "HTTP/1.1 304\r\n\r\n") != 411);
ASSERT(fetch(&mgr, buf, url, "HTTP/1.1 305\r\n\r\n") == 411);
ASSERT(fetch(&mgr, buf, url, post_req) != 411);
mg_mgr_free(&mgr);
2020-12-05 19:26:32 +08:00
ASSERT(mgr.conns == NULL);
}
static void f5(struct mg_connection *c, int ev, void *ev_data) {
2020-12-05 19:26:32 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
mg_http_reply(c, 200, "", "%.*s", (int) hm->uri.len, hm->uri.ptr);
(*(int *) c->fn_data)++;
}
2020-12-05 19:26:32 +08:00
}
static void f6(struct mg_connection *c, int ev, void *ev_data) {
if (ev == MG_EV_HTTP_MSG) {
(*(int *) c->fn_data)++;
(void) ev_data;
}
}
2020-12-05 19:26:32 +08:00
static void test_http_pipeline(void) {
struct mg_mgr mgr;
const char *url = "http://127.0.0.1:12377";
struct mg_connection *c;
int i, ok = 0, ok2 = 0;
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
mg_http_listen(&mgr, url, f5, (void *) &ok);
c = mg_http_connect(&mgr, url, f6, &ok2);
2020-12-05 19:26:32 +08:00
mg_printf(c, "POST / HTTP/1.0\nContent-Length: 5\n\n12345GET / HTTP/1.0\n\n");
for (i = 0; i < 20; i++) mg_mgr_poll(&mgr, 1);
ASSERT(ok == 2);
ASSERT(ok2 == 2);
// Fire a valid, then invalid request, see #2592. First one should serve
ok = ok2 = 0;
c = mg_http_connect(&mgr, url, f6, (void *) &ok2);
mg_printf(c, "GET / HTTP/1.1\n\nInvalid\n\n");
for (i = 0; i < 20; i++) mg_mgr_poll(&mgr, 1);
ASSERT(ok == 1);
ASSERT(ok2 == 1);
2024-02-14 20:32:26 +08:00
// MG_INFO(("-----> [%d] [%d]", ok, ok2));
2020-12-05 19:26:32 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2020-12-05 19:26:32 +08:00
static void test_http_parse(void) {
struct mg_str *v;
struct mg_http_message req;
{
2020-12-05 19:26:32 +08:00
const char *s = "GET / HTTP/1.0\n\n";
ASSERT(mg_http_parse("\b23", 3, &req) == -1);
ASSERT(mg_http_parse("get\n\n", 5, &req) == -1);
ASSERT(mg_http_parse(s, strlen(s) - 1, &req) == 0);
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT(req.message.len == strlen(s));
ASSERT(req.body.len == 0);
}
{
2020-12-05 19:26:32 +08:00
const char *s = "GET /blah HTTP/1.0\r\nFoo: bar \r\n\r\n";
size_t idx, len = strlen(s);
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) len);
ASSERT(mg_vcmp(&req.headers[0].name, "Foo") == 0);
ASSERT(mg_vcmp(&req.headers[0].value, "bar") == 0);
ASSERT(req.headers[1].name.len == 0);
ASSERT(req.headers[1].name.ptr == NULL);
ASSERT(req.query.len == 0);
ASSERT(req.message.len == len);
ASSERT(req.body.len == 0);
ASSERT(mg_vcmp(&req.method, "GET") == 0);
ASSERT(mg_vcmp(&req.uri, "/blah") == 0);
ASSERT(mg_vcmp(&req.proto, "HTTP/1.0") == 0);
2020-12-05 19:26:32 +08:00
for (idx = 0; idx < len; idx++) ASSERT(mg_http_parse(s, idx, &req) == 0);
}
{
const char *s = "get b c\nb: t\nv:vv\n\n xx";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s) - 3);
}
{
const char *s = "get b c\nb: t\nv:\n\n xx";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s) - 3);
}
{
const char *s = "get b c\nb: t\nv v\n\n xx";
ASSERT(mg_http_parse(s, strlen(s), &req) == -1);
}
{
const char *s = "get b c\nb: t\n : aa\n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == -1);
}
{
const char *s = "get b c\nz: k \nb: t\nv:k\n\n xx";
2020-12-05 19:26:32 +08:00
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s) - 3);
ASSERT(req.headers[3].name.len == 0);
ASSERT(mg_vcmp(&req.headers[0].name, "z") == 0);
2020-12-05 19:26:32 +08:00
ASSERT(mg_vcmp(&req.headers[0].value, "k") == 0);
ASSERT(mg_vcmp(&req.headers[1].name, "b") == 0);
2020-12-05 19:26:32 +08:00
ASSERT(mg_vcmp(&req.headers[1].value, "t") == 0);
ASSERT(mg_vcmp(&req.headers[2].name, "v") == 0);
ASSERT(mg_vcmp(&req.headers[2].value, "k") == 0);
2020-12-05 19:26:32 +08:00
ASSERT(req.body.len == 0);
}
// #2292: fail on stray \r inside the headers
ASSERT(mg_http_parse("a є\n\n", 6, &req) == 6);
ASSERT(mg_http_parse("a b\n\n", 5, &req) == 5);
ASSERT(mg_http_parse("a b\na:\n\n", 8, &req) > 0);
ASSERT(mg_http_parse("a b\na:\r\n\n", 9, &req) > 0);
ASSERT(mg_http_parse("a b\n\ra:\r\n\n", 10, &req) == -1);
ASSERT(mg_http_parse("a b\na:\r1\n\n", 10, &req) == -1);
ASSERT(mg_http_parse("a b\na: \r1\n\n", 11, &req) == -1);
ASSERT(mg_http_parse("a b\na: \rb:\n\n", 12, &req) == -1);
ASSERT(mg_http_parse("a b\na: \nb:\n\n", 12, &req) > 0);
{
const char *s = "ґєт /слеш вах вах\nмісто: кіїв \n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT(req.body.len == 0);
ASSERT(req.headers[1].name.len == 0);
ASSERT(mg_vcmp(&req.headers[0].name, "місто") == 0);
ASSERT(mg_vcmp(&req.headers[0].value, "кіїв") == 0);
ASSERT((v = mg_http_get_header(&req, "місто")) != NULL);
ASSERT(mg_vcmp(&req.method, "ґєт") == 0);
ASSERT(mg_vcmp(&req.uri, "/слеш") == 0);
ASSERT(mg_vcmp(&req.proto, "вах вах") == 0);
}
{
const char *s = "a b c\r\nContent-Length: 21 \r\nb: t\r\nv:v\r\n\r\nabc";
2020-12-05 19:26:32 +08:00
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s) - 3);
ASSERT(req.body.len == 21);
ASSERT(req.message.len == 21 - 3 + strlen(s));
ASSERT(mg_http_get_header(&req, "foo") == NULL);
ASSERT((v = mg_http_get_header(&req, "contENT-Length")) != NULL);
ASSERT(mg_vcmp(v, "21") == 0);
ASSERT((v = mg_http_get_header(&req, "B")) != NULL);
ASSERT(mg_vcmp(v, "t") == 0);
}
{
2020-12-05 19:26:32 +08:00
const char *s = "GET /foo?a=b&c=d HTTP/1.0\n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT(mg_vcmp(&req.uri, "/foo") == 0);
ASSERT(mg_vcmp(&req.query, "a=b&c=d") == 0);
}
{
2020-12-05 19:26:32 +08:00
const char *s = "POST /x HTTP/1.0\n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT(req.body.len == (size_t) ~0);
}
{
2020-12-05 19:26:32 +08:00
const char *s = "WOHOO /x HTTP/1.0\n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT(req.body.len == 0);
}
{
2020-12-05 19:26:32 +08:00
const char *s = "HTTP/1.0 200 OK\n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT(mg_vcmp(&req.method, "HTTP/1.0") == 0);
ASSERT(mg_vcmp(&req.uri, "200") == 0);
ASSERT(mg_vcmp(&req.proto, "OK") == 0);
ASSERT(req.body.len == (size_t) ~0);
}
{
2020-12-05 19:26:32 +08:00
static const char *s = "HTTP/1.0 999 OMGWTFBBQ\n\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
}
{
2020-12-05 19:26:32 +08:00
const char *s =
"GET / HTTP/1.0\r\nhost:127.0.0.1:18888\r\nCookie:\r\nX-PlayID: "
"45455\r\nRange: 0-1 \r\n\r\n";
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT((v = mg_http_get_header(&req, "Host")) != NULL);
ASSERT(mg_vcmp(v, "127.0.0.1:18888") == 0);
ASSERT((v = mg_http_get_header(&req, "Cookie")) != NULL);
ASSERT(v->len == 0);
ASSERT((v = mg_http_get_header(&req, "X-PlayID")) != NULL);
ASSERT(mg_vcmp(v, "45455") == 0);
ASSERT((v = mg_http_get_header(&req, "Range")) != NULL);
ASSERT(mg_vcmp(v, "0-1") == 0);
}
{
static const char *s = "a b c\na:1\nb:2\nc:3\nd:4\ne:5\nf:6\ng:7\nh:8\n\n";
2020-12-05 19:26:32 +08:00
ASSERT(mg_http_parse(s, strlen(s), &req) == (int) strlen(s));
ASSERT((v = mg_http_get_header(&req, "e")) != NULL);
ASSERT(mg_vcmp(v, "5") == 0);
ASSERT((v = mg_http_get_header(&req, "h")) == NULL);
}
2020-12-12 06:58:50 +08:00
{
struct mg_connection c;
struct mg_str s,
res = mg_str("GET /\r\nAuthorization: Basic Zm9vOmJhcg==\r\n\r\n");
memset(&c, 0, sizeof(c));
mg_printf(&c, "%s", "GET /\r\n");
mg_http_bauth(&c, "foo", "bar");
mg_printf(&c, "%s", "\r\n");
s = mg_str_n((char *) c.send.buf, c.send.len);
ASSERT(mg_strcmp(s, res) == 0);
mg_iobuf_free(&c.send);
}
2020-12-24 22:33:23 +08:00
{
struct mg_http_message hm;
const char *s = "GET /foo?bar=baz HTTP/1.0\n\n ";
ASSERT(mg_http_parse(s, strlen(s), &hm) == (int) strlen(s) - 1);
2020-12-24 22:33:23 +08:00
ASSERT(mg_strcmp(hm.uri, mg_str("/foo")) == 0);
ASSERT(mg_strcmp(hm.query, mg_str("bar=baz")) == 0);
}
2021-02-10 05:16:33 +08:00
{
struct mg_http_message hm;
const char *s = "a b c\n\n";
ASSERT(mg_http_parse(s, strlen(s), &hm) == (int) strlen(s));
s = "a b\nc:d\n\n";
ASSERT(mg_http_parse(s, strlen(s), &hm) == (int) strlen(s));
s = "a\nb:b\nc:c\n\n";
ASSERT(mg_http_parse(s, strlen(s), &hm) < 0);
s = "a b\nc: \xc0\n\n"; // Invalid UTF in the header value: accept
ASSERT(mg_http_parse(s, strlen(s), &hm) == (int) strlen(s));
ASSERT((v = mg_http_get_header(&hm, "c")) != NULL);
ASSERT(mg_vcmp(v, "\xc0") == 0);
s = "a b\n\xc0: 2\n\n"; // Invalid UTF in the header name: do NOT accept
ASSERT(mg_http_parse(s, strlen(s), &hm) == -1);
2021-02-10 05:16:33 +08:00
}
}
static void ehr(struct mg_connection *c, int ev, void *ev_data) {
2021-04-29 18:00:57 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
struct mg_http_serve_opts opts;
memset(&opts, 0, sizeof(opts));
opts.root_dir = "./test/data";
2021-04-29 18:00:57 +08:00
mg_http_serve_dir(c, hm, &opts);
}
}
2020-12-05 19:26:32 +08:00
static void test_http_range(void) {
struct mg_mgr mgr;
2020-12-05 19:26:32 +08:00
const char *url = "http://127.0.0.1:12349";
struct mg_http_message hm;
char buf[FETCH_BUF_SIZE];
2020-12-05 19:26:32 +08:00
mg_mgr_init(&mgr);
2021-04-29 18:00:57 +08:00
mg_http_listen(&mgr, url, ehr, NULL);
2020-12-05 19:26:32 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /range.txt HTTP/1.0\n\n") == 200);
2021-02-10 05:16:33 +08:00
ASSERT(mg_http_parse(buf, strlen(buf), &hm) > 0);
2020-12-05 19:26:32 +08:00
ASSERT(hm.body.len == 312);
2021-04-29 18:00:57 +08:00
fetch(&mgr, buf, url, "%s", "GET /range.txt HTTP/1.0\nRange: bytes=5-10\n\n");
ASSERT(mg_http_parse(buf, strlen(buf), &hm) > 0);
printf("%s", buf);
2021-04-29 18:00:57 +08:00
ASSERT(mg_strcmp(hm.uri, mg_str("206")) == 0);
ASSERT(mg_strcmp(hm.proto, mg_str("Partial Content")) == 0);
ASSERT(mg_strcmp(hm.body, mg_str(" of co")) == 0);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "Content-Range"),
mg_str("bytes 5-10/312")) == 0);
// Fetch till EOF
fetch(&mgr, buf, url, "%s", "GET /range.txt HTTP/1.0\nRange: bytes=300-\n\n");
ASSERT(mg_http_parse(buf, strlen(buf), &hm) > 0);
ASSERT(mg_strcmp(hm.uri, mg_str("206")) == 0);
ASSERT(mg_strcmp(hm.body, mg_str("is disease.\n")) == 0);
// MG_INFO(("----%d\n[%s]", (int) hm.body.len, buf));
2021-04-29 18:00:57 +08:00
// Fetch past EOF, must trigger 416 response
fetch(&mgr, buf, url, "%s", "GET /range.txt HTTP/1.0\nRange: bytes=999-\n\n");
ASSERT(mg_http_parse(buf, strlen(buf), &hm) > 0);
ASSERT(mg_strcmp(hm.uri, mg_str("416")) == 0);
ASSERT(hm.body.len == 0);
ASSERT(mg_strcmp(*mg_http_get_header(&hm, "Content-Range"),
mg_str("bytes */312")) == 0);
fetch(&mgr, buf, url, "%s",
"GET /range.txt HTTP/1.0\nRange: bytes=0-312\n\n");
ASSERT(mg_http_parse(buf, strlen(buf), &hm) > 0);
ASSERT(mg_strcmp(hm.uri, mg_str("416")) == 0);
mg_mgr_free(&mgr);
2020-12-05 19:26:32 +08:00
ASSERT(mgr.conns == NULL);
}
static void f1(void *arg) {
(*(int *) arg)++;
}
2020-12-05 19:26:32 +08:00
static void test_timer(void) {
int v1 = 0, v2 = 0, v3 = 0;
struct mg_timer t1, t2, t3, *head = NULL;
mg_timer_init(&head, &t1, 5, MG_TIMER_REPEAT, f1, &v1);
2022-05-15 21:29:34 +08:00
mg_timer_init(&head, &t2, 15, MG_TIMER_ONCE, f1, &v2);
mg_timer_init(&head, &t3, 10, MG_TIMER_REPEAT | MG_TIMER_RUN_NOW, f1, &v3);
2020-12-05 19:26:32 +08:00
ASSERT(head == &t3);
ASSERT(head->next == &t2);
2020-12-05 19:26:32 +08:00
mg_timer_poll(&head, 0);
2022-05-15 21:29:34 +08:00
ASSERT(v1 == 0);
ASSERT(v2 == 0);
ASSERT(v3 == 1);
mg_timer_poll(&head, 1);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 0);
ASSERT(v2 == 0);
ASSERT(v3 == 1);
mg_timer_poll(&head, 5);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 1);
ASSERT(v2 == 0);
ASSERT(v3 == 1);
// Simulate long delay - timers must invalidate expiration times
mg_timer_poll(&head, 100);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 2);
ASSERT(v2 == 1);
ASSERT(v3 == 2);
mg_timer_poll(&head, 107);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 3);
ASSERT(v2 == 1);
ASSERT(v3 == 2);
mg_timer_poll(&head, 114);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 4);
ASSERT(v2 == 1);
ASSERT(v3 == 3);
mg_timer_poll(&head, 115);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 5);
ASSERT(v2 == 1);
ASSERT(v3 == 3);
2022-05-15 21:29:34 +08:00
mg_timer_free(&head, &t2);
mg_timer_init(&head, &t2, 3, 0, f1, &v2);
ASSERT(head == &t2);
ASSERT(head->next == &t3);
ASSERT(head->next->next == &t1);
ASSERT(head->next->next->next == NULL);
2020-12-05 19:26:32 +08:00
mg_timer_poll(&head, 120);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 6);
ASSERT(v2 == 1);
ASSERT(v3 == 4);
mg_timer_poll(&head, 125);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 7);
ASSERT(v2 == 2);
ASSERT(v3 == 4);
// Test millisecond counter wrap - when time goes back.
mg_timer_poll(&head, 0);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 7);
ASSERT(v2 == 2);
ASSERT(v3 == 4);
mg_timer_poll(&head, 7);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 8);
ASSERT(v2 == 2);
ASSERT(v3 == 4);
mg_timer_poll(&head, 11);
2020-12-05 19:26:32 +08:00
ASSERT(v1 == 9);
ASSERT(v2 == 2);
ASSERT(v3 == 5);
mg_timer_free(&head, &t1);
2022-05-15 21:29:34 +08:00
ASSERT(head == &t2);
ASSERT(head->next == &t3);
ASSERT(head->next->next == NULL);
2020-12-05 19:26:32 +08:00
mg_timer_free(&head, &t2);
ASSERT(head == &t3);
ASSERT(head->next == NULL);
2020-12-05 19:26:32 +08:00
mg_timer_free(&head, &t3);
ASSERT(head == NULL);
2022-05-10 14:38:03 +08:00
// Start a timer, then shift system time a long time back and long time forth
v1 = 0;
mg_timer_init(&head, &t1, 5, MG_TIMER_REPEAT, f1, &v1);
mg_timer_poll(&head, 0);
ASSERT(v1 == 0);
// Shift a long time forth, make sure it ticks
mg_timer_poll(&head, 100);
ASSERT(v1 == 1);
mg_timer_poll(&head, 101);
ASSERT(v1 == 1);
mg_timer_poll(&head, 102);
ASSERT(v1 == 1);
mg_timer_poll(&head, 103);
ASSERT(v1 == 1);
mg_timer_poll(&head, 104);
ASSERT(v1 == 1);
mg_timer_poll(&head, 105);
ASSERT(v1 == 2);
// Shift a long time back, make sure it ticks
mg_timer_poll(&head, 50);
ASSERT(v1 == 2);
mg_timer_poll(&head, 60);
ASSERT(v1 == 3);
mg_timer_free(&head, &t1);
ASSERT(head == NULL);
2022-05-10 14:38:03 +08:00
// Test proper timer deallocation, see #1539
{
struct mg_mgr mgr;
mg_mgr_init(&mgr);
mg_timer_add(&mgr, 1, MG_TIMER_REPEAT, f1, NULL);
ASSERT(mgr.timers != NULL);
2022-05-10 14:38:03 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.timers == NULL);
2022-05-10 14:38:03 +08:00
ASSERT(mgr.conns == NULL);
}
2020-12-05 19:26:32 +08:00
}
2022-02-12 02:15:34 +08:00
static bool sn(const char *fmt, ...) {
char buf[100], tmp[1] = {0}, buf2[sizeof(buf)];
size_t n, n2, n1;
va_list ap;
bool result;
va_start(ap, fmt);
n = (size_t) vsnprintf(buf2, sizeof(buf2), fmt, ap);
va_end(ap);
va_start(ap, fmt);
2022-06-10 16:38:36 +08:00
n1 = mg_vsnprintf(buf, sizeof(buf), fmt, &ap);
2022-02-12 02:15:34 +08:00
va_end(ap);
va_start(ap, fmt);
2022-06-10 16:38:36 +08:00
n2 = mg_vsnprintf(tmp, 0, fmt, &ap);
2022-02-12 02:15:34 +08:00
va_end(ap);
result = n1 == n2 && n1 == n && strcmp(buf, buf2) == 0;
if (!result)
MG_ERROR(("[%s] -> [%s] != [%s] %d %d %d\n", fmt, buf, buf2, (int) n1,
(int) n2, (int) n));
2022-02-12 02:15:34 +08:00
return result;
}
static bool sccmp(const char *s1, const char *s2, int expected) {
2022-02-23 04:20:56 +08:00
int n1 = mg_casecmp(s1, s2);
// MG_INFO(("[%s] [%s] %d %d", s1, s2, n1, expected));
return n1 == expected;
2022-02-23 04:20:56 +08:00
}
2022-06-28 07:45:52 +08:00
static size_t pf1(void (*out)(char, void *), void *ptr, va_list *ap) {
2022-06-10 16:38:36 +08:00
int a = va_arg(*ap, int);
int b = va_arg(*ap, int);
2022-08-14 03:34:20 +08:00
return mg_xprintf(out, ptr, "%d", a + b);
2022-06-10 16:38:36 +08:00
}
2022-06-28 07:45:52 +08:00
static size_t pf2(void (*out)(char, void *), void *ptr, va_list *ap) {
2022-06-10 19:14:42 +08:00
int cnt = va_arg(*ap, int);
size_t n = 0;
2022-08-14 03:34:20 +08:00
while (cnt-- > 0) n += mg_xprintf(out, ptr, "%d", cnt);
2022-06-10 19:14:42 +08:00
return n;
}
2022-08-14 08:25:38 +08:00
static bool chkdbl(struct mg_str s, double val) {
double d, tolerance = 1e-14;
return mg_json_get_num(s, "$", &d) && fabs(val - d) < tolerance;
}
2020-12-07 16:54:58 +08:00
static void test_str(void) {
{
struct mg_str s = mg_strdup(mg_str("a"));
ASSERT(mg_strcmp(s, mg_str("a")) == 0);
free((void *) s.ptr);
}
2023-02-26 09:32:03 +08:00
{
const char *s;
struct mg_str a = mg_str("hello"), b = mg_str("a"), c = mg_str(NULL);
ASSERT((s = mg_strstr(a, b)) == NULL);
ASSERT((s = mg_strstr(a, c)) != NULL);
ASSERT(s == a.ptr);
}
2020-12-07 16:54:58 +08:00
ASSERT(mg_strcmp(mg_str(""), mg_str(NULL)) == 0);
ASSERT(mg_strcmp(mg_str("a"), mg_str("b")) < 0);
ASSERT(mg_strcmp(mg_str("b"), mg_str("a")) > 0);
ASSERT(mg_strstr(mg_str("abc"), mg_str("d")) == NULL);
ASSERT(mg_strstr(mg_str("abc"), mg_str("b")) != NULL);
ASSERT(mg_strcmp(mg_str("hi"), mg_strstrip(mg_str(" \thi\r\n"))) == 0);
2022-02-10 19:56:55 +08:00
ASSERT(sccmp("", "", 0));
ASSERT(sccmp("", "1", -49));
ASSERT(sccmp("a", "A", 0));
ASSERT(sccmp("a1", "A", 49));
ASSERT(sccmp("a", "A1", -49));
2022-02-23 04:20:56 +08:00
2022-06-09 19:39:48 +08:00
{
2022-08-14 08:25:38 +08:00
ASSERT(chkdbl(mg_str_n("1.23", 3), 1.2));
ASSERT(chkdbl(mg_str("1.23 "), 1.23));
ASSERT(chkdbl(mg_str("-0.01 "), -0.01));
ASSERT(chkdbl(mg_str("-0.5e2"), -50.0));
ASSERT(chkdbl(mg_str("123e-3"), 0.123));
2022-06-09 19:39:48 +08:00
}
2022-02-12 02:15:34 +08:00
ASSERT(sn("%d", 0));
ASSERT(sn("%d", 1));
ASSERT(sn("%d", -1));
ASSERT(sn("%.*s", 0, "ab"));
2022-02-12 02:15:34 +08:00
ASSERT(sn("%.*s", 1, "ab"));
ASSERT(sn("%.1s", "ab"));
ASSERT(sn("%.99s", "a"));
ASSERT(sn("%11s", "a"));
ASSERT(sn("%s", "a\0b"));
ASSERT(sn("%2s", "a"));
ASSERT(sn("%.*s", 3, "a\0b"));
ASSERT(sn("%d", 7));
ASSERT(sn("%d", 123));
#if MG_ARCH == MG_ARCH_UNIX
ASSERT(sn("%lld", (uint64_t) 0xffffffffff));
ASSERT(sn("%lld", (uint64_t) -1));
ASSERT(sn("%llu", (uint64_t) -1));
ASSERT(sn("%llx", (uint64_t) 0xffffffffff));
ASSERT(sn("%p", (void *) (size_t) 7));
#endif
ASSERT(sn("%lx", (unsigned long) 0x6204d754));
ASSERT(sn("ab"));
ASSERT(sn("%dx", 1));
ASSERT(sn("%sx", "a"));
ASSERT(sn("%cx", 32));
ASSERT(sn("%x", 15));
ASSERT(sn("%2x", 15));
ASSERT(sn("%02x", 15));
ASSERT(sn("%hx:%hhx", (short) 1, (char) 2));
ASSERT(sn("%hx:%hhx", (short) 1, (char) 2));
ASSERT(sn("%%"));
ASSERT(sn("%x", 15));
ASSERT(sn("%#x", 15));
ASSERT(sn("%#6x", 15));
ASSERT(sn("%#06x", 15));
ASSERT(sn("%#-6x", 15));
ASSERT(sn("%-2s!", "a"));
ASSERT(sn("%s %s", "a", "b"));
ASSERT(sn("%s %s", "a", "b"));
ASSERT(sn("ab%dc", 123));
ASSERT(sn("%s ", "a"));
ASSERT(sn("%s %s", "a", "b"));
ASSERT(sn("%2s %s", "a", "b"));
// Non-standard formatting
{
char buf[100], *p = NULL;
struct mg_iobuf io = {0, 0, 0, 16};
const char *expected;
expected = "\"\"";
2023-04-14 21:11:12 +08:00
mg_snprintf(buf, sizeof(buf), "%m", mg_print_esc, 0, "");
ASSERT(strcmp(buf, expected) == 0);
2022-08-14 03:34:20 +08:00
expected = "";
mg_snprintf(buf, 1, "%s", "abc");
ASSERT(strcmp(buf, expected) == 0);
expected = "a";
mg_snprintf(buf, 2, "%s", "abc");
ASSERT(strcmp(buf, expected) == 0);
expected = "\"hi, \\\"\"";
2023-04-14 21:11:12 +08:00
mg_snprintf(buf, sizeof(buf), "\"hi, %M\"", mg_print_esc, 0, "\"");
MG_INFO(("[%s] [%s]", buf, expected));
ASSERT(strcmp(buf, expected) == 0);
expected = "\"a'b\"";
2023-04-14 21:11:12 +08:00
mg_snprintf(buf, sizeof(buf), "%m", mg_print_esc, 0, "a'b");
ASSERT(strcmp(buf, expected) == 0);
expected = "\"a\\b\\n\\f\\r\\t\\\"\"";
2023-04-14 21:11:12 +08:00
mg_snprintf(buf, sizeof(buf), "%m", mg_print_esc, 0, "a\b\n\f\r\t\"");
ASSERT(strcmp(buf, expected) == 0);
expected = "\"abc\"";
2023-04-14 21:11:12 +08:00
mg_snprintf(buf, sizeof(buf), "%m", mg_print_esc, 3, "abcdef");
ASSERT(strcmp(buf, expected) == 0);
2022-06-10 16:38:36 +08:00
p = mg_mprintf("[%s,%M,%s]", "null", pf1, 2, 3, "hi");
ASSERT(strcmp(p, "[null,5,hi]") == 0);
free(p);
2022-06-10 19:14:42 +08:00
p = mg_mprintf("[%M,%d]", pf2, 10, 7);
2022-06-28 07:45:52 +08:00
ASSERT(strcmp(p, "[9876543210,7]") == 0);
free(p);
2022-08-14 03:34:20 +08:00
mg_xprintf(mg_pfn_iobuf, &io, "[%M", pf2, 10);
mg_xprintf(mg_pfn_iobuf, &io, ",");
mg_xprintf(mg_pfn_iobuf, &io, "%d]", 7);
ASSERT(strcmp((char *) io.buf, "[9876543210,7]") == 0);
mg_iobuf_free(&io);
2022-06-10 16:38:36 +08:00
}
{
2023-03-30 17:49:58 +08:00
#if MG_ARCH == MG_ARCH_WIN32
bool is_windows = true;
#else
bool is_windows = false;
#endif
2022-06-10 16:38:36 +08:00
#define DBLWIDTH(a, b) a, b
2023-03-30 17:49:58 +08:00
#define TESTDOUBLE(fmt_, num_, res_) \
do { \
char t1[40] = "", t2[40] = ""; \
const char *N = #num_; \
mg_snprintf(t1, sizeof(t1), fmt_, num_); \
snprintf(t2, sizeof(t2), fmt_, num_); \
printf("[%s,%s] : [%s] [%s] [%s]\n", fmt_, N, res_, t2, t1); \
ASSERT(strcmp(t1, res_) == 0); \
if (!is_windows) ASSERT(strcmp(t1, t2) == 0); \
2022-06-10 16:38:36 +08:00
} while (0)
TESTDOUBLE("%g", 0.0, "0");
TESTDOUBLE("%g", 0.123, "0.123");
TESTDOUBLE("%g", 0.00123, "0.00123");
TESTDOUBLE("%g", 0.123456333, "0.123456");
TESTDOUBLE("%g", 123.0, "123");
TESTDOUBLE("%g", 11.5454, "11.5454");
TESTDOUBLE("%g", 11.0001, "11.0001");
TESTDOUBLE("%g", 0.999, "0.999");
TESTDOUBLE("%g", 0.999999, "0.999999");
TESTDOUBLE("%g", 0.9999999, "1");
TESTDOUBLE("%g", 10.9, "10.9");
TESTDOUBLE("%g", 10.01, "10.01");
TESTDOUBLE("%g", 1.0, "1");
TESTDOUBLE("%g", 10.0, "10");
TESTDOUBLE("%g", 100.0, "100");
TESTDOUBLE("%g", 1000.0, "1000");
TESTDOUBLE("%g", 10000.0, "10000");
TESTDOUBLE("%g", 100000.0, "100000");
TESTDOUBLE("%g", 1000000.0, "1e+06");
TESTDOUBLE("%g", 10000000.0, "1e+07");
TESTDOUBLE("%g", 100000001.0, "1e+08");
TESTDOUBLE("%g", 10.5454, "10.5454");
TESTDOUBLE("%g", 999999.0, "999999");
TESTDOUBLE("%g", 9999999.0, "1e+07");
TESTDOUBLE("%g", 44556677.0, "4.45567e+07");
TESTDOUBLE("%g", 1234567.2, "1.23457e+06");
TESTDOUBLE("%g", -987.65432, "-987.654");
TESTDOUBLE("%g", 0.0000000001, "1e-10");
TESTDOUBLE("%g", 2.34567e-57, "2.34567e-57");
TESTDOUBLE("%.*g", DBLWIDTH(7, 9999999.0), "9999999");
TESTDOUBLE("%.*g", DBLWIDTH(10, 0.123456333), "0.123456333");
TESTDOUBLE("%g", 123.456222, "123.456");
TESTDOUBLE("%.*g", DBLWIDTH(10, 123.456222), "123.456222");
TESTDOUBLE("%g", 600.1234, "600.123");
TESTDOUBLE("%g", -600.1234, "-600.123");
TESTDOUBLE("%g", 599.1234, "599.123");
TESTDOUBLE("%g", -599.1234, "-599.123");
2023-03-30 17:49:58 +08:00
TESTDOUBLE("%g", 0.14, "0.14");
TESTDOUBLE("%f", 0.14, "0.140000");
TESTDOUBLE("%.*f", DBLWIDTH(4, 0.14), "0.1400");
TESTDOUBLE("%.*f", DBLWIDTH(3, 0.14), "0.140");
TESTDOUBLE("%.*f", DBLWIDTH(2, 0.14), "0.14");
TESTDOUBLE("%.*f", DBLWIDTH(1, 0.14), "0.1");
TESTDOUBLE("%.*f", DBLWIDTH(1, 0.19), "0.2");
TESTDOUBLE("%.*f", DBLWIDTH(1, 0.16), "0.2");
// TESTDOUBLE("%.*f", DBLWIDTH(1, 0.15), "0.1");
2022-06-10 16:38:36 +08:00
#ifndef _WIN32
TESTDOUBLE("%g", (double) INFINITY, "inf");
TESTDOUBLE("%g", (double) -INFINITY, "-inf");
TESTDOUBLE("%g", (double) NAN, "nan");
#else
TESTDOUBLE("%g", HUGE_VAL, "inf");
TESTDOUBLE("%g", -HUGE_VAL, "-inf");
#endif
}
{
const char *expected = "[\"MA==\",\"MAo=\",\"MAr+\",\"MAr+Zw==\"]";
char tmp[100], s[] = "0\n\xfeg";
2023-04-14 21:11:12 +08:00
ASSERT(mg_snprintf(tmp, sizeof(tmp), "[%m,%m,%m,%m]", mg_print_base64, 1, s,
mg_print_base64, 2, s, mg_print_base64, 3, s,
mg_print_base64, 4, s) == 33);
ASSERT(strcmp(tmp, expected) == 0);
}
{
const char *expected = "\"002001200220616263\"";
char tmp[100], s[] = "\x00 \x01 \x02 abc";
2023-04-14 21:11:12 +08:00
ASSERT(mg_snprintf(tmp, sizeof(tmp), "%m", mg_print_hex, 9, s) == 20);
ASSERT(strcmp(tmp, expected) == 0);
}
2022-08-01 19:53:25 +08:00
{
char tmp[3];
ASSERT(mg_snprintf(tmp, sizeof(tmp), "%s", "0123456789") == 10);
ASSERT(strcmp(tmp, "01") == 0);
ASSERT(tmp[2] == '\0');
}
{
char buf[100];
2023-06-09 20:56:08 +08:00
struct mg_addr a;
uint32_t addr = mg_htonl(0x2000001);
memcpy(a.ip, &addr, sizeof(uint32_t));
a.port = mg_htons(3);
a.is_ip6 = false;
ASSERT(mg_snprintf(buf, sizeof(buf), "%M %d", mg_print_ip, &a, 7) == 9);
ASSERT(strcmp(buf, "2.0.0.1 7") == 0);
ASSERT(mg_snprintf(buf, sizeof(buf), "%M %d", mg_print_ip_port, &a, 7) ==
11);
ASSERT(strcmp(buf, "2.0.0.1:3 7") == 0);
2023-06-09 20:56:08 +08:00
memset(a.ip, 0, sizeof(a.ip));
a.ip[0] = 1, a.ip[1] = 100, a.ip[2] = 33;
a.is_ip6 = true;
ASSERT(mg_snprintf(buf, sizeof(buf), "%M %d", mg_print_ip, &a, 7) == 24);
ASSERT(strcmp(buf, "[164:2100:0:0:0:0:0:0] 7") == 0);
ASSERT(mg_snprintf(buf, sizeof(buf), "%M %d", mg_print_ip_port, &a, 7) ==
26);
ASSERT(strcmp(buf, "[164:2100:0:0:0:0:0:0]:3 7") == 0);
}
ASSERT(mg_path_is_sane(".") == true);
ASSERT(mg_path_is_sane("") == true);
ASSERT(mg_path_is_sane("a.b") == true);
ASSERT(mg_path_is_sane("a..b") == true);
ASSERT(mg_path_is_sane(".a") == true);
ASSERT(mg_path_is_sane(".a.") == true);
ASSERT(mg_path_is_sane("./") == true);
ASSERT(mg_path_is_sane("a..") == true);
ASSERT(mg_path_is_sane("././a..") == true);
ASSERT(mg_path_is_sane("..") == false);
ASSERT(mg_path_is_sane("../") == false);
ASSERT(mg_path_is_sane("./..") == false);
ASSERT(mg_path_is_sane("a/../") == false);
ASSERT(mg_path_is_sane("a/../b") == false);
2020-12-07 16:54:58 +08:00
}
static void fn1(struct mg_connection *c, int ev, void *ev_data) {
2023-07-02 03:23:59 +08:00
if (ev == MG_EV_ERROR) {
free(*(char **) c->fn_data); // See #2263
*(char **) c->fn_data = mg_mprintf("%s", (char *) ev_data);
2023-07-02 03:23:59 +08:00
}
2021-01-21 18:00:18 +08:00
(void) c;
}
2022-05-15 21:29:34 +08:00
static void test_dns_error(const char *dns_server_url, const char *errstr) {
// Test timeout
struct mg_mgr mgr;
char *buf = NULL;
int i;
mg_mgr_init(&mgr);
mgr.dns4.url = dns_server_url;
mgr.dnstimeout = 10;
2022-05-15 21:29:34 +08:00
MG_DEBUG(("opening dummy DNS listener @ [%s]...", dns_server_url));
mg_listen(&mgr, mgr.dns4.url, NULL, NULL); // Just discard our queries
mg_http_connect(&mgr, "http://google.com", fn1, &buf);
for (i = 0; i < 50 && buf == NULL; i++) mg_mgr_poll(&mgr, 1);
mg_mgr_free(&mgr);
2022-05-15 21:29:34 +08:00
// MG_DEBUG(("buf: [%s] [%s]", buf, errstr));
ASSERT(buf != NULL && strcmp(buf, errstr) == 0);
free(buf);
}
2020-12-11 21:16:51 +08:00
static void test_dns(void) {
struct mg_dns_message dm;
2020-12-14 17:31:23 +08:00
// txid flags numQ numA numAP numOP
// 0000 00 01 81 80 00 01 00 01 00 00 00 00 07 63 65 73 .............ces
// 0010 61 6e 74 61 03 63 6f 6d 00 00 01 00 01 c0 0c 00 anta.com........
// 0020 01 00 01 00 00 02 57 00 04 94 fb 36 ec ......W....6.
uint8_t data[] = {0, 1, 0x81, 0x80, 0, 1, 0, 1, 0,
0, 0, 0, 7, 0x63, 0x65, 0x73, 0x61, 0x6e,
0x74, 0x61, 0x03, 0x63, 0x6f, 0x6d, 0, 0, 1,
0, 1, 0xc0, 0x0c, 0, 1, 0, 1, 0,
0, 2, 0x57, 0, 4, 0x94, 0xfb, 0x36, 0xec};
2020-12-11 21:16:51 +08:00
ASSERT(mg_dns_parse(NULL, 0, &dm) == 0);
2020-12-14 17:31:23 +08:00
ASSERT(mg_dns_parse(data, sizeof(data), &dm) == 1);
ASSERT(strcmp(dm.name, "cesanta.com") == 0);
data[30] = 29; // Point a pointer to itself
memset(&dm, 0, sizeof(dm));
ASSERT(mg_dns_parse(data, sizeof(data), &dm) == 1);
ASSERT(strcmp(dm.name, "") == 0);
2022-03-20 22:20:47 +08:00
{
// 0000 00 01 81 80 00 01 00 04 00 00 00 00 05 79 61 68 .............yah
// 0010 6f 6f 05 63 31 31 32 36 03 63 6f 6d 00 00 01 00 oo.c1126.com....
// 0020 01 c0 0c 00 05 00 01 00 00 0d 34 00 0c 03 77 77 ..........4...ww
// 0030 77 05 79 61 68 6f 6f c0 18 c0 2d 00 05 00 01 00 w.yahoo...-.....
// 0040 00 00 01 00 14 0b 6e 65 77 2d 66 70 2d 73 68 65 ......new-fp-she
// 0050 64 03 77 67 31 01 62 c0 31 c0 45 00 01 00 01 00 d.wg1.b.1.E.....
// 0060 00 00 0a 00 04 57 f8 64 d8 c0 45 00 01 00 01 00 .....W.d..E.....
// 0070 00 00 0a 00 04 57 f8 64 d7 .....W.d.
uint8_t d[] = {
0x00, 0x01, 0x81, 0x80, 0x00, 0x01, 0x00, 0x04, 0x00, 0x00, 0x00,
0x00, 0x05, 0x79, 0x61, 0x68, 0x6f, 0x6f, 0x05, 0x63, 0x31, 0x31,
0x32, 0x36, 0x03, 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00, 0x01,
0xc0, 0x0c, 0x00, 0x05, 0x00, 0x01, 0x00, 0x00, 0x0d, 0x34, 0x00,
0x0c, 0x03, 0x77, 0x77, 0x77, 0x05, 0x79, 0x61, 0x68, 0x6f, 0x6f,
0xc0, 0x18, 0xc0, 0x2d, 0x00, 0x05, 0x00, 0x01, 0x00, 0x00, 0x00,
0x01, 0x00, 0x14, 0x0b, 0x6e, 0x65, 0x77, 0x2d, 0x66, 0x70, 0x2d,
0x73, 0x68, 0x65, 0x64, 0x03, 0x77, 0x67, 0x31, 0x01, 0x62, 0xc0,
0x31, 0xc0, 0x45, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00, 0x0a,
0x00, 0x04, 0x57, 0xf8, 0x64, 0xd8, 0xc0, 0x45, 0x00, 0x01, 0x00,
0x01, 0x00, 0x00, 0x00, 0x0a, 0x00, 0x04, 0x57, 0xf8, 0x64, 0xd7,
};
ASSERT(mg_dns_parse(d, sizeof(d), &dm) == 1);
2022-03-22 00:29:27 +08:00
// MG_INFO(("[%s]", dm.name));
ASSERT(strcmp(dm.name, "new-fp-shed.wg1.b.yahoo.com") == 0);
2022-03-20 22:20:47 +08:00
}
2022-05-15 21:29:34 +08:00
test_dns_error("udp://127.0.0.1:12345", "DNS timeout");
test_dns_error("", "resolver");
test_dns_error("tcp://0.0.0.0:0", "DNS error");
2020-12-11 21:16:51 +08:00
}
2020-12-07 16:54:58 +08:00
static void test_util(void) {
const char *e;
char buf[100], *s;
2020-12-21 00:55:33 +08:00
struct mg_addr a;
2023-06-09 20:56:08 +08:00
uint32_t ipv4;
struct mg_str data;
memset(&a, 0xa5, sizeof(a));
ASSERT(mg_file_printf(&mg_fs_posix, "data.txt", "%s", "hi") == true);
// if (system("ls -l") != 0) (void) 0;
data = mg_file_read(&mg_fs_posix, "data.txt");
ASSERT(data.ptr != NULL);
ASSERT(strcmp(data.ptr, "hi") == 0);
free((void *) data.ptr);
2021-01-27 00:30:44 +08:00
remove("data.txt");
2020-12-21 00:55:33 +08:00
ASSERT(mg_aton(mg_str("0"), &a) == false);
ASSERT(mg_aton(mg_str("0.0.0."), &a) == false);
ASSERT(mg_aton(mg_str("0.0.0.256"), &a) == false);
ASSERT(mg_aton(mg_str("0.0.0.-1"), &a) == false);
ASSERT(mg_aton(mg_str("127.0.0.1"), &a) == true);
2020-12-22 17:44:59 +08:00
ASSERT(a.is_ip6 == false);
2023-06-09 20:56:08 +08:00
memcpy(&ipv4, a.ip, sizeof(ipv4));
ASSERT(ipv4 == mg_htonl(0x7f000001));
2020-12-22 17:44:59 +08:00
memset(a.ip, 0xa5, sizeof(a.ip));
2020-12-22 17:44:59 +08:00
ASSERT(mg_aton(mg_str("1:2:3:4:5:6:7:8"), &a) == true);
ASSERT(a.is_ip6 == true);
e = "\x00\x01\x00\x02\x00\x03\x00\x04\x00\x05\x00\x06\x00\x07\x00\x08";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
memset(a.ip, 0xa5, sizeof(a.ip));
ASSERT(mg_aton(mg_str("1:2::3"), &a) == true);
ASSERT(a.is_ip6 == true);
e = "\x00\x01\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
2020-12-22 17:44:59 +08:00
2023-06-09 20:56:08 +08:00
memset(a.ip, 0xaa, sizeof(a.ip));
2020-12-22 17:44:59 +08:00
ASSERT(mg_aton(mg_str("1::1"), &a) == true);
ASSERT(a.is_ip6 == true);
e = "\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
2020-12-22 17:44:59 +08:00
2023-06-09 20:56:08 +08:00
memset(a.ip, 0xaa, sizeof(a.ip));
2021-09-19 02:38:59 +08:00
ASSERT(mg_aton(mg_str("::fFff:1.2.3.4"), &a) == true);
ASSERT(a.is_ip6 == true);
e = "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x01\x02\x03\x04";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
2021-09-19 02:38:59 +08:00
2023-06-09 20:56:08 +08:00
memset(a.ip, 0xaa, sizeof(a.ip));
2020-12-22 17:44:59 +08:00
ASSERT(mg_aton(mg_str("::1"), &a) == true);
ASSERT(a.is_ip6 == true);
ASSERT(a.scope_id == 0);
e = "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
2020-12-22 17:44:59 +08:00
2023-06-09 20:56:08 +08:00
memset(a.ip, 0xaa, sizeof(a.ip));
2020-12-22 17:44:59 +08:00
ASSERT(mg_aton(mg_str("1::"), &a) == true);
ASSERT(a.is_ip6 == true);
e = "\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
2020-12-22 17:44:59 +08:00
2023-06-09 20:56:08 +08:00
memset(a.ip, 0xaa, sizeof(a.ip));
2020-12-22 17:44:59 +08:00
ASSERT(mg_aton(mg_str("2001:4860:4860::8888"), &a) == true);
ASSERT(a.is_ip6 == true);
e = "\x20\x01\x48\x60\x48\x60\x00\x00\x00\x00\x00\x00\x00\x00\x88\x88";
ASSERT(memcmp(a.ip, e, sizeof(a.ip)) == 0);
2020-12-22 17:44:59 +08:00
2020-12-07 16:54:58 +08:00
ASSERT(strcmp(mg_hex("abc", 3, buf), "616263") == 0);
2020-12-12 18:03:19 +08:00
ASSERT(mg_url_decode("a=%", 3, buf, sizeof(buf), 0) < 0);
ASSERT(mg_url_decode("&&&a=%", 6, buf, sizeof(buf), 0) < 0);
2021-02-15 22:42:55 +08:00
memset(a.ip, 0xaa, sizeof(a.ip));
ASSERT(mg_aton(mg_str("::1%1"), &a) == true);
ASSERT(a.is_ip6 == true);
ASSERT(a.scope_id == 1);
memset(a.ip, 0xaa, sizeof(a.ip));
ASSERT(mg_aton(mg_str("abcd::aabb:ccdd%17"), &a) == true);
ASSERT(a.is_ip6 == true);
ASSERT(a.scope_id == 17);
memset(a.ip, 0xaa, sizeof(a.ip));
ASSERT(mg_aton(mg_str("::1%17"), &a) == true);
ASSERT(a.is_ip6 == true);
ASSERT(a.scope_id == 17);
memset(a.ip, 0xaa, sizeof(a.ip));
ASSERT(mg_aton(mg_str("::1%255"), &a) == true);
ASSERT(a.is_ip6 == true);
ASSERT(a.scope_id == 255);
2021-02-15 22:42:55 +08:00
{
size_t n;
2021-02-15 22:42:55 +08:00
ASSERT((n = mg_url_encode("", 0, buf, sizeof(buf))) == 0);
ASSERT((n = mg_url_encode("a", 1, buf, 0)) == 0);
ASSERT((n = mg_url_encode("a", 1, buf, sizeof(buf))) == 1);
ASSERT(strncmp(buf, "a", n) == 0);
ASSERT((n = mg_url_encode("._-~", 4, buf, sizeof(buf))) == 4);
ASSERT(strncmp(buf, "._-~", n) == 0);
ASSERT((n = mg_url_encode("a@%>", 4, buf, sizeof(buf))) == 10);
ASSERT(strncmp(buf, "a%40%25%3e", n) == 0);
ASSERT((n = mg_url_encode("a@b.c", 5, buf, sizeof(buf))) == 7);
ASSERT(strncmp(buf, "a%40b.c", n) == 0);
}
{
2022-08-13 22:37:55 +08:00
s = mg_mprintf("%3d", 123);
ASSERT(strcmp(s, "123") == 0);
free(s);
2021-02-15 22:42:55 +08:00
}
2021-03-22 00:18:44 +08:00
{
extern bool mg_to_size_t(struct mg_str, size_t *);
size_t val, max = (size_t) -1;
ASSERT(mg_to_size_t(mg_str("0"), &val) && val == 0);
ASSERT(mg_to_size_t(mg_str("123"), &val) && val == 123);
ASSERT(mg_to_size_t(mg_str(" 123 \t"), &val) && val == 123);
ASSERT(mg_to_size_t(mg_str(""), &val) == false);
ASSERT(mg_to_size_t(mg_str(" 123x"), &val) == false);
ASSERT(mg_to_size_t(mg_str("-"), &val) == false);
mg_snprintf(buf, sizeof(buf), sizeof(max) == 8 ? "%llu" : "%lu", max);
ASSERT(mg_to_size_t(mg_str(buf), &val) && val == max);
}
{
size_t i;
memset(buf, ' ', sizeof(buf));
mg_random_str(buf, 0);
ASSERT(buf[0] == ' ');
mg_random_str(buf, 1);
ASSERT(buf[0] == '\0');
ASSERT(buf[1] == ' ');
mg_random_str(buf, sizeof(buf));
ASSERT(buf[sizeof(buf) - 1] == '\0');
for (i = 0; i < sizeof(buf) - 1; i++) ASSERT(isalnum((uint8_t) buf[i]));
}
2020-12-07 16:54:58 +08:00
}
static void test_crc32(void) {
// echo -n aaa | cksum -o3
ASSERT(mg_crc32(0, 0, 0) == 0);
ASSERT(mg_crc32(0, "a", 1) == 3904355907);
ASSERT(mg_crc32(0, "abc", 3) == 891568578);
ASSERT(mg_crc32(mg_crc32(0, "ab", 2), "c", 1) == 891568578);
}
static void us(struct mg_connection *c, int ev, void *ev_data) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
2023-09-27 02:59:42 +08:00
if (ev == MG_EV_HTTP_MSG && mg_http_match_uri(hm, "/upload")) {
MG_DEBUG(("Got all %lu bytes!", (unsigned long) hm->body.len));
MG_DEBUG(("Query string: [%.*s]", (int) hm->query.len, hm->query.ptr));
// MG_DEBUG(("Body:\n%.*s", (int) hm->body.len, hm->body.ptr));
mg_http_reply(c, 200, "", "ok (%d %.*s)\n", (int) hm->body.len,
(int) hm->body.len, hm->body.ptr);
} else if (ev == MG_EV_HTTP_MSG) {
mg_http_reply(c, 200, "", "ok\n");
}
}
static void uc(struct mg_connection *c, int ev, void *ev_data) {
const char **s = (const char **) c->fn_data;
if (ev == MG_EV_OPEN) {
2023-09-27 02:59:42 +08:00
// c->is_hexdumping = 1;
} else if (ev == MG_EV_CONNECT) {
mg_printf(c,
"POST /upload HTTP/1.0\r\n"
"Transfer-Encoding: chunked\r\n\r\n");
mg_http_printf_chunk(c, "%s", "foo\n");
mg_http_printf_chunk(c, "%s", "bar\n");
mg_http_printf_chunk(c, "");
} else if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
2023-09-27 02:59:42 +08:00
// MG_INFO(("---> [%s] [%.*s]", *s, hm->body.len, hm->body.ptr));
ASSERT(mg_strcmp(hm->body, mg_str(*s)) == 0);
*s = NULL;
}
}
static void test_http_upload(void) {
struct mg_mgr mgr;
const char *url = "http://127.0.0.1:12352";
2023-09-27 02:59:42 +08:00
int i;
const char *s = "ok (8 foo\nbar\n)\n";
mg_mgr_init(&mgr);
2023-09-27 02:59:42 +08:00
mg_http_listen(&mgr, url, us, NULL);
mg_http_connect(&mgr, url, uc, (void *) &s);
for (i = 0; i < 20; i++) mg_mgr_poll(&mgr, 5);
2023-09-27 02:59:42 +08:00
ASSERT(s == NULL);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
#define LONG_CHUNK "chunk with length taking up more than two hex digits"
static void eX(struct mg_connection *c, int ev, void *ev_data) {
if (ev == MG_EV_HTTP_MSG) {
mg_printf(c, "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n");
c->data[0] = 1;
2023-09-27 02:59:42 +08:00
// c->is_hexdumping = 1;
} else if (ev == MG_EV_POLL && c->data[0] != 0) {
c->data[0]++;
if (c->data[0] == 10) mg_http_printf_chunk(c, "a");
if (c->data[0] == 20) {
2022-08-05 00:08:30 +08:00
mg_http_printf_chunk(c, "b");
mg_http_printf_chunk(c, "c");
}
if (c->data[0] == 30) {
2022-08-05 00:08:30 +08:00
mg_http_printf_chunk(c, "d");
mg_http_printf_chunk(c, "");
c->data[0] = 0;
}
}
(void) ev_data;
}
static void eY(struct mg_connection *c, int ev, void *ev_data) {
2022-08-05 00:08:30 +08:00
if (ev == MG_EV_HTTP_MSG) {
mg_printf(c, "HTTP/1.1 200 OK\r\nContent-Length: 4\r\n\r\n");
c->data[0] = 1;
} else if (ev == MG_EV_POLL && c->data[0] != 0) {
c->data[0]++;
if (c->data[0] == 10) mg_send(c, "a", 1);
if (c->data[0] == 12) mg_send(c, "bc", 2);
if (c->data[0] == 30) mg_send(c, "d", 1), c->is_resp = 0, c->data[0] = 0;
}
(void) ev_data;
}
static void eZ(struct mg_connection *c, int ev, void *ev_data) {
if (ev == MG_EV_HTTP_MSG) {
mg_http_reply(c, 200, "", "abcd");
}
(void) ev_data;
}
2022-08-05 00:08:30 +08:00
// Do not delete chunks as they arrive
static void eh4(struct mg_connection *c, int ev, void *ev_data) {
uint32_t *crc = (uint32_t *) c->fn_data;
2023-09-27 02:59:42 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
*crc = mg_crc32(*crc, hm->body.ptr, hm->body.len);
MG_INFO(("%lu M [%.*s]", c->id, (int) hm->body.len, hm->body.ptr));
}
}
2022-08-05 00:08:30 +08:00
static void test_http_chunked_case(mg_event_handler_t s, mg_event_handler_t c,
int req_count, const char *expected) {
2023-05-23 06:13:48 +08:00
char url[100];
struct mg_mgr mgr;
uint32_t i, crc = 0, expected_crc = mg_crc32(0, expected, strlen(expected));
struct mg_connection *conn;
2023-06-01 08:23:26 +08:00
static uint16_t port = 32344; // To prevent bind errors on Windows
2023-05-23 06:13:48 +08:00
mg_snprintf(url, sizeof(url), "http://127.0.0.1:%d", port++);
mg_mgr_init(&mgr);
2022-08-05 00:08:30 +08:00
mg_http_listen(&mgr, url, s, NULL);
conn = mg_http_connect(&mgr, url, c, &crc);
while (conn != NULL && req_count-- > 0) {
mg_printf(conn, "GET / HTTP/1.0\n\n");
}
for (i = 0; i < 100 && crc != expected_crc; i++) {
2022-08-05 00:08:30 +08:00
mg_mgr_poll(&mgr, 1);
}
2023-09-27 02:59:42 +08:00
// MG_INFO(("-------- %d [%s]", i, expected));
ASSERT(i < 100);
ASSERT(crc == expected_crc);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2022-08-05 00:08:30 +08:00
static void test_http_chunked(void) {
// Non-chunked encoding
2023-09-27 02:59:42 +08:00
test_http_chunked_case(eY, eh4, 1, "abcd"); // Chunks not deleted
test_http_chunked_case(eY, eh4, 2, "abcdabcd");
test_http_chunked_case(eZ, eh4, 1, "abcd"); // Not deleted
test_http_chunked_case(eZ, eh4, 2, "abcdabcd");
2022-08-05 00:08:30 +08:00
// Chunked encoding
2023-09-27 02:59:42 +08:00
test_http_chunked_case(eX, eh4, 1, "abcd"); // Chunks not deleted
test_http_chunked_case(eX, eh4, 2, "abcdabcd");
}
static void test_invalid_listen_addr(void) {
struct mg_mgr mgr;
struct mg_connection *c;
mg_mgr_init(&mgr);
c = mg_http_listen(&mgr, "invalid:31:14", eh1, NULL);
ASSERT(c == NULL);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
struct stream_status {
uint32_t polls;
size_t sent;
size_t received;
uint32_t send_crc;
uint32_t recv_crc;
};
// Consume recv buffer after letting it reach MG_MAX_RECV_SIZE
static void eh8(struct mg_connection *c, int ev, void *ev_data) {
struct stream_status *status = (struct stream_status *) c->fn_data;
if (c->is_listening) return;
ASSERT(c->recv.len <= MG_MAX_RECV_SIZE);
if (ev == MG_EV_ACCEPT) {
// Optimize recv buffer size near max to speed up test
mg_iobuf_resize(&c->recv, MG_MAX_RECV_SIZE - MG_IO_SIZE);
status->received = 0;
status->recv_crc = 0;
}
if (ev == MG_EV_CLOSE) {
ASSERT(status->received == status->sent);
}
// Let buffer fill up and start consuming after 10 full buffer poll events
if (status->polls >= 10 && ev == MG_EV_POLL) {
// consume at most a third of MG_MAX_RECV_SIZE on each poll
size_t consume;
if (MG_MAX_RECV_SIZE / 3 >= c->recv.len)
consume = c->recv.len;
else
consume = MG_MAX_RECV_SIZE / 3;
status->received += consume;
status->recv_crc =
mg_crc32(status->recv_crc, (const char *) c->recv.buf, consume);
mg_iobuf_del(&c->recv, 0, consume);
}
// count polls with full buffer to ensure c->is_full prevents reads
if (ev == MG_EV_POLL && c->recv.len == MG_MAX_RECV_SIZE) status->polls += 1;
(void) ev_data;
}
// Toggle c->is_full to prevent max_recv_buf_size reached read errors
static void eh10(struct mg_connection *c, int ev, void *ev_data) {
if (c->recv.len >= MG_MAX_RECV_SIZE && ev == MG_EV_READ) c->is_full = true;
eh8(c, ev, ev_data);
if (c->recv.len < MG_MAX_RECV_SIZE && ev == MG_EV_POLL) c->is_full = false;
}
// Send buffer larger than MG_MAX_RECV_SIZE to server
static void eh11(struct mg_connection *c, int ev, void *ev_data) {
struct stream_status *status = (struct stream_status *) c->fn_data;
if (ev == MG_EV_CONNECT) {
size_t len = MG_MAX_RECV_SIZE * 2;
2022-08-01 19:53:25 +08:00
struct mg_iobuf buf = {NULL, 0, 0, 0};
mg_iobuf_init(&buf, len, 0);
mg_random(buf.buf, buf.size);
buf.len = buf.size;
mg_send(c, buf.buf, buf.len);
status->sent = buf.len;
status->send_crc = mg_crc32(0, (const char *) buf.buf, buf.len);
mg_iobuf_free(&buf);
}
(void) ev_data;
}
static void test_http_stream_buffer(void) {
struct mg_mgr mgr;
const char *url = "tcp://127.0.0.1:12344";
uint32_t i;
struct stream_status status;
mg_mgr_init(&mgr);
mg_listen(&mgr, url, eh10, &status);
status.polls = 0;
mg_connect(&mgr, url, eh11, &status);
for (i = 0; i < (MG_MAX_RECV_SIZE / MG_IO_SIZE) * 50; i++) {
mg_mgr_poll(&mgr, 1);
if (status.polls >= 10 && status.sent == status.received) break;
}
ASSERT(status.sent == status.received);
ASSERT(status.send_crc == status.recv_crc);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2021-03-17 15:43:29 +08:00
static void test_multipart(void) {
struct mg_http_part part;
size_t ofs;
const char *s =
"--xyz\r\n"
"Content-Disposition: form-data; name=\"val\"\r\n"
"\r\n"
2021-03-17 21:28:36 +08:00
"abc\r\ndef\r\n"
2021-03-17 15:43:29 +08:00
"--xyz\r\n"
2021-03-17 21:28:36 +08:00
"Content-Disposition: form-data; name=\"foo\"; filename=\"a b.txt\"\r\n"
2021-03-17 15:43:29 +08:00
"Content-Type: text/plain\r\n"
"\r\n"
"hello world\r\n"
"\r\n"
"--xyz--\r\n";
ASSERT(mg_http_next_multipart(mg_str(""), 0, NULL) == 0);
2021-03-17 15:46:39 +08:00
ASSERT((ofs = mg_http_next_multipart(mg_str(s), 0, &part)) > 0);
2021-03-17 15:43:29 +08:00
ASSERT(mg_strcmp(part.name, mg_str("val")) == 0);
// MG_INFO(("--> [%.*s]", (int) part.body.len, part.body.ptr));
2021-03-17 21:28:36 +08:00
ASSERT(mg_strcmp(part.body, mg_str("abc\r\ndef")) == 0);
2021-03-17 15:43:29 +08:00
ASSERT(part.filename.len == 0);
2021-03-17 15:46:39 +08:00
ASSERT((ofs = mg_http_next_multipart(mg_str(s), ofs, &part)) > 0);
2021-03-17 15:43:29 +08:00
ASSERT(mg_strcmp(part.name, mg_str("foo")) == 0);
// MG_INFO(("--> [%.*s]", (int) part.filename.len, part.filename.ptr));
2021-03-17 21:28:36 +08:00
ASSERT(mg_strcmp(part.filename, mg_str("a b.txt")) == 0);
ASSERT(mg_strcmp(part.body, mg_str("hello world\r\n")) == 0);
2021-03-17 15:43:29 +08:00
ASSERT(mg_http_next_multipart(mg_str(s), ofs, &part) == 0);
}
static void eh7(struct mg_connection *c, int ev, void *ev_data) {
2021-07-29 21:21:20 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
struct mg_http_serve_opts sopts;
memset(&sopts, 0, sizeof(sopts));
2022-02-15 01:44:43 +08:00
sopts.root_dir = "/";
2021-07-29 21:21:20 +08:00
sopts.fs = &mg_fs_packed;
mg_http_serve_dir(c, hm, &sopts);
}
}
2021-07-26 16:05:34 +08:00
static void test_packed(void) {
2021-07-29 04:11:07 +08:00
struct mg_mgr mgr;
const char *url = "http://127.0.0.1:12351";
char buf[FETCH_BUF_SIZE];
struct mg_str data = mg_file_read(&mg_fs_posix, "Makefile");
2021-07-29 04:11:07 +08:00
mg_mgr_init(&mgr);
2021-07-29 21:21:20 +08:00
mg_http_listen(&mgr, url, eh7, NULL);
2021-07-31 00:41:23 +08:00
// Load top level file directly
2022-08-05 00:08:30 +08:00
// fetch(&mgr, buf, url, "GET /Makefile HTTP/1.0\n\n");
2022-06-14 04:12:11 +08:00
// printf("---> %s\n", buf);
2021-07-30 20:19:20 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /Makefile HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, data.ptr) == 0);
free((void *) data.ptr);
2021-07-31 00:41:23 +08:00
// Load file deeper in the FS tree directly
data = mg_file_read(&mg_fs_posix, "src/ssi.h");
2021-07-31 00:41:23 +08:00
ASSERT(fetch(&mgr, buf, url, "GET /src/ssi.h HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, data.ptr) == 0);
free((void *) data.ptr);
2021-07-31 00:41:23 +08:00
// List root dir
2021-07-30 20:19:20 +08:00
ASSERT(fetch(&mgr, buf, url, "GET / HTTP/1.0\n\n") == 200);
// printf("--------\n%s\n", buf);
2021-07-31 00:41:23 +08:00
// List nested dir
ASSERT(fetch(&mgr, buf, url, "GET /test HTTP/1.0\n\n") == 301);
ASSERT(fetch(&mgr, buf, url, "GET /test/ HTTP/1.0\n\n") == 200);
2021-07-31 00:41:23 +08:00
// printf("--------\n%s\n", buf);
2021-07-29 04:11:07 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
2021-07-26 16:05:34 +08:00
}
#if (MG_ENABLE_SOCKET == 0)
2022-04-22 21:42:07 +08:00
int send(int sock, const void *buf, size_t len, int flags);
int send(int sock, const void *buf, size_t len, int flags) {
(void) sock, (void) buf, (void) len, (void) flags;
return -1;
}
#endif
static void u1(struct mg_connection *c, int ev, void *ev_data) {
2021-08-08 05:55:13 +08:00
if (ev == MG_EV_CONNECT) {
((int *) c->fn_data)[0] += 1;
2021-08-08 05:55:13 +08:00
mg_send(c, "hi", 2);
2022-01-05 21:03:56 +08:00
} else if (ev == MG_EV_WRITE) {
((int *) c->fn_data)[0] += 100;
2021-08-08 05:55:13 +08:00
} else if (ev == MG_EV_READ) {
((int *) c->fn_data)[0] += 10;
2021-08-08 05:55:13 +08:00
mg_iobuf_free(&c->recv);
}
(void) ev_data;
}
static void test_udp(void) {
struct mg_mgr mgr;
const char *url = "udp://127.0.0.1:12353";
int i, done = 0;
mg_mgr_init(&mgr);
mg_listen(&mgr, url, u1, (void *) &done);
mg_connect(&mgr, url, u1, (void *) &done);
for (i = 0; i < 5; i++) mg_mgr_poll(&mgr, 1);
// MG_INFO(("%d", done));
2022-01-05 21:03:56 +08:00
ASSERT(done == 111);
2021-08-08 05:55:13 +08:00
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
2021-08-25 14:59:31 +08:00
static void test_check_ip_acl(void) {
struct mg_addr ip = {{1, 2, 3, 4}, 0, 0, false}; // 1.2.3.4
2023-07-05 22:42:40 +08:00
ASSERT(mg_check_ip_acl(mg_str(NULL), &ip) == 1);
ASSERT(mg_check_ip_acl(mg_str(""), &ip) == 1);
ASSERT(mg_check_ip_acl(mg_str("invalid"), &ip) == -1);
ASSERT(mg_check_ip_acl(mg_str("+hi"), &ip) == -2);
ASSERT(mg_check_ip_acl(mg_str("+//"), &ip) == -2);
ASSERT(mg_check_ip_acl(mg_str("-0.0.0.0/0"), &ip) == 0);
ASSERT(mg_check_ip_acl(mg_str("-0.0.0.0/0,+1.0.0.0/8"), &ip) == 1);
ASSERT(mg_check_ip_acl(mg_str("-0.0.0.0/0,+1.2.3.4"), &ip) == 1);
ASSERT(mg_check_ip_acl(mg_str("-0.0.0.0/0,+1.0.0.0/16"), &ip) == 0);
ip.is_ip6 = true;
ASSERT(mg_check_ip_acl(mg_str("-0.0.0.0/0"), &ip) ==
-1); // not yet supported
2021-08-25 14:59:31 +08:00
}
static void w3(struct mg_connection *c, int ev, void *ev_data) {
// MG_INFO(("ev %d", ev));
2021-08-27 17:25:24 +08:00
if (ev == MG_EV_WS_OPEN) {
char buf[8192];
memset(buf, 'A', sizeof(buf));
2021-08-27 17:25:24 +08:00
mg_ws_send(c, "hi there!", 9, WEBSOCKET_OP_TEXT);
mg_ws_printf(c, WEBSOCKET_OP_TEXT, "%s", "hi there2!");
mg_printf(c, "%s", "boo");
mg_ws_wrap(c, 3, WEBSOCKET_OP_TEXT);
mg_ws_send(c, buf, sizeof(buf), WEBSOCKET_OP_TEXT);
2021-08-27 17:25:24 +08:00
} else if (ev == MG_EV_WS_MSG) {
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
ASSERT(mg_strcmp(wm->data, mg_str("lebowski")) == 0);
((int *) c->fn_data)[0]++;
2021-08-27 17:25:24 +08:00
} else if (ev == MG_EV_CLOSE) {
((int *) c->fn_data)[0] += 10;
2021-08-27 17:25:24 +08:00
}
}
static void w2(struct mg_connection *c, int ev, void *ev_data) {
2021-08-27 17:25:24 +08:00
struct mg_str msg = mg_str_n("lebowski", 8);
if (ev == MG_EV_HTTP_MSG) {
mg_ws_upgrade(c, (struct mg_http_message *) ev_data, NULL);
} else if (ev == MG_EV_WS_OPEN) {
mg_ws_send(c, "x", 1, WEBSOCKET_OP_PONG);
} else if (ev == MG_EV_POLL && c->is_websocket) {
size_t ofs, n = (size_t) c->fn_data;
2021-08-27 17:25:24 +08:00
if (n < msg.len) {
// Send "msg" char by char using fragmented frames
// mg_ws_send() sets the FIN flag in the WS header. Clean it
// to send fragmented packet. Insert PONG messages between frames
uint8_t op = n == 0 ? WEBSOCKET_OP_TEXT : WEBSOCKET_OP_CONTINUE;
mg_ws_send(c, ":->", 3, WEBSOCKET_OP_PING);
ofs = c->send.len;
mg_ws_send(c, &msg.ptr[n], 1, op);
if (n < msg.len - 1) c->send.buf[ofs] = op; // Clear FIN flag
c->fn_data = (void *) (n + 1); // Point to the next char
} else {
mg_ws_send(c, "", 0, WEBSOCKET_OP_CLOSE);
}
} else if (ev == MG_EV_WS_MSG) {
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
MG_INFO(("Got WS, %lu", wm->data.len));
2022-11-30 22:47:37 +08:00
// mg_hexdump(wm->data.ptr, wm->data.len);
if (wm->data.len == 9) {
ASSERT(mg_strcmp(wm->data, mg_str("hi there!")) == 0);
} else if (wm->data.len == 10) {
ASSERT(mg_strcmp(wm->data, mg_str("hi there2!")) == 0);
} else if (wm->data.len == 3) {
ASSERT(mg_strcmp(wm->data, mg_str("boo")) == 0);
} else {
MG_INFO(("%lu", wm->data.len));
ASSERT(wm->data.len == 8192);
}
2021-08-27 17:25:24 +08:00
}
}
static void test_ws_fragmentation(void) {
const char *url = "ws://localhost:12357/ws";
struct mg_mgr mgr;
int i, done = 0;
mg_mgr_init(&mgr);
ASSERT(mg_http_listen(&mgr, url, w2, NULL) != NULL);
mg_ws_connect(&mgr, url, w3, &done, "%s", "Sec-WebSocket-Protocol: echo\r\n");
for (i = 0; i < 25; i++) mg_mgr_poll(&mgr, 1);
// MG_INFO(("--> %d", done));
2021-08-27 17:25:24 +08:00
ASSERT(done == 11);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
static void h7(struct mg_connection *c, int ev, void *ev_data) {
2021-09-20 19:24:44 +08:00
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
struct mg_http_serve_opts opts;
memset(&opts, 0, sizeof(opts));
opts.root_dir = "./test/data,/foo=./src";
mg_http_serve_dir(c, hm, &opts);
}
}
static void test_rewrites(void) {
char buf[FETCH_BUF_SIZE];
const char *url = "http://LOCALHOST:12358";
const char *expected = "#define MG_VERSION \"" MG_VERSION "\"\n";
struct mg_mgr mgr;
mg_mgr_init(&mgr);
ASSERT(mg_http_listen(&mgr, url, h7, NULL) != NULL);
ASSERT(fetch(&mgr, buf, url, "GET /a.txt HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, "hello\n") == 0);
ASSERT(fetch(&mgr, buf, url, "GET /foo/version.h HTTP/1.0\n\n") == 200);
ASSERT(cmpbody(buf, expected) == 0);
ASSERT(fetch(&mgr, buf, url, "GET /foo HTTP/1.0\n\n") == 301);
ASSERT(fetch(&mgr, buf, url, "GET /foo/ HTTP/1.0\n\n") == 200);
2021-09-20 19:24:44 +08:00
// printf("-->[%s]\n", buf);
mg_mgr_free(&mgr);
ASSERT(mgr.conns == NULL);
}
static void test_get_header_var(void) {
struct mg_str empty = mg_str(""), bar = mg_str("bar"), baz = mg_str("baz");
struct mg_str header = mg_str("Digest foo=\"bar\", blah,boo=baz, x=\"yy\"");
struct mg_str yy = mg_str("yy");
// struct mg_str x = mg_http_get_header_var(header, mg_str("x"));
// MG_INFO(("--> [%d] [%d]", (int) x.len, yy.len));
ASSERT(mg_strcmp(empty, mg_http_get_header_var(empty, empty)) == 0);
ASSERT(mg_strcmp(empty, mg_http_get_header_var(header, empty)) == 0);
ASSERT(mg_strcmp(empty, mg_http_get_header_var(header, mg_str("fooo"))) == 0);
ASSERT(mg_strcmp(empty, mg_http_get_header_var(header, mg_str("fo"))) == 0);
ASSERT(mg_strcmp(empty, mg_http_get_header_var(header, mg_str("blah"))) == 0);
ASSERT(mg_strcmp(bar, mg_http_get_header_var(header, mg_str("foo"))) == 0);
ASSERT(mg_strcmp(baz, mg_http_get_header_var(header, mg_str("boo"))) == 0);
ASSERT(mg_strcmp(yy, mg_http_get_header_var(header, mg_str("x"))) == 0);
}
static void json_scan(struct mg_str json, int depth) {
int i, n = 0, o = mg_json_get(json, "$", &n);
for (i = 0; i < depth; i++) printf(" ");
printf("%.*s\n", n, json.ptr + o);
if (json.ptr[o] == '{' || json.ptr[o] == '[') { // Iterate over elems
struct mg_str key, val, sub = mg_str_n(json.ptr + o, (size_t) n);
size_t ofs = 0;
while ((ofs = mg_json_next(sub, ofs, &key, &val)) > 0) {
for (i = 0; i < depth; i++) printf(" ");
printf("KEY: %.*s VAL: %.*s\n", (int) key.len, key.ptr, (int) val.len,
val.ptr);
if (*val.ptr == '[' || *val.ptr == '{') json_scan(val, depth + 1);
}
}
}
2022-06-09 19:39:48 +08:00
static void test_json(void) {
const char *s1 = "{\"a\":{},\"b\":7,\"c\":[[],2]}";
2022-07-22 22:50:23 +08:00
const char *s2 = "{\"a\":{\"b1\":{}},\"c\":7,\"d\":{\"b2\":{}}}";
2022-07-30 14:55:26 +08:00
int n;
2024-03-06 08:57:31 +08:00
struct mg_str json;
2022-07-30 14:55:26 +08:00
ASSERT(mg_json_get(mg_str_n(" true ", 6), "", &n) == MG_JSON_INVALID);
ASSERT(mg_json_get(mg_str_n(" true ", 6), "$", &n) == 1 && n == 4);
ASSERT(mg_json_get(mg_str_n("null ", 5), "$", &n) == 0 && n == 4);
json = mg_str(" \"hi\\nthere\"");
ASSERT(mg_json_get(json, "$", &n) == 2 && n == 11);
ASSERT(mg_json_get(mg_str_n(" { } ", 5), "$", &n) == 1);
ASSERT(mg_json_get(mg_str_n(" [[]]", 5), "$", &n) == 1);
ASSERT(mg_json_get(mg_str_n(" [ ] ", 5), "$", &n) == 1);
ASSERT(mg_json_get(mg_str_n("[1,2]", 5), "$", &n) == 0 && n == 5);
ASSERT(mg_json_get(mg_str_n("[1,2]", 5), "$[0]", &n) == 1 && n == 1);
ASSERT(mg_json_get(mg_str_n("[1,2]", 5), "$[1]", &n) == 3 && n == 1);
ASSERT(mg_json_get(mg_str_n("[1,2]", 5), "$[3]", &n) == MG_JSON_NOT_FOUND);
json = mg_str("{\"a\":[]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 2);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":[1,2]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 5);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":[1,[1]]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 7);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":[[]]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 4);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":[[1,2]]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 7);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":{}}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 2);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":{\"a\":{}}}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 8);
2022-07-30 14:55:26 +08:00
json = mg_str("{\"a\":{\"a\":[]}}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 8);
2022-07-30 14:55:26 +08:00
json = mg_str("[[1,[2,3]],4]");
ASSERT(mg_json_get(json, "$", &n) == 0 && n == 13);
ASSERT(mg_json_get(json, "$[0]", &n) == 1 && n == 9);
ASSERT(mg_json_get(json, "$[1]", &n) == 11);
ASSERT(mg_json_get(json, "$[1]", &n) == 11 && n == 1);
ASSERT(mg_json_get(json, "$[2]", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$[0][0]", &n) == 2 && n == 1);
ASSERT(mg_json_get(json, "$[0][1]", &n) == 4 && n == 5);
ASSERT(mg_json_get(json, "$[0][2]", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$[0][1][0]", &n) == 5 && n == 1);
ASSERT(mg_json_get(json, "$[0][1][1]", &n) == 7 && n == 1);
json = mg_str("[[1,2],3]");
ASSERT(mg_json_get(json, "$", &n) == 0 && n == 9);
ASSERT(mg_json_get(json, "$[0][0]", &n) == 2 && n == 1);
ASSERT(mg_json_get(json, "$[0][1]", &n) == 4 && n == 1);
ASSERT(mg_json_get(json, "$[0][2]", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$[1][0]", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$[1]", &n) == 7 && n == 1);
ASSERT(mg_json_get(json, "$[1][0]", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$", &n) == 0 && n == 9);
ASSERT(mg_json_get(json, "$[1][0]", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$[0][1]", &n) == 4 && n == 1);
json = mg_str(s1);
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 2);
ASSERT(mg_json_get(json, "$.b", &n) == 12 && n == 1);
ASSERT(mg_json_get(json, "$.c", &n) == 18 && n == 6);
ASSERT(mg_json_get(json, "$.c[0]", &n) == 19 && n == 2);
ASSERT(mg_json_get(json, "$.c[1]", &n) == 22 && n == 1);
ASSERT(mg_json_get(json, "$.c[3]", &n) == MG_JSON_NOT_FOUND);
json = mg_str(s2);
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 9);
ASSERT(mg_json_get(json, "$.a.b1", &n) == 11 && n == 2);
ASSERT(mg_json_get(json, "$.a.b2", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$.a.b", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$.a1", &n) == MG_JSON_NOT_FOUND);
ASSERT(mg_json_get(json, "$.c", &n) == 19 && n == 1);
2022-06-09 19:39:48 +08:00
{
double d = 0;
bool b = false;
int len;
2022-07-30 14:55:26 +08:00
char *str = NULL;
json = mg_str("{\"a\":\"b\"}");
str = mg_json_get_str(json, "$.a");
ASSERT(str != NULL);
// printf("---> [%s]\n", str);
ASSERT(strcmp(str, "b") == 0);
free(str);
json = mg_str("{\"a\": \"hi\\nthere\",\"b\": [12345, true]}");
str = mg_json_get_str(json, "$.a");
2022-06-09 19:39:48 +08:00
ASSERT(str != NULL);
ASSERT(strcmp(str, "hi\nthere") == 0);
free(str);
2022-07-30 14:55:26 +08:00
ASSERT(mg_json_get_long(json, "$.foo", -42) == -42);
ASSERT(mg_json_get_long(json, "$.b[0]", -42) == 12345);
2022-07-01 03:03:02 +08:00
2022-07-30 14:55:26 +08:00
ASSERT(mg_json_get_num(json, "$.a", &d) == false);
ASSERT(mg_json_get_num(json, "$.c", &d) == false);
ASSERT(mg_json_get_num(json, "$.b[0]", &d) == true);
2022-06-09 19:39:48 +08:00
ASSERT(d == 12345);
2022-07-30 14:55:26 +08:00
ASSERT(mg_json_get_bool(json, "$.b", &b) == false);
ASSERT(mg_json_get_bool(json, "$.b[0]", &b) == false);
ASSERT(mg_json_get_bool(json, "$.b[1]", &b) == true);
2022-06-09 19:39:48 +08:00
ASSERT(b == true);
2022-09-02 23:23:29 +08:00
ASSERT(mg_json_get(json, "$.b[2]", &len) < 0);
2022-07-30 14:55:26 +08:00
json = mg_str("[\"YWJj\", \"0100026869\"]");
ASSERT((str = mg_json_get_b64(json, "$[0]", &len)) != NULL);
ASSERT(len == 3 && memcmp(str, "abc", (size_t) len) == 0);
free(str);
2022-07-30 14:55:26 +08:00
ASSERT((str = mg_json_get_hex(json, "$[1]", &len)) != NULL);
ASSERT(len == 5 && memcmp(str, "\x01\x00\x02hi", (size_t) len) == 0);
free(str);
json = mg_str("{\"a\":[1,2,3], \"ab\": 2}");
ASSERT(mg_json_get_long(json, "$.a[0]", -42) == 1);
ASSERT(mg_json_get_long(json, "$.ab", -42) == 2);
ASSERT(mg_json_get_long(json, "$.ac", -42) == -42);
2022-08-30 14:22:58 +08:00
json = mg_str("{\"a\":[],\"b\":[1,2]}");
ASSERT(mg_json_get_long(json, "$.a[0]", -42) == -42);
ASSERT(mg_json_get_long(json, "$.b[0]", -42) == 1);
ASSERT(mg_json_get_long(json, "$.b[1]", -42) == 2);
ASSERT(mg_json_get_long(json, "$.b[2]", -42) == -42);
json = mg_str("[{\"a\":1,\"b\":2},{\"a\":3, \"b\":4}]");
ASSERT(mg_json_get_long(json, "$[0].a", -42) == 1);
ASSERT(mg_json_get_long(json, "$[0].b", -42) == 2);
ASSERT(mg_json_get_long(json, "$[1].a", -42) == 3);
ASSERT(mg_json_get_long(json, "$[1].b", -42) == 4);
ASSERT(mg_json_get_long(json, "$[2].a", -42) == -42);
json = mg_str("{\"a\":[1],\"b\":[2,3]}");
ASSERT(mg_json_get_long(json, "$.a[0]", -42) == 1);
ASSERT(mg_json_get_long(json, "$.a[1]", -42) == -42);
2022-09-02 23:23:29 +08:00
json = mg_str("{\"a\":[1,[2,3], 4]}");
ASSERT(mg_json_get_long(json, "$.a[0]", -42) == 1);
ASSERT(mg_json_get_long(json, "$.a[1][0]", -42) == 2);
ASSERT(mg_json_get_long(json, "$.a[1][1]", -42) == 3);
ASSERT(mg_json_get_long(json, "$.a[1][2]", -42) == -42);
ASSERT(mg_json_get_long(json, "$.a[2]", -42) == 4);
ASSERT(mg_json_get_long(json, "$.a[3]", -42) == -42);
2022-06-09 19:39:48 +08:00
}
json = mg_str("{\"a\":[],\"b\":[1,2]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 2);
ASSERT(mg_json_get(json, "$.a[0]", &n) < 0 && n == 0);
json = mg_str("{\"a\":{},\"b\":[1,2]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 2);
ASSERT(mg_json_get(json, "$.a[0]", &n) < 0 && n == 0);
json = mg_str("{\"a\":true,\"b\":[1,2]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 4);
ASSERT(mg_json_get(json, "$.a[0]", &n) < 0 && n == 0);
json = mg_str("{\"a\":1,\"b\":[1,2]}");
ASSERT(mg_json_get(json, "$.a", &n) == 5 && n == 1);
ASSERT(mg_json_get(json, "$.a[0]", &n) < 0 && n == 0);
ASSERT(mg_json_get_long(mg_str("[0, 42]"), "$[1]", 0) == 42);
ASSERT(mg_json_get_long(mg_str("[[], 42]"), "$[1]", 0) == 42);
ASSERT(mg_json_get_long(mg_str("[{}, 42]"), "$[1]", 0) == 42);
json = mg_str("[{\"a\":1},{\"a\":2}]");
ASSERT(mg_json_get_long(json, "$[0]", -1) == -1);
ASSERT(mg_json_get_long(json, "$[0].a", -1) == 1);
ASSERT(mg_json_get_long(json, "$[1].a", -1) == 2);
ASSERT(mg_json_get_long(json, "$[2].a", -1) == -1);
json = mg_str("[{\"a1\":1},{\"a\":2}]");
ASSERT(mg_json_get_long(json, "$[0]", -1) == -1);
ASSERT(mg_json_get_long(json, "$[0].a", -1) == -1);
ASSERT(mg_json_get_long(json, "$[1].a", -1) == 2);
ASSERT(mg_json_get_long(json, "$[2].a", -1) == -1);
// mg_json_next()
json = mg_str("[1,true,{\"a\":[3],\"b\":42}]");
json_scan(json, 0);
{
struct mg_str k, v, sub = mg_str_n(json.ptr + 8, json.len - 8);
const char *a = "\"a\"", *b = "\"b\"";
ASSERT(mg_json_next(sub, 0, &k, &v) == 9);
ASSERT(mg_vcmp(&k, a) == 0);
ASSERT(mg_vcmp(&v, "[3]") == 0);
ASSERT(mg_json_next(sub, 9, &k, &v) == 15);
ASSERT(mg_vcmp(&k, b) == 0);
ASSERT(mg_vcmp(&v, "42") == 0);
ASSERT(mg_json_next(sub, 15, &k, &v) == 0);
}
2024-03-06 08:49:55 +08:00
2024-03-06 08:57:31 +08:00
{
struct mg_str expected = mg_str("\"b:c\""), val;
json = mg_str("{\"a\":\"b:c\"}");
val = mg_json_get_tok(json, "$.a");
ASSERT(mg_strcmp(val, expected) == 0);
}
2022-06-09 19:39:48 +08:00
}
2022-08-27 00:41:13 +08:00
static void resp_rpc(struct mg_rpc_req *r) {
int len = 0, off = mg_json_get(r->frame, "$.result", &len);
mg_xprintf(r->pfn, r->pfn_data, "%.*s", len, &r->frame.ptr[off]);
}
2022-07-27 07:46:05 +08:00
static void test_rpc(void) {
2022-08-01 05:51:59 +08:00
struct mg_rpc *head = NULL;
struct mg_iobuf io = {0, 0, 0, 256};
struct mg_rpc_req req = {&head, 0, mg_pfn_iobuf, &io, 0, {0, 0}};
2022-07-30 14:55:26 +08:00
mg_rpc_add(&head, mg_str("rpc.list"), mg_rpc_list, NULL);
2022-07-27 07:46:05 +08:00
{
2022-07-30 14:55:26 +08:00
req.frame = mg_str("{\"method\":\"rpc.list\"}");
mg_rpc_process(&req);
ASSERT(io.buf == NULL);
2022-07-27 07:46:05 +08:00
}
{
const char *resp = "{\"id\":1,\"result\":[\"rpc.list\"]}";
2022-07-30 14:55:26 +08:00
req.frame = mg_str("{\"id\": 1,\"method\":\"rpc.list\"}");
mg_rpc_process(&req);
ASSERT(strcmp((char *) io.buf, resp) == 0);
mg_iobuf_free(&io);
2022-07-27 07:46:05 +08:00
}
{
const char *resp =
"{\"id\":true,\"error\":{\"code\":-32601,\"message\":\"foo not "
"found\"}}";
2022-07-30 14:55:26 +08:00
req.frame = mg_str("{\"id\": true,\"method\":\"foo\"}");
mg_rpc_process(&req);
// MG_INFO(("-> %s", io.buf));
ASSERT(strcmp((char *) io.buf, resp) == 0);
mg_iobuf_free(&io);
2022-07-27 07:46:05 +08:00
}
2022-08-19 22:07:50 +08:00
{
const char *resp =
"{\"id\":true,\"error\":{\"code\":-32601,\"message\":\"foo not "
"found\"}}";
req.frame = mg_str("{\"id\": true,\"method\":\"foo\"}");
req.head = NULL;
mg_rpc_process(&req);
// MG_INFO(("-> %s", io.buf));
ASSERT(strcmp((char *) io.buf, resp) == 0);
mg_iobuf_free(&io);
req.head = &head;
}
2022-07-27 07:46:05 +08:00
{
const char *resp = "{\"error\":{\"code\":-32700,\"message\":\"haha\"}}";
2022-07-30 14:55:26 +08:00
req.frame = mg_str("haha");
mg_rpc_process(&req);
2022-08-27 00:41:13 +08:00
// MG_INFO(("-> %s", io.buf));
ASSERT(strcmp((char *) io.buf, resp) == 0);
mg_iobuf_free(&io);
}
{
const char *resp =
"{\"id\":1,\"error\":{\"code\":-32601,\"message\":\" not found\"}}";
req.frame = mg_str("{\"id\":1,\"result\":123}");
mg_rpc_process(&req);
// MG_INFO(("-> %s", io.buf));
ASSERT(strcmp((char *) io.buf, resp) == 0);
mg_iobuf_free(&io);
2022-07-27 07:46:05 +08:00
}
2022-08-27 00:41:13 +08:00
{
req.frame = mg_str("{\"id\":1,\"result\":123}");
mg_rpc_add(&head, mg_str(""), resp_rpc, NULL);
mg_rpc_process(&req);
MG_INFO(("-> %s", io.buf));
ASSERT(strcmp((char *) io.buf, "123") == 0);
mg_iobuf_free(&io);
}
2022-07-31 04:13:30 +08:00
mg_rpc_del(&head, NULL);
2022-07-27 07:46:05 +08:00
ASSERT(head == NULL);
}
static void ph(struct mg_connection *c, int ev, void *ev_data) {
if (ev == MG_EV_POLL) ++(*(int *) c->fn_data);
2022-11-30 22:47:37 +08:00
(void) c, (void) ev_data;
}
static void test_poll(void) {
int count = 0, i;
struct mg_mgr mgr;
mg_mgr_init(&mgr);
2023-06-01 08:23:26 +08:00
mg_http_listen(&mgr, "http://127.0.0.1:42346", ph,
&count); // To prevent bind errors on Windows
2022-11-30 22:47:37 +08:00
for (i = 0; i < 10; i++) mg_mgr_poll(&mgr, 0);
ASSERT(count == 10);
mg_mgr_free(&mgr);
}
2023-02-13 04:30:18 +08:00
#define NMESSAGES 99999
static uint32_t s_qcrc = 0;
2023-02-13 04:30:18 +08:00
static int s_out, s_in;
static void producer(void *param) {
struct mg_queue *q = (struct mg_queue *) param;
2023-02-13 04:30:18 +08:00
char tmp[64 * 1024], *buf;
size_t len, ofs = sizeof(tmp);
for (s_out = 0; s_out < NMESSAGES; s_out++) {
if (ofs >= sizeof(tmp)) mg_random(tmp, sizeof(tmp)), ofs = 0;
len = ((uint8_t *) tmp)[ofs] % 55U + 1U;
2023-02-13 04:30:18 +08:00
if (ofs + len > sizeof(tmp)) len = sizeof(tmp) - ofs;
while ((mg_queue_book(q, &buf, len)) < len) (void) 0;
memcpy(buf, &tmp[ofs], len);
s_qcrc = mg_crc32(s_qcrc, buf, len);
ofs += len;
#if 0
fprintf(stderr, "-->prod %3d %8x %-3lu %zu/%zu/%lu\n", s_out, s_qcrc, len, q->tail,
q->head, buf - q->buf);
#endif
mg_queue_add(q, len);
}
}
static uint32_t consumer(struct mg_queue *q) {
2023-02-13 04:30:18 +08:00
uint32_t crc = 0;
for (s_in = 0; s_in < NMESSAGES; s_in++) {
char *buf;
size_t len;
while ((len = mg_queue_next(q, &buf)) == 0) (void) 0;
crc = mg_crc32(crc, buf, len);
#if 0
fprintf(stderr, "-->cons %3u %8x %-3lu %zu/%zu/%lu\n", s_in, crc, len, q->tail,
q->head, buf - q->buf);
#endif
mg_queue_del(q, len);
}
return crc;
}
#if MG_ARCH == MG_ARCH_WIN32
static void start_thread(void (*f)(void *), void *p) {
_beginthread((void(__cdecl *)(void *)) f, 0, p);
}
#elif MG_ARCH == MG_ARCH_UNIX
#include <pthread.h>
static void start_thread(void (*f)(void *), void *p) {
union {
void (*f1)(void *);
void *(*f2)(void *);
} u = {f};
pthread_t thread_id = (pthread_t) 0;
pthread_attr_t attr;
(void) pthread_attr_init(&attr);
(void) pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
pthread_create(&thread_id, &attr, u.f2, p);
pthread_attr_destroy(&attr);
}
#else
static void start_thread(void (*f)(void *), void *p) {
(void) f, (void) p;
}
#endif
static void test_queue(void) {
2023-02-13 04:30:18 +08:00
char buf[512];
struct mg_queue queue;
uint32_t crc;
memset(buf, 0x55, sizeof(buf));
mg_queue_init(&queue, buf, sizeof(buf));
start_thread(producer, &queue); // Start producer in a separate thread
crc = consumer(&queue); // Consumer eats data in this thread
MG_INFO(("CRC1 %8x", s_qcrc)); // Show CRCs
MG_INFO(("CRC2 %8x", crc));
ASSERT(s_qcrc == crc);
}
2023-10-13 18:32:12 +08:00
static void test_md5_str(const char *string,
const unsigned char *expected_hash) {
mg_md5_ctx ctx;
unsigned char digest[16];
mg_md5_init(&ctx);
mg_md5_update(&ctx, (unsigned char *) string, strlen(string));
mg_md5_final(&ctx, digest);
ASSERT((memcmp(digest, expected_hash, 16) == 0));
}
static void test_md5(void) {
const unsigned char expected_hash_1[] = {0xe5, 0x45, 0x14, 0x96, 0xe1, 0x1d,
0x7d, 0xa9, 0x62, 0x9f, 0xe0, 0x64,
0xcb, 0x3d, 0x2b, 0x54};
const unsigned char expected_hash_2[] = {0x99, 0x33, 0xf6, 0x4d, 0x7a, 0xb5,
0x0b, 0x0f, 0xf4, 0x35, 0xdc, 0x61,
0x1d, 0xef, 0x20, 0xff};
const unsigned char expected_hash_3[] = {0xf7, 0x94, 0xc3, 0xa4, 0x56, 0x6d,
0xc1, 0x10, 0x95, 0xfc, 0x56, 0x87,
0xf8, 0xb1, 0x69, 0xf2};
test_md5_str("#&*%$DHFH(0x12345)^&*(^!@$%^^&&*", expected_hash_1);
test_md5_str("1298**&^%DHKSHFLS)(*)&^^%$#!!!!", expected_hash_2);
test_md5_str(")_)+_)!&^*%$#>>>{}}}{{{][[[[]]]", expected_hash_3);
}
static void test_sha1_str(const char *string,
const unsigned char *expected_hash) {
mg_sha1_ctx ctx;
unsigned char digest[20];
mg_sha1_init(&ctx);
mg_sha1_update(&ctx, (unsigned char *) string, strlen(string));
mg_sha1_final(digest, &ctx);
ASSERT((memcmp(digest, expected_hash, 20) == 0));
}
static void test_sha1(void) {
const unsigned char expected_hash_1[] = {
0x02, 0xaf, 0x27, 0x00, 0xf7, 0xba, 0xb5, 0xf5, 0xf3, 0x69,
0xd8, 0x80, 0x01, 0x0d, 0x6a, 0x28, 0x31, 0x63, 0x1f, 0x92};
const unsigned char expected_hash_2[] = {
0xaa, 0xe4, 0x39, 0xe8, 0xb4, 0x72, 0x47, 0xe5, 0x1a, 0x6d,
0x82, 0x25, 0x5e, 0x9f, 0x32, 0xd9, 0x93, 0x0a, 0x5f, 0xfb};
const unsigned char expected_hash_3[] = {
0xa0, 0xdd, 0xd2, 0xa1, 0x52, 0xdf, 0xa9, 0xb8, 0x7e, 0x73,
0x32, 0x6a, 0x31, 0x28, 0xe9, 0x6d, 0x3a, 0x90, 0x82, 0x58};
test_sha1_str("#&*%$DHFH(0x12345)^&*(^!@$%^^&&*", expected_hash_1);
test_sha1_str("1298**&^%DHKSHFLS)(*)&^^%$#!!!!", expected_hash_2);
test_sha1_str(")_)+_)!&^*%$#>>>{}}}{{{][[[[]]]", expected_hash_3);
}
2024-02-02 02:21:46 +08:00
static void test_split(void) {
struct mg_str a, b, s;
2024-02-02 02:21:46 +08:00
s = mg_str("");
ASSERT(mg_span(s, &a, &s, '.') == false);
ASSERT(mg_span(s, &a, NULL, '.') == false);
ASSERT(mg_span(s, NULL, &b, '.') == false);
ASSERT(mg_span(s, NULL, NULL, '.') == false);
ASSERT(mg_span(s, &a, &b, '.') == false);
s = mg_str("aa.bb.cc"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, &a, &b, '.') == true);
ASSERT(mg_strcmp(a, mg_str("aa")) == 0);
ASSERT(mg_strcmp(b, mg_str("bb.cc")) == 0);
ASSERT(mg_strcmp(s, mg_str("aa.bb.cc")) == 0);
s = mg_str("aa.bb.cc"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, &a, NULL, '.') == true);
ASSERT(mg_strcmp(a, mg_str("aa")) == 0);
ASSERT(mg_strcmp(s, mg_str("aa.bb.cc")) == 0);
s = mg_str("aa.bb.cc"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, NULL, &b, '.') == true);
ASSERT(mg_strcmp(b, mg_str("bb.cc")) == 0);
ASSERT(mg_strcmp(s, mg_str("aa.bb.cc")) == 0);
s = mg_str("aa.bb.cc"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, NULL, NULL, '.') == true);
ASSERT(mg_strcmp(s, mg_str("aa.bb.cc")) == 0);
s = mg_str("aa.bb.cc"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, &a, &s, '.') == true);
ASSERT(mg_strcmp(a, mg_str("aa")) == 0);
ASSERT(mg_strcmp(s, mg_str("bb.cc")) == 0);
s = mg_str(".aa"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, &a, &b, '.') == true);
ASSERT(mg_strcmp(a, mg_str("")) == 0);
ASSERT(mg_strcmp(b, mg_str("aa")) == 0);
s = mg_str("aa."), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, &a, &b, '.') == true);
ASSERT(mg_strcmp(a, mg_str("aa")) == 0);
ASSERT(mg_strcmp(b, mg_str("")) == 0);
s = mg_str("aa"), a = mg_str_n(NULL, 0), b = mg_str_n(NULL, 0);
ASSERT(mg_span(s, &a, &b, '.') == true);
ASSERT(mg_strcmp(a, mg_str("aa")) == 0);
ASSERT(mg_strcmp(b, mg_str("")) == 0);
2024-02-02 02:21:46 +08:00
}
2020-12-05 19:26:32 +08:00
int main(void) {
2022-02-11 01:11:03 +08:00
const char *debug_level = getenv("V");
if (debug_level == NULL) debug_level = "3";
2022-08-01 18:19:32 +08:00
mg_log_set(atoi(debug_level));
2022-02-11 01:11:03 +08:00
2024-02-02 02:21:46 +08:00
test_split();
2022-06-09 19:39:48 +08:00
test_json();
test_queue();
2022-07-27 07:46:05 +08:00
test_rpc();
2022-02-10 19:56:55 +08:00
test_str();
test_globmatch();
test_get_header_var();
test_http_parse();
2022-10-31 18:40:34 +08:00
test_rewrites();
2021-08-25 14:59:31 +08:00
test_check_ip_acl();
2021-08-08 05:55:13 +08:00
test_udp();
2022-10-31 18:40:34 +08:00
test_packed();
test_crc32();
2021-03-17 15:43:29 +08:00
test_multipart();
test_invalid_listen_addr();
2022-10-31 18:40:34 +08:00
test_http_chunked();
test_http_upload();
test_http_stream_buffer();
2020-12-07 16:54:58 +08:00
test_util();
2020-12-11 21:16:51 +08:00
test_dns();
2020-12-05 19:26:32 +08:00
test_timer();
test_url();
test_iobuf();
test_base64();
test_http_get_var();
test_tls();
2022-10-31 18:40:34 +08:00
test_ws();
test_ws_fragmentation();
2023-09-12 03:33:43 +08:00
test_host_validation();
2022-10-31 18:40:34 +08:00
test_http_server();
test_http_404();
test_http_no_content_length();
test_http_pipeline();
test_http_range();
#ifndef LOCALHOST_ONLY
test_sntp();
2022-10-31 18:40:34 +08:00
test_mqtt();
test_http_client();
#else
(void) test_sntp, (void) test_mqtt, (void) test_http_client;
#endif
2022-11-30 22:47:37 +08:00
test_poll();
2023-10-13 18:32:12 +08:00
test_md5();
test_sha1();
2020-12-05 19:26:32 +08:00
printf("SUCCESS. Total tests: %d\n", s_num_tests);
2022-08-14 03:09:51 +08:00
2020-12-05 19:26:32 +08:00
return EXIT_SUCCESS;
}