/*
* This file is part of DisOrder.
- * Copyright (C) 2005, 2007 Richard Kettlewell
+ * Copyright (C) 2005, 2007, 2008 Richard Kettlewell
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
#include <signal.h>
#include <sys/wait.h>
#include <stddef.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <netinet/in.h>
+#include <sys/un.h>
+#include <pcre.h>
-#include "utf8.h"
#include "mem.h"
#include "log.h"
#include "vector.h"
#include "printf.h"
#include "basen.h"
#include "split.h"
+#include "configuration.h"
+#include "addr.h"
+#include "base64.h"
+#include "url.h"
+#include "regsub.h"
static int tests, errors;
static int fail_first;
static void test_mime(void) {
char *t, *n, *v;
struct vector parts[1];
+ struct kvp *k;
fprintf(stderr, "test_mime\n");
- t = n = v = 0;
- insist(!mime_content_type("text/plain", &t, &n, &v));
+ t = 0;
+ k = 0;
+ insist(!mime_content_type("text/plain", &t, &k));
check_string(t, "text/plain");
- insist(n == 0);
- insist(v == 0);
+ insist(k == 0);
- insist(mime_content_type("TEXT ((broken) comment", &t, &n, &v) < 0);
- insist(mime_content_type("TEXT ((broken) comment\\", &t, &n, &v) < 0);
+ insist(mime_content_type("TEXT ((broken) comment", &t, &k) < 0);
+ insist(mime_content_type("TEXT ((broken) comment\\", &t, &k) < 0);
- t = n = v = 0;
- insist(!mime_content_type("TEXT ((nested)\\ comment) /plain", &t, &n, &v));
+ t = 0;
+ k = 0;
+ insist(!mime_content_type("TEXT ((nested)\\ comment) /plain", &t, &k));
check_string(t, "text/plain");
- insist(n == 0);
- insist(v == 0);
+ insist(k == 0);
- t = n = v = 0;
- insist(!mime_content_type(" text/plain ; Charset=\"utf-\\8\"", &t, &n, &v));
+ t = 0;
+ k = 0;
+ insist(!mime_content_type(" text/plain ; Charset=\"utf-\\8\"", &t, &k));
check_string(t, "text/plain");
- check_string(n, "charset");
- check_string(v, "utf-8");
+ insist(k != 0);
+ insist(k->next == 0);
+ check_string(k->name, "charset");
+ check_string(k->value, "utf-8");
- t = n = v = 0;
- insist(!mime_content_type("text/plain;charset = ISO-8859-1 ", &t, &n, &v));
+ t = 0;
+ k = 0;
+ insist(!mime_content_type("text/plain;charset = ISO-8859-1 ", &t, &k));
+ insist(k != 0);
+ insist(k->next == 0);
check_string(t, "text/plain");
- check_string(n, "charset");
- check_string(v, "ISO-8859-1");
+ check_string(k->name, "charset");
+ check_string(k->value, "ISO-8859-1");
t = n = v = 0;
insist(!mime_rfc2388_content_disposition("form-data; name=\"field1\"", &t, &n, &v));
check_string(mime_qp("x =\r\ny"), "x y");
check_string(mime_qp("x = \r\ny"), "x y");
+ check_string(mime_to_qp(""), "");
+ check_string(mime_to_qp("foobar\n"), "foobar\n");
+ check_string(mime_to_qp("foobar \n"), "foobar=20\n");
+ check_string(mime_to_qp("foobar\t\n"), "foobar=09\n");
+ check_string(mime_to_qp("foobar \t \n"), "foobar=20=09=20\n");
+ check_string(mime_to_qp(" foo=bar"), " foo=3Dbar\n");
+ check_string(mime_to_qp("copyright \xC2\xA9"), "copyright =C2=A9\n");
+ check_string(mime_to_qp("foo\nbar\nbaz\n"), "foo\nbar\nbaz\n");
+ check_string(mime_to_qp("wibble wobble wibble wobble wibble wobble wibble wobble wibble wobble wibble"), "wibble wobble wibble wobble wibble wobble wibble wobble wibble wobble wibb=\nle\n");
+
/* from RFC2045 */
check_string(mime_qp("Now's the time =\r\n"
"for all folk to come=\r\n"
" to the aid of their country."),
"Now's the time for all folk to come to the aid of their country.");
- check_string(mime_base64(""), "");
- check_string(mime_base64("BBBB"), "\x04\x10\x41");
- check_string(mime_base64("////"), "\xFF\xFF\xFF");
- check_string(mime_base64("//BB"), "\xFF\xF0\x41");
- check_string(mime_base64("BBBB//BB////"),
- "\x04\x10\x41" "\xFF\xF0\x41" "\xFF\xFF\xFF");
- check_string(mime_base64("B B B B / / B B / / / /"),
- "\x04\x10\x41" "\xFF\xF0\x41" "\xFF\xFF\xFF");
- check_string(mime_base64("B\r\nBBB.// B-B//~//"),
+#define check_base64(encoded, decoded) do { \
+ check_string(mime_base64(encoded, 0), decoded); \
+ check_string(mime_to_base64((const uint8_t *)decoded, \
+ (sizeof decoded) - 1), \
+ encoded); \
+ } while(0)
+
+
+ check_base64("", "");
+ check_base64("BBBB", "\x04\x10\x41");
+ check_base64("////", "\xFF\xFF\xFF");
+ check_base64("//BB", "\xFF\xF0\x41");
+ check_base64("BBBB//BB////",
+ "\x04\x10\x41" "\xFF\xF0\x41" "\xFF\xFF\xFF");
+ check_base64("BBBBBA==",
+ "\x04\x10\x41" "\x04");
+ check_base64("BBBBBBA=",
+ "\x04\x10\x41" "\x04\x10");
+
+ /* Check that decoding handles various kinds of rubbish OK */
+ check_string(mime_base64("B B B B / / B B / / / /", 0),
+ "\x04\x10\x41" "\xFF\xF0\x41" "\xFF\xFF\xFF");
+ check_string(mime_base64("B\r\nBBB.// B-B//~//", 0),
"\x04\x10\x41" "\xFF\xF0\x41" "\xFF\xFF\xFF");
- check_string(mime_base64("BBBB="),
- "\x04\x10\x41");
- check_string(mime_base64("BBBBx="), /* not actually valid base64 */
+ check_string(mime_base64("BBBB BB==", 0),
+ "\x04\x10\x41" "\x04");
+ check_string(mime_base64("BBBB BB = =", 0),
+ "\x04\x10\x41" "\x04");
+ check_string(mime_base64("BBBB BBB=", 0),
+ "\x04\x10\x41" "\x04\x10");
+ check_string(mime_base64("BBBB BBB = ", 0),
+ "\x04\x10\x41" "\x04\x10");
+ check_string(mime_base64("BBBB=", 0),
"\x04\x10\x41");
- check_string(mime_base64("BBBB BB=="),
+ check_string(mime_base64("BBBBBB==", 0),
"\x04\x10\x41" "\x04");
- check_string(mime_base64("BBBB BBB="),
+ check_string(mime_base64("BBBBBBB=", 0),
"\x04\x10\x41" "\x04\x10");
+ /* Not actually valid base64 */
+ check_string(mime_base64("BBBBx=", 0),
+ "\x04\x10\x41");
+}
+
+static void test_cookies(void) {
+ struct cookiedata cd[1];
+
+ fprintf(stderr, "test_cookies\n");
+
+ /* These are the examples from RFC2109 */
+ insist(!parse_cookie("$Version=\"1\"; Customer=\"WILE_E_COYOTE\"; $Path=\"/acme\"", cd));
+ insist(!strcmp(cd->version, "1"));
+ insist(cd->ncookies = 1);
+ insist(find_cookie(cd, "Customer") == &cd->cookies[0]);
+ check_string(cd->cookies[0].value, "WILE_E_COYOTE");
+ check_string(cd->cookies[0].path, "/acme");
+ insist(cd->cookies[0].domain == 0);
+ insist(!parse_cookie("$Version=\"1\";\n"
+ "Customer=\"WILE_E_COYOTE\"; $Path=\"/acme\";\n"
+ "Part_Number=\"Rocket_Launcher_0001\"; $Path=\"/acme\"",
+ cd));
+ insist(cd->ncookies = 2);
+ insist(find_cookie(cd, "Customer") == &cd->cookies[0]);
+ insist(find_cookie(cd, "Part_Number") == &cd->cookies[1]);
+ check_string(cd->cookies[0].value, "WILE_E_COYOTE");
+ check_string(cd->cookies[0].path, "/acme");
+ insist(cd->cookies[0].domain == 0);
+ check_string(cd->cookies[1].value, "Rocket_Launcher_0001");
+ check_string(cd->cookies[1].path, "/acme");
+ insist(cd->cookies[1].domain == 0);
+ insist(!parse_cookie("$Version=\"1\";\n"
+ "Customer=\"WILE_E_COYOTE\"; $Path=\"/acme\";\n"
+ "Part_Number=\"Rocket_Launcher_0001\"; $Path=\"/acme\";\n"
+ "Shipping=\"FedEx\"; $Path=\"/acme\"",
+ cd));
+ insist(cd->ncookies = 3);
+ insist(find_cookie(cd, "Customer") == &cd->cookies[0]);
+ insist(find_cookie(cd, "Part_Number") == &cd->cookies[1]);
+ insist(find_cookie(cd, "Shipping") == &cd->cookies[2]);
+ check_string(cd->cookies[0].value, "WILE_E_COYOTE");
+ check_string(cd->cookies[0].path, "/acme");
+ insist(cd->cookies[0].domain == 0);
+ check_string(cd->cookies[1].value, "Rocket_Launcher_0001");
+ check_string(cd->cookies[1].path, "/acme");
+ insist(cd->cookies[1].domain == 0);
+ check_string(cd->cookies[2].value, "FedEx");
+ check_string(cd->cookies[2].path, "/acme");
+ insist(cd->cookies[2].domain == 0);
}
static void test_hex(void) {
static void test_filepart(void) {
fprintf(stderr, "test_filepart\n");
check_string(d_dirname("/"), "/");
+ check_string(d_dirname("////"), "/");
check_string(d_dirname("/spong"), "/");
+ check_string(d_dirname("////spong"), "/");
check_string(d_dirname("/foo/bar"), "/foo");
+ check_string(d_dirname("////foo/////bar"), "////foo");
check_string(d_dirname("./bar"), ".");
+ check_string(d_dirname(".//bar"), ".");
check_string(d_dirname("."), ".");
check_string(d_dirname(".."), ".");
check_string(d_dirname("../blat"), "..");
+ check_string(d_dirname("..//blat"), "..");
check_string(d_dirname("wibble"), ".");
check_string(extension("foo.c"), ".c");
check_string(extension(".c"), ".c");
check_string(extension("foo"), "");
check_string(extension("./foo"), "");
check_string(extension("./foo.c"), ".c");
+ check_string(strip_extension("foo.c"), "foo");
+ check_string(strip_extension("foo.mp3"), "foo");
+ check_string(strip_extension("foo.---"), "foo.---");
+ check_string(strip_extension("foo.---xyz"), "foo.---xyz");
+ check_string(strip_extension("foo.bar/wibble.spong"), "foo.bar/wibble");
}
static void test_selection(void) {
insist(inputline("tmpfile", fp, &l, '\n') == 0);
check_string(l, "wibble: foobar");
insist(inputline("tmpfile", fp, &l, '\n') == -1);
+
+ fp = tmpfile();
+ assert(fp != 0);
+ fprintf(fp, "foo\rbar\nwibble\r\n");
+ fprintf(fp, "second\n\rspong\r\n");
+ rewind(fp);
+ insist(inputline("tmpfile", fp, &l, CRLF) == 0);
+ check_string(l, "foo\rbar\nwibble");
+ insist(inputline("tmpfile", fp, &l, CRLF) == 0);
+ check_string(l, "second\n\rspong");
+ insist(inputline("tmpfile", fp, &l, CRLF) == -1);
dynstr_init(d);
s = sink_dynstr(d);
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("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) {
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 *sin4;
+ struct sockaddr_in s4;
+ struct sockaddr_un su;
+
+ static const struct addrinfo pref = {
+ AI_PASSIVE,
+ PF_INET,
+ SOCK_STREAM,
+ 0,
+ 0,
+ 0,
+ 0,
+ 0
+ };
+
+ printf("test_addr\n");
+
+ 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));
+ sin4 = (const struct sockaddr_in *)ai->ai_addr;
+ check_integer(sin4->sin_family, AF_INET);
+ check_integer(sin4->sin_addr.s_addr, 0);
+ check_integer(ntohs(sin4->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));
+ sin4 = (const struct sockaddr_in *)ai->ai_addr;
+ check_integer(sin4->sin_family, AF_INET);
+ check_integer(ntohl(sin4->sin_addr.s_addr), 0x7F000001);
+ check_integer(ntohs(sin4->sin_port), 119);
+ check_string(name, "host localhost service nntp");
+
+ memset(&s4, 0, sizeof s4);
+ s4.sin_family = AF_INET;
+ s4.sin_addr.s_addr = 0;
+ s4.sin_port = 0;
+ check_string(format_sockaddr((struct sockaddr *)&s4),
+ "0.0.0.0");
+ check_integer(multicast((struct sockaddr *)&s4), 0);
+ s4.sin_addr.s_addr = htonl(0x7F000001);
+ s4.sin_port = htons(1000);
+ check_string(format_sockaddr((struct sockaddr *)&s4),
+ "127.0.0.1 port 1000");
+ check_integer(multicast((struct sockaddr *)&s4), 0);
+ s4.sin_addr.s_addr = htonl(0xE0000001);
+ check_string(format_sockaddr((struct sockaddr *)&s4),
+ "224.0.0.1 port 1000");
+ check_integer(multicast((struct sockaddr *)&s4), 1);
+
+ memset(&su, 0, sizeof su);
+ su.sun_family = AF_UNIX;
+ strcpy(su.sun_path, "/wibble/wobble");
+ check_string(format_sockaddr((struct sockaddr *)&su),
+ "/wibble/wobble");
+ check_integer(multicast((struct sockaddr *)&su), 0);
+}
+
+static void test_url(void) {
+ struct url p;
+
+ printf("test_url\n");
+
+ insist(parse_url("http://www.example.com/example/path", &p) == 0);
+ check_string(p.scheme, "http");
+ check_string(p.host, "www.example.com");
+ insist(p.port == -1);
+ check_string(p.path, "/example/path");
+ insist(p.query == 0);
+
+ insist(parse_url("https://www.example.com:82/example%2fpath?+query+", &p) == 0);
+ check_string(p.scheme, "https");
+ check_string(p.host, "www.example.com");
+ insist(p.port == 82);
+ check_string(p.path, "/example/path");
+ check_string(p.query, "+query+");
+
+ insist(parse_url("//www.example.com/example/path", &p) == 0);
+ insist(p.scheme == 0);
+ check_string(p.host, "www.example.com");
+ insist(p.port == -1);
+ check_string(p.path, "/example/path");
+ insist(p.query == 0);
+
+ insist(parse_url("http://www.example.com:100000/", &p) == -1);
+ insist(parse_url("http://www.example.com:1000000000000/", &p) == -1);
+ insist(parse_url("http://www.example.com/example%2zpath", &p) == -1);
+}
+
+static void test_regsub(void) {
+ pcre *re;
+ const char *errstr;
+ int erroffset;
+
+ printf("test_regsub\n");
+
+ check_integer(regsub_flags(""), 0);
+ check_integer(regsub_flags("g"), REGSUB_GLOBAL);
+ check_integer(regsub_flags("i"), REGSUB_CASE_INDEPENDENT);
+ check_integer(regsub_flags("gi"), REGSUB_GLOBAL|REGSUB_CASE_INDEPENDENT);
+ check_integer(regsub_flags("iiggxx"), REGSUB_GLOBAL|REGSUB_CASE_INDEPENDENT);
+ check_integer(regsub_compile_options(0), 0);
+ check_integer(regsub_compile_options(REGSUB_CASE_INDEPENDENT), PCRE_CASELESS);
+ check_integer(regsub_compile_options(REGSUB_GLOBAL|REGSUB_CASE_INDEPENDENT), PCRE_CASELESS);
+ check_integer(regsub_compile_options(REGSUB_GLOBAL), 0);
+
+ re = pcre_compile("foo", PCRE_UTF8, &errstr, &erroffset, 0);
+ assert(re != 0);
+ check_string(regsub(re, "wibble-foo-foo-bar", "spong", 0),
+ "wibble-spong-foo-bar");
+ check_string(regsub(re, "wibble-foo-foo-bar", "spong", REGSUB_GLOBAL),
+ "wibble-spong-spong-bar");
+ check_string(regsub(re, "wibble-x-x-bar", "spong", REGSUB_GLOBAL),
+ "wibble-x-x-bar");
+ insist(regsub(re, "wibble-x-x-bar", "spong", REGSUB_MUST_MATCH) == 0);
+
+ re = pcre_compile("a+", PCRE_UTF8, &errstr, &erroffset, 0);
+ assert(re != 0);
+ check_string(regsub(re, "baaaaa", "spong", 0),
+ "bspong");
+ check_string(regsub(re, "baaaaa", "spong", REGSUB_GLOBAL),
+ "bspong");
+ check_string(regsub(re, "baaaaa", "foo-$&-bar", 0),
+ "bfoo-aaaaa-bar");
+
+ re = pcre_compile("(a+)(b+)", PCRE_UTF8|PCRE_CASELESS, &errstr, &erroffset, 0);
+ assert(re != 0);
+ check_string(regsub(re, "foo-aaaabbb-bar", "spong", 0),
+ "foo-spong-bar");
+ check_string(regsub(re, "foo-aaaabbb-bar", "x:$2/$1:y", 0),
+ "foo-x:bbb/aaaa:y-bar");
+ check_string(regsub(re, "foo-aAaAbBb-bar", "x:$2$$$1:y", 0),
+ "foo-x:bBb$aAaA:y-bar");
+}
+
int main(void) {
mem_init();
fail_first = !!getenv("FAIL_FIRST");
insist('a' == 0x61);
insist('z' == 0x7A);
/* addr.c */
+ test_addr();
/* asprintf.c */
/* authhash.c */
/* basen.c */
/* mem.c */
/* mime.c */
test_mime();
+ test_cookies();
/* mixer.c */
/* plugin.c */
/* printf.c */
/* selection.c */
test_selection();
test_hash();
+ test_url();
+ test_regsub();
fprintf(stderr, "%d errors out of %d tests\n", errors, tests);
return !!errors;
}
~mdw / disorder / blobdiff