]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.sbin/pw/pwupd.c
Merge bmake-20121111
[FreeBSD/FreeBSD.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
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 <pwd.h>
38 #include <libutil.h>
39 #include <errno.h>
40 #include <err.h>
41 #include <sys/types.h>
42 #include <sys/stat.h>
43 #include <sys/param.h>
44 #include <sys/wait.h>
45
46 #include "pwupd.h"
47
48 #define HAVE_PWDB_C     1
49 #define HAVE_PWDB_U     1
50
51 static char pathpwd[] = _PATH_PWD;
52 static char * pwpath = pathpwd;
53  
54 int
55 setpwdir(const char * dir)
56 {
57         if (dir == NULL)
58                 return -1;
59         else {
60                 char * d = malloc(strlen(dir)+1);
61                 if (d == NULL)
62                         return -1;
63                 pwpath = strcpy(d, dir);
64         }
65         return 0;
66 }
67
68 char *
69 getpwpath(char const * file)
70 {
71         static char pathbuf[MAXPATHLEN];
72
73         snprintf(pathbuf, sizeof pathbuf, "%s/%s", pwpath, file);
74         return pathbuf;
75 }
76
77 static int
78 pwdb(char *arg,...)
79 {
80         int             i = 0;
81         pid_t           pid;
82         va_list         ap;
83         char           *args[10];
84
85         args[i++] = _PATH_PWD_MKDB;
86         va_start(ap, arg);
87         while (i < 6 && arg != NULL) {
88                 args[i++] = arg;
89                 arg = va_arg(ap, char *);
90         }
91         if (pwpath != pathpwd) {
92                 args[i++] = "-d";
93                 args[i++] = pwpath;
94         }
95         args[i++] = getpwpath(_MASTERPASSWD);
96         args[i] = NULL;
97
98         if ((pid = fork()) == -1)       /* Error (errno set) */
99                 i = errno;
100         else if (pid == 0) {    /* Child */
101                 execv(args[0], args);
102                 _exit(1);
103         } else {                /* Parent */
104                 waitpid(pid, &i, 0);
105                 if (WEXITSTATUS(i))
106                         i = EIO;
107         }
108         va_end(ap);
109         return i;
110 }
111
112 static int
113 pw_update(struct passwd * pwd, char const * user)
114 {
115         int             rc = 0;
116
117         /*
118          * First, let's check the see if the database is alright
119          * Note: -C is only available in FreeBSD 2.2 and above
120          */
121 #ifdef HAVE_PWDB_C
122         rc = pwdb("-C", (char *)NULL);  /* Check only */
123         if (rc == 0) {
124 #else
125         {                               /* No -C */
126 #endif
127                 int pfd, tfd;
128                 struct passwd *pw = NULL;
129                 struct passwd *old_pw = NULL;
130
131                 if (pwd != NULL)
132                        pw = pw_dup(pwd);
133
134                 if (user != NULL)
135                         old_pw = GETPWNAM(user);
136
137                 if (pw_init(pwpath, NULL))
138                         err(1, "pw_init()");
139                 if ((pfd = pw_lock()) == -1) {
140                         pw_fini();
141                         err(1, "pw_lock()");
142                 }
143                 if ((tfd = pw_tmp(-1)) == -1) {
144                         pw_fini();
145                         err(1, "pw_tmp()");
146                 }
147                 if (pw_copy(pfd, tfd, pw, old_pw) == -1) {
148                         pw_fini();
149                         err(1, "pw_copy()");
150                 }
151                 if (pw_mkdb(user) == -1) {
152                         pw_fini();
153                         err(1, "pw_mkdb()");
154                 }
155                 free(pw);
156                 pw_fini();
157         }
158         return 0;
159 }
160
161 int
162 addpwent(struct passwd * pwd)
163 {
164         return pw_update(pwd, NULL);
165 }
166
167 int
168 chgpwent(char const * login, struct passwd * pwd)
169 {
170         return pw_update(pwd, login);
171 }
172
173 int
174 delpwent(struct passwd * pwd)
175 {
176         char login[MAXLOGNAME];
177         
178         strlcpy(login, pwd->pw_name, MAXLOGNAME);
179         return pw_update(NULL, login);
180 }