| /* Exercise low-level query functions with different QTYPEs. |
| Copyright (C) 2016-2018 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 |
| <http://www.gnu.org/licenses/>. */ |
| |
| #include <resolv.h> |
| #include <string.h> |
| #include <support/check.h> |
| #include <support/check_nss.h> |
| #include <support/resolv_test.h> |
| #include <support/support.h> |
| #include <support/test-driver.h> |
| #include <support/xmemstream.h> |
| |
| /* If ture, the response function will send the actual response packet |
| over TCP instead of UDP. */ |
| static volatile bool force_tcp; |
| |
| /* Send back a fake resource record matching the QTYPE. */ |
| static void |
| response (const struct resolv_response_context *ctx, |
| struct resolv_response_builder *b, |
| const char *qname, uint16_t qclass, uint16_t qtype) |
| { |
| if (force_tcp && ctx->tcp) |
| { |
| resolv_response_init (b, (struct resolv_response_flags) { .tc = 1 }); |
| resolv_response_add_question (b, qname, qclass, qtype); |
| return; |
| } |
| |
| resolv_response_init (b, (struct resolv_response_flags) { }); |
| resolv_response_add_question (b, qname, qclass, qtype); |
| resolv_response_section (b, ns_s_an); |
| resolv_response_open_record (b, qname, qclass, qtype, 0); |
| resolv_response_add_data (b, &qtype, sizeof (qtype)); |
| resolv_response_close_record (b); |
| } |
| |
| static const char domain[] = "www.example.com"; |
| |
| static int |
| wrap_res_query (int type, unsigned char *answer, int answer_length) |
| { |
| return res_query (domain, C_IN, type, answer, answer_length); |
| } |
| |
| static int |
| wrap_res_search (int type, unsigned char *answer, int answer_length) |
| { |
| return res_query (domain, C_IN, type, answer, answer_length); |
| } |
| |
| static int |
| wrap_res_querydomain (int type, unsigned char *answer, int answer_length) |
| { |
| return res_querydomain ("www", "example.com", C_IN, type, |
| answer, answer_length); |
| } |
| |
| static int |
| wrap_res_send (int type, unsigned char *answer, int answer_length) |
| { |
| unsigned char buf[512]; |
| int ret = res_mkquery (QUERY, domain, C_IN, type, |
| (const unsigned char *) "", 0, NULL, |
| buf, sizeof (buf)); |
| if (type < 0 || type >= 65536) |
| { |
| /* res_mkquery fails for out-of-range record types. */ |
| TEST_VERIFY_EXIT (ret == -1); |
| return -1; |
| } |
| TEST_VERIFY_EXIT (ret > 12); /* DNS header length. */ |
| return res_send (buf, ret, answer, answer_length); |
| } |
| |
| static int |
| wrap_res_nquery (int type, unsigned char *answer, int answer_length) |
| { |
| return res_nquery (&_res, domain, C_IN, type, answer, answer_length); |
| } |
| |
| static int |
| wrap_res_nsearch (int type, unsigned char *answer, int answer_length) |
| { |
| return res_nquery (&_res, domain, C_IN, type, answer, answer_length); |
| } |
| |
| static int |
| wrap_res_nquerydomain (int type, unsigned char *answer, int answer_length) |
| { |
| return res_nquerydomain (&_res, "www", "example.com", C_IN, type, |
| answer, answer_length); |
| } |
| |
| static int |
| wrap_res_nsend (int type, unsigned char *answer, int answer_length) |
| { |
| unsigned char buf[512]; |
| int ret = res_nmkquery (&_res, QUERY, domain, C_IN, type, |
| (const unsigned char *) "", 0, NULL, |
| buf, sizeof (buf)); |
| if (type < 0 || type >= 65536) |
| { |
| /* res_mkquery fails for out-of-range record types. */ |
| TEST_VERIFY_EXIT (ret == -1); |
| return -1; |
| } |
| TEST_VERIFY_EXIT (ret > 12); /* DNS header length. */ |
| return res_nsend (&_res, buf, ret, answer, answer_length); |
| } |
| |
| static void |
| test_function (const char *fname, |
| int (*func) (int type, |
| unsigned char *answer, int answer_length)) |
| { |
| unsigned char buf[512]; |
| for (int tcp = 0; tcp < 2; ++tcp) |
| { |
| force_tcp = tcp; |
| for (unsigned int type = 1; type <= 65535; ++type) |
| { |
| if (test_verbose) |
| printf ("info: sending QTYPE %d with %s (tcp=%d)\n", |
| type, fname, tcp); |
| int ret = func (type, buf, sizeof (buf)); |
| if (ret != 47) |
| FAIL_EXIT1 ("%s tcp=%d qtype=%d return value %d", |
| fname,tcp, type, ret); |
| /* One question, one answer record. */ |
| TEST_VERIFY (memcmp (buf + 4, "\0\1\0\1\0\0\0\0", 8) == 0); |
| /* Question section. */ |
| static const char qname[] = "\3www\7example\3com"; |
| size_t qname_length = sizeof (qname); |
| TEST_VERIFY (memcmp (buf + 12, qname, qname_length) == 0); |
| /* RDATA part of answer. */ |
| uint16_t type16 = type; |
| TEST_VERIFY (memcmp (buf + ret - 2, &type16, sizeof (type16)) == 0); |
| } |
| } |
| |
| TEST_VERIFY (func (-1, buf, sizeof (buf) == -1)); |
| TEST_VERIFY (func (65536, buf, sizeof (buf) == -1)); |
| } |
| |
| static int |
| do_test (void) |
| { |
| struct resolv_redirect_config config = |
| { |
| .response_callback = response, |
| }; |
| struct resolv_test *obj = resolv_test_start (config); |
| |
| test_function ("res_query", &wrap_res_query); |
| test_function ("res_search", &wrap_res_search); |
| test_function ("res_querydomain", &wrap_res_querydomain); |
| test_function ("res_send", &wrap_res_send); |
| |
| test_function ("res_nquery", &wrap_res_nquery); |
| test_function ("res_nsearch", &wrap_res_nsearch); |
| test_function ("res_nquerydomain", &wrap_res_nquerydomain); |
| test_function ("res_nsend", &wrap_res_nsend); |
| |
| resolv_test_end (obj); |
| return 0; |
| } |
| |
| #define TIMEOUT 300 |
| #include <support/test-driver.c> |