1 /* Test the creation of many struct __res_state objects.
2    Copyright (C) 2017-2022 Free Software Foundation, Inc.
3    This file is part of the GNU C Library.
4 
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9 
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14 
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, see
17    <https://www.gnu.org/licenses/>.  */
18 
19 #include <mcheck.h>
20 #include <resolv.h>
21 #include <resolv/resolv_context.h>
22 #include <stdlib.h>
23 #include <support/check.h>
24 
25 /* Order the resolver states by their extended resolver state
26    index.  */
27 static int
sort_res_state(const void * a,const void * b)28 sort_res_state (const void *a, const void *b)
29 {
30   res_state left = (res_state) a;
31   res_state right = (res_state) b;
32   return memcmp (&left->_u._ext.__glibc_extension_index,
33                  &right->_u._ext.__glibc_extension_index,
34                  sizeof (left->_u._ext.__glibc_extension_index));
35 }
36 
37 static int
do_test(void)38 do_test (void)
39 {
40   mtrace ();
41 
42   enum { count = 100 * 1000 };
43   res_state array = calloc (count, sizeof (*array));
44   const struct resolv_conf *conf = NULL;
45   for (size_t i = 0; i < count; ++i)
46     {
47       TEST_VERIFY (res_ninit (array + i) == 0);
48       TEST_VERIFY (array[i].nscount > 0);
49       struct resolv_context *ctx = __resolv_context_get_override (array + i);
50       TEST_VERIFY_EXIT (ctx != NULL);
51       TEST_VERIFY (ctx->resp == array + i);
52       if (i == 0)
53         {
54           conf = ctx->conf;
55           TEST_VERIFY (conf != NULL);
56         }
57       else
58         /* The underyling configuration should be identical across all
59            res_state opjects because resolv.conf did not change.  */
60         TEST_VERIFY (ctx->conf == conf);
61     }
62   qsort (array, count, sizeof (*array), sort_res_state);
63   for (size_t i = 1; i < count; ++i)
64     /* All extension indices should be different.  */
65     TEST_VERIFY (sort_res_state (array + i - 1, array + i) < 0);
66   for (size_t i = 0; i < count; ++i)
67     res_nclose (array + i);
68   free (array);
69 
70   TEST_VERIFY (res_init () == 0);
71   return 0;
72 }
73 
74 #define TIMEOUT 50
75 #include <support/test-driver.c>
76