google3/third_party/grte/v5_src/glibc-2.27/resolv/tst-resolv-binary.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>