pwupd.c revision 180565 
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
28static const char rcsid[] =
29  "$FreeBSD: head/usr.sbin/pw/pwupd.c 180565 2008-07-17 13:47:59Z jhb $";
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
48static char pathpwd[] = _PATH_PWD;
49static char * pwpath = pathpwd;
50
51int
52setpwdir(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
65char *
66getpwpath(char const * file)
67{
68	static char pathbuf[MAXPATHLEN];
69
70	snprintf(pathbuf, sizeof pathbuf, "%s/%s", pwpath, file);
71	return pathbuf;
72}
73
74int
75pwdb(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
108int
109fmtpwentry(char *buf, struct passwd * pwd, int type)
110{
111	int             l;
112	char           *pw;
113
114	pw = (type == PWF_MASTER) ?
115	    ((pwd->pw_passwd == NULL) ? "" : pwd->pw_passwd) : "*";
116
117	if (type == PWF_PASSWD)
118		l = sprintf(buf, "%s:*:%ld:%ld:%s:%s:%s\n",
119		       pwd->pw_name, (long) pwd->pw_uid, (long) pwd->pw_gid,
120			    pwd->pw_gecos ? pwd->pw_gecos : "User &",
121			    pwd->pw_dir, pwd->pw_shell);
122	else
123		l = sprintf(buf, "%s:%s:%ld:%ld:%s:%lu:%lu:%s:%s:%s\n",
124		   pwd->pw_name, pw, (long) pwd->pw_uid, (long) pwd->pw_gid,
125			    pwd->pw_class ? pwd->pw_class : "",
126			    (unsigned long) pwd->pw_change,
127			    (unsigned long) pwd->pw_expire,
128			    pwd->pw_gecos, pwd->pw_dir, pwd->pw_shell);
129	return l;
130}
131
132
133int
134fmtpwent(char *buf, struct passwd * pwd)
135{
136	return fmtpwentry(buf, pwd, PWF_STANDARD);
137}
138
139static int
140pw_update(struct passwd * pwd, char const * user, int mode)
141{
142	int             rc = 0;
143
144	ENDPWENT();
145
146	/*
147	 * First, let's check the see if the database is alright
148	 * Note: -C is only available in FreeBSD 2.2 and above
149	 */
150#ifdef HAVE_PWDB_C
151	rc = pwdb("-C", (char *)NULL);	/* Check only */
152	if (rc == 0) {
153#else
154	{				/* No -C */
155#endif
156		char            pfx[PWBUFSZ];
157		char            pwbuf[PWBUFSZ];
158		int             l = snprintf(pfx, PWBUFSZ, "%s:", user);
159#ifdef HAVE_PWDB_U
160		int		isrename = pwd!=NULL && strcmp(user, pwd->pw_name);
161#endif
162
163		/*
164		 * Update the passwd file first
165		 */
166		if (pwd == NULL)
167			*pwbuf = '\0';
168		else
169			fmtpwentry(pwbuf, pwd, PWF_PASSWD);
170
171		if (l < 0)
172			l = 0;
173		rc = fileupdate(getpwpath(_PASSWD), 0644, pwbuf, pfx, l, mode);
174		if (rc == 0) {
175
176			/*
177			 * Then the master.passwd file
178			 */
179			if (pwd != NULL)
180				fmtpwentry(pwbuf, pwd, PWF_MASTER);
181			rc = fileupdate(getpwpath(_MASTERPASSWD), 0600, pwbuf, pfx, l, mode);
182			if (rc == 0) {
183#ifdef HAVE_PWDB_U
184				if (mode == UPD_DELETE || isrename)
185#endif
186					rc = pwdb(NULL);
187#ifdef HAVE_PWDB_U
188				else
189					rc = pwdb("-u", user, (char *)NULL);
190#endif
191			}
192		}
193	}
194	return rc;
195}
196
197int
198addpwent(struct passwd * pwd)
199{
200	return pw_update(pwd, pwd->pw_name, UPD_CREATE);
201}
202
203int
204chgpwent(char const * login, struct passwd * pwd)
205{
206	return pw_update(pwd, login, UPD_REPLACE);
207}
208
209int
210delpwent(struct passwd * pwd)
211{
212	return pw_update(NULL, pwd->pw_name, UPD_DELETE);
213}
214