find: improve usage text (Natanael Copa <natanael.copa@gmail.com>)
[oweals/busybox.git] / procps / kill.c
index b7a2985c88147ebbecbe1b009c5422dd107d3ca9..9a6e936655fdb09e26e824d0155fea159cad4a91 100644 (file)
 
 #include "busybox.h"
 
+int kill_main(int argc, char **argv);
 int kill_main(int argc, char **argv)
 {
-       int killall, signo = SIGTERM, errors = 0, quiet=0;
-
-       killall = (ENABLE_KILLALL && bb_applet_name[4]=='a') ? 1 : 0;
+       char *arg;
+       pid_t pid;
+       int signo = SIGTERM, errors = 0, quiet = 0;
+       const int killall = (ENABLE_KILLALL && applet_name[4] == 'a'
+                      && (!ENABLE_KILLALL5 || applet_name[7] != '5'));
+       const int killall5 = (ENABLE_KILLALL5 && applet_name[4] == 'a'
+                         && (!ENABLE_KILLALL || applet_name[7] == '5'));
 
        /* Parse any options */
-       if (argc < 2)
-               bb_show_usage();
+       argc--;
+       arg = *++argv;
 
-       if(argv[1][0] != '-'){
-               argv++;
-               argc--;
+       if (argc < 1 || arg[0] != '-') {
                goto do_it_now;
        }
 
        /* The -l option, which prints out signal names. */
-       if(argv[1][1]=='l' && argv[1][2]=='\0'){
-               if(argc==2) {
+       if (arg[1] == 'l' && arg[2] == '\0') {
+               const char *name;
+               if (argc == 1) {
                        /* Print the whole signal list */
                        int col = 0;
-
-                       for(signo = 0;;) {
-                               char *name = get_signame(++signo);
-                               if (isdigit(*name)) break;
-
-                               if (col > 60) {
-                                       printf("\n");
+                       for (signo = 1; signo < 32; signo++) {
+                               name = get_signame(signo);
+                               if (isdigit(name[0])) continue;
+                               if (col > 66) {
+                                       puts("");
                                        col = 0;
                                }
-                               col += printf("%2d) %-16s", signo, name);
+                               col += printf("%2d) %-6s", signo, name);
                        }
-                       printf("\n");
-               } else {
-                       for(argv++; *argv; argv++) {
-                               char *name;
-
-                               if (isdigit(**argv)) name = get_signame(atoi(*argv));
-                               else {
-                                       int temp = get_signum(*argv);
-                                       if (temp<0)
-                                               bb_error_msg_and_die("unknown signal %s", *argv);
-                                       name = get_signame(temp);
+                       puts("");
+               } else { /* -l <sig list> */
+                       while ((arg = *++argv)) {
+                               if (isdigit(arg[0])) {
+                                       signo = xatoi_u(arg);
+                                       name = get_signame(signo);
+                               else {
+                                       signo = get_signum(arg);
+                                       if (signo < 0)
+                                               bb_error_msg_and_die("unknown signal '%s'", arg);
+                                       name = get_signame(signo);
                                }
-                               puts(name);
+                               printf("%2d) %s\n", signo, name);
                        }
                }
-               /* If they specified -l, were all done */
+               /* If they specified -l, we are all done */
                return EXIT_SUCCESS;
        }
 
        /* The -q quiet option */
-       if(killall && argv[1][1]=='q' && argv[1][2]=='\0'){
-               quiet++;
-               argv++;
+       if (killall && arg[1] == 'q' && arg[2] == '\0') {
+               quiet = 1;
+               arg = *++argv;
                argc--;
-               if(argc<2 || argv[1][0] != '-'){
-                       goto do_it_now;
-               }
+               if (argc < 1) bb_show_usage();
+               if (arg[0] != '-') goto do_it_now;
        }
 
-       if(0>(signo = get_signum(argv[1]+1)))
-               bb_error_msg_and_die( "bad signal name '%s'", argv[1]+1);
-       argv+=2;
-       argc-=2;
+       /* -SIG */
+       signo = get_signum(&arg[1]);
+       if (signo < 0)
+               bb_error_msg_and_die("bad signal name '%s'", &arg[1]);
+       arg = *++argv;
+       argc--;
 
 do_it_now:
 
-       /* Pid or name required */
-       if (argc <= 0)
-               bb_show_usage();
-
-       if (!killall) {
-               /* Looks like they want to do a kill. Do that */
-               while (--argc >= 0) {
-                       int pid;
-
-                       if (!isdigit(**argv) && **argv != '-')
-                               bb_error_msg_and_die( "Bad PID '%s'", *argv);
-                       pid = strtol(*argv, NULL, 0);
-                       if (kill(pid, signo) != 0) {
-                               bb_perror_msg( "Could not kill pid '%d'", pid);
-                               errors++;
-                       }
-                       argv++;
+       if (killall5) {
+               pid_t sid;
+               procps_status_t* p = NULL;
+
+// Cannot happen anyway? We don't TERM ourself, we STOP
+//             /* kill(-1, sig) on Linux (at least 2.1.x)
+//              * might send signal to the calling process too */
+//             signal(SIGTERM, SIG_IGN);
+               /* Now stop all processes */
+               kill(-1, SIGSTOP);
+               /* Find out our own session id */
+               pid = getpid();
+               sid = getsid(pid);
+               /* Now kill all processes except our session */
+               while ((p = procps_scan(p, PSSCAN_PID|PSSCAN_SID))) {
+                       if (p->sid != sid && p->pid != pid && p->pid != 1)
+                               kill(p->pid, signo);
                }
+               /* And let them continue */
+               kill(-1, SIGCONT);
+               return 0;
+       }
 
-       } else {
-               pid_t myPid=getpid();
+       /* Pid or name required for kill/killall */
+       if (argc < 1)
+               bb_show_usage();
 
+       if (killall) {
                /* Looks like they want to do a killall.  Do that */
-               while (--argc >= 0) {
-                       long* pidList;
+               pid = getpid();
+               while (arg) {
+                       pid_t* pidList;
 
-                       pidList = find_pid_by_name(*argv);
-                       if (!pidList || *pidList<=0) {
+                       pidList = find_pid_by_name(arg);
+                       if (*pidList == 0) {
                                errors++;
-                               if (quiet==0)
-                                       bb_error_msg( "%s: no process killed", *argv);
+                               if (!quiet)
+                                       bb_error_msg("%s: no process killed", arg);
                        } else {
-                               long *pl;
+                               pid_t *pl;
 
-                               for(pl = pidList; *pl !=0 ; pl++) {
-                                       if (*pl==myPid)
+                               for (pl = pidList; *pl; pl++) {
+                                       if (*pl == pid)
+                                               continue;
+                                       if (kill(*pl, signo) == 0)
                                                continue;
-                                       if (kill(*pl, signo) != 0) {
-                                               errors++;
-                                               if (quiet==0)
-                                                       bb_perror_msg( "Could not kill pid '%ld'", *pl);
-                                       }
+                                       errors++;
+                                       if (!quiet)
+                                               bb_perror_msg("cannot kill pid %u", (unsigned)*pl);
                                }
                        }
                        free(pidList);
-                       argv++;
+                       arg = *++argv;
                }
+               return errors;
        }
 
+       /* Looks like they want to do a kill. Do that */
+       while (arg) {
+               /* Huh?
+               if (!isdigit(arg[0]) && arg[0] != '-')
+                       bb_error_msg_and_die("bad pid '%s'", arg);
+               */
+               pid = xatou(arg);
+               /* FIXME: better overflow check? */
+               if (kill(pid, signo) != 0) {
+                       bb_perror_msg("cannot kill pid %u", (unsigned)pid);
+                       errors++;
+               }
+               arg = *++argv;
+       }
        return errors;
 }