END_TEST
START_TEST(test_mg_url_decode)
{
char buf[20];
int ret;
ret = mg_url_decode("abc", 3, buf, sizeof(buf), 0);
ck_assert_int_eq(ret, 3);
ck_assert_str_eq(buf, "abc");
ret = mg_url_decode("abcdef", 3, buf, sizeof(buf), 0);
ck_assert_int_eq(ret, 3);
ck_assert_str_eq(buf, "abc");
ret = mg_url_decode("x+y", 3, buf, sizeof(buf), 0);
ck_assert_int_eq(ret, 3);
ck_assert_str_eq(buf, "x+y");
ret = mg_url_decode("x+y", 3, buf, sizeof(buf), 1);
ck_assert_int_eq(ret, 3);
ck_assert_str_eq(buf, "x y");
ret = mg_url_decode("%25", 3, buf, sizeof(buf), 1);
ck_assert_int_eq(ret, 1);
ck_assert_str_eq(buf, "%");
}
static void test_url_decode(void) {
char buf[100];
ASSERT(mg_url_decode("foo", 3, buf, 3, 0) == -1); // No space for \0
ASSERT(mg_url_decode("foo", 3, buf, 4, 0) == 3);
ASSERT(strcmp(buf, "foo") == 0);
ASSERT(mg_url_decode("a+", 2, buf, sizeof(buf), 0) == 2);
ASSERT(strcmp(buf, "a+") == 0);
ASSERT(mg_url_decode("a+", 2, buf, sizeof(buf), 1) == 2);
ASSERT(strcmp(buf, "a ") == 0);
ASSERT(mg_url_decode("%61", 1, buf, sizeof(buf), 1) == 1);
ASSERT(strcmp(buf, "%") == 0);
ASSERT(mg_url_decode("%61", 2, buf, sizeof(buf), 1) == 2);
ASSERT(strcmp(buf, "%6") == 0);
ASSERT(mg_url_decode("%61", 3, buf, sizeof(buf), 1) == 1);
ASSERT(strcmp(buf, "a") == 0);
}
END_TEST
START_TEST(test_encode_decode)
{
char buf[128];
const char *alpha = "abcdefghijklmnopqrstuvwxyz";
const char *nonalpha = " !\"#$%&'()*+,-./0123456789:;<=>?@";
const char *nonalpha_url_enc1 =
"%20%21%22%23$%25%26%27()%2a%2b,-.%2f0123456789%3a;%3c%3d%3e%3f%40";
const char *nonalpha_url_enc2 =
"%20!%22%23%24%25%26'()*%2B%2C-.%2F0123456789%3A%3B%3C%3D%3E%3F%40";
int ret;
size_t len;
#if defined(USE_WEBSOCKET) || defined(USE_LUA)
const char *alpha_b64_enc = "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXo=";
const char *nonalpha_b64_enc =
"ICEiIyQlJicoKSorLC0uLzAxMjM0NTY3ODk6Ozw9Pj9A";
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)"a", 1, buf);
ck_assert_str_eq(buf, "YQ==");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)"ab", 1, buf);
ck_assert_str_eq(buf, "YQ==");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)"ab", 2, buf);
ck_assert_str_eq(buf, "YWI=");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)alpha, 3, buf);
ck_assert_str_eq(buf, "YWJj");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)alpha, 4, buf);
ck_assert_str_eq(buf, "YWJjZA==");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)alpha, 5, buf);
ck_assert_str_eq(buf, "YWJjZGU=");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)alpha, 6, buf);
ck_assert_str_eq(buf, "YWJjZGVm");
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)alpha, (int)strlen(alpha), buf);
ck_assert_str_eq(buf, alpha_b64_enc);
memset(buf, 77, sizeof(buf));
base64_encode((unsigned char *)nonalpha, (int)strlen(nonalpha), buf);
ck_assert_str_eq(buf, nonalpha_b64_enc);
#endif
#if defined(USE_LUA)
memset(buf, 77, sizeof(buf));
len = 9999;
ret = base64_decode((unsigned char *)alpha_b64_enc,
(int)strlen(alpha_b64_enc),
buf,
&len);
ck_assert_int_eq(ret, -1);
ck_assert_uint_eq((unsigned int)len, (unsigned int)strlen(alpha));
ck_assert_str_eq(buf, alpha);
memset(buf, 77, sizeof(buf));
len = 9999;
ret = base64_decode((unsigned char *)"AAA*AAA", 7, buf, &len);
ck_assert_int_eq(ret, 3);
#endif
memset(buf, 77, sizeof(buf));
ret = mg_url_encode(alpha, buf, sizeof(buf));
ck_assert_int_eq(ret, (int)strlen(buf));
ck_assert_int_eq(ret, (int)strlen(alpha));
ck_assert_str_eq(buf, alpha);
memset(buf, 77, sizeof(buf));
ret = mg_url_encode(nonalpha, buf, sizeof(buf));
ck_assert_int_eq(ret, (int)strlen(buf));
ck_assert_int_eq(ret, (int)strlen(nonalpha_url_enc1));
ck_assert_str_eq(buf, nonalpha_url_enc1);
memset(buf, 77, sizeof(buf));
ret = mg_url_decode(alpha, (int)strlen(alpha), buf, sizeof(buf), 0);
ck_assert_int_eq(ret, (int)strlen(buf));
ck_assert_int_eq(ret, (int)strlen(alpha));
ck_assert_str_eq(buf, alpha);
memset(buf, 77, sizeof(buf));
ret = mg_url_decode(
nonalpha_url_enc1, (int)strlen(nonalpha_url_enc1), buf, sizeof(buf), 0);
ck_assert_int_eq(ret, (int)strlen(buf));
ck_assert_int_eq(ret, (int)strlen(nonalpha));
ck_assert_str_eq(buf, nonalpha);
memset(buf, 77, sizeof(buf));
ret = mg_url_decode(
nonalpha_url_enc2, (int)strlen(nonalpha_url_enc2), buf, sizeof(buf), 0);
ck_assert_int_eq(ret, (int)strlen(buf));
ck_assert_int_eq(ret, (int)strlen(nonalpha));
ck_assert_str_eq(buf, nonalpha);
/* len could be unused, if base64_decode is not tested because USE_LUA is
* not defined */
(void)len;
}
