Add PCICAP_{ID,NEXTPTR} to avoid using magic number
[dragonfly.git] / usr.sbin / pw / pwupd.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  * $FreeBSD: src/usr.sbin/pw/pwupd.c,v 1.12.2.4 2001/12/21 15:21:32 nectar Exp $
27  * $DragonFly: src/usr.sbin/pw/pwupd.c,v 1.2 2003/06/17 04:30:02 dillon Exp $
28  */
29
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <unistd.h>
34 #include <stdarg.h>
35 #include <errno.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #include <sys/param.h>
39 #include <sys/wait.h>
40
41 #include "pwupd.h"
42
43 #define HAVE_PWDB_C     1
44 #define HAVE_PWDB_U     1
45
46 static char pathpwd[] = _PATH_PWD;
47 static char * pwpath = pathpwd;
48  
49 int
50 setpwdir(const char * dir)
51 {
52         if (dir == NULL)
53                 return -1;
54         else {
55                 char * d = malloc(strlen(dir)+1);
56                 if (d == NULL)
57                         return -1;
58                 pwpath = strcpy(d, dir);
59         }
60         return 0;
61 }
62
63 char *
64 getpwpath(char const * file)
65 {
66         static char pathbuf[MAXPATHLEN];
67
68         snprintf(pathbuf, sizeof pathbuf, "%s/%s", pwpath, file);
69         return pathbuf;
70 }
71
72 int
73 pwdb(char *arg,...)
74 {
75         int             i = 0;
76         pid_t           pid;
77         va_list         ap;
78         char           *args[10];
79
80         args[i++] = _PATH_PWD_MKDB;
81         va_start(ap, arg);
82         while (i < 6 && arg != NULL) {
83                 args[i++] = arg;
84                 arg = va_arg(ap, char *);
85         }
86         if (pwpath != pathpwd) {
87                 args[i++] = "-d";
88                 args[i++] = pwpath;
89         }
90         args[i++] = getpwpath(_MASTERPASSWD);
91         args[i] = NULL;
92
93         if ((pid = fork()) == -1)       /* Error (errno set) */
94                 i = errno;
95         else if (pid == 0) {    /* Child */
96                 execv(args[0], args);
97                 _exit(1);
98         } else {                /* Parent */
99                 waitpid(pid, &i, 0);
100                 if (WEXITSTATUS(i))
101                         i = EIO;
102         }
103         return i;
104 }
105
106 int
107 fmtpwentry(char *buf, struct passwd * pwd, int type)
108 {
109         int             l;
110         char           *pw;
111
112         pw = (pwd->pw_passwd == NULL || !*pwd->pw_passwd) ? "" : (type == PWF_MASTER) ? pwd->pw_passwd : "*";
113
114         if (type == PWF_PASSWD)
115                 l = sprintf(buf, "%s:*:%ld:%ld:%s:%s:%s\n",
116                        pwd->pw_name, (long) pwd->pw_uid, (long) pwd->pw_gid,
117                             pwd->pw_gecos ? pwd->pw_gecos : "User &",
118                             pwd->pw_dir, pwd->pw_shell);
119         else
120                 l = sprintf(buf, "%s:%s:%ld:%ld:%s:%lu:%lu:%s:%s:%s\n",
121                    pwd->pw_name, pw, (long) pwd->pw_uid, (long) pwd->pw_gid,
122                             pwd->pw_class ? pwd->pw_class : "",
123                             (unsigned long) pwd->pw_change,
124                             (unsigned long) pwd->pw_expire,
125                             pwd->pw_gecos, pwd->pw_dir, pwd->pw_shell);
126         return l;
127 }
128
129
130 int
131 fmtpwent(char *buf, struct passwd * pwd)
132 {
133         return fmtpwentry(buf, pwd, PWF_STANDARD);
134 }
135
136 static int
137 pw_update(struct passwd * pwd, char const * user, int mode)
138 {
139         int             rc = 0;
140
141         ENDPWENT();
142
143         /*
144          * First, let's check the see if the database is alright
145          * Note: -C is only available in FreeBSD 2.2 and above
146          */
147 #ifdef HAVE_PWDB_C
148         if (pwdb("-C", NULL) == 0) {    /* Check only */
149 #else
150         {                               /* No -C */
151 #endif
152                 char            pfx[PWBUFSZ];
153                 char            pwbuf[PWBUFSZ];
154                 int             l = snprintf(pfx, PWBUFSZ, "%s:", user);
155 #ifdef HAVE_PWDB_U
156                 int             isrename = pwd!=NULL && strcmp(user, pwd->pw_name);
157 #endif
158
159                 /*
160                  * Update the passwd file first
161                  */
162                 if (pwd == NULL)
163                         *pwbuf = '\0';
164                 else
165                         fmtpwentry(pwbuf, pwd, PWF_PASSWD);
166
167                 rc = fileupdate(getpwpath(_PASSWD), 0644, pwbuf, pfx, l, mode);
168                 if (rc == 0) {
169
170                         /*
171                          * Then the master.passwd file
172                          */
173                         if (pwd != NULL)
174                                 fmtpwentry(pwbuf, pwd, PWF_MASTER);
175                         rc = fileupdate(getpwpath(_MASTERPASSWD), 0600, pwbuf, pfx, l, mode);
176                         if (rc == 0) {
177 #ifdef HAVE_PWDB_U
178                                 if (mode == UPD_DELETE || isrename)
179 #endif
180                                         rc = pwdb(NULL);
181 #ifdef HAVE_PWDB_U
182                                 else
183                                         rc = pwdb("-u", user, NULL);
184 #endif
185                         }
186                 }
187         }
188         return rc;
189 }
190
191 int
192 addpwent(struct passwd * pwd)
193 {
194         return pw_update(pwd, pwd->pw_name, UPD_CREATE);
195 }
196
197 int
198 chgpwent(char const * login, struct passwd * pwd)
199 {
200         return pw_update(pwd, login, UPD_REPLACE);
201 }
202
203 int
204 delpwent(struct passwd * pwd)
205 {
206         return pw_update(NULL, pwd->pw_name, UPD_DELETE);
207 }