Further harden glibc malloc metadata against 1-byte overflows.
[glibc.git] / nss / tst-nss-getpwent.c
blob7bd69fc378ed304f3eba4a345a3fe70a503a1834
1 /* Copyright (C) 2015-2017 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public
6 License as published by the Free Software Foundation; either
7 version 2.1 of the License, or (at your option) any later version.
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 Lesser General Public License for more details.
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; if not, see
16 <http://www.gnu.org/licenses/>. */
18 #include <pwd.h>
19 #include <stdbool.h>
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <string.h>
24 int
25 do_test (void)
27 /* Count the number of entries in the password database, and fetch
28 data from the first and last entries. */
29 size_t count = 0;
30 struct passwd * pw;
31 char *first_name = NULL;
32 uid_t first_uid = 0;
33 char *last_name = NULL;
34 uid_t last_uid = 0;
35 setpwent ();
36 while ((pw = getpwent ()) != NULL)
38 if (first_name == NULL)
40 first_name = strdup (pw->pw_name);
41 if (first_name == NULL)
43 printf ("strdup: %m\n");
44 return 1;
46 first_uid = pw->pw_uid;
49 free (last_name);
50 last_name = strdup (pw->pw_name);
51 if (last_name == NULL)
53 printf ("strdup: %m\n");
54 return 1;
56 last_uid = pw->pw_uid;
57 ++count;
59 endpwent ();
61 if (count == 0)
63 printf ("No entries in the password database.\n");
64 return 0;
67 /* Try again, this time interleaving with name-based and UID-based
68 lookup operations. The counts do not match if the interleaved
69 lookups affected the enumeration. */
70 size_t new_count = 0;
71 setpwent ();
72 while ((pw = getpwent ()) != NULL)
74 if (new_count == count)
76 printf ("Additional entry in the password database.\n");
77 return 1;
79 ++new_count;
80 struct passwd *pw2 = getpwnam (first_name);
81 if (pw2 == NULL)
83 printf ("getpwnam (%s) failed: %m\n", first_name);
84 return 1;
86 pw2 = getpwnam (last_name);
87 if (pw2 == NULL)
89 printf ("getpwnam (%s) failed: %m\n", last_name);
90 return 1;
92 pw2 = getpwuid (first_uid);
93 if (pw2 == NULL)
95 printf ("getpwuid (%llu) failed: %m\n",
96 (unsigned long long) first_uid);
97 return 1;
99 pw2 = getpwuid (last_uid);
100 if (pw2 == NULL)
102 printf ("getpwuid (%llu) failed: %m\n",
103 (unsigned long long) last_uid);
104 return 1;
107 endpwent ();
108 if (new_count < count)
110 printf ("Missing entry in the password database.\n");
111 return 1;
114 return 0;
117 #define TIMEOUT 300
118 #define TEST_FUNCTION do_test ()
119 #include "../test-skeleton.c"