nss_wrapper: restructure parts of the testsuite.
[ira/wip.git] / lib / nss_wrapper / testsuite.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    local testing of the nss wrapper
5
6    Copyright (C) Guenther Deschner 2009
7
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.
12
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.
17
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/>.
20 */
21
22 #include "includes.h"
23 #include "torture/torture.h"
24 #include "lib/replace/system/passwd.h"
25 #include "lib/nss_wrapper/nss_wrapper.h"
26
27 static void print_passwd(struct passwd *pwd)
28 {
29         printf("%s:%s:%lu:%lu:%s:%s:%s\n",
30                pwd->pw_name,
31                pwd->pw_passwd,
32                (unsigned long)pwd->pw_uid,
33                (unsigned long)pwd->pw_gid,
34                pwd->pw_gecos,
35                pwd->pw_dir,
36                pwd->pw_shell);
37 }
38
39
40 static bool test_nwrap_getpwnam(struct torture_context *tctx,
41                                 const char *name)
42 {
43         struct passwd *pwd;
44
45         torture_comment(tctx, "Testing getpwnam: %s\n", name);
46
47         pwd = getpwnam(name);
48         if (pwd) {
49                 print_passwd(pwd);
50         }
51
52         return pwd ? true : false;
53 }
54
55 static bool test_nwrap_getpwuid(struct torture_context *tctx,
56                                 uid_t uid)
57 {
58         struct passwd *pwd;
59
60         torture_comment(tctx, "Testing getpwuid: %lu\n", (unsigned long)uid);
61
62         pwd = getpwuid(uid);
63         if (pwd) {
64                 print_passwd(pwd);
65         }
66
67         return pwd ? true : false;
68 }
69
70 static void print_group(struct group *grp)
71 {
72         int i;
73         printf("%s:%s:%lu:",
74                grp->gr_name,
75                grp->gr_passwd,
76                (unsigned long)grp->gr_gid);
77
78         if (!grp->gr_mem[0]) {
79                 printf("\n");
80                 return;
81         }
82
83         for (i=0; grp->gr_mem[i+1]; i++) {
84                 printf("%s,", grp->gr_mem[i]);
85         }
86         printf("%s\n", grp->gr_mem[i]);
87 }
88
89 static bool test_nwrap_getgrnam(struct torture_context *tctx,
90                                 const char *name)
91 {
92         struct group *grp;
93
94         torture_comment(tctx, "Testing getgrnam: %s\n", name);
95
96         grp = getgrnam(name);
97         if (grp) {
98                 print_group(grp);
99         }
100
101         return grp ? true : false;
102 }
103
104 static bool test_nwrap_getgrgid(struct torture_context *tctx,
105                                 gid_t gid)
106 {
107         struct group *grp;
108
109         torture_comment(tctx, "Testing getgrgid: %lu\n", (unsigned long)gid);
110
111         grp = getgrgid(gid);
112         if (grp) {
113                 print_group(grp);
114         }
115
116         return grp ? true : false;
117 }
118
119 static bool test_nwrap_enum_passwd(struct torture_context *tctx,
120                                    struct passwd **pwd_array_p,
121                                    size_t *num_pwd_p)
122 {
123         struct passwd *pwd;
124         struct passwd *pwd_array = NULL;
125         size_t num_pwd = 0;
126
127         torture_comment(tctx, "Testing setpwent\n");
128         setpwent();
129
130         while ((pwd = getpwent()) != NULL) {
131                 torture_comment(tctx, "Testing getpwent\n");
132
133                 print_passwd(pwd);
134                 if (pwd_array_p && num_pwd_p) {
135                         pwd_array = talloc_realloc(tctx, pwd_array, struct passwd, num_pwd+1);
136                         torture_assert(tctx, pwd_array, "out of memory");
137                         pwd_array[num_pwd].pw_name = talloc_strdup(tctx, pwd->pw_name);
138                         pwd_array[num_pwd].pw_uid = pwd->pw_uid;
139                         pwd_array[num_pwd].pw_gid = pwd->pw_gid;
140                         num_pwd++;
141                 }
142         }
143
144         torture_comment(tctx, "Testing endpwent\n");
145         endpwent();
146
147         if (pwd_array_p) {
148                 *pwd_array_p = pwd_array;
149         }
150         if (num_pwd_p) {
151                 *num_pwd_p = num_pwd;
152         }
153
154         return true;
155 }
156
157 static bool test_nwrap_passwd(struct torture_context *tctx)
158 {
159         int i;
160         struct passwd *pwd;
161         size_t num_pwd;
162
163         torture_assert(tctx, test_nwrap_enum_passwd(tctx, &pwd, &num_pwd),
164                                                     "failed to enumerate passwd");
165
166         for (i=0; i < num_pwd; i++) {
167                 torture_assert(tctx, test_nwrap_getpwnam(tctx, pwd[i].pw_name),
168                         "failed to call getpwnam for enumerated user");
169                 torture_assert(tctx, test_nwrap_getpwuid(tctx, pwd[i].pw_uid),
170                         "failed to call getpwuid for enumerated user");
171         }
172
173         return true;
174 }
175
176 static bool test_nwrap_enum_group(struct torture_context *tctx,
177                                   struct group **grp_array_p,
178                                   size_t *num_grp_p)
179 {
180         struct group *grp;
181         struct group *grp_array = NULL;
182         size_t num_grp = 0;
183
184         torture_comment(tctx, "Testing setgrent\n");
185         setgrent();
186
187         while ((grp = getgrent()) != NULL) {
188                 torture_comment(tctx, "Testing getgrent\n");
189
190                 print_group(grp);
191                 if (grp_array_p && num_grp_p) {
192                         grp_array = talloc_realloc(tctx, grp_array, struct group, num_grp+1);
193                         torture_assert(tctx, grp_array, "out of memory");
194                         grp_array[num_grp].gr_name = talloc_strdup(tctx, grp->gr_name);
195                         grp_array[num_grp].gr_gid = grp->gr_gid;
196                         num_grp++;
197                 }
198         }
199
200         torture_comment(tctx, "Testing endgrent\n");
201         endgrent();
202
203         if (grp_array_p) {
204                 *grp_array_p = grp_array;
205         }
206         if (num_grp_p) {
207                 *num_grp_p = num_grp;
208         }
209
210
211         return true;
212 }
213
214 static bool test_nwrap_group(struct torture_context *tctx)
215 {
216         int i;
217         struct group *grp;
218         size_t num_grp;
219
220         torture_assert(tctx, test_nwrap_enum_group(tctx, &grp, &num_grp),
221                                                    "failed to enumerate group");
222
223         for (i=0; i < num_grp; i++) {
224                 torture_assert(tctx, test_nwrap_getgrnam(tctx, grp[i].gr_name),
225                         "failed to call getgrnam for enumerated user");
226                 torture_assert(tctx, test_nwrap_getgrgid(tctx, grp[i].gr_gid),
227                         "failed to call getgrgid for enumerated user");
228         }
229
230         return true;
231 }
232
233 static bool test_nwrap_env(struct torture_context *tctx)
234 {
235         const char *old_pwd = getenv("NSS_WRAPPER_PASSWD");
236         const char *old_group = getenv("NSS_WRAPPER_GROUP");
237
238         if (!old_pwd || !old_group) {
239                 torture_skip(tctx, "nothing to test\n");
240                 return true;
241         }
242
243         torture_assert(tctx, test_nwrap_passwd(tctx),
244                         "failed to test users");
245         torture_assert(tctx, test_nwrap_group(tctx),
246                         "failed to test groups");
247
248         return true;
249 }
250
251 struct torture_suite *torture_local_nss_wrapper(TALLOC_CTX *mem_ctx)
252 {
253         struct torture_suite *suite = torture_suite_create(mem_ctx, "NSS-WRAPPER");
254
255         torture_suite_add_simple_test(suite, "env", test_nwrap_env);
256
257         return suite;
258 }