random small size optimizations
[oweals/busybox.git] / loginutils / deluser.c
index ad62d41e0d6e197125ae1310119a8776c1e55be3..d07fd22aca3fb3ae2696bc0f32dc54ff9380d227 100644 (file)
  *
  * Copyright (C) 1999 by Lineo, inc. and John Beppu
  * Copyright (C) 1999,2000,2001 by John Beppu <beppu@codepoet.org>
+ * Unified with delgroup by Tito Ragusa <farmatito@tiscali.it>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Licensed under GPL version 2, see file LICENSE in this tarball for details.
  *
  */
 
-#include <sys/stat.h>
-#include <unistd.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
 #include "busybox.h"
 
-
-
-/* where to start and stop deletion */
-typedef struct {
-       size_t start;
-       size_t stop;
-} Bounds;
-
-/* An interesting side-effect of boundary()'s
- * implementation is that the first user (typically root)
- * cannot be removed.  Let's call it a feature. */
-static inline Bounds boundary(const char *buffer, const char *login)
-{
-       char needle[256];
-       char *start;
-       char *stop;
-       Bounds b;
-
-       snprintf(needle, 256, "\n%s:", login);
-       needle[255] = 0;
-       start = strstr(buffer, needle);
-       if (!start) {
-               b.start = 0;
-               b.stop = 0;
-               return b;
-       }
-       start++;
-
-       stop = index(start, '\n');      /* index is a BSD-ism */
-       b.start = start - buffer;
-       b.stop = stop - buffer;
-       return b;
-}
-
-/* grep -v ^login (except it only deletes the first match) */
-/* ...in fact, I think I'm going to simplify this later */
-static int del_line_matching(const char *login, const char *filename)
+static void del_line_matching(const char *login, const char *filename)
 {
-       char *buffer;
+       char *line;
        FILE *passwd;
-       size_t len;
-       Bounds b;
-       struct stat statbuf;
-
-       /* load into buffer */
-       passwd = fopen(filename, "r");
-       if (!passwd) {
-               return 1;
-       }
-       stat(filename, &statbuf);
-       len = statbuf.st_size;
-       buffer = (char *) malloc(len * sizeof(char));
-
-       if (!buffer) {
-               fclose(passwd);
-               return 1;
-       }
-       fread(buffer, len, sizeof(char), passwd);
-
-       fclose(passwd);
-
-       /* find the user to remove */
-       b = boundary(buffer, login);
-       if (b.stop == 0) {
-               free(buffer);
-               return 1;
-       }
-
-       /* write the file w/o the user */
-       passwd = fopen(filename, "w");
-       if (!passwd) {
-               return 1;
+       int len = strlen(login);
+       int found = 0;
+       llist_t *plist = NULL;
+
+       passwd = fopen_or_warn(filename, "r");
+       if (!passwd) return;
+
+       while ((line = xmalloc_fgets(passwd))) {
+               if (!strncmp(line, login, len)
+                && line[len] == ':'
+               ) {
+                       found++;
+                       free(line);
+               } else {
+                       llist_add_to_end(&plist, line);
+               }
        }
-       fwrite(buffer, (b.start - 1), sizeof(char), passwd);
-       fwrite(&buffer[b.stop], (len - b.stop), sizeof(char), passwd);
-
-       fclose(passwd);
-
-       return 0;
-}
-
-/* ________________________________________________________________________ */
-int delgroup_main(int argc, char **argv)
-{
-       /* int successful; */
-       int failure;
 
-       if (argc != 2) {
-               bb_show_usage();
+       if (!ENABLE_FEATURE_CLEAN_UP) {
+               if (!found) {
+                       bb_error_msg("can't find '%s' in '%s'", login, filename);
+                       return;
+               }
+               passwd = fopen_or_warn(filename, "w");
+               if (passwd)
+                       while ((line = llist_pop(&plist)))
+                               fputs(line, passwd);
        } else {
-
-               failure = del_line_matching(argv[1], bb_path_group_file);
-#ifdef CONFIG_FEATURE_SHADOWPASSWDS
-               if (access(bb_path_gshadow_file, W_OK) == 0) {
-                       /* EDR the |= works if the error is not 0, so he had it wrong */
-                       failure |= del_line_matching(argv[1], bb_path_gshadow_file);
+               if (!found) {
+                       bb_error_msg("can't find '%s' in '%s'", login, filename);
+                       goto clean_up;
                }
-#endif                                                 /* CONFIG_FEATURE_SHADOWPASSWDS */
-               /* if (!successful) { */
-               if (failure) {
-                       bb_error_msg_and_die("%s: Group could not be removed\n", argv[1]);
+               fclose(passwd);
+               passwd = fopen_or_warn(filename, "w");
+               if (passwd) {
+ clean_up:
+                       while ((line = llist_pop(&plist))) {
+                               if (found) fputs(line, passwd);
+                               free(line);
+                       }
+                       fclose(passwd);
                }
-
        }
-       return (EXIT_SUCCESS);
 }
 
-/* ________________________________________________________________________ */
 int deluser_main(int argc, char **argv)
 {
-       /* int successful; */
-       int failure;
-
-       if (argc != 2) {
+       if (argc != 2)
                bb_show_usage();
-       } else {
-
-               failure = del_line_matching(argv[1], bb_path_passwd_file);
-               /* if (!successful) { */
-               if (failure) {
-                       bb_error_msg_and_die("%s: User could not be removed from %s\n",
-                                                         argv[1], bb_path_passwd_file);
-               }
-#ifdef CONFIG_FEATURE_SHADOWPASSWDS
-               failure = del_line_matching(argv[1], bb_path_shadow_file);
-               /* if (!successful) { */
-               if (failure) {
-                       bb_error_msg_and_die("%s: User could not be removed from %s\n",
-                                                         argv[1], bb_path_shadow_file);
-               }
-               failure = del_line_matching(argv[1], bb_path_gshadow_file);
-               /* if (!successful) { */
-               if (failure) {
-                       bb_error_msg_and_die("%s: User could not be removed from %s\n",
-                                                         argv[1], bb_path_gshadow_file);
-               }
-#endif                                                 /* CONFIG_FEATURE_SHADOWPASSWDS */
-               failure = del_line_matching(argv[1], bb_path_group_file);
-               /* if (!successful) { */
-               if (failure) {
-                       bb_error_msg_and_die("%s: User could not be removed from %s\n",
-                                                         argv[1], bb_path_group_file);
-               }
 
+       if (ENABLE_DELUSER
+        && (!ENABLE_DELGROUP || applet_name[3] == 'u')
+       ) {
+               del_line_matching(argv[1], bb_path_passwd_file);
+               if (ENABLE_FEATURE_SHADOWPASSWDS)
+                       del_line_matching(argv[1], bb_path_shadow_file);
        }
-       return (EXIT_SUCCESS);
+       del_line_matching(argv[1], bb_path_group_file);
+       if (ENABLE_FEATURE_SHADOWPASSWDS)
+               del_line_matching(argv[1], bb_path_gshadow_file);
+
+       return EXIT_SUCCESS;
 }
 
-/* $Id: deluser.c,v 1.3 2003/03/19 09:12:20 mjn3 Exp $ */
+/* $Id: deluser.c,v 1.4 2003/07/14 20:20:45 andersen Exp $ */