227 lines
8.1 KiB
C++
227 lines
8.1 KiB
C++
/*
|
|
* nghttp2 - HTTP/2 C Library
|
|
*
|
|
* Copyright (c) 2013 Tatsuhiro Tsujikawa
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
#include "util_test.h"
|
|
|
|
#include <cstring>
|
|
#include <iostream>
|
|
|
|
#include <CUnit/CUnit.h>
|
|
|
|
#include <nghttp2/nghttp2.h>
|
|
|
|
#include "util.h"
|
|
|
|
using namespace nghttp2;
|
|
|
|
namespace shrpx {
|
|
|
|
void test_util_streq(void) {
|
|
CU_ASSERT(util::streq("alpha", (const uint8_t *)"alpha", 5));
|
|
CU_ASSERT(util::streq("alpha", (const uint8_t *)"alphabravo", 5));
|
|
CU_ASSERT(!util::streq("alpha", (const uint8_t *)"alphabravo", 6));
|
|
CU_ASSERT(!util::streq("alphabravo", (const uint8_t *)"alpha", 5));
|
|
CU_ASSERT(!util::streq("alpha", (const uint8_t *)"alphA", 5));
|
|
CU_ASSERT(!util::streq("", (const uint8_t *)"a", 1));
|
|
CU_ASSERT(util::streq("", (const uint8_t *)"", 0));
|
|
CU_ASSERT(!util::streq("alpha", (const uint8_t *)"", 0));
|
|
|
|
CU_ASSERT(
|
|
util::streq((const uint8_t *)"alpha", 5, (const uint8_t *)"alpha", 5));
|
|
CU_ASSERT(
|
|
!util::streq((const uint8_t *)"alpha", 4, (const uint8_t *)"alpha", 5));
|
|
CU_ASSERT(
|
|
!util::streq((const uint8_t *)"alpha", 5, (const uint8_t *)"alpha", 4));
|
|
CU_ASSERT(
|
|
!util::streq((const uint8_t *)"alpha", 5, (const uint8_t *)"alphA", 5));
|
|
char *a = nullptr;
|
|
char *b = nullptr;
|
|
CU_ASSERT(util::streq(a, 0, b, 0));
|
|
}
|
|
|
|
void test_util_strieq(void) {
|
|
CU_ASSERT(util::strieq(std::string("alpha"), std::string("alpha")));
|
|
CU_ASSERT(util::strieq(std::string("alpha"), std::string("AlPhA")));
|
|
CU_ASSERT(util::strieq(std::string(), std::string()));
|
|
CU_ASSERT(!util::strieq(std::string("alpha"), std::string("AlPhA ")));
|
|
CU_ASSERT(!util::strieq(std::string(), std::string("AlPhA ")));
|
|
}
|
|
|
|
void test_util_inp_strlower(void) {
|
|
std::string a("alPha");
|
|
util::inp_strlower(a);
|
|
CU_ASSERT("alpha" == a);
|
|
|
|
a = "ALPHA123BRAVO";
|
|
util::inp_strlower(a);
|
|
CU_ASSERT("alpha123bravo" == a);
|
|
|
|
a = "";
|
|
util::inp_strlower(a);
|
|
CU_ASSERT("" == a);
|
|
}
|
|
|
|
void test_util_to_base64(void) {
|
|
std::string x = "AAA--B_";
|
|
util::to_base64(x);
|
|
CU_ASSERT("AAA++B/=" == x);
|
|
|
|
x = "AAA--B_B";
|
|
util::to_base64(x);
|
|
CU_ASSERT("AAA++B/B" == x);
|
|
}
|
|
|
|
void test_util_percent_encode_token(void) {
|
|
CU_ASSERT("h2" == util::percent_encode_token("h2"));
|
|
CU_ASSERT("h3~" == util::percent_encode_token("h3~"));
|
|
CU_ASSERT("100%25" == util::percent_encode_token("100%"));
|
|
CU_ASSERT("http%202" == util::percent_encode_token("http 2"));
|
|
}
|
|
|
|
void test_util_quote_string(void) {
|
|
CU_ASSERT("alpha" == util::quote_string("alpha"));
|
|
CU_ASSERT("" == util::quote_string(""));
|
|
CU_ASSERT("\\\"alpha\\\"" == util::quote_string("\"alpha\""));
|
|
}
|
|
|
|
void test_util_utox(void) {
|
|
CU_ASSERT("0" == util::utox(0));
|
|
CU_ASSERT("1" == util::utox(1));
|
|
CU_ASSERT("F" == util::utox(15));
|
|
CU_ASSERT("10" == util::utox(16));
|
|
CU_ASSERT("3B9ACA07" == util::utox(1000000007));
|
|
CU_ASSERT("100000000" == util::utox(1LL << 32));
|
|
}
|
|
|
|
void test_util_http_date(void) {
|
|
CU_ASSERT("Thu, 01 Jan 1970 00:00:00 GMT" == util::http_date(0));
|
|
CU_ASSERT("Wed, 29 Feb 2012 09:15:16 GMT" == util::http_date(1330506916));
|
|
}
|
|
|
|
void test_util_select_h2(void) {
|
|
const unsigned char *out = NULL;
|
|
unsigned char outlen = 0;
|
|
|
|
// Check single entry and select it.
|
|
const unsigned char t1[] = "\x5h2-14";
|
|
CU_ASSERT(util::select_h2(&out, &outlen, t1, sizeof(t1) - 1));
|
|
CU_ASSERT(
|
|
memcmp(NGHTTP2_PROTO_VERSION_ID, out, NGHTTP2_PROTO_VERSION_ID_LEN) == 0);
|
|
CU_ASSERT(NGHTTP2_PROTO_VERSION_ID_LEN == outlen);
|
|
|
|
out = NULL;
|
|
outlen = 0;
|
|
|
|
// Check the case where id is correct but length is invalid and too
|
|
// long.
|
|
const unsigned char t2[] = "\x6h2-14";
|
|
CU_ASSERT(!util::select_h2(&out, &outlen, t2, sizeof(t2) - 1));
|
|
|
|
// Check the case where h2-14 is located after bogus ID.
|
|
const unsigned char t3[] = "\x2h3\x5h2-14";
|
|
CU_ASSERT(util::select_h2(&out, &outlen, t3, sizeof(t3) - 1));
|
|
CU_ASSERT(
|
|
memcmp(NGHTTP2_PROTO_VERSION_ID, out, NGHTTP2_PROTO_VERSION_ID_LEN) == 0);
|
|
CU_ASSERT(NGHTTP2_PROTO_VERSION_ID_LEN == outlen);
|
|
|
|
out = NULL;
|
|
outlen = 0;
|
|
|
|
// Check the case that last entry's length is invalid and too long.
|
|
const unsigned char t4[] = "\x2h3\x6h2-14";
|
|
CU_ASSERT(!util::select_h2(&out, &outlen, t4, sizeof(t4) - 1));
|
|
|
|
// Check the case that all entries are not supported.
|
|
const unsigned char t5[] = "\x2h3\x2h4";
|
|
CU_ASSERT(!util::select_h2(&out, &outlen, t5, sizeof(t5) - 1));
|
|
|
|
// Check the case where 2 values are eligible, but last one is
|
|
// picked up because it has precedence over the other.
|
|
const unsigned char t6[] = "\x5h2-14\x5h2-16";
|
|
CU_ASSERT(util::select_h2(&out, &outlen, t6, sizeof(t6) - 1));
|
|
CU_ASSERT(memcmp(NGHTTP2_H2_PROTO_ALIAS, out, NGHTTP2_H2_PROTO_ALIAS_LEN) ==
|
|
0);
|
|
CU_ASSERT(NGHTTP2_H2_PROTO_ALIAS_LEN == outlen);
|
|
}
|
|
|
|
void test_util_ipv6_numeric_addr(void) {
|
|
CU_ASSERT(util::ipv6_numeric_addr("::1"));
|
|
CU_ASSERT(util::ipv6_numeric_addr("2001:0db8:85a3:0042:1000:8a2e:0370:7334"));
|
|
// IPv4
|
|
CU_ASSERT(!util::ipv6_numeric_addr("127.0.0.1"));
|
|
// not numeric address
|
|
CU_ASSERT(!util::ipv6_numeric_addr("localhost"));
|
|
}
|
|
|
|
void test_util_utos_with_unit(void) {
|
|
CU_ASSERT("0" == util::utos_with_unit(0));
|
|
CU_ASSERT("1023" == util::utos_with_unit(1023));
|
|
CU_ASSERT("1K" == util::utos_with_unit(1024));
|
|
CU_ASSERT("1K" == util::utos_with_unit(1025));
|
|
CU_ASSERT("1M" == util::utos_with_unit(1 << 20));
|
|
CU_ASSERT("1G" == util::utos_with_unit(1 << 30));
|
|
CU_ASSERT("1024G" == util::utos_with_unit(1LL << 40));
|
|
}
|
|
|
|
void test_util_parse_uint_with_unit(void) {
|
|
CU_ASSERT(0 == util::parse_uint_with_unit("0"));
|
|
CU_ASSERT(1023 == util::parse_uint_with_unit("1023"));
|
|
CU_ASSERT(1024 == util::parse_uint_with_unit("1k"));
|
|
CU_ASSERT(2048 == util::parse_uint_with_unit("2K"));
|
|
CU_ASSERT(1 << 20 == util::parse_uint_with_unit("1m"));
|
|
CU_ASSERT(1 << 21 == util::parse_uint_with_unit("2M"));
|
|
CU_ASSERT(1 << 30 == util::parse_uint_with_unit("1g"));
|
|
CU_ASSERT(1LL << 31 == util::parse_uint_with_unit("2G"));
|
|
CU_ASSERT(9223372036854775807LL ==
|
|
util::parse_uint_with_unit("9223372036854775807"));
|
|
// check overflow case
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("9223372036854775808"));
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("10000000000000000000"));
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("9223372036854775807G"));
|
|
// bad characters
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("1.1"));
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("1a"));
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("a1"));
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit("1T"));
|
|
CU_ASSERT(-1 == util::parse_uint_with_unit(""));
|
|
}
|
|
|
|
void test_util_parse_uint(void) {
|
|
CU_ASSERT(0 == util::parse_uint("0"));
|
|
CU_ASSERT(1023 == util::parse_uint("1023"));
|
|
CU_ASSERT(-1 == util::parse_uint("1k"));
|
|
CU_ASSERT(9223372036854775807LL == util::parse_uint("9223372036854775807"));
|
|
// check overflow case
|
|
CU_ASSERT(-1 == util::parse_uint("9223372036854775808"));
|
|
CU_ASSERT(-1 == util::parse_uint("10000000000000000000"));
|
|
// bad characters
|
|
CU_ASSERT(-1 == util::parse_uint("1.1"));
|
|
CU_ASSERT(-1 == util::parse_uint("1a"));
|
|
CU_ASSERT(-1 == util::parse_uint("a1"));
|
|
CU_ASSERT(-1 == util::parse_uint("1T"));
|
|
CU_ASSERT(-1 == util::parse_uint(""));
|
|
}
|
|
|
|
} // namespace shrpx
|