usr.sbin/pw/pw_vpw.c

   1 /*-
   2  * Copyright (C) 1996
   3  *      David L. Nugent.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY DAVID L. NUGENT AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL DAVID L. NUGENT OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  */
  27
  28 #ifndef lint
  29 static const char rcsid[] =
  30   "$FreeBSD$";
  31 #endif /* not lint */
  32
  33 #include <pwd.h>
  34 #include <grp.h>
  35 #include <libutil.h>
  36 #define _WITH_GETLINE
  37 #include <stdio.h>
  38 #include <string.h>
  39 #include <stdlib.h>
  40 #include <sys/param.h>
  41 #include <err.h>
  42
  43 #include "pwupd.h"
  44
  45 static FILE * pwd_fp = NULL;
  46
  47 void
  48 vendpwent(void)
  49 {
  50         if (pwd_fp != NULL) {
  51                 fclose(pwd_fp);
  52                 pwd_fp = NULL;
  53         }
  54 }
  55
  56 void
  57 vsetpwent(void)
  58 {
  59         vendpwent();
  60 }
  61
  62 static struct passwd *
  63 vnextpwent(char const *nam, uid_t uid, int doclose)
  64 {
  65         struct passwd *pw;
  66         char *line;
  67         size_t linecap;
  68         ssize_t linelen;
  69
  70         pw = NULL;
  71         line = NULL;
  72         linecap = 0;
  73
  74         if (pwd_fp != NULL || (pwd_fp = fopen(getpwpath(_MASTERPASSWD), "r")) != NULL) {
  75                 while ((linelen = getline(&line, &linecap, pwd_fp)) > 0) {
  76                         /* Skip comments and empty lines */
  77                         if (*line == '\n' || *line == '#')
  78                                 continue;
  79                         /* trim latest \n */
  80                         if (line[linelen - 1 ] == '\n')
  81                                 line[linelen - 1] = '\0';
  82                         pw = pw_scan(line, PWSCAN_MASTER);
  83                         if (pw == NULL)
  84                                 errx(EXIT_FAILURE, "Invalid user entry in '%s':"
  85                                     " '%s'", getpwpath(_MASTERPASSWD), line);
  86                         if (uid != (uid_t)-1) {
  87                                 if (uid == pw->pw_uid)
  88                                         break;
  89                         } else if (nam != NULL) {
  90                                 if (strcmp(nam, pw->pw_name) == 0)
  91                                         break;
  92                         } else
  93                                 break;
  94                         free(pw);
  95                         pw = NULL;
  96                 }
  97                 if (doclose)
  98                         vendpwent();
  99         }
 100         free(line);
 101
 102         return (pw);
 103 }
 104
 105 struct passwd *
 106 vgetpwent(void)
 107 {
 108   return vnextpwent(NULL, -1, 0);
 109 }
 110
 111 struct passwd *
 112 vgetpwuid(uid_t uid)
 113 {
 114   return vnextpwent(NULL, uid, 1);
 115 }
 116
 117 struct passwd *
 118 vgetpwnam(const char * nam)
 119 {
 120   return vnextpwent(nam, -1, 1);
 121 }
 122
 123
 124 static FILE * grp_fp = NULL;
 125
 126 void
 127 vendgrent(void)
 128 {
 129         if (grp_fp != NULL) {
 130                 fclose(grp_fp);
 131                 grp_fp = NULL;
 132         }
 133 }
 134
 135 RET_SETGRENT
 136 vsetgrent(void)
 137 {
 138         vendgrent();
 139 #if defined(__FreeBSD__)
 140         return 0;
 141 #endif
 142 }
 143
 144 static struct group *
 145 vnextgrent(char const *nam, gid_t gid, int doclose)
 146 {
 147         struct group *gr;
 148         char *line;
 149         size_t linecap;
 150         ssize_t linelen;
 151
 152         gr = NULL;
 153         line = NULL;
 154         linecap = 0;
 155
 156         if (grp_fp != NULL || (grp_fp = fopen(getgrpath(_GROUP), "r")) != NULL) {
 157                 while ((linelen = getline(&line, &linecap, grp_fp)) > 0) {
 158                         /* Skip comments and empty lines */
 159                         if (*line == '\n' || *line == '#')
 160                                 continue;
 161                         /* trim latest \n */
 162                         if (line[linelen - 1 ] == '\n')
 163                                 line[linelen - 1] = '\0';
 164                         gr = gr_scan(line);
 165                         if (gr == NULL)
 166                                 errx(EXIT_FAILURE, "Invalid group entry in '%s':"
 167                                     " '%s'", getgrpath(_GROUP), line);
 168                         if (gid != (gid_t)-1) {
 169                                 if (gid == gr->gr_gid)
 170                                         break;
 171                         } else if (nam != NULL) {
 172                                 if (strcmp(nam, gr->gr_name) == 0)
 173                                         break;
 174                         } else
 175                                 break;
 176                         free(gr);
 177                         gr = NULL;
 178                 }
 179                 if (doclose)
 180                         vendgrent();
 181         }
 182         free(line);
 183
 184         return (gr);
 185 }
 186
 187 struct group *
 188 vgetgrent(void)
 189 {
 190   return vnextgrent(NULL, -1, 0);
 191 }
 192
 193
 194 struct group *
 195 vgetgrgid(gid_t gid)
 196 {
 197   return vnextgrent(NULL, gid, 1);
 198 }
 199
 200 struct group *
 201 vgetgrnam(const char * nam)
 202 {
 203   return vnextgrent(nam, -1, 1);
 204 }
 205