mirror of
https://sourceware.org/git/glibc.git
synced 2024-12-14 23:30:06 +00:00
401311cfba
The old implementation based on hsearch_r used an ad-hoc C string encoding and produced an incorrect format on the wire for domain names which contained bytes which needed escaping when printed. This commit switches to ns_name_pton for the wire format conversion (now that we have separate tests for it) and uses a tsearch tree with a suitable comparison function to locate compression targets.
121 lines
4.0 KiB
C
121 lines
4.0 KiB
C
/* Test handling of binary domain names with res_send.
|
|
Copyright (C) 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/resolv_test.h>
|
|
|
|
static void
|
|
response (const struct resolv_response_context *ctx,
|
|
struct resolv_response_builder *b,
|
|
const char *qname, uint16_t qclass, uint16_t qtype)
|
|
{
|
|
TEST_COMPARE (qclass, C_IN);
|
|
TEST_COMPARE (qtype, T_TXT);
|
|
TEST_VERIFY (strlen (qname) <= 255);
|
|
|
|
struct resolv_response_flags flags = { 0 };
|
|
resolv_response_init (b, flags);
|
|
resolv_response_add_question (b, qname, qclass, qtype);
|
|
resolv_response_section (b, ns_s_an);
|
|
resolv_response_open_record (b, qname, qclass, T_TXT, 0x12345678);
|
|
unsigned char qnamelen = strlen (qname);
|
|
resolv_response_add_data (b, &qnamelen, 1);
|
|
resolv_response_add_data (b, qname, qnamelen);
|
|
resolv_response_close_record (b);
|
|
}
|
|
|
|
static int
|
|
do_test (void)
|
|
{
|
|
struct resolv_test *aux = resolv_test_start
|
|
((struct resolv_redirect_config)
|
|
{
|
|
.response_callback = response,
|
|
});
|
|
|
|
for (int b = 0; b <= 255; ++b)
|
|
{
|
|
unsigned char query[] =
|
|
{
|
|
b, b, /* Transaction ID. */
|
|
1, 0, /* Query with RD flag. */
|
|
0, 1, /* One question. */
|
|
0, 0, 0, 0, 0, 0, /* The other sections are empty. */
|
|
1, b, 7, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 0,
|
|
0, T_TXT, /* TXT query. */
|
|
0, 1, /* Class IN. */
|
|
};
|
|
unsigned char response[512];
|
|
int ret = res_send (query, sizeof (query), response, sizeof (response));
|
|
|
|
char expected_name[20];
|
|
/* The name is uncompressed in the query, so we can reference it
|
|
directly. */
|
|
TEST_VERIFY_EXIT (ns_name_ntop (query + 12, expected_name,
|
|
sizeof (expected_name)) >= 0);
|
|
TEST_COMPARE (ret,
|
|
(ssize_t) sizeof (query)
|
|
+ 2 /* Compression reference. */
|
|
+ 2 + 2 + 4 + 2 /* Type, class, TTL, RDATA length. */
|
|
+ 1 /* Pascal-style string length. */
|
|
+ strlen (expected_name));
|
|
|
|
/* Mark as answer, with recursion available, and one answer. */
|
|
query[2] = 0x81;
|
|
query[3] = 0x80;
|
|
query[7] = 1;
|
|
|
|
/* Prefix of the response must match the query. */
|
|
TEST_COMPARE (memcmp (response, query, sizeof (query)), 0);
|
|
|
|
/* The actual answer follows, starting with the compression
|
|
reference. */
|
|
unsigned char *p = response + sizeof (query);
|
|
TEST_COMPARE (*p++, 0xc0);
|
|
TEST_COMPARE (*p++, 0x0c);
|
|
|
|
/* Type and class. */
|
|
TEST_COMPARE (*p++, 0);
|
|
TEST_COMPARE (*p++, T_TXT);
|
|
TEST_COMPARE (*p++, 0);
|
|
TEST_COMPARE (*p++, C_IN);
|
|
|
|
/* TTL. */
|
|
TEST_COMPARE (*p++, 0x12);
|
|
TEST_COMPARE (*p++, 0x34);
|
|
TEST_COMPARE (*p++, 0x56);
|
|
TEST_COMPARE (*p++, 0x78);
|
|
|
|
/* RDATA length. */
|
|
TEST_COMPARE (*p++, 0);
|
|
TEST_COMPARE (*p++, 1 + strlen (expected_name));
|
|
|
|
/* RDATA. */
|
|
TEST_COMPARE (*p++, strlen (expected_name));
|
|
TEST_COMPARE (memcmp (p, expected_name, strlen (expected_name)), 0);
|
|
}
|
|
|
|
resolv_test_end (aux);
|
|
|
|
return 0;
|
|
}
|
|
|
|
#include <support/test-driver.c>
|