From f2b1a1897537d1915ac2f840b13a8991018755c4 Mon Sep 17 00:00:00 2001 From: Daniel Stenberg Date: Tue, 19 Jun 2018 16:08:05 +0200 Subject: DOH: add test case 1650 and 2100 --- tests/data/Makefile.inc | 4 + tests/data/test1650 | 26 +++++ tests/data/test2100 | Bin 0 -> 1613 bytes tests/unit/Makefile.inc | 6 +- tests/unit/unit1650.c | 283 ++++++++++++++++++++++++++++++++++++++++++++++++ 5 files changed, 318 insertions(+), 1 deletion(-) create mode 100644 tests/data/test1650 create mode 100644 tests/data/test2100 create mode 100644 tests/unit/unit1650.c (limited to 'tests') diff --git a/tests/data/Makefile.inc b/tests/data/Makefile.inc index d3f7e39b2..0980ff003 100644 --- a/tests/data/Makefile.inc +++ b/tests/data/Makefile.inc @@ -182,6 +182,8 @@ test1590 \ test1600 test1601 test1602 test1603 test1604 test1605 test1606 test1607 \ test1608 test1609 test1620 \ \ +test1650 \ +\ test1700 test1701 test1702 \ \ test1800 test1801 \ @@ -201,4 +203,6 @@ test2064 test2065 test2066 test2067 test2068 test2069 \ test2070 test2071 test2072 test2073 \ test2074 test2075 \ \ +test2100 \ +\ test3000 test3001 diff --git a/tests/data/test1650 b/tests/data/test1650 new file mode 100644 index 000000000..60bfb6df3 --- /dev/null +++ b/tests/data/test1650 @@ -0,0 +1,26 @@ + + + +unittest +DOH + + + +# +# Client-side + + +none + + +unittest + + +DOH + + +unit1650 + + + + diff --git a/tests/data/test2100 b/tests/data/test2100 new file mode 100644 index 000000000..4b4c0724a Binary files /dev/null and b/tests/data/test2100 differ diff --git a/tests/unit/Makefile.inc b/tests/unit/Makefile.inc index 6d88157a2..5811ab476 100644 --- a/tests/unit/Makefile.inc +++ b/tests/unit/Makefile.inc @@ -10,7 +10,8 @@ UNITPROGS = unit1300 unit1301 unit1302 unit1303 unit1304 unit1305 unit1307 \ unit1330 unit1394 unit1395 unit1396 unit1397 unit1398 \ unit1399 \ unit1600 unit1601 unit1602 unit1603 unit1604 unit1605 unit1606 unit1607 \ - unit1608 unit1609 unit1620 + unit1608 unit1609 unit1620 \ + unit1650 unit1300_SOURCES = unit1300.c $(UNITFILES) unit1300_CPPFLAGS = $(AM_CPPFLAGS) @@ -98,3 +99,6 @@ unit1609_CPPFLAGS = $(AM_CPPFLAGS) unit1620_SOURCES = unit1620.c $(UNITFILES) unit1620_CPPFLAGS = $(AM_CPPFLAGS) + +unit1650_SOURCES = unit1650.c $(UNITFILES) +unit1650_CPPFLAGS = $(AM_CPPFLAGS) diff --git a/tests/unit/unit1650.c b/tests/unit/unit1650.c new file mode 100644 index 000000000..a6278dd3c --- /dev/null +++ b/tests/unit/unit1650.c @@ -0,0 +1,283 @@ +/*************************************************************************** + * _ _ ____ _ + * Project ___| | | | _ \| | + * / __| | | | |_) | | + * | (__| |_| | _ <| |___ + * \___|\___/|_| \_\_____| + * + * Copyright (C) 2018, Daniel Stenberg, , et al. + * + * This software is licensed as described in the file COPYING, which + * you should have received as part of this distribution. The terms + * are also available at https://curl.haxx.se/docs/copyright.html. + * + * You may opt to use, copy, modify, merge, publish, distribute and/or sell + * copies of the Software, and permit persons to whom the Software is + * furnished to do so, under the terms of the COPYING file. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ***************************************************************************/ +#include "curlcheck.h" + +#include "doh.h" + +static CURLcode unit_setup(void) +{ + return CURLE_OK; +} + +static void unit_stop(void) +{ + +} + +#define DNS_PREAMBLE "\x00\x00\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00" +#define LABEL_TEST "\x04\x74\x65\x73\x74" +#define LABEL_HOST "\x04\x68\x6f\x73\x74" +#define LABEL_NAME "\x04\x6e\x61\x6d\x65" +#define DNSA_TYPE "\x01" +#define DNSAAAA_TYPE "\x1c" +#define DNSA_EPILOGUE "\x00\x00" DNSA_TYPE "\x00\x01" +#define DNSAAAA_EPILOGUE "\x00\x00" DNSAAAA_TYPE "\x00\x01" + +#define DNS_Q1 DNS_PREAMBLE LABEL_TEST LABEL_HOST LABEL_NAME DNSA_EPILOGUE +#define DNS_Q2 DNS_PREAMBLE LABEL_TEST LABEL_HOST LABEL_NAME DNSAAAA_EPILOGUE + +struct dohrequest { + /* input */ + const char *name; + DNStype type; + + /* output */ + const char *packet; + size_t size; + int rc; +}; + + +static struct dohrequest req[] = { + {"test.host.name", DNS_TYPE_A, DNS_Q1, sizeof(DNS_Q1)-1, 0 }, + {"test.host.name", DNS_TYPE_AAAA, DNS_Q2, sizeof(DNS_Q2)-1, 0 }, + {"zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz" + ".host.name", + DNS_TYPE_AAAA, NULL, 0, DOH_DNS_BAD_LABEL } +}; + +struct dohresp { + /* input */ + const char *packet; + size_t size; + DNStype type; + + /* output */ + int rc; + const char *out; +}; + +#define DNS_FOO_EXAMPLE_COM \ + "\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x03\x66\x6f\x6f" \ + "\x07\x65\x78\x61\x6d\x70\x6c\x65\x03\x63\x6f\x6d\x00\x00\x01\x00" \ + "\x01\xc0\x0c\x00\x01\x00\x01\x00\x00\x00\x37\x00\x04\x7f\x00\x00" \ + "\x01" + +static const char full49[] = DNS_FOO_EXAMPLE_COM; + +static struct dohresp resp[] = { + {"\x00\x00", 2, DNS_TYPE_A, DOH_TOO_SMALL_BUFFER, NULL }, + {"\x00\x01\x00\x01\x00\x01\x00\x01\x00\x01\x00\x01", 12, + DNS_TYPE_A, DOH_DNS_BAD_ID, NULL }, + {"\x00\x00\x00\x01\x00\x01\x00\x01\x00\x01\x00\x01", 12, + DNS_TYPE_A, DOH_DNS_BAD_RCODE, NULL }, + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x03\x66\x6f\x6f", 16, + DNS_TYPE_A, DOH_DNS_OUT_OF_RANGE, NULL }, + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x03\x66\x6f\x6f\x00", 17, + DNS_TYPE_A, DOH_DNS_OUT_OF_RANGE, NULL }, + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x03\x66\x6f\x6f\x00" + "\x00\x01\x00\x01", 21, + DNS_TYPE_A, DOH_DNS_OUT_OF_RANGE, NULL }, + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x03\x66\x6f\x6f\x00" + "\x00\x01\x00\x01" + "\x04", 18, + DNS_TYPE_A, DOH_DNS_OUT_OF_RANGE, NULL }, + + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x04\x63\x75\x72" + "\x6c\x04\x63\x75\x72\x6c\x00\x00\x05\x00\x01\xc0\x0c\x00\x05\x00" + "\x01\x00\x00\x00\x37\x00\x11\x08\x61\x6e\x79\x77\x68\x65\x72\x65" + "\x06\x72\x65\x61\x6c\x6c\x79\x00", 56, + DNS_TYPE_A, DOH_OK, "anywhere.really "}, + + {DNS_FOO_EXAMPLE_COM, 49, DNS_TYPE_A, DOH_OK, "127.0.0.1 "}, + + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x04\x61\x61\x61" + "\x61\x07\x65\x78\x61\x6d\x70\x6c\x65\x03\x63\x6f\x6d\x00\x00\x1c" + "\x00\x01\xc0\x0c\x00\x1c\x00\x01\x00\x00\x00\x37\x00\x10\x20\x20" + "\x20\x20\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x20\x20", 62, + DNS_TYPE_AAAA, DOH_OK, + "2020:2020:0000:0000:0000:0000:0000:2020 " }, + + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x00\x04\x63\x75\x72" + "\x6c\x04\x63\x75\x72\x6c\x00\x00\x05\x00\x01\xc0\x0c\x00\x05\x00" + "\x01\x00\x00\x00\x37\x00" + "\x07\x03\x61\x6e\x79\xc0\x27\x00", 46, + DNS_TYPE_A, DOH_DNS_LABEL_LOOP, NULL}, + + /* packet with NSCOUNT == 1 */ + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x01\x00\x00\x04\x61\x61\x61" + "\x61\x07\x65\x78\x61\x6d\x70\x6c\x65\x03\x63\x6f\x6d\x00\x00\x1c" + "\x00\x01\xc0\x0c\x00\x1c\x00\x01\x00\x00\x00\x37\x00\x10\x20\x20" + "\x20\x20\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x20\x20" + LABEL_TEST LABEL_HOST LABEL_NAME DNSAAAA_EPILOGUE "\x00\x00\x00\x01" + "\00\x04\x01\x01\x01\x01", /* RDDATA */ + + 62 + 30, + DNS_TYPE_AAAA, DOH_OK, + "2020:2020:0000:0000:0000:0000:0000:2020 " }, + + /* packet with ARCOUNT == 1 */ + {"\x00\x00\x01\x00\x00\x01\x00\x01\x00\x00\x00\x01\x04\x61\x61\x61" + "\x61\x07\x65\x78\x61\x6d\x70\x6c\x65\x03\x63\x6f\x6d\x00\x00\x1c" + "\x00\x01\xc0\x0c\x00\x1c\x00\x01\x00\x00\x00\x37\x00\x10\x20\x20" + "\x20\x20\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x20\x20" + LABEL_TEST LABEL_HOST LABEL_NAME DNSAAAA_EPILOGUE "\x00\x00\x00\x01" + "\00\x04\x01\x01\x01\x01", /* RDDATA */ + + 62 + 30, + DNS_TYPE_AAAA, DOH_OK, + "2020:2020:0000:0000:0000:0000:0000:2020 " }, + +}; + +UNITTEST_START +{ + size_t size; + unsigned char buffer[256]; + size_t i; + for(i = 0; i < sizeof(req) / sizeof(req[0]); i++) { + int rc = doh_encode(req[i].name, req[i].type, + buffer, sizeof(buffer), &size); + if(rc != req[i].rc) { + fprintf(stderr, "req %d: Expected return code %d got %d\n", i, + req[i].rc, rc); + return 1; + } + else if(size != req[i].size) { + fprintf(stderr, "req %d: Expected size %d got %d\n", i, + (int)req[i].size, (int)size); + fprintf(stderr, "DNS encode made: %s\n", hexdump(buffer, size)); + return 2; + } + else if(memcmp(req[i].packet, buffer, size)) { + fprintf(stderr, "DNS encode made: %s\n", hexdump(buffer, size)); + fprintf(stderr, "... instead of: %s\n", + hexdump((unsigned char *)req[i].packet, size)); + return 3; + } + } + + for(i = 0; i < sizeof(resp) / sizeof(resp[0]); i++) { + struct dohentry d; + int rc; + char *ptr; + size_t len; + int u; + memset(&d, 0, sizeof(d)); + rc = doh_decode((unsigned char *)resp[i].packet, resp[i].size, + resp[i].type, &d); + if(rc != resp[i].rc) { + fprintf(stderr, "resp %d: Expected return code %d got %d\n", i, + resp[i].rc, rc); + return 4; + } + len = sizeof(buffer); + ptr = (char *)buffer; + for(u = 0; u < d.numaddr; u++) { + size_t o; + struct dohaddr *a; + a = &d.addr[u]; + if(resp[i].type == DNS_TYPE_A) { + snprintf(ptr, len, "%d.%d.%d.%d ", + a->ip.v4 & 0xff, (a->ip.v4>>8) & 0xff, + (a->ip.v4>>16) & 0xff, a->ip.v4 >>24); + o = strlen(ptr); + len -= o; + ptr += o; + } + else { + int j; + for(j = 0; j < 16; j += 2) { + size_t l; + snprintf(ptr, len, "%s%02x%02x", j?":":"", a->ip.v6.byte[j], + a->ip.v6.byte[j + 1]); + l = strlen(ptr); + len -= l; + ptr += l; + } + snprintf(ptr, len, " "); + len--; + ptr++; + } + } + for(u = 0; u < d.numcname; u++) { + size_t o; + snprintf(ptr, len, "%s ", d.cname[u].alloc); + o = strlen(ptr); + len -= o; + ptr += o; + } + de_cleanup(&d); + if(resp[i].out && strcmp((char *)buffer, resp[i].out)) { + fprintf(stderr, "resp %d: Expected %s got %s\n", i, + resp[i].out, buffer); + return 1; + } + } + + { + /* pass all sizes into the decoder until full */ + for(i = 0; i < sizeof(full49)-1; i++) { + struct dohentry d; + int rc; + memset(&d, 0, sizeof(d)); + rc = doh_decode((unsigned char *)full49, i, DNS_TYPE_A, &d); + if(!rc) { + /* none of them should work */ + fprintf(stderr, "%d: %d\n", i, rc); + return 5; + } + } + /* and try all pieces from the other end of the packet */ + for(i = 1; i < sizeof(full49); i++) { + struct dohentry d; + int rc; + memset(&d, 0, sizeof(d)); + rc = doh_decode((unsigned char *)&full49[i], sizeof(full49)-i-1, + DNS_TYPE_A, &d); + if(!rc) { + /* none of them should work */ + fprintf(stderr, "2 %d: %d\n", i, rc); + return 7; + } + } + { + int rc; + struct dohentry d; + struct dohaddr *a; + memset(&d, 0, sizeof(d)); + rc = doh_decode((unsigned char *)full49, sizeof(full49)-1, + DNS_TYPE_A, &d); + fail_if(d.numaddr != 1, "missing address"); + a = &d.addr[i]; + snprintf((char *)buffer, sizeof(buffer), "%d.%d.%d.%d\n", + a->ip.v4 & 0xff, (a->ip.v4>>8) & 0xff, + (a->ip.v4>>16) & 0xff, a->ip.v4 >>24); + if(rc && strcmp((char *)buffer, "127.0.0.1")) { + fprintf(stderr, "bad address decoded\n"); + return 7; + } + fail_if(d.numcname, "bad cname counter"); + } + } +} +UNITTEST_STOP -- cgit v1.2.3