xref: /dragonfly/usr.sbin/pw/pwupd.c (revision f2c43266)
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.21 2008/07/17 13:47:59 jhb Exp $
27  */
28 
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <string.h>
32 #include <unistd.h>
33 #include <stdarg.h>
34 #include <errno.h>
35 #include <sys/types.h>
36 #include <sys/stat.h>
37 #include <sys/param.h>
38 #include <sys/wait.h>
39 
40 #include "pwupd.h"
41 
42 #define HAVE_PWDB_C	1
43 #define	HAVE_PWDB_U	1
44 
45 static char pathpwd[] = _PATH_PWD;
46 static char * pwpath = pathpwd;
47 
48 int
49 setpwdir(const char * dir)
50 {
51 	if (dir == NULL)
52 		return -1;
53 	else {
54 		char * d = malloc(strlen(dir)+1);
55 		if (d == NULL)
56 			return -1;
57 		pwpath = strcpy(d, dir);
58 	}
59 	return 0;
60 }
61 
62 char *
63 getpwpath(char const * file)
64 {
65 	static char pathbuf[MAXPATHLEN];
66 
67 	snprintf(pathbuf, sizeof pathbuf, "%s/%s", pwpath, file);
68 	return pathbuf;
69 }
70 
71 int
72 pwdb(char *arg,...)
73 {
74 	int             i = 0;
75 	pid_t           pid;
76 	va_list         ap;
77 	char           *args[10];
78 
79 	args[i++] = _PATH_PWD_MKDB;
80 	va_start(ap, arg);
81 	while (i < 6 && arg != NULL) {
82 		args[i++] = arg;
83 		arg = va_arg(ap, char *);
84 	}
85 	if (pwpath != pathpwd) {
86 		args[i++] = "-d";
87 		args[i++] = pwpath;
88 	}
89 	args[i++] = getpwpath(_MASTERPASSWD);
90 	args[i] = NULL;
91 
92 	if ((pid = fork()) == -1)	/* Error (errno set) */
93 		i = errno;
94 	else if (pid == 0) {	/* Child */
95 		execv(args[0], args);
96 		_exit(1);
97 	} else {		/* Parent */
98 		waitpid(pid, &i, 0);
99 		if (WEXITSTATUS(i))
100 			i = EIO;
101 	}
102 	return i;
103 }
104 
105 int
106 fmtpwentry(char *buf, struct passwd * pwd, int type)
107 {
108 	int             l;
109 	char           *pw;
110 
111 	pw = (type == PWF_MASTER) ?
112 	    ((pwd->pw_passwd == NULL) ? "" : 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 	rc = pwdb("-C", NULL);	/* Check only */
149 	if (rc == 0) {
150 #else
151 	{				/* No -C */
152 #endif
153 		char            pfx[PWBUFSZ];
154 		char            pwbuf[PWBUFSZ];
155 		int             l = snprintf(pfx, PWBUFSZ, "%s:", user);
156 #ifdef HAVE_PWDB_U
157 		int		isrename = pwd!=NULL && strcmp(user, pwd->pw_name);
158 #endif
159 
160 		/*
161 		 * Update the passwd file first
162 		 */
163 		if (pwd == NULL)
164 			*pwbuf = '\0';
165 		else
166 			fmtpwentry(pwbuf, pwd, PWF_PASSWD);
167 
168 		if (l < 0)
169 			l = 0;
170 		rc = fileupdate(getpwpath(_PASSWD), 0644, pwbuf, pfx, l, mode);
171 		if (rc == 0) {
172 
173 			/*
174 			 * Then the master.passwd file
175 			 */
176 			if (pwd != NULL)
177 				fmtpwentry(pwbuf, pwd, PWF_MASTER);
178 			rc = fileupdate(getpwpath(_MASTERPASSWD), 0600, pwbuf, pfx, l, mode);
179 			if (rc == 0) {
180 #ifdef HAVE_PWDB_U
181 				if (mode == UPD_DELETE || isrename)
182 #endif
183 					rc = pwdb(NULL);
184 #ifdef HAVE_PWDB_U
185 				else
186 					rc = pwdb("-u", user, NULL);
187 #endif
188 			}
189 		}
190 	}
191 	return rc;
192 }
193 
194 int
195 addpwent(struct passwd * pwd)
196 {
197 	return pw_update(pwd, pwd->pw_name, UPD_CREATE);
198 }
199 
200 int
201 chgpwent(char const * login, struct passwd * pwd)
202 {
203 	return pw_update(pwd, login, UPD_REPLACE);
204 }
205 
206 int
207 delpwent(struct passwd * pwd)
208 {
209 	return pw_update(NULL, pwd->pw_name, UPD_DELETE);
210 }
211