deluser: size reduction by 60 bytes.

patch from Tito <farmatito@tiscali.it>


git-svn-id: svn://busybox.net/trunk/busybox@16680 69ca8d6d-28ef-0310-b511-8ec308f3f277

1 files changed, 48 insertions, 62 deletions

diff --git a/loginutils/deluser.c b/loginutils/deluser.c
index 795dae49f..e33710175 100644
--- a/loginutils/deluser.c
+++ b/loginutils/deluser.c
@@ -12,85 +12,71 @@
 
 #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 = strchr(start, '\n');
-        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 void del_line_matching(const char *login, const char *filename)
 {
-        char *buffer;
+        char *line;
         FILE *passwd;
-        Bounds b;
-        struct stat statbuf;
+        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);
+                }
+        }
 
+        if (!found) {
+                bb_error_msg("can't find '%s' in '%s'", login, filename);
+                if (!ENABLE_FEATURE_CLEAN_UP) return;
+                goto clean_up;
+        }
 
-        if ((passwd = fopen_or_warn(filename, "r"))) {
-                // Remove pointless const.
-                xstat((char *)filename, &statbuf);
-                buffer = (char *) xmalloc(statbuf.st_size * sizeof(char));
-                fread(buffer, statbuf.st_size, sizeof(char), passwd);
+        if (ENABLE_FEATURE_CLEAN_UP)
                 fclose(passwd);
-                /* find the user to remove */
-                b = boundary(buffer, login);
-                if (b.stop != 0) {
-                        /* write the file w/o the user */
-                        if ((passwd = fopen_or_warn(filename, "w"))) {
-                                fwrite(buffer, (b.start - 1), sizeof(char), passwd);
-                                fwrite(&buffer[b.stop], (statbuf.st_size - b.stop), sizeof(char), passwd);
-                                fclose(passwd);
+
+        passwd = fopen_or_warn(filename, "w");
+        if (passwd) {
+                if (ENABLE_FEATURE_CLEAN_UP) {
+ clean_up:
+                        while ((line = llist_pop(&plist))) {
+                                if (found) fputs(line, passwd);
+                                free(line);
                         }
+                        fclose(passwd);
                 } else {
-                        bb_error_msg("can't find '%s' in '%s'", login, filename);
+                        /* found != 0 here, no need to check */
+                        while ((line = llist_pop(&plist)))
+                                fputs(line, passwd);
                 }
-                free(buffer);
         }
 }
 
 int deluser_main(int argc, char **argv)
 {
-        if (argc != 2) {
+        if (argc != 2)
                 bb_show_usage();
-        } else {
-                if (ENABLE_DELUSER && 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);
-                }
-                del_line_matching(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_gshadow_file);
+                        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.4 2003/07/14 20:20:45 andersen Exp $ */