X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=procps%2Ffuser.c;h=a1b93d77fe9fcadbff0097467000248f14822b09;hb=9aa599dc9dc076f6fa6e4312e4750a703cf16450;hp=4906797cf075c2b13d5d80ca232bfe04ee60c997;hpb=ea62077b850076c4d7dc3cf78ebd1888928c6ddf;p=platform%2Fupstream%2Fbusybox.git

diff --git a/procps/fuser.c b/procps/fuser.c
index 4906797..a1b93d7 100644
--- a/procps/fuser.c
+++ b/procps/fuser.c
@@ -4,363 +4,314 @@
  *
  * Copyright 2004 Tony J. White
  *
- * May be distributed under the conditions of the
- * GNU Library General Public License
+ * Licensed under GPLv2, see file LICENSE in this source tree.
  */
 
-#include "busybox.h"
+#include "libbb.h"
 
-#define FUSER_PROC_DIR "/proc"
-#define FUSER_MAX_LINE 255
+#define MAX_LINE 255
 
-#define FUSER_OPT_MOUNT  1
-#define FUSER_OPT_KILL   2
-#define FUSER_OPT_SILENT 4
-#define FUSER_OPT_IP6    8
-#define FUSER_OPT_IP4    16
+#define OPTION_STRING "mks64"
+enum {
+	OPT_MOUNT  = (1 << 0),
+	OPT_KILL   = (1 << 1),
+	OPT_SILENT = (1 << 2),
+	OPT_IP6    = (1 << 3),
+	OPT_IP4    = (1 << 4),
+};
 
 typedef struct inode_list {
+	struct inode_list *next;
 	ino_t inode;
 	dev_t dev;
-	struct inode_list *next;
 } inode_list;
 
 typedef struct pid_list {
-	pid_t pid;
 	struct pid_list *next;
+	pid_t pid;
 } pid_list;
 
-static int fuser_option(char *option)
-{
-	int opt = 0;
-
-	if(!(strlen(option))) return 0;
-	if(option[0] != '-') return 0;
-	++option;
-	while(*option != '\0') {
-		if(*option == 'm') opt |= FUSER_OPT_MOUNT;
-		else if(*option == 'k') opt |= FUSER_OPT_KILL;
-		else if(*option == 's') opt |= FUSER_OPT_SILENT;
-		else if(*option == '6') opt |= FUSER_OPT_IP6;
-		else if(*option == '4') opt |= FUSER_OPT_IP4;
-		else {
-			bb_error_msg_and_die(
-				"Unsupported option '%c'", *option);
-		}
-		++option;
-	}
-	return opt;
-}
-
-static int fuser_file_to_dev_inode(const char *filename,
-	 dev_t *dev, ino_t *inode)
-{
-	struct stat f_stat;
-	if((stat(filename, &f_stat)) < 0) return 0;
-	*inode = f_stat.st_ino;
-	*dev = f_stat.st_dev;
-	return 1;
-}
 
-static int fuser_find_socket_dev(dev_t *dev)
-{
-	int fd = socket(PF_INET, SOCK_DGRAM,0);
-	struct stat buf;
+struct globals {
+	pid_list *pid_list_head;
+	inode_list *inode_list_head;
+} FIX_ALIASING;
+#define G (*(struct globals*)&bb_common_bufsiz1)
+#define INIT_G() do { } while (0)
 
-	if (fd >= 0 && (fstat(fd, &buf)) == 0) {
-		*dev =  buf.st_dev;
-		close(fd);
-		return 1;
-	}
-	return 0;
-}
 
-static int fuser_parse_net_arg(const char *filename,
-	const char **proto, int *port)
+static void add_pid(const pid_t pid)
 {
-	char path[sizeof(FUSER_PROC_DIR)+12], tproto[5];
-
-	if((sscanf(filename, "%d/%4s", port, tproto)) != 2) return 0;
-	sprintf(path, "%s/net/%s", FUSER_PROC_DIR, tproto);
-	if((access(path, R_OK)) != 0) return 0;
-	*proto = xstrdup(tproto);
-	return 1;
-}
+	pid_list **curr = &G.pid_list_head;
 
-static int fuser_add_pid(pid_list *plist, pid_t pid)
-{
-	pid_list *curr = NULL, *last = NULL;
-
-	if(plist->pid == 0) plist->pid = pid;
-	curr = plist;
-	while(curr != NULL) {
-		if(curr->pid == pid) return 1;
-		last = curr;
-		curr = curr->next;
+	while (*curr) {
+		if ((*curr)->pid == pid)
+			return;
+		curr = &(*curr)->next;
 	}
-	curr = xmalloc(sizeof(pid_list));
-	last->next = curr;
-	curr->pid = pid;
-	curr->next = NULL;
-	return 1;
+
+	*curr = xzalloc(sizeof(pid_list));
+	(*curr)->pid = pid;
 }
 
-static int fuser_add_inode(inode_list *ilist, dev_t dev, ino_t inode)
+static void add_inode(const struct stat *st)
 {
-	inode_list *curr = NULL, *last = NULL;
+	inode_list **curr = &G.inode_list_head;
 
-	if(!ilist->inode && !ilist->dev) {
-		ilist->dev = dev;
-		ilist->inode = inode;
-	}
-	curr = ilist;
-	while(curr != NULL) {
-		if(curr->inode == inode && curr->dev == dev) return 1;
-		last = curr;
-		curr = curr->next;
+	while (*curr) {
+		if ((*curr)->dev == st->st_dev
+		 && (*curr)->inode == st->st_ino
+		) {
+			return;
+		}
+		curr = &(*curr)->next;
 	}
-	curr = xmalloc(sizeof(inode_list));
-	last->next = curr;
-	curr->dev = dev;
-	curr->inode = inode;
-	curr->next = NULL;
-	return 1;
+
+	*curr = xzalloc(sizeof(inode_list));
+	(*curr)->dev = st->st_dev;
+	(*curr)->inode = st->st_ino;
 }
 
-static int fuser_scan_proc_net(int opts, const char *proto,
-	int port, inode_list *ilist)
+static void scan_proc_net(const char *path, unsigned port)
 {
-	char path[sizeof(FUSER_PROC_DIR)+12], line[FUSER_MAX_LINE+1];
-	char addr[128];
-	ino_t tmp_inode;
-	dev_t tmp_dev;
-	long long  uint64_inode;
-	int tmp_port;
+	char line[MAX_LINE + 1];
+	long long uint64_inode;
+	unsigned tmp_port;
 	FILE *f;
+	struct stat st;
+	int fd;
+
+	/* find socket dev */
+	st.st_dev = 0;
+	fd = socket(AF_INET, SOCK_DGRAM, 0);
+	if (fd >= 0) {
+		fstat(fd, &st);
+		close(fd);
+	}
 
-	if(!fuser_find_socket_dev(&tmp_dev)) tmp_dev = 0;
-	sprintf(path, "%s/net/%s", FUSER_PROC_DIR, proto);
-
-	if (!(f = fopen(path, "r"))) return 0;
-	while(fgets(line, FUSER_MAX_LINE, f)) {
-		if(sscanf(line,
-			"%*d: %64[0-9A-Fa-f]:%x %*x:%*x %*x %*x:%*x "
-			"%*x:%*x %*x %*d %*d %llu",
-			addr, &tmp_port, &uint64_inode) == 3) {
-			if((strlen(addr) == 8) &&
-				(opts & FUSER_OPT_IP6)) continue;
-			else if((strlen(addr) > 8) &&
-				(opts & FUSER_OPT_IP4)) continue;
-			if(tmp_port == port) {
-				tmp_inode = uint64_inode;
-				fuser_add_inode(ilist, tmp_dev, tmp_inode);
+	f = fopen_for_read(path);
+	if (!f)
+		return;
+
+	while (fgets(line, MAX_LINE, f)) {
+		char addr[68];
+		if (sscanf(line, "%*d: %64[0-9A-Fa-f]:%x %*x:%*x %*x %*x:%*x "
+				"%*x:%*x %*x %*d %*d %llu",
+				addr, &tmp_port, &uint64_inode) == 3
+		) {
+			int len = strlen(addr);
+			if (len == 8 && (option_mask32 & OPT_IP6))
+				continue;
+			if (len > 8 && (option_mask32 & OPT_IP4))
+				continue;
+			if (tmp_port == port) {
+				st.st_ino = uint64_inode;
+				add_inode(&st);
 			}
 		}
-
 	}
 	fclose(f);
-	return 1;
 }
 
-static int fuser_search_dev_inode(int opts, inode_list *ilist,
-	dev_t dev, ino_t inode)
+static int search_dev_inode(const struct stat *st)
 {
-	inode_list *curr;
-	curr = ilist;
-
-	while(curr) {
-		if((opts & FUSER_OPT_MOUNT) &&  curr->dev == dev)
-			return 1;
-		if(curr->inode == inode && curr->dev == dev)
-			return 1;
-		curr = curr->next;
+	inode_list *ilist = G.inode_list_head;
+
+	while (ilist) {
+		if (ilist->dev == st->st_dev) {
+			if (option_mask32 & OPT_MOUNT)
+				return 1;
+			if (ilist->inode == st->st_ino)
+				return 1;
+		}
+		ilist = ilist->next;
 	}
 	return 0;
 }
 
-static int fuser_scan_pid_maps(int opts, const char *fname, pid_t pid,
-	inode_list *ilist, pid_list *plist)
+static void scan_pid_maps(const char *fname, pid_t pid)
 {
 	FILE *file;
-	char line[FUSER_MAX_LINE + 1];
+	char line[MAX_LINE + 1];
 	int major, minor;
-	ino_t inode;
 	long long uint64_inode;
-	dev_t dev;
+	struct stat st;
 
-	if (!(file = fopen(fname, "r"))) return 0;
-	while (fgets(line, FUSER_MAX_LINE, file)) {
-		if(sscanf(line, "%*s %*s %*s %x:%x %llu",
-			&major, &minor, &uint64_inode) != 3) continue;
-		inode = uint64_inode;
-		if(major == 0 && minor == 0 && inode == 0) continue;
-		dev = makedev(major, minor);
-		if(fuser_search_dev_inode(opts, ilist, dev, inode)) {
-			fuser_add_pid(plist, pid);
-		}
+	file = fopen_for_read(fname);
+	if (!file)
+		return;
 
+	while (fgets(line, MAX_LINE, file)) {
+		if (sscanf(line, "%*s %*s %*s %x:%x %llu", &major, &minor, &uint64_inode) != 3)
+			continue;
+		st.st_ino = uint64_inode;
+		if (major == 0 && minor == 0 && st.st_ino == 0)
+			continue;
+		st.st_dev = makedev(major, minor);
+		if (search_dev_inode(&st))
+			add_pid(pid);
 	}
 	fclose(file);
-	return 1;
 }
 
-static int fuser_scan_link(int opts, const char *lname, pid_t pid,
-	inode_list *ilist, pid_list *plist)
+static void scan_link(const char *lname, pid_t pid)
 {
-	ino_t inode;
-	dev_t dev;
+	struct stat st;
 
-	if(!fuser_file_to_dev_inode(lname, &dev, &inode)) return 0;
-	if(fuser_search_dev_inode(opts, ilist, dev, inode))
-		fuser_add_pid(plist, pid);
-	return 1;
+	if (stat(lname, &st) >= 0) {
+		if (search_dev_inode(&st))
+			add_pid(pid);
+	}
 }
 
-static int fuser_scan_dir_links(int opts, const char *dname, pid_t pid,
-	inode_list *ilist, pid_list *plist)
+static void scan_dir_links(const char *dname, pid_t pid)
 {
 	DIR *d;
 	struct dirent *de;
 	char *lname;
 
-	if((d = opendir(dname))) {
-		while((de = readdir(d)) != NULL) {
-			lname = concat_subpath_file(dname, de->d_name);
-			if(lname == NULL)
-				continue;
-			fuser_scan_link(opts, lname, pid, ilist, plist);
-			free(lname);
-		}
-		closedir(d);
-	}
-	else return 0;
-	return 1;
+	d = opendir(dname);
+	if (!d)
+		return;
 
+	while ((de = readdir(d)) != NULL) {
+		lname = concat_subpath_file(dname, de->d_name);
+		if (lname == NULL)
+			continue;
+		scan_link(lname, pid);
+		free(lname);
+	}
+	closedir(d);
 }
 
-static int fuser_scan_proc_pids(int opts, inode_list *ilist, pid_list *plist)
+/* NB: does chdir internally */
+static void scan_proc_pids(void)
 {
 	DIR *d;
 	struct dirent *de;
 	pid_t pid;
-	char *dname;
-
-	if(!(d = opendir(FUSER_PROC_DIR))) return 0;
-	while((de = readdir(d)) != NULL) {
-		pid = (pid_t)atoi(de->d_name);
-		if(!pid) continue;
-		dname = concat_subpath_file(FUSER_PROC_DIR, de->d_name);
-		if(chdir(dname) < 0) {
-			free(dname);
+
+	xchdir("/proc");
+	d = opendir("/proc");
+	if (!d)
+		return;
+
+	while ((de = readdir(d)) != NULL) {
+		pid = (pid_t)bb_strtou(de->d_name, NULL, 10);
+		if (errno)
 			continue;
-		}
-		free(dname);
-		fuser_scan_link(opts, "cwd", pid, ilist, plist);
-		fuser_scan_link(opts, "exe", pid, ilist, plist);
-		fuser_scan_link(opts, "root", pid, ilist, plist);
-		fuser_scan_dir_links(opts, "fd", pid, ilist, plist);
-		fuser_scan_dir_links(opts, "lib", pid, ilist, plist);
-		fuser_scan_dir_links(opts, "mmap", pid, ilist, plist);
-		fuser_scan_pid_maps(opts, "maps", pid, ilist, plist);
-		chdir("..");
+		if (chdir(de->d_name) < 0)
+			continue;
+		scan_link("cwd", pid);
+		scan_link("exe", pid);
+		scan_link("root", pid);
+
+		scan_dir_links("fd", pid);
+		scan_dir_links("lib", pid);
+		scan_dir_links("mmap", pid);
+
+		scan_pid_maps("maps", pid);
+		xchdir("/proc");
 	}
 	closedir(d);
-	return 1;
 }
 
-static int fuser_print_pid_list(pid_list *plist)
+int fuser_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int fuser_main(int argc UNUSED_PARAM, char **argv)
 {
-	pid_list *curr = plist;
-
-	if(plist == NULL) return 0;
-	while(curr != NULL) {
-		if(curr->pid > 0) printf("%d ", curr->pid);
-		curr = curr->next;
+	pid_list *plist;
+	pid_t mypid;
+	char **pp;
+	struct stat st;
+	unsigned port;
+	int opt;
+	int exitcode;
+	int killsig;
+/*
+fuser [OPTIONS] FILE or PORT/PROTO
+Find processes which use FILEs or PORTs
+        -m      Find processes which use same fs as FILEs
+        -4      Search only IPv4 space
+        -6      Search only IPv6 space
+        -s      Don't display PIDs
+        -k      Kill found processes
+        -SIGNAL Signal to send (default: KILL)
+*/
+	/* Handle -SIGNAL. Oh my... */
+	killsig = SIGKILL; /* yes, the default is not SIGTERM */
+	pp = argv;
+	while (*++pp) {
+		char *arg = *pp;
+		if (arg[0] != '-')
+			continue;
+		if (arg[1] == '-' && arg[2] == '\0') /* "--" */
+			break;
+		if ((arg[1] == '4' || arg[1] == '6') && arg[2] == '\0')
+			continue; /* it's "-4" or "-6" */
+		opt = get_signum(&arg[1]);
+		if (opt < 0)
+			continue;
+		/* "-SIGNAL" option found. Remove it and bail out */
+		killsig = opt;
+		do {
+			pp[0] = arg = pp[1];
+			pp++;
+		} while (arg);
+		break;
 	}
-	printf("\n");
-	return 1;
-}
 
-static int fuser_kill_pid_list(pid_list *plist, int sig)
-{
-	pid_list *curr = plist;
-	pid_t mypid = getpid();
-	int success = 1;
-
-	if(plist == NULL) return 0;
-	while(curr != NULL) {
-		if(curr->pid > 0 && curr->pid != mypid) {
-			if (kill(curr->pid, sig) != 0) {
-				bb_perror_msg(
-					"cannot kill pid '%d'", curr->pid);
-				success = 0;
-			}
+	opt_complementary = "-1"; /* at least one param */
+	opt = getopt32(argv, OPTION_STRING);
+	argv += optind;
+
+	pp = argv;
+	while (*pp) {
+		/* parse net arg */
+		char path[20], tproto[5];
+		if (sscanf(*pp, "%u/%4s", &port, tproto) != 2)
+			goto file;
+		sprintf(path, "/proc/net/%s", tproto);
+		if (access(path, R_OK) != 0) { /* PORT/PROTO */
+			scan_proc_net(path, port);
+		} else { /* FILE */
+ file:
+			xstat(*pp, &st);
+			add_inode(&st);
 		}
-		curr = curr->next;
+		pp++;
 	}
-	return success;
-}
 
-int fuser_main(int argc, char **argv)
-{
-	int port, i, optn;
-	int* fni; /* file name indexes of argv */
-	int fnic = 0;  /* file name index count */
-	const char *proto;
-	static int opt = 0; /* FUSER_OPT_ */
-	dev_t dev;
-	ino_t inode;
-	pid_list *pids;
-	inode_list *inodes;
-	int killsig = SIGTERM;
-	int success = 1;
-
-	if (argc < 2)
-		bb_show_usage();
-
-	fni = xmalloc(sizeof(int));
-	for(i=1;i<argc;i++) {
-		optn = fuser_option(argv[i]);
-		if(optn) opt |= optn;
-		else if(argv[i][0] == '-') {
-			if(0>(killsig = get_signum(argv[i]+1)))
-				killsig = SIGTERM;
-		}
-		else {
-			fni = xrealloc(fni, sizeof(int) * (fnic+2));
-			fni[fnic++] = i;
-		}
+	scan_proc_pids(); /* changes dir to "/proc" */
+
+	mypid = getpid();
+	plist = G.pid_list_head;
+	while (1) {
+		if (!plist)
+			return EXIT_FAILURE;
+		if (plist->pid != mypid)
+			break;
+		plist = plist->next;
 	}
-	if(!fnic) return 1;
 
-	inodes = xmalloc(sizeof(inode_list));
-	for(i=0;i<fnic;i++) {
-		if(fuser_parse_net_arg(argv[fni[i]], &proto, &port)) {
-			fuser_scan_proc_net(opt, proto, port, inodes);
-		}
-		else {
-			if(!fuser_file_to_dev_inode(
-				argv[fni[i]], &dev, &inode)) {
-				if (ENABLE_FEATURE_CLEAN_UP) free(inodes);
-				bb_perror_msg_and_die("cannot open '%s'", argv[fni[i]]);
+	exitcode = EXIT_SUCCESS;
+	do {
+		if (plist->pid != mypid) {
+			if (opt & OPT_KILL) {
+				if (kill(plist->pid, killsig) != 0) {
+					bb_perror_msg("kill pid %u", (unsigned)plist->pid);
+					exitcode = EXIT_FAILURE;
+				}
+			}
+			if (!(opt & OPT_SILENT)) {
+				printf("%u ", (unsigned)plist->pid);
 			}
-			fuser_add_inode(inodes, dev, inode);
-		}
-	}
-	pids = xmalloc(sizeof(pid_list));
-	success = fuser_scan_proc_pids(opt, inodes, pids);
-	/* if the first pid in the list is 0, none have been found */
-	if(pids->pid == 0) success = 0;
-	if(success) {
-		if(opt & FUSER_OPT_KILL) {
-			success = fuser_kill_pid_list(pids, killsig);
-		}
-		else if(!(opt & FUSER_OPT_SILENT)) {
-			success = fuser_print_pid_list(pids);
 		}
+		plist = plist->next;
+	} while (plist);
+
+	if (!(opt & (OPT_SILENT))) {
+		bb_putchar('\n');
 	}
-	free(pids);
-	free(inodes);
-	/* return 0 on (success == 1) 1 otherwise */
-	return (success != 1);
+
+	return exitcode;
 }