s3-spoolss: Added a winreg_add_driver function.
[Samba/ekacnet.git] / lib / nss_wrapper / testsuite.c
blobda1a2675a2726c930804327fbf46bf0bf9a64e31
1 /*
2 Unix SMB/CIFS implementation.
4 local testing of the nss wrapper
6 Copyright (C) Guenther Deschner 2009-2010
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "torture/torture.h"
24 #include "lib/replace/system/passwd.h"
25 #include "lib/nss_wrapper/nss_wrapper.h"
27 static bool copy_passwd(struct torture_context *tctx,
28 const struct passwd *pwd,
29 struct passwd *p)
31 p->pw_name = talloc_strdup(tctx, pwd->pw_name);
32 p->pw_passwd = talloc_strdup(tctx, pwd->pw_passwd);
33 p->pw_uid = pwd->pw_uid;
34 p->pw_gid = pwd->pw_gid;
35 p->pw_gecos = talloc_strdup(tctx, pwd->pw_gecos);
36 p->pw_dir = talloc_strdup(tctx, pwd->pw_dir);
37 p->pw_shell = talloc_strdup(tctx, pwd->pw_shell);
39 return true;
42 static void print_passwd(struct passwd *pwd)
44 printf("%s:%s:%lu:%lu:%s:%s:%s\n",
45 pwd->pw_name,
46 pwd->pw_passwd,
47 (unsigned long)pwd->pw_uid,
48 (unsigned long)pwd->pw_gid,
49 pwd->pw_gecos,
50 pwd->pw_dir,
51 pwd->pw_shell);
55 static bool test_nwrap_getpwnam(struct torture_context *tctx,
56 const char *name,
57 struct passwd *pwd_p)
59 struct passwd *pwd;
61 torture_comment(tctx, "Testing getpwnam: %s\n", name);
63 pwd = getpwnam(name);
64 if (pwd) {
65 print_passwd(pwd);
68 if (pwd_p) {
69 copy_passwd(tctx, pwd, pwd_p);
72 return pwd ? true : false;
75 static bool test_nwrap_getpwnam_r(struct torture_context *tctx,
76 const char *name,
77 struct passwd *pwd_p)
79 struct passwd pwd, *pwdp;
80 char buffer[4096];
81 int ret;
83 torture_comment(tctx, "Testing getpwnam_r: %s\n", name);
85 ret = getpwnam_r(name, &pwd, buffer, sizeof(buffer), &pwdp);
86 if (ret != 0) {
87 if (ret != ENOENT) {
88 torture_comment(tctx, "got %d return code\n", ret);
90 return false;
93 print_passwd(&pwd);
95 if (pwd_p) {
96 copy_passwd(tctx, &pwd, pwd_p);
99 return true;
102 static bool test_nwrap_getpwuid(struct torture_context *tctx,
103 uid_t uid,
104 struct passwd *pwd_p)
106 struct passwd *pwd;
108 torture_comment(tctx, "Testing getpwuid: %lu\n", (unsigned long)uid);
110 pwd = getpwuid(uid);
111 if (pwd) {
112 print_passwd(pwd);
115 if (pwd_p) {
116 copy_passwd(tctx, pwd, pwd_p);
119 return pwd ? true : false;
122 static bool test_nwrap_getpwuid_r(struct torture_context *tctx,
123 uid_t uid,
124 struct passwd *pwd_p)
126 struct passwd pwd, *pwdp;
127 char buffer[4096];
128 int ret;
130 torture_comment(tctx, "Testing getpwuid_r: %lu\n", (unsigned long)uid);
132 ret = getpwuid_r(uid, &pwd, buffer, sizeof(buffer), &pwdp);
133 if (ret != 0) {
134 if (ret != ENOENT) {
135 torture_comment(tctx, "got %d return code\n", ret);
137 return false;
140 print_passwd(&pwd);
142 if (pwd_p) {
143 copy_passwd(tctx, &pwd, pwd_p);
146 return true;
150 static bool copy_group(struct torture_context *tctx,
151 const struct group *grp,
152 struct group *g)
154 int i;
156 g->gr_name = talloc_strdup(tctx, grp->gr_name);
157 g->gr_passwd = talloc_strdup(tctx, grp->gr_passwd);
158 g->gr_gid = grp->gr_gid;
159 g->gr_mem = NULL;
161 for (i=0; grp->gr_mem && grp->gr_mem[i]; i++) {
162 g->gr_mem = talloc_realloc(tctx, g->gr_mem, char *, i + 2);
163 g->gr_mem[i] = talloc_strdup(g->gr_mem, grp->gr_mem[i]);
164 g->gr_mem[i+1] = NULL;
167 return true;
170 static void print_group(struct group *grp)
172 int i;
173 printf("%s:%s:%lu:",
174 grp->gr_name,
175 grp->gr_passwd,
176 (unsigned long)grp->gr_gid);
178 if ((grp->gr_mem == NULL) || !grp->gr_mem[0]) {
179 printf("\n");
180 return;
183 for (i=0; grp->gr_mem[i+1]; i++) {
184 printf("%s,", grp->gr_mem[i]);
186 printf("%s\n", grp->gr_mem[i]);
189 static bool test_nwrap_getgrnam(struct torture_context *tctx,
190 const char *name,
191 struct group *grp_p)
193 struct group *grp;
195 torture_comment(tctx, "Testing getgrnam: %s\n", name);
197 grp = getgrnam(name);
198 if (grp) {
199 print_group(grp);
202 if (grp_p) {
203 copy_group(tctx, grp, grp_p);
206 return grp ? true : false;
209 static bool test_nwrap_getgrnam_r(struct torture_context *tctx,
210 const char *name,
211 struct group *grp_p)
213 struct group grp, *grpp;
214 char buffer[4096];
215 int ret;
217 torture_comment(tctx, "Testing getgrnam_r: %s\n", name);
219 ret = getgrnam_r(name, &grp, buffer, sizeof(buffer), &grpp);
220 if (ret != 0) {
221 if (ret != ENOENT) {
222 torture_comment(tctx, "got %d return code\n", ret);
224 return false;
227 print_group(&grp);
229 if (grp_p) {
230 copy_group(tctx, &grp, grp_p);
233 return true;
237 static bool test_nwrap_getgrgid(struct torture_context *tctx,
238 gid_t gid,
239 struct group *grp_p)
241 struct group *grp;
243 torture_comment(tctx, "Testing getgrgid: %lu\n", (unsigned long)gid);
245 grp = getgrgid(gid);
246 if (grp) {
247 print_group(grp);
250 if (grp_p) {
251 copy_group(tctx, grp, grp_p);
254 return grp ? true : false;
257 static bool test_nwrap_getgrgid_r(struct torture_context *tctx,
258 gid_t gid,
259 struct group *grp_p)
261 struct group grp, *grpp;
262 char buffer[4096];
263 int ret;
265 torture_comment(tctx, "Testing getgrgid_r: %lu\n", (unsigned long)gid);
267 ret = getgrgid_r(gid, &grp, buffer, sizeof(buffer), &grpp);
268 if (ret != 0) {
269 if (ret != ENOENT) {
270 torture_comment(tctx, "got %d return code\n", ret);
272 return false;
275 print_group(&grp);
277 if (grp_p) {
278 copy_group(tctx, &grp, grp_p);
281 return true;
284 static bool test_nwrap_enum_passwd(struct torture_context *tctx,
285 struct passwd **pwd_array_p,
286 size_t *num_pwd_p)
288 struct passwd *pwd;
289 struct passwd *pwd_array = NULL;
290 size_t num_pwd = 0;
292 torture_comment(tctx, "Testing setpwent\n");
293 setpwent();
295 while ((pwd = getpwent()) != NULL) {
296 torture_comment(tctx, "Testing getpwent\n");
298 print_passwd(pwd);
299 if (pwd_array_p && num_pwd_p) {
300 pwd_array = talloc_realloc(tctx, pwd_array, struct passwd, num_pwd+1);
301 torture_assert(tctx, pwd_array, "out of memory");
302 copy_passwd(tctx, pwd, &pwd_array[num_pwd]);
303 num_pwd++;
307 torture_comment(tctx, "Testing endpwent\n");
308 endpwent();
310 if (pwd_array_p) {
311 *pwd_array_p = pwd_array;
313 if (num_pwd_p) {
314 *num_pwd_p = num_pwd;
317 return true;
320 static bool test_nwrap_enum_r_passwd(struct torture_context *tctx,
321 struct passwd **pwd_array_p,
322 size_t *num_pwd_p)
324 struct passwd pwd, *pwdp;
325 struct passwd *pwd_array = NULL;
326 size_t num_pwd = 0;
327 char buffer[4096];
328 int ret;
330 torture_comment(tctx, "Testing setpwent\n");
331 setpwent();
333 while (1) {
334 torture_comment(tctx, "Testing getpwent_r\n");
336 ret = getpwent_r(&pwd, buffer, sizeof(buffer), &pwdp);
337 if (ret != 0) {
338 if (ret != ENOENT) {
339 torture_comment(tctx, "got %d return code\n", ret);
341 break;
343 print_passwd(&pwd);
344 if (pwd_array_p && num_pwd_p) {
345 pwd_array = talloc_realloc(tctx, pwd_array, struct passwd, num_pwd+1);
346 torture_assert(tctx, pwd_array, "out of memory");
347 copy_passwd(tctx, &pwd, &pwd_array[num_pwd]);
348 num_pwd++;
352 torture_comment(tctx, "Testing endpwent\n");
353 endpwent();
355 if (pwd_array_p) {
356 *pwd_array_p = pwd_array;
358 if (num_pwd_p) {
359 *num_pwd_p = num_pwd;
362 return true;
365 static bool torture_assert_passwd_equal(struct torture_context *tctx,
366 const struct passwd *p1,
367 const struct passwd *p2,
368 const char *comment)
370 torture_assert_str_equal(tctx, p1->pw_name, p2->pw_name, comment);
371 torture_assert_str_equal(tctx, p1->pw_passwd, p2->pw_passwd, comment);
372 torture_assert_int_equal(tctx, p1->pw_uid, p2->pw_uid, comment);
373 torture_assert_int_equal(tctx, p1->pw_gid, p2->pw_gid, comment);
374 torture_assert_str_equal(tctx, p1->pw_gecos, p2->pw_gecos, comment);
375 torture_assert_str_equal(tctx, p1->pw_dir, p2->pw_dir, comment);
376 torture_assert_str_equal(tctx, p1->pw_shell, p2->pw_shell, comment);
378 return true;
381 static bool test_nwrap_passwd(struct torture_context *tctx)
383 int i;
384 struct passwd *pwd, pwd1, pwd2;
385 size_t num_pwd;
387 torture_assert(tctx, test_nwrap_enum_passwd(tctx, &pwd, &num_pwd),
388 "failed to enumerate passwd");
390 for (i=0; i < num_pwd; i++) {
391 torture_assert(tctx, test_nwrap_getpwnam(tctx, pwd[i].pw_name, &pwd1),
392 "failed to call getpwnam for enumerated user");
393 torture_assert_passwd_equal(tctx, &pwd[i], &pwd1,
394 "getpwent and getpwnam gave different results");
395 torture_assert(tctx, test_nwrap_getpwuid(tctx, pwd[i].pw_uid, &pwd2),
396 "failed to call getpwuid for enumerated user");
397 torture_assert_passwd_equal(tctx, &pwd[i], &pwd2,
398 "getpwent and getpwuid gave different results");
399 torture_assert_passwd_equal(tctx, &pwd1, &pwd2,
400 "getpwnam and getpwuid gave different results");
403 return true;
406 static bool test_nwrap_passwd_r(struct torture_context *tctx)
408 int i;
409 struct passwd *pwd, pwd1, pwd2;
410 size_t num_pwd;
412 torture_assert(tctx, test_nwrap_enum_r_passwd(tctx, &pwd, &num_pwd),
413 "failed to enumerate passwd");
415 for (i=0; i < num_pwd; i++) {
416 torture_assert(tctx, test_nwrap_getpwnam_r(tctx, pwd[i].pw_name, &pwd1),
417 "failed to call getpwnam_r for enumerated user");
418 torture_assert_passwd_equal(tctx, &pwd[i], &pwd1,
419 "getpwent_r and getpwnam_r gave different results");
420 torture_assert(tctx, test_nwrap_getpwuid_r(tctx, pwd[i].pw_uid, &pwd2),
421 "failed to call getpwuid_r for enumerated user");
422 torture_assert_passwd_equal(tctx, &pwd[i], &pwd2,
423 "getpwent_r and getpwuid_r gave different results");
424 torture_assert_passwd_equal(tctx, &pwd1, &pwd2,
425 "getpwnam_r and getpwuid_r gave different results");
428 return true;
431 static bool test_nwrap_passwd_r_cross(struct torture_context *tctx)
433 int i;
434 struct passwd *pwd, pwd1, pwd2, pwd3, pwd4;
435 size_t num_pwd;
437 torture_assert(tctx, test_nwrap_enum_r_passwd(tctx, &pwd, &num_pwd),
438 "failed to enumerate passwd");
440 for (i=0; i < num_pwd; i++) {
441 torture_assert(tctx, test_nwrap_getpwnam_r(tctx, pwd[i].pw_name, &pwd1),
442 "failed to call getpwnam_r for enumerated user");
443 torture_assert_passwd_equal(tctx, &pwd[i], &pwd1,
444 "getpwent_r and getpwnam_r gave different results");
445 torture_assert(tctx, test_nwrap_getpwuid_r(tctx, pwd[i].pw_uid, &pwd2),
446 "failed to call getpwuid_r for enumerated user");
447 torture_assert_passwd_equal(tctx, &pwd[i], &pwd2,
448 "getpwent_r and getpwuid_r gave different results");
449 torture_assert_passwd_equal(tctx, &pwd1, &pwd2,
450 "getpwnam_r and getpwuid_r gave different results");
451 torture_assert(tctx, test_nwrap_getpwnam(tctx, pwd[i].pw_name, &pwd3),
452 "failed to call getpwnam for enumerated user");
453 torture_assert_passwd_equal(tctx, &pwd[i], &pwd3,
454 "getpwent_r and getpwnam gave different results");
455 torture_assert(tctx, test_nwrap_getpwuid(tctx, pwd[i].pw_uid, &pwd4),
456 "failed to call getpwuid for enumerated user");
457 torture_assert_passwd_equal(tctx, &pwd[i], &pwd4,
458 "getpwent_r and getpwuid gave different results");
459 torture_assert_passwd_equal(tctx, &pwd3, &pwd4,
460 "getpwnam and getpwuid gave different results");
463 return true;
466 static bool test_nwrap_enum_group(struct torture_context *tctx,
467 struct group **grp_array_p,
468 size_t *num_grp_p)
470 struct group *grp;
471 struct group *grp_array = NULL;
472 size_t num_grp = 0;
474 torture_comment(tctx, "Testing setgrent\n");
475 setgrent();
477 while ((grp = getgrent()) != NULL) {
478 torture_comment(tctx, "Testing getgrent\n");
480 print_group(grp);
481 if (grp_array_p && num_grp_p) {
482 grp_array = talloc_realloc(tctx, grp_array, struct group, num_grp+1);
483 torture_assert(tctx, grp_array, "out of memory");
484 copy_group(tctx, grp, &grp_array[num_grp]);
485 num_grp++;
489 torture_comment(tctx, "Testing endgrent\n");
490 endgrent();
492 if (grp_array_p) {
493 *grp_array_p = grp_array;
495 if (num_grp_p) {
496 *num_grp_p = num_grp;
499 return true;
502 static bool test_nwrap_enum_r_group(struct torture_context *tctx,
503 struct group **grp_array_p,
504 size_t *num_grp_p)
506 struct group grp, *grpp;
507 struct group *grp_array = NULL;
508 size_t num_grp = 0;
509 char buffer[4096];
510 int ret;
512 torture_comment(tctx, "Testing setgrent\n");
513 setgrent();
515 while (1) {
516 torture_comment(tctx, "Testing getgrent_r\n");
518 ret = getgrent_r(&grp, buffer, sizeof(buffer), &grpp);
519 if (ret != 0) {
520 if (ret != ENOENT) {
521 torture_comment(tctx, "got %d return code\n", ret);
523 break;
525 print_group(&grp);
526 if (grp_array_p && num_grp_p) {
527 grp_array = talloc_realloc(tctx, grp_array, struct group, num_grp+1);
528 torture_assert(tctx, grp_array, "out of memory");
529 copy_group(tctx, &grp, &grp_array[num_grp]);
530 num_grp++;
534 torture_comment(tctx, "Testing endgrent\n");
535 endgrent();
537 if (grp_array_p) {
538 *grp_array_p = grp_array;
540 if (num_grp_p) {
541 *num_grp_p = num_grp;
544 return true;
547 static bool torture_assert_group_equal(struct torture_context *tctx,
548 const struct group *g1,
549 const struct group *g2,
550 const char *comment)
552 int i;
553 torture_assert_str_equal(tctx, g1->gr_name, g2->gr_name, comment);
554 torture_assert_str_equal(tctx, g1->gr_passwd, g2->gr_passwd, comment);
555 torture_assert_int_equal(tctx, g1->gr_gid, g2->gr_gid, comment);
556 if (g1->gr_mem && !g2->gr_mem) {
557 return false;
559 if (!g1->gr_mem && g2->gr_mem) {
560 return false;
562 if (!g1->gr_mem && !g2->gr_mem) {
563 return true;
565 for (i=0; g1->gr_mem[i] && g2->gr_mem[i]; i++) {
566 torture_assert_str_equal(tctx, g1->gr_mem[i], g2->gr_mem[i], comment);
569 return true;
572 static bool test_nwrap_group(struct torture_context *tctx)
574 int i;
575 struct group *grp, grp1, grp2;
576 size_t num_grp;
578 torture_assert(tctx, test_nwrap_enum_group(tctx, &grp, &num_grp),
579 "failed to enumerate group");
581 for (i=0; i < num_grp; i++) {
582 torture_assert(tctx, test_nwrap_getgrnam(tctx, grp[i].gr_name, &grp1),
583 "failed to call getgrnam for enumerated user");
584 torture_assert_group_equal(tctx, &grp[i], &grp1,
585 "getgrent and getgrnam gave different results");
586 torture_assert(tctx, test_nwrap_getgrgid(tctx, grp[i].gr_gid, &grp2),
587 "failed to call getgrgid for enumerated user");
588 torture_assert_group_equal(tctx, &grp[i], &grp2,
589 "getgrent and getgrgid gave different results");
590 torture_assert_group_equal(tctx, &grp1, &grp2,
591 "getgrnam and getgrgid gave different results");
594 return true;
597 static bool test_nwrap_group_r(struct torture_context *tctx)
599 int i;
600 struct group *grp, grp1, grp2;
601 size_t num_grp;
603 torture_assert(tctx, test_nwrap_enum_r_group(tctx, &grp, &num_grp),
604 "failed to enumerate group");
606 for (i=0; i < num_grp; i++) {
607 torture_assert(tctx, test_nwrap_getgrnam_r(tctx, grp[i].gr_name, &grp1),
608 "failed to call getgrnam_r for enumerated user");
609 torture_assert_group_equal(tctx, &grp[i], &grp1,
610 "getgrent_r and getgrnam_r gave different results");
611 torture_assert(tctx, test_nwrap_getgrgid_r(tctx, grp[i].gr_gid, &grp2),
612 "failed to call getgrgid_r for enumerated user");
613 torture_assert_group_equal(tctx, &grp[i], &grp2,
614 "getgrent_r and getgrgid_r gave different results");
615 torture_assert_group_equal(tctx, &grp1, &grp2,
616 "getgrnam_r and getgrgid_r gave different results");
619 return true;
622 static bool test_nwrap_group_r_cross(struct torture_context *tctx)
624 int i;
625 struct group *grp, grp1, grp2, grp3, grp4;
626 size_t num_grp;
628 torture_assert(tctx, test_nwrap_enum_r_group(tctx, &grp, &num_grp),
629 "failed to enumerate group");
631 for (i=0; i < num_grp; i++) {
632 torture_assert(tctx, test_nwrap_getgrnam_r(tctx, grp[i].gr_name, &grp1),
633 "failed to call getgrnam_r for enumerated user");
634 torture_assert_group_equal(tctx, &grp[i], &grp1,
635 "getgrent_r and getgrnam_r gave different results");
636 torture_assert(tctx, test_nwrap_getgrgid_r(tctx, grp[i].gr_gid, &grp2),
637 "failed to call getgrgid_r for enumerated user");
638 torture_assert_group_equal(tctx, &grp[i], &grp2,
639 "getgrent_r and getgrgid_r gave different results");
640 torture_assert_group_equal(tctx, &grp1, &grp2,
641 "getgrnam_r and getgrgid_r gave different results");
642 torture_assert(tctx, test_nwrap_getgrnam(tctx, grp[i].gr_name, &grp3),
643 "failed to call getgrnam for enumerated user");
644 torture_assert_group_equal(tctx, &grp[i], &grp3,
645 "getgrent_r and getgrnam gave different results");
646 torture_assert(tctx, test_nwrap_getgrgid(tctx, grp[i].gr_gid, &grp4),
647 "failed to call getgrgid for enumerated user");
648 torture_assert_group_equal(tctx, &grp[i], &grp4,
649 "getgrent_r and getgrgid gave different results");
650 torture_assert_group_equal(tctx, &grp3, &grp4,
651 "getgrnam and getgrgid gave different results");
654 return true;
657 static bool test_nwrap_getgrouplist(struct torture_context *tctx,
658 const char *user,
659 gid_t gid,
660 gid_t **gids_p,
661 int *num_gids_p)
663 int ret;
664 int num_groups = 0;
665 gid_t *groups = NULL;
667 torture_comment(tctx, "Testing getgrouplist: %s\n", user);
669 ret = getgrouplist(user, gid, NULL, &num_groups);
670 if (ret == -1 || num_groups != 0) {
672 groups = talloc_array(tctx, gid_t, num_groups);
673 torture_assert(tctx, groups, "out of memory\n");
675 ret = getgrouplist(user, gid, groups, &num_groups);
678 torture_assert(tctx, (ret != -1), "failed to call getgrouplist");
680 torture_comment(tctx, "%s is member in %d groups\n", user, num_groups);
682 if (gids_p) {
683 *gids_p = groups;
685 if (num_gids_p) {
686 *num_gids_p = num_groups;
689 return true;
692 static bool test_nwrap_user_in_group(struct torture_context *tctx,
693 const struct passwd *pwd,
694 const struct group *grp)
696 int i;
698 for (i=0; grp->gr_mem && grp->gr_mem[i] != NULL; i++) {
699 if (strequal(grp->gr_mem[i], pwd->pw_name)) {
700 return true;
704 return false;
707 static bool test_nwrap_membership_user(struct torture_context *tctx,
708 const struct passwd *pwd,
709 struct group *grp_array,
710 size_t num_grp)
712 int num_user_groups = 0;
713 int num_user_groups_from_enum = 0;
714 gid_t *user_groups = NULL;
715 int g, i;
716 bool primary_group_had_user_member = false;
718 torture_assert(tctx, test_nwrap_getgrouplist(tctx,
719 pwd->pw_name,
720 pwd->pw_gid,
721 &user_groups,
722 &num_user_groups),
723 "failed to test getgrouplist");
725 for (g=0; g < num_user_groups; g++) {
726 torture_assert(tctx, test_nwrap_getgrgid(tctx, user_groups[g], NULL),
727 "failed to find the group the user is a member of");
731 for (i=0; i < num_grp; i++) {
733 struct group grp = grp_array[i];
735 if (test_nwrap_user_in_group(tctx, pwd, &grp)) {
737 struct group current_grp;
738 num_user_groups_from_enum++;
740 torture_assert(tctx, test_nwrap_getgrnam(tctx, grp.gr_name, &current_grp),
741 "failed to find the group the user is a member of");
743 if (current_grp.gr_gid == pwd->pw_gid) {
744 torture_comment(tctx, "primary group %s of user %s lists user as member\n",
745 current_grp.gr_name,
746 pwd->pw_name);
747 primary_group_had_user_member = true;
750 continue;
754 if (!primary_group_had_user_member) {
755 num_user_groups_from_enum++;
758 torture_assert_int_equal(tctx, num_user_groups, num_user_groups_from_enum,
759 "getgrouplist and real inspection of grouplist gave different results\n");
761 return true;
764 static bool test_nwrap_membership(struct torture_context *tctx)
766 const char *old_pwd = getenv("NSS_WRAPPER_PASSWD");
767 const char *old_group = getenv("NSS_WRAPPER_GROUP");
768 struct passwd *pwd;
769 size_t num_pwd;
770 struct group *grp;
771 size_t num_grp;
772 int i;
774 if (!old_pwd || !old_group) {
775 torture_comment(tctx, "ENV NSS_WRAPPER_PASSWD or NSS_WRAPPER_GROUP not set\n");
776 torture_skip(tctx, "nothing to test\n");
779 torture_assert(tctx, test_nwrap_enum_passwd(tctx, &pwd, &num_pwd),
780 "failed to enumerate passwd");
781 torture_assert(tctx, test_nwrap_enum_group(tctx, &grp, &num_grp),
782 "failed to enumerate group");
784 for (i=0; i < num_pwd; i++) {
786 torture_assert(tctx, test_nwrap_membership_user(tctx, &pwd[i], grp, num_grp),
787 "failed to test membership for user");
791 return true;
794 static bool test_nwrap_enumeration(struct torture_context *tctx)
796 const char *old_pwd = getenv("NSS_WRAPPER_PASSWD");
797 const char *old_group = getenv("NSS_WRAPPER_GROUP");
799 if (!old_pwd || !old_group) {
800 torture_comment(tctx, "ENV NSS_WRAPPER_PASSWD or NSS_WRAPPER_GROUP not set\n");
801 torture_skip(tctx, "nothing to test\n");
804 torture_assert(tctx, test_nwrap_passwd(tctx),
805 "failed to test users");
806 torture_assert(tctx, test_nwrap_group(tctx),
807 "failed to test groups");
809 return true;
812 static bool test_nwrap_reentrant_enumeration(struct torture_context *tctx)
814 const char *old_pwd = getenv("NSS_WRAPPER_PASSWD");
815 const char *old_group = getenv("NSS_WRAPPER_GROUP");
817 if (!old_pwd || !old_group) {
818 torture_comment(tctx, "ENV NSS_WRAPPER_PASSWD or NSS_WRAPPER_GROUP not set\n");
819 torture_skip(tctx, "nothing to test\n");
822 torture_comment(tctx, "Testing re-entrant calls\n");
824 torture_assert(tctx, test_nwrap_passwd_r(tctx),
825 "failed to test users");
826 torture_assert(tctx, test_nwrap_group_r(tctx),
827 "failed to test groups");
829 return true;
832 static bool test_nwrap_reentrant_enumeration_crosschecks(struct torture_context *tctx)
834 const char *old_pwd = getenv("NSS_WRAPPER_PASSWD");
835 const char *old_group = getenv("NSS_WRAPPER_GROUP");
837 if (!old_pwd || !old_group) {
838 torture_comment(tctx, "ENV NSS_WRAPPER_PASSWD or NSS_WRAPPER_GROUP not set\n");
839 torture_skip(tctx, "nothing to test\n");
842 torture_comment(tctx, "Testing re-entrant calls with cross checks\n");
844 torture_assert(tctx, test_nwrap_passwd_r_cross(tctx),
845 "failed to test users");
846 torture_assert(tctx, test_nwrap_group_r_cross(tctx),
847 "failed to test groups");
849 return true;
852 static bool test_nwrap_passwd_duplicates(struct torture_context *tctx)
854 int i, d;
855 struct passwd *pwd;
856 size_t num_pwd;
857 int duplicates = 0;
859 torture_assert(tctx, test_nwrap_enum_passwd(tctx, &pwd, &num_pwd),
860 "failed to enumerate passwd");
862 for (i=0; i < num_pwd; i++) {
863 const char *current_name = pwd[i].pw_name;
864 for (d=0; d < num_pwd; d++) {
865 const char *dup_name = pwd[d].pw_name;
866 if (d == i) {
867 continue;
869 if (!strequal(current_name, dup_name)) {
870 continue;
873 torture_warning(tctx, "found duplicate names:");
874 print_passwd(&pwd[d]);
875 print_passwd(&pwd[i]);
876 duplicates++;
880 if (duplicates) {
881 torture_fail(tctx, talloc_asprintf(tctx, "found %d duplicate names", duplicates));
884 return true;
887 static bool test_nwrap_group_duplicates(struct torture_context *tctx)
889 int i, d;
890 struct group *grp;
891 size_t num_grp;
892 int duplicates = 0;
894 torture_assert(tctx, test_nwrap_enum_group(tctx, &grp, &num_grp),
895 "failed to enumerate group");
897 for (i=0; i < num_grp; i++) {
898 const char *current_name = grp[i].gr_name;
899 for (d=0; d < num_grp; d++) {
900 const char *dup_name = grp[d].gr_name;
901 if (d == i) {
902 continue;
904 if (!strequal(current_name, dup_name)) {
905 continue;
908 torture_warning(tctx, "found duplicate names:");
909 print_group(&grp[d]);
910 print_group(&grp[i]);
911 duplicates++;
915 if (duplicates) {
916 torture_fail(tctx, talloc_asprintf(tctx, "found %d duplicate names", duplicates));
919 return true;
923 static bool test_nwrap_duplicates(struct torture_context *tctx)
925 const char *old_pwd = getenv("NSS_WRAPPER_PASSWD");
926 const char *old_group = getenv("NSS_WRAPPER_GROUP");
928 if (!old_pwd || !old_group) {
929 torture_comment(tctx, "ENV NSS_WRAPPER_PASSWD or NSS_WRAPPER_GROUP not set\n");
930 torture_skip(tctx, "nothing to test\n");
933 torture_assert(tctx, test_nwrap_passwd_duplicates(tctx),
934 "failed to test users");
935 torture_assert(tctx, test_nwrap_group_duplicates(tctx),
936 "failed to test groups");
938 return true;
942 struct torture_suite *torture_local_nss_wrapper(TALLOC_CTX *mem_ctx)
944 struct torture_suite *suite = torture_suite_create(mem_ctx, "NSS-WRAPPER");
946 torture_suite_add_simple_test(suite, "enumeration", test_nwrap_enumeration);
947 torture_suite_add_simple_test(suite, "reentrant enumeration", test_nwrap_reentrant_enumeration);
948 torture_suite_add_simple_test(suite, "reentrant enumeration crosschecks", test_nwrap_reentrant_enumeration_crosschecks);
949 torture_suite_add_simple_test(suite, "membership", test_nwrap_membership);
950 torture_suite_add_simple_test(suite, "duplicates", test_nwrap_duplicates);
952 return suite;