1 /* $NetBSD: getid.c,v 1.8 2013/10/16 17:27:42 christos 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.8 2013/10/16 17:27:42 christos 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 /* skip comments */ 234 if (pwline[0] == '#') 235 continue; 236 if (grmatchline(search, gid, name)) 237 return 1; 238 } 239 /* NOTREACHED */ 240 } 241 242 static int 243 grmatchline(int search, gid_t gid, const char *name) 244 { 245 unsigned long id; 246 char **m; 247 char *cp, *bp, *ep; 248 249 /* name may be NULL if search is nonzero */ 250 251 bp = grline; 252 memset(&_gr_group, 0, sizeof(_gr_group)); 253 _gr_group.gr_name = strsep(&bp, ":\n"); 254 if (search && name && strcmp(_gr_group.gr_name, name)) 255 return 0; 256 _gr_group.gr_passwd = strsep(&bp, ":\n"); 257 if (!(cp = strsep(&bp, ":\n"))) 258 return 0; 259 id = strtoul(cp, &ep, 10); 260 if (id > GID_MAX || *ep != '\0') 261 return 0; 262 _gr_group.gr_gid = (gid_t)id; 263 if (search && name == NULL && _gr_group.gr_gid != gid) 264 return 0; 265 cp = NULL; 266 if (bp == NULL) 267 return 0; 268 for (_gr_group.gr_mem = m = members;; bp++) { 269 if (m == &members[MAXGRP - 1]) 270 break; 271 if (*bp == ',') { 272 if (cp) { 273 *bp = '\0'; 274 *m++ = cp; 275 cp = NULL; 276 } 277 } else if (*bp == '\0' || *bp == '\n' || *bp == ' ') { 278 if (cp) { 279 *bp = '\0'; 280 *m++ = cp; 281 } 282 break; 283 } else if (cp == NULL) 284 cp = bp; 285 } 286 *m = NULL; 287 return 1; 288 } 289 290 291 /* 292 * user lookup functions 293 */ 294 295 static struct passwd * 296 gi_getpwnam(const char *name) 297 { 298 int rval; 299 300 if (!pwstart()) 301 return NULL; 302 rval = pwscan(1, 0, name); 303 if (!_pw_stayopen) 304 endpwent(); 305 return (rval) ? &_pw_passwd : NULL; 306 } 307 308 static struct passwd * 309 gi_getpwuid(uid_t uid) 310 { 311 int rval; 312 313 if (!pwstart()) 314 return NULL; 315 rval = pwscan(1, uid, NULL); 316 if (!_pw_stayopen) 317 endpwent(); 318 return (rval) ? &_pw_passwd : NULL; 319 } 320 321 static int 322 gi_setpassent(int stayopen) 323 { 324 325 if (!pwstart()) 326 return 0; 327 _pw_stayopen = stayopen; 328 return 1; 329 } 330 331 static void 332 gi_endpwent(void) 333 { 334 335 _pw_filesdone = 0; 336 if (_pw_fp) { 337 (void)fclose(_pw_fp); 338 _pw_fp = NULL; 339 } 340 } 341 342 static int 343 pwstart(void) 344 { 345 346 _pw_filesdone = 0; 347 if (_pw_fp) { 348 rewind(_pw_fp); 349 return 1; 350 } 351 if (pwfile[0] == '\0') /* sanity check */ 352 return 0; 353 return (_pw_fp = fopen(pwfile, "r")) ? 1 : 0; 354 } 355 356 357 static int 358 pwscan(int search, uid_t uid, const char *name) 359 { 360 361 if (_pw_filesdone) 362 return 0; 363 for (;;) { 364 if (!fgets(pwline, sizeof(pwline), _pw_fp)) { 365 if (!search) 366 _pw_filesdone = 1; 367 return 0; 368 } 369 /* skip lines that are too big */ 370 if (!strchr(pwline, '\n')) { 371 int ch; 372 373 while ((ch = getc(_pw_fp)) != '\n' && ch != EOF) 374 ; 375 continue; 376 } 377 /* skip comments */ 378 if (pwline[0] == '#') 379 continue; 380 if (pwmatchline(search, uid, name)) 381 return 1; 382 } 383 /* NOTREACHED */ 384 } 385 386 static int 387 pwmatchline(int search, uid_t uid, const char *name) 388 { 389 unsigned long id; 390 char *cp, *bp, *ep; 391 392 /* name may be NULL if search is nonzero */ 393 394 bp = pwline; 395 memset(&_pw_passwd, 0, sizeof(_pw_passwd)); 396 _pw_passwd.pw_name = strsep(&bp, ":\n"); /* name */ 397 if (search && name && strcmp(_pw_passwd.pw_name, name)) 398 return 0; 399 400 _pw_passwd.pw_passwd = strsep(&bp, ":\n"); /* passwd */ 401 402 if (!(cp = strsep(&bp, ":\n"))) /* uid */ 403 return 0; 404 id = strtoul(cp, &ep, 10); 405 if (id > UID_MAX || *ep != '\0') 406 return 0; 407 _pw_passwd.pw_uid = (uid_t)id; 408 if (search && name == NULL && _pw_passwd.pw_uid != uid) 409 return 0; 410 411 if (!(cp = strsep(&bp, ":\n"))) /* gid */ 412 return 0; 413 id = strtoul(cp, &ep, 10); 414 if (id > GID_MAX || *ep != '\0') 415 return 0; 416 _pw_passwd.pw_gid = (gid_t)id; 417 418 if (!(ep = strsep(&bp, ":"))) /* class */ 419 return 0; 420 if (!(ep = strsep(&bp, ":"))) /* change */ 421 return 0; 422 if (!(ep = strsep(&bp, ":"))) /* expire */ 423 return 0; 424 425 if (!(_pw_passwd.pw_gecos = strsep(&bp, ":\n"))) /* gecos */ 426 return 0; 427 if (!(_pw_passwd.pw_dir = strsep(&bp, ":\n"))) /* directory */ 428 return 0; 429 if (!(_pw_passwd.pw_shell = strsep(&bp, ":\n"))) /* shell */ 430 return 0; 431 432 if (strchr(bp, ':') != NULL) 433 return 0; 434 435 return 1; 436 } 437 438