/* Test the RES_NOAAAA resolver option. Copyright (C) 2022 Free Software Foundation, Inc. This file is part of the GNU C Library. The GNU C Library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. The GNU C Library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with the GNU C Library; if not, see . */ #include #include #include #include #include #include #include #include /* Used to keep track of the number of queries. */ static volatile unsigned int queries; static void response (const struct resolv_response_context *ctx, struct resolv_response_builder *b, const char *qname, uint16_t qclass, uint16_t qtype) { /* Each test should only send one query. */ ++queries; TEST_COMPARE (queries, 1); /* AAAA queries are supposed to be disabled. */ TEST_VERIFY (qtype != T_AAAA); TEST_COMPARE (qclass, C_IN); /* The only other query type besides A is PTR. */ if (qtype != T_A) TEST_COMPARE (qtype, T_PTR); int an, ns, ar; char *tail; if (sscanf (qname, "an%d.ns%d.ar%d.%ms", &an, &ns, &ar, &tail) != 4) FAIL_EXIT1 ("invalid QNAME: %s\n", qname); TEST_COMPARE_STRING (tail, "example"); free (tail); if (an < 0 || ns < 0 || ar < 0) { struct resolv_response_flags flags = { .rcode = NXDOMAIN, }; resolv_response_init (b, flags); resolv_response_add_question (b, qname, qclass, qtype); return; } struct resolv_response_flags flags = {}; resolv_response_init (b, flags); resolv_response_add_question (b, qname, qclass, qtype); resolv_response_section (b, ns_s_an); for (int i = 0; i < an; ++i) { resolv_response_open_record (b, qname, qclass, qtype, 60); switch (qtype) { case T_A: { char ipv4[4] = {192, 0, 2, i + 1}; resolv_response_add_data (b, &ipv4, sizeof (ipv4)); } break; case T_PTR: { char *name = xasprintf ("ptr-%d", i); resolv_response_add_name (b, name); free (name); } break; } resolv_response_close_record (b); } resolv_response_section (b, ns_s_ns); for (int i = 0; i < ns; ++i) { resolv_response_open_record (b, qname, qclass, T_NS, 60); char *name = xasprintf ("ns%d.example.net", i); resolv_response_add_name (b, name); free (name); resolv_response_close_record (b); } resolv_response_section (b, ns_s_ar); int addr = 1; for (int i = 0; i < ns; ++i) { char *name = xasprintf ("ns%d.example.net", i); for (int j = 0; j < ar; ++j) { resolv_response_open_record (b, name, qclass, T_A, 60); char ipv4[4] = {192, 0, 2, addr}; resolv_response_add_data (b, &ipv4, sizeof (ipv4)); resolv_response_close_record (b); resolv_response_open_record (b, name, qclass, T_AAAA, 60); char ipv6[16] = {0x20, 0x01, 0xd, 0xb8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, addr}; resolv_response_add_data (b, &ipv6, sizeof (ipv6)); resolv_response_close_record (b); ++addr; } free (name); } } /* Number of modes. Lowest bit encodes *n* function vs implicit _res argument. The mode numbers themselves are arbitrary. */ enum { mode_count = 8 }; /* res_send-like modes do not perform error translation. */ enum { first_send_mode = 6 }; static int libresolv_query (unsigned int mode, const char *qname, uint16_t qtype, unsigned char *buf, size_t buflen) { int saved_errno = errno; TEST_VERIFY_EXIT (mode < mode_count); switch (mode) { case 0: return res_query (qname, C_IN, qtype, buf, buflen); case 1: return res_nquery (&_res, qname, C_IN, qtype, buf, buflen); case 2: return res_search (qname, C_IN, qtype, buf, buflen); case 3: return res_nsearch (&_res, qname, C_IN, qtype, buf, buflen); case 4: return res_querydomain (qname, "", C_IN, qtype, buf, buflen); case 5: return res_nquerydomain (&_res, qname, "", C_IN, qtype, buf, buflen); case 6: { unsigned char querybuf[512]; int ret = res_mkquery (QUERY, qname, C_IN, qtype, NULL, 0, NULL, querybuf, sizeof (querybuf)); TEST_VERIFY_EXIT (ret > 0); errno = saved_errno; return res_send (querybuf, ret, buf, buflen); } case 7: { unsigned char querybuf[512]; int ret = res_nmkquery (&_res, QUERY, qname, C_IN, qtype, NULL, 0, NULL, querybuf, sizeof (querybuf)); TEST_VERIFY_EXIT (ret > 0); errno = saved_errno; return res_nsend (&_res, querybuf, ret, buf, buflen); } } __builtin_unreachable (); } static int do_test (void) { struct resolv_test *obj = resolv_test_start ((struct resolv_redirect_config) { .response_callback = response }); _res.options |= RES_NOAAAA; check_hostent ("an1.ns2.ar1.example", gethostbyname ("an1.ns2.ar1.example"), "name: an1.ns2.ar1.example\n" "address: 192.0.2.1\n"); queries = 0; check_hostent ("an0.ns2.ar1.example", gethostbyname ("an0.ns2.ar1.example"), "error: NO_ADDRESS\n"); queries = 0; check_hostent ("an-1.ns2.ar1.example", gethostbyname ("an-1.ns2.ar1.example"), "error: HOST_NOT_FOUND\n"); queries = 0; check_hostent ("an1.ns2.ar1.example AF_INET", gethostbyname2 ("an1.ns2.ar1.example", AF_INET), "name: an1.ns2.ar1.example\n" "address: 192.0.2.1\n"); queries = 0; check_hostent ("an0.ns2.ar1.example AF_INET", gethostbyname2 ("an0.ns2.ar1.example", AF_INET), "error: NO_ADDRESS\n"); queries = 0; check_hostent ("an-1.ns2.ar1.example AF_INET", gethostbyname2 ("an-1.ns2.ar1.example", AF_INET), "error: HOST_NOT_FOUND\n"); queries = 0; check_hostent ("an1.ns2.ar1.example AF_INET6", gethostbyname2 ("an1.ns2.ar1.example", AF_INET6), "error: NO_ADDRESS\n"); queries = 0; check_hostent ("an0.ns2.ar1.example AF_INET6", gethostbyname2 ("an0.ns2.ar1.example", AF_INET6), "error: NO_ADDRESS\n"); queries = 0; check_hostent ("an-1.ns2.ar1.example AF_INET6", gethostbyname2 ("an-1.ns2.ar1.example", AF_INET6), "error: HOST_NOT_FOUND\n"); queries = 0; /* Multiple addresses. */ check_hostent ("an2.ns0.ar0.example", gethostbyname ("an2.ns0.ar0.example"), "name: an2.ns0.ar0.example\n" "address: 192.0.2.1\n" "address: 192.0.2.2\n"); queries = 0; check_hostent ("an2.ns0.ar0.example AF_INET6", gethostbyname2 ("an2.ns0.ar0.example", AF_INET6), "error: NO_ADDRESS\n"); queries = 0; /* getaddrinfo checks with one address. */ struct addrinfo *ai; int ret; ret = getaddrinfo ("an1.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an1.ns2.ar1.example (AF_INET)", ai, ret, "address: STREAM/TCP 192.0.2.1 80\n"); freeaddrinfo (ai); queries = 0; ret = getaddrinfo ("an1.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET6, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an1.ns2.ar1.example (AF_INET6)", ai, ret, "error: No address associated with hostname\n"); queries = 0; ret = getaddrinfo ("an1.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_UNSPEC, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an1.ns2.ar1.example (AF_UNSPEC)", ai, ret, "address: STREAM/TCP 192.0.2.1 80\n"); freeaddrinfo (ai); queries = 0; /* getaddrinfo checks with three addresses. */ ret = getaddrinfo ("an3.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an3.ns2.ar1.example (AF_INET)", ai, ret, "address: STREAM/TCP 192.0.2.1 80\n" "address: STREAM/TCP 192.0.2.2 80\n" "address: STREAM/TCP 192.0.2.3 80\n"); freeaddrinfo (ai); queries = 0; ret = getaddrinfo ("an3.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET6, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an3.ns2.ar1.example (AF_INET6)", ai, ret, "error: No address associated with hostname\n"); queries = 0; ret = getaddrinfo ("an3.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_UNSPEC, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an3.ns2.ar1.example (AF_UNSPEC)", ai, ret, "address: STREAM/TCP 192.0.2.1 80\n" "address: STREAM/TCP 192.0.2.2 80\n" "address: STREAM/TCP 192.0.2.3 80\n"); freeaddrinfo (ai); queries = 0; /* getaddrinfo checks with no address. */ ret = getaddrinfo ("an0.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an0.ns2.ar1.example (AF_INET)", ai, ret, "error: No address associated with hostname\n"); queries = 0; ret = getaddrinfo ("an0.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET6, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an0.ns2.ar1.example (AF_INET6)", ai, ret, "error: No address associated with hostname\n"); queries = 0; ret = getaddrinfo ("an0.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_UNSPEC, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an-1.ns2.ar1.example (AF_UNSPEC)", ai, ret, "error: No address associated with hostname\n"); queries = 0; /* getaddrinfo checks with NXDOMAIN. */ ret = getaddrinfo ("an-1.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an-1.ns2.ar1.example (AF_INET)", ai, ret, "error: Name or service not known\n"); queries = 0; ret = getaddrinfo ("an-1.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_INET6, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an-1.ns2.ar1.example (AF_INET6)", ai, ret, "error: Name or service not known\n"); queries = 0; ret = getaddrinfo ("an-1.ns2.ar1.example", "80", &(struct addrinfo) { .ai_family = AF_UNSPEC, .ai_socktype = SOCK_STREAM, }, &ai); check_addrinfo ("an-1.ns2.ar1.example (AF_UNSPEC)", ai, ret, "error: Name or service not known\n"); queries = 0; for (unsigned int mode = 0; mode < mode_count; ++mode) { unsigned char *buf; int ret; /* Response for A. */ buf = malloc (512); ret = libresolv_query (mode, "an1.ns2.ar1.example", T_A, buf, 512); TEST_VERIFY_EXIT (ret > 0); check_dns_packet ("an1.ns2.ar1.example A", buf, ret, "name: an1.ns2.ar1.example\n" "address: 192.0.2.1\n"); free (buf); queries = 0; /* NODATA response for A. */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an0.ns2.ar1.example", T_A, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, NO_ADDRESS); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, 0); check_dns_packet ("an1.ns2.ar1.example A", buf, ret, "name: an0.ns2.ar1.example\n"); } free (buf); queries = 0; /* NXDOMAIN response for A. */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an-1.ns2.ar1.example", T_A, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, HOST_NOT_FOUND); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, NXDOMAIN); check_dns_packet ("an1.ns2.ar1.example A", buf, ret, "name: an-1.ns2.ar1.example\n"); } free (buf); queries = 0; /* Response for PTR. */ buf = malloc (512); ret = libresolv_query (mode, "an1.ns2.ar1.example", T_PTR, buf, 512); TEST_VERIFY_EXIT (ret > 0); check_dns_packet ("an1.ns2.ar1.example PTR", buf, ret, "name: an1.ns2.ar1.example\n" "data: an1.ns2.ar1.example PTR ptr-0\n"); free (buf); queries = 0; /* NODATA response for PTR. */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an0.ns2.ar1.example", T_PTR, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, NO_ADDRESS); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, 0); check_dns_packet ("an1.ns2.ar1.example PTR", buf, ret, "name: an0.ns2.ar1.example\n"); } free (buf); queries = 0; /* NXDOMAIN response for PTR. */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an-1.ns2.ar1.example", T_PTR, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, HOST_NOT_FOUND); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, NXDOMAIN); check_dns_packet ("an1.ns2.ar1.example PTR", buf, ret, "name: an-1.ns2.ar1.example\n"); } free (buf); queries = 0; /* NODATA response for AAAA. */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an1.ns2.ar1.example", T_AAAA, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, NO_ADDRESS); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, 0); check_dns_packet ("an1.ns2.ar1.example A", buf, ret, "name: an1.ns2.ar1.example\n"); } free (buf); queries = 0; /* NODATA response for AAAA (original is already NODATA). */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an0.ns2.ar1.example", T_AAAA, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, NO_ADDRESS); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, 0); check_dns_packet ("an0.ns2.ar1.example A", buf, ret, "name: an0.ns2.ar1.example\n"); } free (buf); queries = 0; /* NXDOMAIN response. */ buf = malloc (512); errno = 0; ret = libresolv_query (mode, "an-1.ns2.ar1.example", T_AAAA, buf, 512); if (mode < first_send_mode) { TEST_COMPARE (ret, -1); TEST_COMPARE (errno, 0); TEST_COMPARE (h_errno, HOST_NOT_FOUND); } else { TEST_VERIFY_EXIT (ret > 0); TEST_COMPARE (((HEADER *)buf)->rcode, NXDOMAIN); check_dns_packet ("an-1.ns2.ar1.example A", buf, ret, "name: an-1.ns2.ar1.example\n"); } free (buf); queries = 0; } resolv_test_end (obj); return 0; } #include