From c9bb67787a0d48af33045a2d813faf44ed5d59b4 Mon Sep 17 00:00:00 2001 From: Denis Vlasenko Date: Mon, 27 Nov 2006 00:54:59 +0000 Subject: [PATCH] deluser: size reduction by 60 bytes. patch from Tito --- loginutils/deluser.c | 110 +++++++++++++++++++------------------------ 1 file 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 $ */ -- 2.25.1