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