From 7c6209d6c336df74984e62aa88679c06641ae309 Mon Sep 17 00:00:00 2001
From: Elie De Brauwer <eliedebrauwer@gmail.com>
Date: Sat, 8 Dec 2012 20:10:05 +0100
Subject: Adding -s (single shot) and -o (omit pids) options to pidof

---
 lib/lib.c          |  7 ++++---
 lib/lib.h          |  2 +-
 toys/lsb/killall.c |  3 ++-
 toys/lsb/pidof.c   | 34 +++++++++++++++++++++++++++++++---
 4 files changed, 38 insertions(+), 8 deletions(-)

diff --git a/lib/lib.c b/lib/lib.c
index f452bf3c..148a7652 100644
--- a/lib/lib.c
+++ b/lib/lib.c
@@ -975,7 +975,7 @@ int yesno(char *prompt, int def)
 }
 
 // Execute a callback for each PID that matches a process name from a list.
-void for_each_pid_with_name_in(char **names, void (*callback)(pid_t pid))
+void for_each_pid_with_name_in(char **names, int (*callback)(pid_t pid))
 {
   DIR *dp;
   struct dirent *entry;
@@ -998,9 +998,10 @@ void for_each_pid_with_name_in(char **names, void (*callback)(pid_t pid))
     if (n<1) continue;
 
     for (curname = names; *curname; curname++)
-      if (!strcmp(basename(cmd), *curname)) callback(atol(entry->d_name));
+      if (!strcmp(basename(cmd), *curname)) 
+          if (!callback(atol(entry->d_name))) goto done;
   }
-
+done:
   closedir(dp);
 }
 
diff --git a/lib/lib.h b/lib/lib.h
index 1c0350e0..7cc4d5ca 100644
--- a/lib/lib.h
+++ b/lib/lib.h
@@ -144,7 +144,7 @@ void replace_tempfile(int fdin, int fdout, char **tempname);
 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, void (*callback)(pid_t pid));
+void for_each_pid_with_name_in(char **names, int (*callback)(pid_t pid));
 unsigned long xstrtoul(const char *nptr, char **endptr, int base);
 
 // getmountlist.c
diff --git a/toys/lsb/killall.c b/toys/lsb/killall.c
index debc3dfc..2294f61d 100644
--- a/toys/lsb/killall.c
+++ b/toys/lsb/killall.c
@@ -25,7 +25,7 @@ GLOBALS(
   int signum;
 )
 
-static void kill_process(pid_t pid)
+static int kill_process(pid_t pid)
 {
   int ret;
 
@@ -33,6 +33,7 @@ static void kill_process(pid_t pid)
   ret = kill(pid, TT.signum);
 
   if (ret == -1 && !(toys.optflags & FLAG_q)) perror("kill");
+  return 1;
 }
 
 void killall_main(void)
diff --git a/toys/lsb/pidof.c b/toys/lsb/pidof.c
index 03643015..8aeb33e4 100644
--- a/toys/lsb/pidof.c
+++ b/toys/lsb/pidof.c
@@ -1,26 +1,54 @@
 /* pidof.c - Print the Process IDs of all processes with the given names.
  *
  * Copyright 2012 Andreas Heck <aheck@gmx.de>
+ * Copyright 2012 Elie De Brauwer <eliedebrauwer@gmail.com>
  *
  * http://refspecs.linuxfoundation.org/LSB_4.1.0/LSB-Core-generic/LSB-Core-generic/pidof.html
 
-USE_PIDOF(NEWTOY(pidof, "<1", TOYFLAG_USR|TOYFLAG_BIN))
+USE_PIDOF(NEWTOY(pidof, "so:<1", TOYFLAG_USR|TOYFLAG_BIN))
 
 config PIDOF
   bool "pidof"
   default y
   help
-    usage: pidof [NAME]...
+    usage: pidof [-s] [-o omitpid[,omitpid..]] [NAME]...
 
     Print the PIDs of all processes with the given names.
+    -s	single shot, only return one pid.
+    -o	omits processes with specified PID
 */
 
+#define FOR_pidof
 #include "toys.h"
 
-static void print_pid(pid_t pid)
+GLOBALS(
+  char *omit;
+)
+
+static int print_pid(pid_t pid)
 {
+
+  if (toys.optflags & FLAG_o)
+  {
+      char * res;
+      int len;
+      snprintf(toybuf, sizeof(toybuf), "%d", pid);
+      len = strlen(toybuf);
+      res = strstr(TT.omit, toybuf);
+      if (res &&
+          (res == TT.omit || res[-1] == ',') &&
+          (res[len] == ',' || res[len] == 0))
+          // Found in omit string
+          return 1;
+  }
+
   xprintf("%s%ld", toys.exitval ? "" : " ", (long)pid);
   toys.exitval = 0;
+
+  if (toys.optflags & FLAG_s)
+      return 0;
+
+  return 1;
 }
 
 void pidof_main(void)
-- 
cgit v1.2.3