+static void test_wstat(void) {
+ pid_t pid;
+ int w;
+
+ fprintf(stderr, "test_wstat\n");
+ if(!(pid = xfork())) {
+ _exit(1);
+ }
+ while(waitpid(pid, &w, 0) < 0 && errno == EINTR)
+ ;
+ check_string(wstat(w), "exited with status 1");
+ if(!(pid = xfork())) {
+ kill(getpid(), SIGTERM);
+ _exit(-1);
+ }
+ while(waitpid(pid, &w, 0) < 0 && errno == EINTR)
+ ;
+ check_string_prefix(wstat(w), "terminated by signal 15");
+}
+
+static void test_kvp(void) {
+ struct kvp *k;
+ size_t n;
+
+ fprintf(stderr, "test_kvp\n");
+ /* decoding */
+#define KVP_URLDECODE(S) kvp_urldecode((S), strlen(S))
+ insist(KVP_URLDECODE("=%zz") == 0);
+ insist(KVP_URLDECODE("=%0") == 0);
+ insist(KVP_URLDECODE("=%0z") == 0);
+ insist(KVP_URLDECODE("=%%") == 0);
+ insist(KVP_URLDECODE("==%") == 0);
+ insist(KVP_URLDECODE("wibble") == 0);
+ insist(KVP_URLDECODE("") == 0);
+ insist(KVP_URLDECODE("wibble&") == 0);
+ insist((k = KVP_URLDECODE("one=bl%61t+foo")) != 0);
+ check_string(kvp_get(k, "one"), "blat foo");
+ insist(kvp_get(k, "ONE") == 0);
+ insist(k->next == 0);
+ insist((k = KVP_URLDECODE("wibble=splat&bar=spong")) != 0);
+ check_string(kvp_get(k, "wibble"), "splat");
+ check_string(kvp_get(k, "bar"), "spong");
+ insist(kvp_get(k, "ONE") == 0);
+ insist(k->next->next == 0);
+ /* encoding */
+ insist(kvp_set(&k, "bar", "spong") == 0);
+ insist(kvp_set(&k, "bar", "foo") == 1);
+ insist(kvp_set(&k, "zog", "%") == 1);
+ insist(kvp_set(&k, "wibble", 0) == 1);
+ insist(kvp_set(&k, "wibble", 0) == 0);
+ check_string(kvp_urlencode(k, 0),
+ "bar=foo&zog=%25");
+ check_string(kvp_urlencode(k, &n),
+ "bar=foo&zog=%25");
+ insist(n == strlen("bar=foo&zog=%25"));
+ check_string(urlencodestring("abc% +\n"),
+ "abc%25%20%2b%0a");
+}
+
+static void test_sink(void) {
+ struct sink *s;
+ struct dynstr d[1];
+ FILE *fp;
+ char *l;
+
+ fprintf(stderr, "test_sink\n");
+
+ fp = tmpfile();
+ assert(fp != 0);
+ s = sink_stdio("tmpfile", fp);
+ insist(sink_printf(s, "test: %d\n", 999) == 10);
+ insist(sink_printf(s, "wibble: %s\n", "foobar") == 15);
+ rewind(fp);
+ insist(inputline("tmpfile", fp, &l, '\n') == 0);
+ check_string(l, "test: 999");
+ insist(inputline("tmpfile", fp, &l, '\n') == 0);
+ check_string(l, "wibble: foobar");
+ insist(inputline("tmpfile", fp, &l, '\n') == -1);
+
+ dynstr_init(d);
+ s = sink_dynstr(d);
+ insist(sink_printf(s, "test: %d\n", 999) == 10);
+ insist(sink_printf(s, "wibble: %s\n", "foobar") == 15);
+ dynstr_terminate(d);
+ check_string(d->vec, "test: 999\nwibble: foobar\n");
+}
+
+static const char *do_printf(const char *fmt, ...) {
+ va_list ap;
+ char *s;
+ int rc;
+
+ va_start(ap, fmt);
+ rc = byte_vasprintf(&s, fmt, ap);
+ va_end(ap);
+ if(rc < 0)
+ return 0;
+ return s;
+}
+
+static void test_printf(void) {
+ char c;
+ short s;
+ int i;
+ long l;
+ long long ll;
+ intmax_t m;
+ ssize_t ssz;
+ ptrdiff_t p;
+ char *cp;
+ char buffer[16];
+
+ fprintf(stderr, "test_printf\n");
+ check_string(do_printf("%d", 999), "999");
+ check_string(do_printf("%d", -999), "-999");
+ check_string(do_printf("%i", 999), "999");
+ check_string(do_printf("%i", -999), "-999");
+ check_string(do_printf("%u", 999), "999");
+ check_string(do_printf("%2u", 999), "999");
+ check_string(do_printf("%10u", 999), " 999");
+ check_string(do_printf("%-10u", 999), "999 ");
+ check_string(do_printf("%010u", 999), "0000000999");
+ check_string(do_printf("%-10d", -999), "-999 ");
+ check_string(do_printf("%-010d", -999), "-999 "); /* "-" beats "0" */
+ check_string(do_printf("%66u", 999), " 999");
+ check_string(do_printf("%o", 999), "1747");
+ check_string(do_printf("%#o", 999), "01747");
+ check_string(do_printf("%#o", 0), "0");
+ check_string(do_printf("%x", 999), "3e7");
+ check_string(do_printf("%#x", 999), "0x3e7");
+ check_string(do_printf("%#X", 999), "0X3E7");
+ check_string(do_printf("%#x", 0), "0");
+ check_string(do_printf("%hd", (short)999), "999");
+ check_string(do_printf("%hhd", (short)99), "99");
+ check_string(do_printf("%ld", 100000L), "100000");
+ check_string(do_printf("%lld", 10000000000LL), "10000000000");
+ check_string(do_printf("%qd", 10000000000LL), "10000000000");
+ check_string(do_printf("%jd", (intmax_t)10000000000LL), "10000000000");
+ check_string(do_printf("%zd", (ssize_t)2000000000), "2000000000");
+ check_string(do_printf("%td", (ptrdiff_t)2000000000), "2000000000");
+ check_string(do_printf("%hu", (short)999), "999");
+ check_string(do_printf("%hhu", (short)99), "99");
+ check_string(do_printf("%lu", 100000L), "100000");
+ check_string(do_printf("%llu", 10000000000LL), "10000000000");
+ check_string(do_printf("%ju", (uintmax_t)10000000000LL), "10000000000");
+ check_string(do_printf("%zu", (size_t)2000000000), "2000000000");
+ check_string(do_printf("%tu", (ptrdiff_t)2000000000), "2000000000");
+ check_string(do_printf("%p", (void *)0x100), "0x100");
+ check_string(do_printf("%s", "wibble"), "wibble");
+ check_string(do_printf("%s-%s", "wibble", "wobble"), "wibble-wobble");
+ check_string(do_printf("%10s", "wibble"), " wibble");
+ check_string(do_printf("%010s", "wibble"), " wibble"); /* 0 ignored for %s */
+ check_string(do_printf("%-10s", "wibble"), "wibble ");
+ check_string(do_printf("%2s", "wibble"), "wibble");
+ check_string(do_printf("%.2s", "wibble"), "wi");
+ check_string(do_printf("%.2s", "w"), "w");
+ check_string(do_printf("%4.2s", "wibble"), " wi");
+ check_string(do_printf("%c", 'a'), "a");
+ check_string(do_printf("%4c", 'a'), " a");
+ check_string(do_printf("%-4c", 'a'), "a ");
+ check_string(do_printf("%*c", 0, 'a'), "a");
+ check_string(do_printf("x%hhny", &c), "xy");
+ insist(c == 1);
+ check_string(do_printf("xx%hnyy", &s), "xxyy");
+ insist(s == 2);
+ check_string(do_printf("xxx%nyyy", &i), "xxxyyy");
+ insist(i == 3);
+ check_string(do_printf("xxxx%lnyyyy", &l), "xxxxyyyy");
+ insist(l == 4);
+ check_string(do_printf("xxxxx%llnyyyyy", &ll), "xxxxxyyyyy");
+ insist(ll == 5);
+ check_string(do_printf("xxxxxx%jnyyyyyy", &m), "xxxxxxyyyyyy");
+ insist(m == 6);
+ check_string(do_printf("xxxxxxx%znyyyyyyy", &ssz), "xxxxxxxyyyyyyy");
+ insist(ssz == 7);
+ check_string(do_printf("xxxxxxxx%tnyyyyyyyy", &p), "xxxxxxxxyyyyyyyy");
+ insist(p == 8);
+ check_string(do_printf("%*d", 5, 99), " 99");
+ check_string(do_printf("%*d", -5, 99), "99 ");
+ check_string(do_printf("%.*d", 5, 99), "00099");
+ check_string(do_printf("%.*d", -5, 99), "99");
+ check_string(do_printf("%.0d", 0), "");
+ check_string(do_printf("%.d", 0), "");
+ check_string(do_printf("%.d", 0), "");
+ check_string(do_printf("%%"), "%");
+ check_string(do_printf("wibble"), "wibble");
+ insist(do_printf("%") == 0);
+ insist(do_printf("%=") == 0);
+ i = byte_asprintf(&cp, "xyzzy %d", 999);
+ insist(i == 9);
+ check_string(cp, "xyzzy 999");
+ i = byte_snprintf(buffer, sizeof buffer, "xyzzy %d", 999);
+ insist(i == 9);
+ check_string(buffer, "xyzzy 999");
+ i = byte_snprintf(buffer, sizeof buffer, "%*d", 32, 99);
+ insist(i == 32);
+ check_string(buffer, " ");
+ {
+ /* bizarre workaround for compiler checking of format strings */
+ char f[] = "xyzzy %";
+ i = byte_asprintf(&cp, f);
+ insist(i == -1);
+ }
+}
+
+static void test_basen(void) {
+ unsigned long v[64];
+ char buffer[1024];
+
+ fprintf(stderr, "test_basen\n");
+ v[0] = 999;
+ insist(basen(v, 1, buffer, sizeof buffer, 10) == 0);
+ check_string(buffer, "999");
+
+ v[0] = 1+2*7+3*7*7+4*7*7*7;
+ insist(basen(v, 1, buffer, sizeof buffer, 7) == 0);
+ check_string(buffer, "4321");
+
+ v[0] = 0x00010203;
+ v[1] = 0x04050607;
+ v[2] = 0x08090A0B;
+ v[3] = 0x0C0D0E0F;
+ insist(basen(v, 4, buffer, sizeof buffer, 256) == 0);
+ check_string(buffer, "123456789abcdef");
+
+ v[0] = 0x00010203;
+ v[1] = 0x04050607;
+ v[2] = 0x08090A0B;
+ v[3] = 0x0C0D0E0F;
+ insist(basen(v, 4, buffer, sizeof buffer, 16) == 0);
+ check_string(buffer, "102030405060708090a0b0c0d0e0f");
+
+ v[0] = 0x00010203;
+ v[1] = 0x04050607;
+ v[2] = 0x08090A0B;
+ v[3] = 0x0C0D0E0F;
+ insist(basen(v, 4, buffer, 10, 16) == -1);
+}
+
+static void test_split(void) {
+ char **v;
+ int nv;
+
+ fprintf(stderr, "test_split\n");
+ insist(split("\"misquoted", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0) == 0);
+ insist(split("\'misquoted", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0) == 0);
+ insist(split("\'misquoted\\", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0) == 0);
+ insist(split("\'misquoted\\\"", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0) == 0);
+ insist(split("\'mis\\escaped\'", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0) == 0);
+
+ insist((v = split("", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 0);
+ insist(*v == 0);
+
+ insist((v = split("wibble", &nv, SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 1);
+ check_string(v[0], "wibble");
+ insist(v[1] == 0);
+
+ insist((v = split(" wibble \t\r\n wobble ", &nv,
+ SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 2);
+ check_string(v[0], "wibble");
+ check_string(v[1], "wobble");
+ insist(v[2] == 0);
+
+ insist((v = split("wibble wobble #splat", &nv,
+ SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 2);
+ check_string(v[0], "wibble");
+ check_string(v[1], "wobble");
+ insist(v[2] == 0);
+
+ insist((v = split("\"wibble wobble\" #splat", &nv,
+ SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 1);
+ check_string(v[0], "wibble wobble");
+ insist(v[1] == 0);
+
+ insist((v = split("\"wibble \\\"\\nwobble\"", &nv,
+ SPLIT_COMMENTS|SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 1);
+ check_string(v[0], "wibble \"\nwobble");
+ insist(v[1] == 0);
+
+ insist((v = split("\"wibble wobble\" #splat", &nv,
+ SPLIT_QUOTES, 0, 0)));
+ check_integer(nv, 2);
+ check_string(v[0], "wibble wobble");
+ check_string(v[1], "#splat");
+ insist(v[2] == 0);
+
+ insist((v = split("\"wibble wobble\" #splat", &nv,
+ SPLIT_COMMENTS, 0, 0)));
+ check_integer(nv, 2);
+ check_string(v[0], "\"wibble");
+ check_string(v[1], "wobble\"");
+ insist(v[2] == 0);
+
+ check_string(quoteutf8("wibble"), "wibble");
+ check_string(quoteutf8(" wibble "), "\" wibble \"");
+ check_string(quoteutf8("wibble wobble"), "\"wibble wobble\"");
+ check_string(quoteutf8("wibble\"wobble"), "\"wibble\\\"wobble\"");
+ check_string(quoteutf8("wibble\nwobble"), "\"wibble\\nwobble\"");
+ check_string(quoteutf8("wibble\\wobble"), "\"wibble\\\\wobble\"");
+ check_string(quoteutf8("wibble'wobble"), "\"wibble'wobble\"");
+}
+
+static void test_hash(void) {
+ hash *h;
+ int i, *ip;
+ char **keys;
+
+ fprintf(stderr, "test_hash\n");
+ h = hash_new(sizeof(int));
+ for(i = 0; i < 10000; ++i)
+ insist(hash_add(h, do_printf("%d", i), &i, HASH_INSERT) == 0);
+ check_integer(hash_count(h), 10000);
+ for(i = 0; i < 10000; ++i) {
+ insist((ip = hash_find(h, do_printf("%d", i))) != 0);
+ check_integer(*ip, i);
+ insist(hash_add(h, do_printf("%d", i), &i, HASH_REPLACE) == 0);
+ }
+ check_integer(hash_count(h), 10000);
+ keys = hash_keys(h);
+ for(i = 0; i < 10000; ++i)
+ insist(keys[i] != 0);
+ insist(keys[10000] == 0);
+ for(i = 0; i < 10000; ++i)
+ insist(hash_remove(h, do_printf("%d", i)) == 0);
+ check_integer(hash_count(h), 0);
+}
+
+static void test_addr(void) {
+ struct stringlist a;
+ const char *s[2];
+ struct addrinfo *ai;
+ char *name;
+ const struct sockaddr_in *sin;
+
+ static const struct addrinfo pref = {
+ AI_PASSIVE,
+ PF_INET,
+ SOCK_STREAM,
+ 0,
+ 0,
+ 0,
+ 0,
+ 0
+ };
+
+ a.n = 1;
+ a.s = (char **)s;
+ s[0] = "smtp";
+ ai = get_address(&a, &pref, &name);
+ insist(ai != 0);
+ check_integer(ai->ai_family, PF_INET);
+ check_integer(ai->ai_socktype, SOCK_STREAM);
+ check_integer(ai->ai_protocol, IPPROTO_TCP);
+ check_integer(ai->ai_addrlen, sizeof(struct sockaddr_in));
+ sin = (const struct sockaddr_in *)ai->ai_addr;
+ check_integer(sin->sin_family, AF_INET);
+ check_integer(sin->sin_addr.s_addr, 0);
+ check_integer(ntohs(sin->sin_port), 25);
+ check_string(name, "host * service smtp");
+
+ a.n = 2;
+ s[0] = "localhost";
+ s[1] = "nntp";
+ ai = get_address(&a, &pref, &name);
+ insist(ai != 0);
+ check_integer(ai->ai_family, PF_INET);
+ check_integer(ai->ai_socktype, SOCK_STREAM);
+ check_integer(ai->ai_protocol, IPPROTO_TCP);
+ check_integer(ai->ai_addrlen, sizeof(struct sockaddr_in));
+ sin = (const struct sockaddr_in *)ai->ai_addr;
+ check_integer(sin->sin_family, AF_INET);
+ check_integer(ntohl(sin->sin_addr.s_addr), 0x7F000001);
+ check_integer(ntohs(sin->sin_port), 119);
+ check_string(name, "host localhost service nntp");
+}
+