]> git.cameronkatri.com Git - pw-darwin.git/blobdiff - libutil/pw_util.c
Modify pw_copy:
[pw-darwin.git] / libutil / pw_util.c
index 7741388ca0c7f5b07d6e88b838bd6639a8eea46d..1068eff6ea9d38e47ae7fbbebb7999fcf1c24df4 100644 (file)
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
- * 3. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *     This product includes software developed by the University of
- *     California, Berkeley and its contributors.
  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
@@ -71,7 +67,7 @@ static const char rcsid[] =
 #include <string.h>
 #include <unistd.h>
 
-#include <libutil.h>
+#include "libutil.h"
 
 static pid_t editpid = -1;
 static int lockfd = -1;
@@ -293,7 +289,7 @@ int
 pw_edit(int notsetuid)
 {
        struct sigaction sa, sa_int, sa_quit;
-       sigset_t oldsigset, sigset;
+       sigset_t oldsigset, nsigset;
        struct stat st1, st2;
        const char *editor;
        int pstat;
@@ -307,9 +303,9 @@ pw_edit(int notsetuid)
        sa.sa_flags = 0;
        sigaction(SIGINT, &sa, &sa_int);
        sigaction(SIGQUIT, &sa, &sa_quit);
-       sigemptyset(&sigset);
-       sigaddset(&sigset, SIGCHLD);
-       sigprocmask(SIG_BLOCK, &sigset, &oldsigset);
+       sigemptyset(&nsigset);
+       sigaddset(&nsigset, SIGCHLD);
+       sigprocmask(SIG_BLOCK, &nsigset, &oldsigset);
        switch ((editpid = fork())) {
        case -1:
                return (-1);
@@ -414,18 +410,25 @@ pw_make(const struct passwd *pw)
 }
 
 /*
- * Copy password file from one descriptor to another, replacing or adding
- * a single record on the way.
+ * Copy password file from one descriptor to another, replacing, deleting
+ * or adding a single record on the way.
  */
 int
 pw_copy(int ffd, int tfd, const struct passwd *pw, struct passwd *old_pw)
 {
        char buf[8192], *end, *line, *p, *q, *r, t;
        struct passwd *fpw;
+       const struct passwd *spw;
        size_t len;
        int eof, readlen;
 
-       if ((line = pw_make(pw)) == NULL)
+       spw = pw;
+       if (pw == NULL) {
+               line = NULL;
+               if (old_pw == NULL)
+                       return (-1);
+               spw = old_pw;
+       } else if ((line = pw_make(pw)) == NULL)
                return (-1);
 
        eof = 0;
@@ -493,7 +496,7 @@ pw_copy(int ffd, int tfd, const struct passwd *pw, struct passwd *old_pw)
                 */
 
                *q = t;
-               if (fpw == NULL || strcmp(fpw->pw_name, pw->pw_name) != 0) {
+               if (fpw == NULL || fpw->pw_uid != spw->pw_uid) {
                        /* nope */
                        if (fpw != NULL)
                                free(fpw);
@@ -510,11 +513,15 @@ pw_copy(int ffd, int tfd, const struct passwd *pw, struct passwd *old_pw)
                }
                free(fpw);
 
-               /* it is, replace it */
-               len = strlen(line);
-               if (write(tfd, line, len) != (int)len)
-                       goto err;
-
+               /* it is, replace or remove it */
+               if (line != NULL) {
+                       len = strlen(line);
+                       if (write(tfd, line, len) != (int)len)
+                               goto err;
+               } else {
+                       /* when removed, avoid the \n */
+                       q++;
+               }
                /* we're done, just copy the rest over */
                for (;;) {
                        if (write(tfd, q, end - q) != end - q)
@@ -532,16 +539,22 @@ pw_copy(int ffd, int tfd, const struct passwd *pw, struct passwd *old_pw)
                goto done;
        }
 
-       /* if we got here, we have a new entry */
+       /* if we got here, we didn't find the old entry */
+       if (line == NULL) {
+               errno = ENOENT;
+               goto err;
+       }
        len = strlen(line);
        if ((size_t)write(tfd, line, len) != len ||
            write(tfd, "\n", 1) != 1)
                goto err;
  done:
-       free(line);
+       if (line != NULL)
+               free(line);
        return (0);
  err:
-       free(line);
+       if (line != NULL)
+               free(line);
        return (-1);
 }
 
@@ -561,43 +574,50 @@ pw_tempname(void)
 struct passwd *
 pw_dup(const struct passwd *pw)
 {
+       char *dst;
        struct passwd *npw;
        ssize_t len;
 
-       len = sizeof(*npw) +
-           (pw->pw_name ? strlen(pw->pw_name) + 1 : 0) +
-           (pw->pw_passwd ? strlen(pw->pw_passwd) + 1 : 0) +
-           (pw->pw_class ? strlen(pw->pw_class) + 1 : 0) +
-           (pw->pw_gecos ? strlen(pw->pw_gecos) + 1 : 0) +
-           (pw->pw_dir ? strlen(pw->pw_dir) + 1 : 0) +
-           (pw->pw_shell ? strlen(pw->pw_shell) + 1 : 0);
+       len = sizeof(*npw);
+       if (pw->pw_name != NULL)
+               len += strlen(pw->pw_name) + 1;
+       if (pw->pw_passwd != NULL)
+               len += strlen(pw->pw_passwd) + 1;
+       if (pw->pw_class != NULL)
+               len += strlen(pw->pw_class) + 1;
+       if (pw->pw_gecos != NULL)
+               len += strlen(pw->pw_gecos) + 1;
+       if (pw->pw_dir != NULL)
+               len += strlen(pw->pw_dir) + 1;
+       if (pw->pw_shell != NULL)
+               len += strlen(pw->pw_shell) + 1;
        if ((npw = malloc((size_t)len)) == NULL)
                return (NULL);
        memcpy(npw, pw, sizeof(*npw));
-       len = sizeof(*npw);
-       if (pw->pw_name) {
-               npw->pw_name = ((char *)npw) + len;
-               len += sprintf(npw->pw_name, "%s", pw->pw_name) + 1;
+       dst = (char *)npw + sizeof(*npw);
+       if (pw->pw_name != NULL) {
+               npw->pw_name = dst;
+               dst = stpcpy(npw->pw_name, pw->pw_name) + 1;
        }
-       if (pw->pw_passwd) {
-               npw->pw_passwd = ((char *)npw) + len;
-               len += sprintf(npw->pw_passwd, "%s", pw->pw_passwd) + 1;
+       if (pw->pw_passwd != NULL) {
+               npw->pw_passwd = dst;
+               dst = stpcpy(npw->pw_passwd, pw->pw_passwd) + 1;
        }
-       if (pw->pw_class) {
-               npw->pw_class = ((char *)npw) + len;
-               len += sprintf(npw->pw_class, "%s", pw->pw_class) + 1;
+       if (pw->pw_class != NULL) {
+               npw->pw_class = dst;
+               dst = stpcpy(npw->pw_class, pw->pw_class) + 1;
        }
-       if (pw->pw_gecos) {
-               npw->pw_gecos = ((char *)npw) + len;
-               len += sprintf(npw->pw_gecos, "%s", pw->pw_gecos) + 1;
+       if (pw->pw_gecos != NULL) {
+               npw->pw_gecos = dst;
+               dst = stpcpy(npw->pw_gecos, pw->pw_gecos) + 1;
        }
-       if (pw->pw_dir) {
-               npw->pw_dir = ((char *)npw) + len;
-               len += sprintf(npw->pw_dir, "%s", pw->pw_dir) + 1;
+       if (pw->pw_dir != NULL) {
+               npw->pw_dir = dst;
+               dst = stpcpy(npw->pw_dir, pw->pw_dir) + 1;
        }
-       if (pw->pw_shell) {
-               npw->pw_shell = ((char *)npw) + len;
-               len += sprintf(npw->pw_shell, "%s", pw->pw_shell) + 1;
+       if (pw->pw_shell != NULL) {
+               npw->pw_shell = dst;
+               dst = stpcpy(npw->pw_shell, pw->pw_shell) + 1;
        }
        return (npw);
 }