1 /* $NetBSD: getid.c,v 1.7 2008/04/28 20:24:17 martin Exp $ */ 2 /* from: NetBSD: getpwent.c,v 1.48 2000/10/03 03:22:26 enami Exp */ 3 /* from: NetBSD: getgrent.c,v 1.41 2002/01/12 23:51:30 lukem Exp */ 4 5 /* 6 * Copyright (c) 1987, 1988, 1989, 1993, 1994, 1995 7 * The Regents of the University of California. All rights reserved. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 3. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 /*- 35 * Copyright (c) 2002 The NetBSD Foundation, Inc. 36 * All rights reserved. 37 * 38 * This code is derived from software contributed to The NetBSD Foundation 39 * by Luke Mewburn of Wasabi Systems. 40 * 41 * Redistribution and use in source and binary forms, with or without 42 * modification, are permitted provided that the following conditions 43 * are met: 44 * 1. Redistributions of source code must retain the above copyright 45 * notice, this list of conditions and the following disclaimer. 46 * 2. Redistributions in binary form must reproduce the above copyright 47 * notice, this list of conditions and the following disclaimer in the 48 * documentation and/or other materials provided with the distribution. 49 * 50 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 51 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 52 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 53 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 54 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 55 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 56 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 57 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 58 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 59 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 60 * POSSIBILITY OF SUCH DAMAGE. 61 */ 62 63 #if HAVE_NBTOOL_CONFIG_H 64 #include "nbtool_config.h" 65 #endif 66 67 #include <sys/cdefs.h> 68 __RCSID("$NetBSD: getid.c,v 1.7 2008/04/28 20:24:17 martin Exp $"); 69 70 #include <sys/param.h> 71 72 #include <grp.h> 73 #include <limits.h> 74 #include <pwd.h> 75 #include <stdlib.h> 76 #include <stdio.h> 77 #include <string.h> 78 #include <time.h> 79 #include <unistd.h> 80 81 #include "extern.h" 82 83 static struct group * gi_getgrnam(const char *); 84 static struct group * gi_getgrgid(gid_t); 85 static int gi_setgroupent(int); 86 static void gi_endgrent(void); 87 static int grstart(void); 88 static int grscan(int, gid_t, const char *); 89 static int grmatchline(int, gid_t, const char *); 90 91 static struct passwd * gi_getpwnam(const char *); 92 static struct passwd * gi_getpwuid(uid_t); 93 static int gi_setpassent(int); 94 static void gi_endpwent(void); 95 static int pwstart(void); 96 static int pwscan(int, uid_t, const char *); 97 static int pwmatchline(int, uid_t, const char *); 98 99 #define MAXGRP 200 100 #define MAXLINELENGTH 1024 101 102 static FILE *_gr_fp; 103 static struct group _gr_group; 104 static int _gr_stayopen; 105 static int _gr_filesdone; 106 static FILE *_pw_fp; 107 static struct passwd _pw_passwd; /* password structure */ 108 static int _pw_stayopen; /* keep fd's open */ 109 static int _pw_filesdone; 110 111 static char grfile[MAXPATHLEN]; 112 static char pwfile[MAXPATHLEN]; 113 114 static char *members[MAXGRP]; 115 static char grline[MAXLINELENGTH]; 116 static char pwline[MAXLINELENGTH]; 117 118 int 119 setup_getid(const char *dir) 120 { 121 if (dir == NULL) 122 return (0); 123 124 /* close existing databases */ 125 gi_endgrent(); 126 gi_endpwent(); 127 128 /* build paths to new databases */ 129 snprintf(grfile, sizeof(grfile), "%s/group", dir); 130 snprintf(pwfile, sizeof(pwfile), "%s/master.passwd", dir); 131 132 /* try to open new databases */ 133 if (!grstart() || !pwstart()) 134 return (0); 135 136 /* switch pwcache(3) lookup functions */ 137 if (pwcache_groupdb(gi_setgroupent, gi_endgrent, 138 gi_getgrnam, gi_getgrgid) == -1 139 || pwcache_userdb(gi_setpassent, gi_endpwent, 140 gi_getpwnam, gi_getpwuid) == -1) 141 return (0); 142 143 return (1); 144 } 145 146 147 /* 148 * group lookup functions 149 */ 150 151 static struct group * 152 gi_getgrnam(const char *name) 153 { 154 int rval; 155 156 if (!grstart()) 157 return NULL; 158 rval = grscan(1, 0, name); 159 if (!_gr_stayopen) 160 endgrent(); 161 return (rval) ? &_gr_group : NULL; 162 } 163 164 static struct group * 165 gi_getgrgid(gid_t gid) 166 { 167 int rval; 168 169 if (!grstart()) 170 return NULL; 171 rval = grscan(1, gid, NULL); 172 if (!_gr_stayopen) 173 endgrent(); 174 return (rval) ? &_gr_group : NULL; 175 } 176 177 static int 178 gi_setgroupent(int stayopen) 179 { 180 181 if (!grstart()) 182 return 0; 183 _gr_stayopen = stayopen; 184 return 1; 185 } 186 187 static void 188 gi_endgrent(void) 189 { 190 191 _gr_filesdone = 0; 192 if (_gr_fp) { 193 (void)fclose(_gr_fp); 194 _gr_fp = NULL; 195 } 196 } 197 198 static int 199 grstart(void) 200 { 201 202 _gr_filesdone = 0; 203 if (_gr_fp) { 204 rewind(_gr_fp); 205 return 1; 206 } 207 if (grfile[0] == '\0') /* sanity check */ 208 return 0; 209 return (_gr_fp = fopen(grfile, "r")) ? 1 : 0; 210 } 211 212 213 static int 214 grscan(int search, gid_t gid, const char *name) 215 { 216 217 if (_gr_filesdone) 218 return 0; 219 for (;;) { 220 if (!fgets(grline, sizeof(grline), _gr_fp)) { 221 if (!search) 222 _gr_filesdone = 1; 223 return 0; 224 } 225 /* skip lines that are too big */ 226 if (!strchr(grline, '\n')) { 227 int ch; 228 229 while ((ch = getc(_gr_fp)) != '\n' && ch != EOF) 230 ; 231 continue; 232 } 233 if (grmatchline(search, gid, name)) 234 return 1; 235 } 236 /* NOTREACHED */ 237 } 238 239 static int 240 grmatchline(int search, gid_t gid, const char *name) 241 { 242 unsigned long id; 243 char **m; 244 char *cp, *bp, *ep; 245 246 /* name may be NULL if search is nonzero */ 247 248 bp = grline; 249 memset(&_gr_group, 0, sizeof(_gr_group)); 250 _gr_group.gr_name = strsep(&bp, ":\n"); 251 if (search && name && strcmp(_gr_group.gr_name, name)) 252 return 0; 253 _gr_group.gr_passwd = strsep(&bp, ":\n"); 254 if (!(cp = strsep(&bp, ":\n"))) 255 return 0; 256 id = strtoul(cp, &ep, 10); 257 if (id > GID_MAX || *ep != '\0') 258 return 0; 259 _gr_group.gr_gid = (gid_t)id; 260 if (search && name == NULL && _gr_group.gr_gid != gid) 261 return 0; 262 cp = NULL; 263 if (bp == NULL) 264 return 0; 265 for (_gr_group.gr_mem = m = members;; bp++) { 266 if (m == &members[MAXGRP - 1]) 267 break; 268 if (*bp == ',') { 269 if (cp) { 270 *bp = '\0'; 271 *m++ = cp; 272 cp = NULL; 273 } 274 } else if (*bp == '\0' || *bp == '\n' || *bp == ' ') { 275 if (cp) { 276 *bp = '\0'; 277 *m++ = cp; 278 } 279 break; 280 } else if (cp == NULL) 281 cp = bp; 282 } 283 *m = NULL; 284 return 1; 285 } 286 287 288 /* 289 * user lookup functions 290 */ 291 292 static struct passwd * 293 gi_getpwnam(const char *name) 294 { 295 int rval; 296 297 if (!pwstart()) 298 return NULL; 299 rval = pwscan(1, 0, name); 300 if (!_pw_stayopen) 301 endpwent(); 302 return (rval) ? &_pw_passwd : NULL; 303 } 304 305 static struct passwd * 306 gi_getpwuid(uid_t uid) 307 { 308 int rval; 309 310 if (!pwstart()) 311 return NULL; 312 rval = pwscan(1, uid, NULL); 313 if (!_pw_stayopen) 314 endpwent(); 315 return (rval) ? &_pw_passwd : NULL; 316 } 317 318 static int 319 gi_setpassent(int stayopen) 320 { 321 322 if (!pwstart()) 323 return 0; 324 _pw_stayopen = stayopen; 325 return 1; 326 } 327 328 static void 329 gi_endpwent(void) 330 { 331 332 _pw_filesdone = 0; 333 if (_pw_fp) { 334 (void)fclose(_pw_fp); 335 _pw_fp = NULL; 336 } 337 } 338 339 static int 340 pwstart(void) 341 { 342 343 _pw_filesdone = 0; 344 if (_pw_fp) { 345 rewind(_pw_fp); 346 return 1; 347 } 348 if (pwfile[0] == '\0') /* sanity check */ 349 return 0; 350 return (_pw_fp = fopen(pwfile, "r")) ? 1 : 0; 351 } 352 353 354 static int 355 pwscan(int search, uid_t uid, const char *name) 356 { 357 358 if (_pw_filesdone) 359 return 0; 360 for (;;) { 361 if (!fgets(pwline, sizeof(pwline), _pw_fp)) { 362 if (!search) 363 _pw_filesdone = 1; 364 return 0; 365 } 366 /* skip lines that are too big */ 367 if (!strchr(pwline, '\n')) { 368 int ch; 369 370 while ((ch = getc(_pw_fp)) != '\n' && ch != EOF) 371 ; 372 continue; 373 } 374 if (pwmatchline(search, uid, name)) 375 return 1; 376 } 377 /* NOTREACHED */ 378 } 379 380 static int 381 pwmatchline(int search, uid_t uid, const char *name) 382 { 383 unsigned long id; 384 char *cp, *bp, *ep; 385 386 /* name may be NULL if search is nonzero */ 387 388 bp = pwline; 389 memset(&_pw_passwd, 0, sizeof(_pw_passwd)); 390 _pw_passwd.pw_name = strsep(&bp, ":\n"); /* name */ 391 if (search && name && strcmp(_pw_passwd.pw_name, name)) 392 return 0; 393 394 _pw_passwd.pw_passwd = strsep(&bp, ":\n"); /* passwd */ 395 396 if (!(cp = strsep(&bp, ":\n"))) /* uid */ 397 return 0; 398 id = strtoul(cp, &ep, 10); 399 if (id > UID_MAX || *ep != '\0') 400 return 0; 401 _pw_passwd.pw_uid = (uid_t)id; 402 if (search && name == NULL && _pw_passwd.pw_uid != uid) 403 return 0; 404 405 if (!(cp = strsep(&bp, ":\n"))) /* gid */ 406 return 0; 407 id = strtoul(cp, &ep, 10); 408 if (id > GID_MAX || *ep != '\0') 409 return 0; 410 _pw_passwd.pw_gid = (gid_t)id; 411 412 if (!(ep = strsep(&bp, ":"))) /* class */ 413 return 0; 414 if (!(ep = strsep(&bp, ":"))) /* change */ 415 return 0; 416 if (!(ep = strsep(&bp, ":"))) /* expire */ 417 return 0; 418 419 if (!(_pw_passwd.pw_gecos = strsep(&bp, ":\n"))) /* gecos */ 420 return 0; 421 if (!(_pw_passwd.pw_dir = strsep(&bp, ":\n"))) /* directory */ 422 return 0; 423 if (!(_pw_passwd.pw_shell = strsep(&bp, ":\n"))) /* shell */ 424 return 0; 425 426 if (strchr(bp, ':') != NULL) 427 return 0; 428 429 return 1; 430 } 431 432