mirror of
				https://sourceware.org/git/glibc.git
				synced 2025-10-31 22:10:34 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			300 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			300 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* Test getnetbyname and getnetbyaddr.
 | |
|    Copyright (C) 2016-2017 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 <netdb.h>
 | |
| #include <stdlib.h>
 | |
| #include <string.h>
 | |
| #include <support/check.h>
 | |
| #include <support/check_nss.h>
 | |
| #include <support/resolv_test.h>
 | |
| #include <support/support.h>
 | |
| #include <support/xmemstream.h>
 | |
| 
 | |
| static void
 | |
| send_ptr (struct resolv_response_builder *b,
 | |
|           const char *qname, uint16_t qclass, uint16_t qtype,
 | |
|           const char *alias)
 | |
| {
 | |
|   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, T_PTR, 0);
 | |
|   resolv_response_add_name (b, alias);
 | |
|   resolv_response_close_record (b);
 | |
| }
 | |
| 
 | |
| static void
 | |
| handle_code (const struct resolv_response_context *ctx,
 | |
|              struct resolv_response_builder *b,
 | |
|              const char *qname, uint16_t qclass, uint16_t qtype,
 | |
|              int code)
 | |
| {
 | |
|   switch (code)
 | |
|     {
 | |
|     case 1:
 | |
|       send_ptr (b, qname, qclass, qtype, "1.in-addr.arpa");
 | |
|       break;
 | |
|     case 2:
 | |
|       send_ptr (b, qname, qclass, qtype, "2.1.in-addr.arpa");
 | |
|       break;
 | |
|     case 3:
 | |
|       send_ptr (b, qname, qclass, qtype, "3.2.1.in-addr.arpa");
 | |
|       break;
 | |
|     case 4:
 | |
|       send_ptr (b, qname, qclass, qtype, "4.3.2.1.in-addr.arpa");
 | |
|       break;
 | |
|     case 5:
 | |
|       /* Test multiple PTR records.  */
 | |
|       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, T_PTR, 0);
 | |
|       resolv_response_add_name (b, "127.in-addr.arpa");
 | |
|       resolv_response_close_record (b);
 | |
|       resolv_response_open_record (b, qname, qclass, T_PTR, 0);
 | |
|       resolv_response_add_name (b, "0.in-addr.arpa");
 | |
|       resolv_response_close_record (b);
 | |
|       break;
 | |
|     case 6:
 | |
|       /* Test skipping of RRSIG record.  */
 | |
|       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, T_PTR, 0);
 | |
|       resolv_response_add_name (b, "127.in-addr.arpa");
 | |
|       resolv_response_close_record (b);
 | |
| 
 | |
|       resolv_response_open_record (b, qname, qclass, 46 /* RRSIG */, 0);
 | |
|       {
 | |
|         char buf[500];
 | |
|         memset (buf, 0x3f, sizeof (buf));
 | |
|         resolv_response_add_data (b, buf, sizeof (buf));
 | |
|       }
 | |
|       resolv_response_close_record (b);
 | |
| 
 | |
|       resolv_response_open_record (b, qname, qclass, T_PTR, 0);
 | |
|       resolv_response_add_name (b, "0.in-addr.arpa");
 | |
|       resolv_response_close_record (b);
 | |
|       break;
 | |
|     case 7:
 | |
|       /* Test CNAME handling.  */
 | |
|       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, T_CNAME, 0);
 | |
|       resolv_response_add_name (b, "cname.example");
 | |
|       resolv_response_close_record (b);
 | |
|       resolv_response_open_record (b, "cname.example", qclass, T_PTR, 0);
 | |
|       resolv_response_add_name (b, "4.3.2.1.in-addr.arpa");
 | |
|       resolv_response_close_record (b);
 | |
|       break;
 | |
| 
 | |
|     case 100:
 | |
|       resolv_response_init (b, (struct resolv_response_flags) { .rcode = 0, });
 | |
|       resolv_response_add_question (b, qname, qclass, qtype);
 | |
|       break;
 | |
|     case 101:
 | |
|       resolv_response_init (b, (struct resolv_response_flags)
 | |
|                             { .rcode = NXDOMAIN, });
 | |
|       resolv_response_add_question (b, qname, qclass, qtype);
 | |
|       break;
 | |
|     case 102:
 | |
|       resolv_response_init (b, (struct resolv_response_flags) {.rcode = SERVFAIL});
 | |
|       resolv_response_add_question (b, qname, qclass, qtype);
 | |
|       break;
 | |
|     case 103:
 | |
|       /* Check response length matching.  */
 | |
|       if (!ctx->tcp)
 | |
|         {
 | |
|           resolv_response_init (b, (struct resolv_response_flags) {.tc = true});
 | |
|           resolv_response_add_question (b, qname, qclass, qtype);
 | |
|         }
 | |
|       else
 | |
|         {
 | |
|           resolv_response_init (b, (struct resolv_response_flags) {.ancount = 1});
 | |
|           resolv_response_add_question (b, qname, qclass, qtype);
 | |
|           resolv_response_section (b, ns_s_an);
 | |
|           resolv_response_open_record (b, qname, qclass, T_PTR, 0);
 | |
|           resolv_response_add_name (b, "127.in-addr.arpa");
 | |
|           resolv_response_close_record (b);
 | |
|           resolv_response_open_record (b, qname, qclass, T_PTR, 0);
 | |
|           resolv_response_add_name (b, "example");
 | |
|           resolv_response_close_record (b);
 | |
| 
 | |
|           resolv_response_open_record (b, qname, qclass, T_PTR, 0);
 | |
|           size_t to_fill = 65535 - resolv_response_length (b)
 | |
|             - 2 /* length, "n" */ - 2 /* compression reference */
 | |
|             - 2 /* RR type */;
 | |
|           for (size_t i = 0; i < to_fill; ++i)
 | |
|             resolv_response_add_data (b, "", 1);
 | |
|           resolv_response_close_record (b);
 | |
|           resolv_response_add_name (b, "n.example");
 | |
|           uint16_t rrtype = htons (T_PTR);
 | |
|           resolv_response_add_data (b, &rrtype, sizeof (rrtype));
 | |
|         }
 | |
|       break;
 | |
|     default:
 | |
|       FAIL_EXIT1 ("invalid QNAME: %s (code %d)", qname, code);
 | |
|     }
 | |
| }
 | |
| 
 | |
| static void
 | |
| response (const struct resolv_response_context *ctx,
 | |
|           struct resolv_response_builder *b,
 | |
|           const char *qname, uint16_t qclass, uint16_t qtype)
 | |
| {
 | |
|   int code;
 | |
|   if (strstr (qname, "in-addr.arpa") == NULL)
 | |
|     {
 | |
|       char *tail;
 | |
|       if (sscanf (qname, "code%d.%ms", &code, &tail) != 2
 | |
|           || strcmp (tail, "example") != 0)
 | |
|         FAIL_EXIT1 ("invalid QNAME: %s", qname);
 | |
|       free (tail);
 | |
|       handle_code (ctx, b, qname, qclass, qtype, code);
 | |
|     }
 | |
|   else
 | |
|     {
 | |
|       /* Reverse lookup.  */
 | |
|       int components[4];
 | |
|       char *tail;
 | |
|       if (sscanf (qname, "%d.%d.%d.%d.%ms",
 | |
|                   components, components + 1, components + 2, components + 3,
 | |
|                   &tail) != 5
 | |
|           || strcmp (tail, "in-addr.arpa") != 0)
 | |
|         FAIL_EXIT1 ("invalid QNAME: %s", qname);
 | |
|       free (tail);
 | |
|       handle_code (ctx, b, qname, qclass, qtype, components[3]);
 | |
|     }
 | |
| }
 | |
| 
 | |
| static void
 | |
| check_reverse (int code, const char *expected)
 | |
| {
 | |
|   char *query = xasprintf ("code=%d", code);
 | |
|   check_netent (query, getnetbyaddr (code, AF_INET), expected);
 | |
|   free (query);
 | |
| }
 | |
| 
 | |
| /* Test for CVE-2016-3075.  */
 | |
| static void
 | |
| check_long_name (void)
 | |
| {
 | |
|   struct xmemstream mem;
 | |
|   xopen_memstream (&mem);
 | |
| 
 | |
|   char label[65];
 | |
|   memset (label, 'x', 63);
 | |
|   label[63] = '.';
 | |
|   label[64] = '\0';
 | |
|   for (unsigned i = 0; i < 64 * 1024 * 1024 / strlen (label); ++i)
 | |
|     fprintf (mem.out, "%s", label);
 | |
| 
 | |
|   xfclose_memstream (&mem);
 | |
| 
 | |
|   check_netent ("long name", getnetbyname (mem.buffer),
 | |
|                 "error: NO_RECOVERY\n");
 | |
| 
 | |
|   free (mem.buffer);
 | |
| }
 | |
| 
 | |
| int
 | |
| main (void)
 | |
| {
 | |
|   struct resolv_test *obj = resolv_test_start
 | |
|     ((struct resolv_redirect_config)
 | |
|      {
 | |
|        .response_callback = response
 | |
|      });
 | |
| 
 | |
|   /* Lookup by name, success cases.  */
 | |
|   check_netent ("code1.example", getnetbyname ("code1.example"),
 | |
|                 "alias: 1.in-addr.arpa\n"
 | |
|                 "net: 0x00000001\n");
 | |
|   check_netent ("code2.example", getnetbyname ("code2.example"),
 | |
|                 "alias: 2.1.in-addr.arpa\n"
 | |
|                 "net: 0x00000102\n");
 | |
|   check_netent ("code3.example", getnetbyname ("code3.example"),
 | |
|                 "alias: 3.2.1.in-addr.arpa\n"
 | |
|                 "net: 0x00010203\n");
 | |
|   check_netent ("code4.example", getnetbyname ("code4.example"),
 | |
|                 "alias: 4.3.2.1.in-addr.arpa\n"
 | |
|                 "net: 0x01020304\n");
 | |
|   check_netent ("code5.example", getnetbyname ("code5.example"),
 | |
|                 "alias: 127.in-addr.arpa\n"
 | |
|                 "alias: 0.in-addr.arpa\n"
 | |
|                 "net: 0x0000007f\n");
 | |
|   check_netent ("code6.example", getnetbyname ("code6.example"),
 | |
|                 "alias: 127.in-addr.arpa\n"
 | |
|                 "alias: 0.in-addr.arpa\n"
 | |
|                 "net: 0x0000007f\n");
 | |
|   check_netent ("code7.example", getnetbyname ("code7.example"),
 | |
|                 "alias: 4.3.2.1.in-addr.arpa\n"
 | |
|                 "net: 0x01020304\n");
 | |
| 
 | |
|   /* Lookup by name, failure cases.  */
 | |
|   check_netent ("code100.example", getnetbyname ("code100.example"),
 | |
|                 "error: NO_ADDRESS\n");
 | |
|   check_netent ("code101.example", getnetbyname ("code101.example"),
 | |
|                 "error: HOST_NOT_FOUND\n");
 | |
|   check_netent ("code102.example", getnetbyname ("code102.example"),
 | |
|                 "error: TRY_AGAIN\n");
 | |
|   check_netent ("code103.example", getnetbyname ("code103.example"),
 | |
|                 "error: NO_RECOVERY\n");
 | |
| 
 | |
|   /* Lookup by address, success cases.  */
 | |
|   check_reverse (1,
 | |
|                  "name: 1.in-addr.arpa\n"
 | |
|                  "net: 0x00000001\n");
 | |
|   check_reverse (2,
 | |
|                  "name: 2.1.in-addr.arpa\n"
 | |
|                  "net: 0x00000002\n");
 | |
|   check_reverse (3,
 | |
|                  "name: 3.2.1.in-addr.arpa\n"
 | |
|                  "net: 0x00000003\n");
 | |
|   check_reverse (4,
 | |
|                  "name: 4.3.2.1.in-addr.arpa\n"
 | |
|                  "net: 0x00000004\n");
 | |
|   check_reverse (5,
 | |
|                  "name: 127.in-addr.arpa\n"
 | |
|                  "alias: 0.in-addr.arpa\n"
 | |
|                  "net: 0x00000005\n");
 | |
|   check_reverse (6,
 | |
|                  "name: 127.in-addr.arpa\n"
 | |
|                  "alias: 0.in-addr.arpa\n"
 | |
|                  "net: 0x00000006\n");
 | |
|   check_reverse (7,
 | |
|                  "name: 4.3.2.1.in-addr.arpa\n"
 | |
|                  "net: 0x00000007\n");
 | |
| 
 | |
|   /* Lookup by address, failure cases.  */
 | |
|   check_reverse (100,
 | |
|                  "error: NO_ADDRESS\n");
 | |
|   check_reverse (101,
 | |
|                  "error: HOST_NOT_FOUND\n");
 | |
|   check_reverse (102,
 | |
|                  "error: TRY_AGAIN\n");
 | |
|   check_reverse (103,
 | |
|                  "error: NO_RECOVERY\n");
 | |
| 
 | |
|   check_long_name ();
 | |
| 
 | |
|   resolv_test_end (obj);
 | |
| }
 |