source: vendor/3.5.2/source3/torture/nsstest.c

Last change on this file was 414, checked in by Herwig Bauernfeind, 16 years ago

Samba 3.5.0: Initial import
File size: 10.7 KB
Line 
1/*
2 Unix SMB/CIFS implementation.
3 nss tester for winbindd
4 Copyright (C) Andrew Tridgell 2001
5 Copyright (C) Tim Potter 2003
6
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
11
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
19*/
20
21#include "includes.h"
22
23#ifdef malloc
24#undef malloc
25#endif
26#ifdef realloc
27#undef realloc
28#endif
29
30static const char *so_path = "/lib/libnss_winbind.so";
31static const char *nss_name = "winbind";
32static int nss_errno;
33static NSS_STATUS last_error;
34static int total_errors;
35
36static void *find_fn(const char *name)
37{
38 char *s;
39 static void *h;
40 void *res;
41
42 if (asprintf(&s, "_nss_%s_%s", nss_name, name) < 0) {
43 exit(1);
44 }
45
46 if (!h) {
47 h = dlopen(so_path, RTLD_LAZY);
48 }
49 if (!h) {
50 printf("Can't open shared library %s\n", so_path);
51 exit(1);
52 }
53 res = dlsym(h, s);
54 if (!res) {
55 printf("Can't find function %s\n", s);
56 total_errors++;
57 SAFE_FREE(s);
58 return NULL;
59 }
60 SAFE_FREE(s);
61 return res;
62}
63
64static void report_nss_error(const char *who, NSS_STATUS status)
65{
66 last_error = status;
67 total_errors++;
68 printf("ERROR %s: NSS_STATUS=%d %d (nss_errno=%d)\n",
69 who, status, NSS_STATUS_SUCCESS, nss_errno);
70}
71
72static struct passwd *nss_getpwent(void)
73{
74 NSS_STATUS (*_nss_getpwent_r)(struct passwd *, char *,
75 size_t , int *) =
76 (NSS_STATUS (*)(struct passwd *, char *,
77 size_t, int *))find_fn("getpwent_r");
78 static struct passwd pwd;
79 static char buf[1000];
80 NSS_STATUS status;
81
82 if (!_nss_getpwent_r)
83 return NULL;
84
85 status = _nss_getpwent_r(&pwd, buf, sizeof(buf), &nss_errno);
86 if (status == NSS_STATUS_NOTFOUND) {
87 return NULL;
88 }
89 if (status != NSS_STATUS_SUCCESS) {
90 report_nss_error("getpwent", status);
91 return NULL;
92 }
93 return &pwd;
94}
95
96static struct passwd *nss_getpwnam(const char *name)
97{
98 NSS_STATUS (*_nss_getpwnam_r)(const char *, struct passwd *, char *,
99 size_t , int *) =
100 (NSS_STATUS (*)(const char *, struct passwd *, char *,
101 size_t, int *))find_fn("getpwnam_r");
102 static struct passwd pwd;
103 static char buf[1000];
104 NSS_STATUS status;
105
106 if (!_nss_getpwnam_r)
107 return NULL;
108
109 status = _nss_getpwnam_r(name, &pwd, buf, sizeof(buf), &nss_errno);
110 if (status == NSS_STATUS_NOTFOUND) {
111 return NULL;
112 }
113 if (status != NSS_STATUS_SUCCESS) {
114 report_nss_error("getpwnam", status);
115 return NULL;
116 }
117 return &pwd;
118}
119
120static struct passwd *nss_getpwuid(uid_t uid)
121{
122 NSS_STATUS (*_nss_getpwuid_r)(uid_t , struct passwd *, char *,
123 size_t , int *) =
124 (NSS_STATUS (*)(uid_t, struct passwd *, char *,
125 size_t, int *))find_fn("getpwuid_r");
126 static struct passwd pwd;
127 static char buf[1000];
128 NSS_STATUS status;
129
130 if (!_nss_getpwuid_r)
131 return NULL;
132
133 status = _nss_getpwuid_r(uid, &pwd, buf, sizeof(buf), &nss_errno);
134 if (status == NSS_STATUS_NOTFOUND) {
135 return NULL;
136 }
137 if (status != NSS_STATUS_SUCCESS) {
138 report_nss_error("getpwuid", status);
139 return NULL;
140 }
141 return &pwd;
142}
143
144static void nss_setpwent(void)
145{
146 NSS_STATUS (*_nss_setpwent)(void) =
147 (NSS_STATUS(*)(void))find_fn("setpwent");
148 NSS_STATUS status;
149
150 if (!_nss_setpwent)
151 return;
152
153 status = _nss_setpwent();
154 if (status != NSS_STATUS_SUCCESS) {
155 report_nss_error("setpwent", status);
156 }
157}
158
159static void nss_endpwent(void)
160{
161 NSS_STATUS (*_nss_endpwent)(void) =
162 (NSS_STATUS (*)(void))find_fn("endpwent");
163 NSS_STATUS status;
164
165 if (!_nss_endpwent)
166 return;
167
168 status = _nss_endpwent();
169 if (status != NSS_STATUS_SUCCESS) {
170 report_nss_error("endpwent", status);
171 }
172}
173
174
175static struct group *nss_getgrent(void)
176{
177 NSS_STATUS (*_nss_getgrent_r)(struct group *, char *,
178 size_t , int *) =
179 (NSS_STATUS (*)(struct group *, char *,
180 size_t, int *))find_fn("getgrent_r");
181 static struct group grp;
182 static char *buf;
183 static int buflen = 1024;
184 NSS_STATUS status;
185
186 if (!_nss_getgrent_r)
187 return NULL;
188
189 if (!buf)
190 buf = (char *)malloc(buflen);
191
192again:
193 status = _nss_getgrent_r(&grp, buf, buflen, &nss_errno);
194 if (status == NSS_STATUS_TRYAGAIN) {
195 buflen *= 2;
196 buf = (char *)realloc(buf, buflen);
197 if (!buf) {
198 return NULL;
199 }
200 goto again;
201 }
202 if (status == NSS_STATUS_NOTFOUND) {
203 SAFE_FREE(buf);
204 return NULL;
205 }
206 if (status != NSS_STATUS_SUCCESS) {
207 report_nss_error("getgrent", status);
208 SAFE_FREE(buf);
209 return NULL;
210 }
211 return &grp;
212}
213
214static struct group *nss_getgrnam(const char *name)
215{
216 NSS_STATUS (*_nss_getgrnam_r)(const char *, struct group *, char *,
217 size_t , int *) =
218 (NSS_STATUS (*)(const char *, struct group *, char *,
219 size_t, int *))find_fn("getgrnam_r");
220 static struct group grp;
221 static char *buf;
222 static int buflen = 1000;
223 NSS_STATUS status;
224
225 if (!_nss_getgrnam_r)
226 return NULL;
227
228 if (!buf)
229 buf = (char *)malloc(buflen);
230again:
231 status = _nss_getgrnam_r(name, &grp, buf, buflen, &nss_errno);
232 if (status == NSS_STATUS_TRYAGAIN) {
233 buflen *= 2;
234 buf = (char *)realloc(buf, buflen);
235 if (!buf) {
236 return NULL;
237 }
238 goto again;
239 }
240 if (status == NSS_STATUS_NOTFOUND) {
241 SAFE_FREE(buf);
242 return NULL;
243 }
244 if (status != NSS_STATUS_SUCCESS) {
245 report_nss_error("getgrnam", status);
246 SAFE_FREE(buf);
247 return NULL;
248 }
249 return &grp;
250}
251
252static struct group *nss_getgrgid(gid_t gid)
253{
254 NSS_STATUS (*_nss_getgrgid_r)(gid_t , struct group *, char *,
255 size_t , int *) =
256 (NSS_STATUS (*)(gid_t, struct group *, char *,
257 size_t, int *))find_fn("getgrgid_r");
258 static struct group grp;
259 static char *buf;
260 static int buflen = 1000;
261 NSS_STATUS status;
262
263 if (!_nss_getgrgid_r)
264 return NULL;
265
266 if (!buf)
267 buf = (char *)malloc(buflen);
268
269again:
270 status = _nss_getgrgid_r(gid, &grp, buf, buflen, &nss_errno);
271 if (status == NSS_STATUS_TRYAGAIN) {
272 buflen *= 2;
273 buf = (char *)realloc(buf, buflen);
274 if (!buf) {
275 return NULL;
276 }
277 goto again;
278 }
279 if (status == NSS_STATUS_NOTFOUND) {
280 SAFE_FREE(buf);
281 return NULL;
282 }
283 if (status != NSS_STATUS_SUCCESS) {
284 report_nss_error("getgrgid", status);
285 SAFE_FREE(buf);
286 return NULL;
287 }
288 return &grp;
289}
290
291static void nss_setgrent(void)
292{
293 NSS_STATUS (*_nss_setgrent)(void) =
294 (NSS_STATUS (*)(void))find_fn("setgrent");
295 NSS_STATUS status;
296
297 if (!_nss_setgrent)
298 return;
299
300 status = _nss_setgrent();
301 if (status != NSS_STATUS_SUCCESS) {
302 report_nss_error("setgrent", status);
303 }
304}
305
306static void nss_endgrent(void)
307{
308 NSS_STATUS (*_nss_endgrent)(void) =
309 (NSS_STATUS (*)(void))find_fn("endgrent");
310 NSS_STATUS status;
311
312 if (!_nss_endgrent)
313 return;
314
315 status = _nss_endgrent();
316 if (status != NSS_STATUS_SUCCESS) {
317 report_nss_error("endgrent", status);
318 }
319}
320
321static int nss_initgroups(char *user, gid_t group, gid_t **groups, long int *start, long int *size)
322{
323 NSS_STATUS (*_nss_initgroups)(char *, gid_t , long int *,
324 long int *, gid_t **, long int , int *) =
325 (NSS_STATUS (*)(char *, gid_t, long int *,
326 long int *, gid_t **,
327 long int, int *))find_fn("initgroups_dyn");
328 NSS_STATUS status;
329
330 if (!_nss_initgroups)
331 return NSS_STATUS_UNAVAIL;
332
333 status = _nss_initgroups(user, group, start, size, groups, 0, &nss_errno);
334 if (status != NSS_STATUS_SUCCESS) {
335 report_nss_error("initgroups", status);
336 }
337 return status;
338}
339
340static void print_passwd(struct passwd *pwd)
341{
342 printf("%s:%s:%lu:%lu:%s:%s:%s\n",
343 pwd->pw_name,
344 pwd->pw_passwd,
345 (unsigned long)pwd->pw_uid,
346 (unsigned long)pwd->pw_gid,
347 pwd->pw_gecos,
348 pwd->pw_dir,
349 pwd->pw_shell);
350}
351
352static void print_group(struct group *grp)
353{
354 int i;
355 printf("%s:%s:%lu:",
356 grp->gr_name,
357 grp->gr_passwd,
358 (unsigned long)grp->gr_gid);
359
360 if (!grp->gr_mem[0]) {
361 printf("\n");
362 return;
363 }
364
365 for (i=0; grp->gr_mem[i+1]; i++) {
366 printf("%s,", grp->gr_mem[i]);
367 }
368 printf("%s\n", grp->gr_mem[i]);
369}
370
371static void nss_test_initgroups(char *name, gid_t gid)
372{
373 long int size = 16;
374 long int start = 1;
375 gid_t *groups = NULL;
376 int i;
377 NSS_STATUS status;
378
379 groups = (gid_t *)malloc(size);
380 groups[0] = gid;
381
382 status = nss_initgroups(name, gid, &groups, &start, &size);
383 if (status == NSS_STATUS_UNAVAIL) {
384 printf("No initgroups fn\n");
385 return;
386 }
387
388 for (i=0; i<start-1; i++) {
389 printf("%lu, ", (unsigned long)groups[i]);
390 }
391 printf("%lu\n", (unsigned long)groups[i]);
392}
393
394
395static void nss_test_users(void)
396{
397 struct passwd *pwd;
398
399 nss_setpwent();
400 /* loop over all users */
401 while ((pwd = nss_getpwent())) {
402 printf("Testing user %s\n", pwd->pw_name);
403 printf("getpwent: "); print_passwd(pwd);
404 pwd = nss_getpwuid(pwd->pw_uid);
405 if (!pwd) {
406 total_errors++;
407 printf("ERROR: can't getpwuid\n");
408 continue;
409 }
410 printf("getpwuid: "); print_passwd(pwd);
411 pwd = nss_getpwnam(pwd->pw_name);
412 if (!pwd) {
413 total_errors++;
414 printf("ERROR: can't getpwnam\n");
415 continue;
416 }
417 printf("getpwnam: "); print_passwd(pwd);
418 printf("initgroups: "); nss_test_initgroups(pwd->pw_name, pwd->pw_gid);
419 printf("\n");
420 }
421 nss_endpwent();
422}
423
424static void nss_test_groups(void)
425{
426 struct group *grp;
427
428 nss_setgrent();
429 /* loop over all groups */
430 while ((grp = nss_getgrent())) {
431 printf("Testing group %s\n", grp->gr_name);
432 printf("getgrent: "); print_group(grp);
433 grp = nss_getgrnam(grp->gr_name);
434 if (!grp) {
435 total_errors++;
436 printf("ERROR: can't getgrnam\n");
437 continue;
438 }
439 printf("getgrnam: "); print_group(grp);
440 grp = nss_getgrgid(grp->gr_gid);
441 if (!grp) {
442 total_errors++;
443 printf("ERROR: can't getgrgid\n");
444 continue;
445 }
446 printf("getgrgid: "); print_group(grp);
447 printf("\n");
448 }
449 nss_endgrent();
450}
451
452static void nss_test_errors(void)
453{
454 struct passwd *pwd;
455 struct group *grp;
456
457 pwd = getpwnam("nosuchname");
458 if (pwd || last_error != NSS_STATUS_NOTFOUND) {
459 total_errors++;
460 printf("ERROR Non existant user gave error %d\n", last_error);
461 }
462
463 pwd = getpwuid(0xFFF0);
464 if (pwd || last_error != NSS_STATUS_NOTFOUND) {
465 total_errors++;
466 printf("ERROR Non existant uid gave error %d\n", last_error);
467 }
468
469 grp = getgrnam("nosuchgroup");
470 if (grp || last_error != NSS_STATUS_NOTFOUND) {
471 total_errors++;
472 printf("ERROR Non existant group gave error %d\n", last_error);
473 }
474
475 grp = getgrgid(0xFFF0);
476 if (grp || last_error != NSS_STATUS_NOTFOUND) {
477 total_errors++;
478 printf("ERROR Non existant gid gave error %d\n", last_error);
479 }
480}
481
482 int main(int argc, char *argv[])
483{
484 if (argc > 1) so_path = argv[1];
485 if (argc > 2) nss_name = argv[2];
486
487 nss_test_users();
488 nss_test_groups();
489 nss_test_errors();
490
491 printf("total_errors=%d\n", total_errors);
492
493 return total_errors;
494}
Note: See TracBrowser for help on using the repository browser.