nss/tst-nss-gai-actions.c

   1 /* Test continue and merge NSS actions for getaddrinfo.
   2    Copyright The GNU Toolchain Authors.
   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 <dlfcn.h>
  20 #include <gnu/lib-names.h>
  21 #include <nss.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25
  26 #include <support/check.h>
  27 #include <support/format_nss.h>
  28 #include <support/support.h>
  29 #include <support/xstdio.h>
  30 #include <support/xunistd.h>
  31
  32 enum
  33 {
  34   ACTION_MERGE = 0,
  35   ACTION_CONTINUE,
  36 };
  37
  38 static const char *
  39 family_str (int family)
  40 {
  41   switch (family)
  42     {
  43     case AF_UNSPEC:
  44       return "AF_UNSPEC";
  45     case AF_INET:
  46       return "AF_INET";
  47     default:
  48       __builtin_unreachable ();
  49     }
  50 }
  51
  52 static const char *
  53 action_str (int action)
  54 {
  55   switch (action)
  56     {
  57     case ACTION_MERGE:
  58       return "merge";
  59     case ACTION_CONTINUE:
  60       return "continue";
  61     default:
  62       __builtin_unreachable ();
  63     }
  64 }
  65
  66 static void
  67 do_one_test (int action, int family, bool canon)
  68 {
  69   struct addrinfo hints =
  70     {
  71       .ai_family = family,
  72     };
  73
  74   struct addrinfo *ai;
  75
  76   if (canon)
  77     hints.ai_flags = AI_CANONNAME;
  78
  79   printf ("***** Testing \"files [SUCCESS=%s] files\" for family %s, %s\n",
  80           action_str (action), family_str (family),
  81           canon ? "AI_CANONNAME" : "");
  82
  83   int ret = getaddrinfo ("example.org", "80", &hints, &ai);
  84
  85   switch (action)
  86     {
  87     case ACTION_MERGE:
  88       if (ret == 0)
  89         {
  90           char *formatted = support_format_addrinfo (ai, ret);
  91
  92           printf ("merge unexpectedly succeeded:\n %s\n", formatted);
  93           support_record_failure ();
  94           free (formatted);
  95         }
  96       else
  97         return;
  98     case ACTION_CONTINUE:
  99         {
 100           char *formatted = support_format_addrinfo (ai, ret);
 101
 102           /* Verify that the result appears exactly once.  */
 103           const char *expected = "address: STREAM/TCP 192.0.0.1 80\n"
 104             "address: DGRAM/UDP 192.0.0.1 80\n"
 105             "address: RAW/IP 192.0.0.1 80\n";
 106
 107           const char *contains = strstr (formatted, expected);
 108           const char *contains2 = NULL;
 109
 110           if (contains != NULL)
 111             contains2 = strstr (contains + strlen (expected), expected);
 112
 113           if (contains == NULL || contains2 != NULL)
 114             {
 115               printf ("continue failed:\n%s\n", formatted);
 116               support_record_failure ();
 117             }
 118
 119           free (formatted);
 120           break;
 121         }
 122     default:
 123       __builtin_unreachable ();
 124     }
 125 }
 126
 127 static void
 128 do_one_test_set (int action)
 129 {
 130   char buf[32];
 131
 132   snprintf (buf, sizeof (buf), "files [SUCCESS=%s] files",
 133             action_str (action));
 134   __nss_configure_lookup ("hosts", buf);
 135
 136   do_one_test (action, AF_UNSPEC, false);
 137   do_one_test (action, AF_INET, false);
 138   do_one_test (action, AF_INET, true);
 139 }
 140
 141 static int
 142 do_test (void)
 143 {
 144   do_one_test_set (ACTION_CONTINUE);
 145   do_one_test_set (ACTION_MERGE);
 146   return 0;
 147 }
 148
 149 #include <support/test-driver.c>