nsswitch: user new internal API (tests)

Testsuite support and new test for new API.

Reviewed-by: Siddhesh Poyarekar <siddhesh@sourceware.org>

1 files changed, 341 insertions, 0 deletions

diff --git a/nss/tst-reload1.c b/nss/tst-reload1.c
new file mode 100644
index 0000000..449874d
--- /dev/null
+++ b/nss/tst-reload1.c
@@ -0,0 +1,341 @@
+/* Test that nsswitch.conf reloading actually works.
+   Copyright (C) 2020 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
+   <https://www.gnu.org/licenses/>.  */
+
+#include <nss.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <errno.h>
+#include <pwd.h>
+
+#include <support/support.h>
+#include <support/check.h>
+
+#include "nss_test.h"
+
+/* Size of buffers used by *_r functions.  */
+#define TESTBUFLEN 4096
+
+static struct passwd pwd_table_1[] = {
+    PWD (100),
+    PWD (30),
+    PWD (200),
+    PWD (60),
+    PWD (20000),
+    PWD_LAST ()
+  };
+
+static const char *hostaddr_5[] =
+  {
+   "ABCD", "abcd", "1234", NULL
+  };
+
+static const char *hostaddr_15[] =
+  {
+   "4321", "ghij", NULL
+  };
+
+static const char *hostaddr_25[] =
+  {
+   "WXYZ", NULL
+  };
+
+
+static struct hostent host_table_1[] = {
+  HOST (5),
+  HOST (15),
+  HOST (25),
+  HOST_LAST ()
+};
+
+void
+_nss_test1_init_hook(test_tables *t)
+{
+  t->pwd_table = pwd_table_1;
+  t->host_table = host_table_1;
+}
+
+/* The first of these must not appear in pwd_table_1.  */
+static struct passwd pwd_table_2[] = {
+    PWD (5),
+    PWD_N(200, "name30"),
+    PWD (16),
+    PWD_LAST ()
+  };
+
+static const char *hostaddr_6[] =
+  {
+   "mnop", NULL
+  };
+
+static const char *hostaddr_16[] =
+  {
+   "7890", "a1b2", NULL
+  };
+
+static const char *hostaddr_26[] =
+  {
+   "qwer", "tyui", NULL
+  };
+
+static struct hostent host_table_2[] = {
+  HOST (6),
+  HOST (16),
+  HOST (26),
+  HOST_LAST ()
+};
+
+void
+_nss_test2_init_hook(test_tables *t)
+{
+  t->pwd_table = pwd_table_2;
+  t->host_table = host_table_2;
+}
+
+static void
+must_be_tests (struct passwd *pt, struct hostent *ht)
+{
+  int i;
+  struct hostent *h;
+
+  struct passwd *p;
+  for (i = 0; !PWD_ISLAST (&pt[i]); ++i)
+    {
+      p = getpwuid (pt[i].pw_uid);
+      TEST_VERIFY (p != NULL);
+      if (p != NULL)
+	{
+	  TEST_VERIFY (strcmp (p->pw_name, pt[i].pw_name) == 0);
+	}
+    }
+
+  setpwent ();
+  for (i = 0; !PWD_ISLAST (&pt[i]); ++i)
+    {
+      p = getpwent ();
+      TEST_VERIFY (p != NULL);
+      if (p != NULL)
+	{
+	  TEST_VERIFY (strcmp (p->pw_name, pt[i].pw_name) == 0);
+	  TEST_VERIFY (p->pw_uid == pt[i].pw_uid);
+	}
+    }
+  endpwent ();
+
+  for (i = 0; !HOST_ISLAST (&ht[i]); ++i)
+    {
+      h = gethostbyname (ht[i].h_name);
+      TEST_VERIFY (h != NULL);
+      if (h != NULL)
+	{
+	  TEST_VERIFY (strcmp (h->h_name, ht[i].h_name) == 0);
+	  TEST_VERIFY (h->h_addr_list[0] != NULL);
+	  if (h->h_addr_list[0])
+	    TEST_VERIFY (strcmp (h->h_addr_list[0], ht[i].h_addr_list[0]) == 0);
+	}
+    }
+
+  for (i = 0; !HOST_ISLAST (&ht[i]); ++i)
+    {
+      struct hostent r, *rp;
+      char buf[TESTBUFLEN];
+      int herrno, res;
+
+      res = gethostbyname2_r (ht[i].h_name, AF_INET,
+			    &r, buf, TESTBUFLEN, &rp, &herrno);
+      TEST_VERIFY (res == 0);
+      if (res == 0)
+	{
+	  TEST_VERIFY (strcmp (r.h_name, ht[i].h_name) == 0);
+	  TEST_VERIFY (r.h_addr_list[0] != NULL);
+	  if (r.h_addr_list[0])
+	    TEST_VERIFY (strcmp (r.h_addr_list[0], ht[i].h_addr_list[0]) == 0);
+	}
+    }
+
+  for (i = 0; !HOST_ISLAST (&ht[i]); ++i)
+    {
+      h = gethostbyaddr (ht[i].h_addr, 4, AF_INET);
+      TEST_VERIFY (h != NULL);
+      if (h != NULL)
+	{
+	  TEST_VERIFY (strcmp (h->h_name, ht[i].h_name) == 0);
+	  TEST_VERIFY (h->h_addr_list[0] != NULL);
+	  if (h->h_addr_list[0])
+	    TEST_VERIFY (strcmp (h->h_addr_list[0], ht[i].h_addr_list[0]) == 0);
+	}
+    }
+
+  /* getaddrinfo */
+
+  for (i = 0; !HOST_ISLAST (&ht[i]); ++i)
+    {
+      struct addrinfo *ap;
+      struct addrinfo hint;
+      int res, j;
+
+      memset (&hint, 0, sizeof (hint));
+      hint.ai_family = AF_INET;
+      hint.ai_socktype = SOCK_STREAM;
+      hint.ai_protocol = 0;
+      hint.ai_flags = 0;
+
+      ap = NULL;
+      res = getaddrinfo (ht[i].h_name, NULL, &hint, &ap);
+      TEST_VERIFY (res == 0);
+      TEST_VERIFY (ap != NULL);
+      if (res == 0 && ap != NULL)
+	{
+	  j = 0; /* which address in the list */
+	  while (ap)
+	    {
+	      struct sockaddr_in *in = (struct sockaddr_in *)ap->ai_addr;
+	      unsigned char *up = (unsigned char *)&in->sin_addr;
+
+	      TEST_VERIFY (memcmp (up, ht[i].h_addr_list[j], 4) == 0);
+
+	      ap = ap->ai_next;
+	      ++j;
+	    }
+	}
+    }
+
+  /* getnameinfo */
+
+  for (i = 0; !HOST_ISLAST (&ht[i]); ++i)
+    {
+      struct sockaddr_in addr;
+      int res;
+      char host_buf[NI_MAXHOST];
+
+      memset (&addr, 0, sizeof (addr));
+      addr.sin_family = AF_INET;
+      addr.sin_port = 80;
+      memcpy (& addr.sin_addr, ht[i].h_addr_list[0], 4);
+
+      res = getnameinfo ((struct sockaddr *) &addr, sizeof(addr),
+			 host_buf, sizeof(host_buf),
+			 NULL, 0, NI_NOFQDN);
+
+      TEST_VERIFY (res == 0);
+      if (res == 0)
+	TEST_VERIFY (strcmp (ht[i].h_name, host_buf) == 0);
+      else
+	printf ("error %s\n", gai_strerror (res));
+    }
+}
+
+static void
+must_be_1 (void)
+{
+  struct passwd *p;
+
+  must_be_tests (pwd_table_1, host_table_1);
+  p = getpwnam("name5");
+  TEST_VERIFY (p == NULL);
+}
+
+static void
+must_be_2 (void)
+{
+  struct passwd *p;
+
+  must_be_tests (pwd_table_2, host_table_2);
+  p = getpwnam("name100");
+  TEST_VERIFY (p == NULL);
+}
+
+static void
+xrename (const char *a, const char *b)
+{
+  int i = rename (a, b);
+  if (i != 0)
+    FAIL_EXIT1 ("rename(%s,%s) failed: %s\n", a, b, strerror(errno));
+}
+
+/* If the actions change while in the midst of doing a series of
+   lookups, make sure they're consistent.  */
+static void
+test_cross_switch_consistency (void)
+{
+  int i;
+  struct passwd *p;
+
+  /* We start by initiating a set/get/end loop on conf1.  */
+  setpwent ();
+  for (i = 0; !PWD_ISLAST (&pwd_table_1[i]); ++i)
+    {
+      p = getpwent ();
+      TEST_VERIFY (p != NULL);
+      if (p != NULL)
+	{
+	  TEST_VERIFY (strcmp (p->pw_name, pwd_table_1[i].pw_name) == 0);
+	  TEST_VERIFY (p->pw_uid == pwd_table_1[i].pw_uid);
+	}
+
+      /* After the first lookup, switch to conf2 and verify */
+      if (i == 0)
+	{
+	  xrename ("/etc/nsswitch.conf", "/etc/nsswitch.conf1");
+	  xrename ("/etc/nsswitch.conf2", "/etc/nsswitch.conf");
+
+	  p = getpwnam (pwd_table_2[0].pw_name);
+	  TEST_VERIFY (p->pw_uid == pwd_table_2[0].pw_uid);
+	}
+
+      /* But the original loop should still be on conf1.  */
+    }
+  endpwent ();
+
+  /* Make sure the set/get/end loop sees conf2 now.  */
+  setpwent ();
+  for (i = 0; !PWD_ISLAST (&pwd_table_2[i]); ++i)
+    {
+      p = getpwent ();
+      TEST_VERIFY (p != NULL);
+      if (p != NULL)
+	{
+	  TEST_VERIFY (strcmp (p->pw_name, pwd_table_2[i].pw_name) == 0);
+	  TEST_VERIFY (p->pw_uid == pwd_table_2[i].pw_uid);
+	}
+    }
+  endpwent ();
+
+}
+
+static int
+do_test (void)
+{
+  /* The test1 module was configured at program start.  */
+  must_be_1 ();
+
+  xrename ("/etc/nsswitch.conf", "/etc/nsswitch.conf1");
+  xrename ("/etc/nsswitch.conf2", "/etc/nsswitch.conf");
+  must_be_2 ();
+
+  xrename ("/etc/nsswitch.conf", "/etc/nsswitch.conf2");
+  xrename ("/etc/nsswitch.conf1", "/etc/nsswitch.conf");
+  must_be_1 ();
+
+  test_cross_switch_consistency ();
+
+  return 0;
+}
+
+#include <support/test-driver.c>