changeset 762:f169d9708518

Extend killall with support for -v and -i
author Elie De Brauwer <eliedebrauwer@gmail.com>
date Sun, 16 Dec 2012 13:43:36 +0100
parents 6a558bf5de65
children d8e2d723ab07
files lib/lib.c lib/lib.h toys/lsb/killall.c toys/lsb/pidof.c
diffstat 4 files changed, 19 insertions(+), 8 deletions(-) [+]
line wrap: on
line diff
--- a/lib/lib.c	Sun Dec 23 01:25:27 2012 -0600
+++ b/lib/lib.c	Sun Dec 16 13:43:36 2012 +0100
@@ -158,7 +158,7 @@
 
 void xputs(char *s)
 {
-  if (EOF == puts(s)) perror_exit("write");
+  if (EOF == puts(s) || fflush(stdout)) perror_exit("write");
 }
 
 void xputc(char c)
@@ -975,7 +975,7 @@
 }
 
 // Execute a callback for each PID that matches a process name from a list.
-void for_each_pid_with_name_in(char **names, int (*callback)(pid_t pid))
+void for_each_pid_with_name_in(char **names, int (*callback)(pid_t pid, char *name))
 {
   DIR *dp;
   struct dirent *entry;
@@ -999,7 +999,7 @@
 
     for (curname = names; *curname; curname++)
       if (!strcmp(basename(cmd), *curname)) 
-          if (!callback(atol(entry->d_name))) goto done;
+          if (!callback(atol(entry->d_name), *curname)) goto done;
   }
 done:
   closedir(dp);
--- a/lib/lib.h	Sun Dec 23 01:25:27 2012 -0600
+++ b/lib/lib.h	Sun Dec 16 13:43:36 2012 +0100
@@ -144,7 +144,7 @@
 void crc_init(unsigned int *crc_table, int little_endian);
 void terminal_size(unsigned *x, unsigned *y);
 int yesno(char *prompt, int def);
-void for_each_pid_with_name_in(char **names, int (*callback)(pid_t pid));
+void for_each_pid_with_name_in(char **names, int (*callback)(pid_t pid, char *name));
 unsigned long xstrtoul(const char *nptr, char **endptr, int base);
 
 // getmountlist.c
--- a/toys/lsb/killall.c	Sun Dec 23 01:25:27 2012 -0600
+++ b/toys/lsb/killall.c	Sun Dec 16 13:43:36 2012 +0100
@@ -1,20 +1,23 @@
+
 /* killall.c - Send signal (default: TERM) to all processes with given names.
  *
  * Copyright 2012 Andreas Heck <aheck@gmx.de>
  *
  * http://refspecs.linuxfoundation.org/LSB_4.1.0/LSB-Core-generic/LSB-Core-generic/killall.html
 
-USE_KILLALL(NEWTOY(killall, "<1?lq", TOYFLAG_USR|TOYFLAG_BIN))
+USE_KILLALL(NEWTOY(killall, "<1?lqvi", TOYFLAG_USR|TOYFLAG_BIN))
 
 config KILLALL
   bool "killall"
   default y
   help
-    usage: killall [-l] [-q] [-SIG] PROCESS_NAME...
+    usage: killall [-l] [-qv] [-SIG] PROCESS_NAME...
 
     Send a signal (default: TERM) to all processes with the given names.
 
     -l	print list of all available signals
+    -i	ask for confirmation before killing
+    -v	report if the signal was successfully sent
     -q	don't print any warnings or error messages
 */
 
@@ -25,12 +28,20 @@
   int signum;
 )
 
-static int kill_process(pid_t pid)
+static int kill_process(pid_t pid, char *name)
 {
   int ret;
 
+  if(toys.optflags & FLAG_i) {
+    snprintf(toybuf, sizeof(toybuf), "Signal %s(%d) ?", name, pid);
+    if (yesno(toybuf, 0) == 0) return 1;
+  }
+
   toys.exitval = 0;
+
   ret = kill(pid, TT.signum);
+  if (toys.optflags & FLAG_v)
+    printf("Killed %s(%d) with signal %d\n", name, pid, TT.signum);
 
   if (ret == -1 && !(toys.optflags & FLAG_q)) perror("kill");
   return 1;
--- a/toys/lsb/pidof.c	Sun Dec 23 01:25:27 2012 -0600
+++ b/toys/lsb/pidof.c	Sun Dec 16 13:43:36 2012 +0100
@@ -25,7 +25,7 @@
   char *omit;
 )
 
-static int print_pid(pid_t pid)
+static int print_pid(pid_t pid, char * name)
 {
   char * res;
   int len;