| /*************************************************************************** | 
 |  *                                  _   _ ____  _ | 
 |  *  Project                     ___| | | |  _ \| | | 
 |  *                             / __| | | | |_) | | | 
 |  *                            | (__| |_| |  _ <| |___ | 
 |  *                             \___|\___/|_| \_\_____| | 
 |  * | 
 |  * Copyright (C) 2018 - 2022, Daniel Stenberg, <daniel@haxx.se>, 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.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. | 
 |  * | 
 |  * SPDX-License-Identifier: curl | 
 |  * | 
 |  ***************************************************************************/ | 
 | #include "curlcheck.h" | 
 |  | 
 | #include "doh.h" | 
 | #include "dynbuf.h" | 
 |  | 
 | static CURLcode unit_setup(void) | 
 | { | 
 |   return CURLE_OK; | 
 | } | 
 |  | 
 | static void unit_stop(void) | 
 | { | 
 |  | 
 | } | 
 |  | 
 | #ifndef CURL_DISABLE_DOH | 
 | #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 const 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 const 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 = 0; | 
 |   unsigned char buffer[256]; | 
 |   size_t i; | 
 |   unsigned char *p; | 
 |   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 %zu: Expected return code %d got %d\n", i, | 
 |               req[i].rc, rc); | 
 |       return 1; | 
 |     } | 
 |     else if(size != req[i].size) { | 
 |       fprintf(stderr, "req %zu: Expected size %zu got %zu\n", i, | 
 |               req[i].size, size); | 
 |       fprintf(stderr, "DNS encode made: %s\n", hexdump(buffer, size)); | 
 |       return 2; | 
 |     } | 
 |     else if(req[i].packet && 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; | 
 |     de_init(&d); | 
 |     rc = doh_decode((const unsigned char *)resp[i].packet, resp[i].size, | 
 |                     resp[i].type, &d); | 
 |     if(rc != resp[i].rc) { | 
 |       fprintf(stderr, "resp %zu: 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) { | 
 |         p = &a->ip.v4[0]; | 
 |         msnprintf(ptr, len, "%u.%u.%u.%u ", p[0], p[1], p[2], p[3]); | 
 |         o = strlen(ptr); | 
 |         len -= o; | 
 |         ptr += o; | 
 |       } | 
 |       else { | 
 |         int j; | 
 |         for(j = 0; j < 16; j += 2) { | 
 |           size_t l; | 
 |           msnprintf(ptr, len, "%s%02x%02x", j?":":"", a->ip.v6[j], | 
 |                    a->ip.v6[j + 1]); | 
 |           l = strlen(ptr); | 
 |           len -= l; | 
 |           ptr += l; | 
 |         } | 
 |         msnprintf(ptr, len, " "); | 
 |         len--; | 
 |         ptr++; | 
 |       } | 
 |     } | 
 |     for(u = 0; u < d.numcname; u++) { | 
 |       size_t o; | 
 |       msnprintf(ptr, len, "%s ", Curl_dyn_ptr(&d.cname[u])); | 
 |       o = strlen(ptr); | 
 |       len -= o; | 
 |       ptr += o; | 
 |     } | 
 |     de_cleanup(&d); | 
 |     if(resp[i].out && strcmp((char *)buffer, resp[i].out)) { | 
 |       fprintf(stderr, "resp %zu: 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((const unsigned char *)full49, i, DNS_TYPE_A, &d); | 
 |       if(!rc) { | 
 |         /* none of them should work */ | 
 |         fprintf(stderr, "%zu: %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((const unsigned char *)&full49[i], sizeof(full49)-i-1, | 
 |                       DNS_TYPE_A, &d); | 
 |       if(!rc) { | 
 |         /* none of them should work */ | 
 |         fprintf(stderr, "2 %zu: %d\n", i, rc); | 
 |         return 7; | 
 |       } | 
 |     } | 
 |     { | 
 |       int rc; | 
 |       struct dohentry d; | 
 |       struct dohaddr *a; | 
 |       memset(&d, 0, sizeof(d)); | 
 |       rc = doh_decode((const unsigned char *)full49, sizeof(full49)-1, | 
 |                       DNS_TYPE_A, &d); | 
 |       fail_if(d.numaddr != 1, "missing address"); | 
 |       a = &d.addr[0]; | 
 |       p = &a->ip.v4[0]; | 
 |       msnprintf((char *)buffer, sizeof(buffer), | 
 |                 "%u.%u.%u.%u", p[0], p[1], p[2], p[3]); | 
 |       if(rc || strcmp((char *)buffer, "127.0.0.1")) { | 
 |         fprintf(stderr, "bad address decoded: %s, rc == %d\n", buffer, rc); | 
 |         return 7; | 
 |       } | 
 |       fail_if(d.numcname, "bad cname counter"); | 
 |     } | 
 |   } | 
 | } | 
 | UNITTEST_STOP | 
 |  | 
 | #else /* CURL_DISABLE_DOH */ | 
 | UNITTEST_START | 
 | { | 
 |   return 1; /* nothing to do, just fail */ | 
 | } | 
 | UNITTEST_STOP | 
 |  | 
 |  | 
 | #endif |