resolv/tst-resolv-res_ninit.c

   1 /* Test the creation of many struct __res_state objects.
   2    Copyright (C) 2017-2023 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
  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
  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 underlying configuration should be identical across all
  59            res_state objects 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>