user-tools/killp

   1 #!/bin/bash
   2
   3 true <<'EOF'
   4
   5 =pod
   6
   7 =head1 NAME
   8
   9 killp - Send signal to processes (kill, terminate, ...) by PID until they end
  10
  11 killpgrp - Send signal to processes (kill, terminate, ...) by PGID until they end
  12
  13 killcmd - Send signal to processes (kill, terminate, ...) by command line until they end
  14
  15 killexe - Send signal to processes (kill, terminate, ...) by executable path until they end
  16
  17 =head1 SYNOPSIS
  18
  19 killp [OPTIONS] <PID> [<PID> [...]]
  20
  21 =head1 DESCRIPTION
  22
  23 Send signal to process(es) by PID, PGID (process group ID), command name, or by executable path
  24 until the selected process(es) exists.
  25 Ie. in usuall invocation, eg. C<killcmd java> tries to SIGTERM all java processes until at
  26 least 1 exists, and returns only afterwards.
  27
  28 =head1 OPTIONS
  29
  30 The following options control how B<killcmd> and B<killexe> finds processes.
  31 Semantics are the same as in grep(1):
  32
  33   -E --extended-regexp
  34   -F --fixed-strings
  35   -G --basic-regexp
  36   -P --perl-regexp
  37   -i --ignore-case
  38   -w --word-regexp
  39   -x --line-regexp
  40
  41 Other options:
  42
  43 =over 4
  44
  45 =item -a
  46
  47 B<killcmd> looks for matching substring in the command's arguments too.
  48 By default, only the command name is considered (first word in the command line).
  49
  50 =item -f
  51
  52 B<killcmd> and B<killexe> look for matching substring in the command's full path too.
  53 By default, only the basename is considered.
  54
  55 =item [--]signal=I<SIG>, [-]s=I<SIG>
  56
  57 Which signal to send.
  58 See kill(1) and signal(7) for valid I<SIG> signal names and numbers.
  59
  60 =item [--]interval=I<IVAL>
  61
  62 How much to wait between attempts.
  63 See sleep(1) for valid I<IVAL> intervals.
  64
  65 =item -q, --quiet
  66
  67 =item -v, --verbose
  68
  69 By default, prints what is being killed on the second attempt onward.
  70 With --verbose, prints the first attempt too.
  71 With --quiet, does not print what is being killed.
  72
  73 =item -n, --dryrun
  74
  75 =back
  76
  77 =head1 SEE ALSO
  78
  79 kill(1), pkill(1), pgrep(1), killall(1), signal(7)
  80
  81 =cut
  82
  83 EOF
  84
  85
  86
  87 set -e
  88 set -o pipefail
  89
  90 . /usr/lib/tool/bash-utils
  91
  92
  93
  94 case "${0##*/}" in
  95 killp)
  96         mode=pid
  97         ;;
  98 killpgrp)
  99         mode=pgid
 100         ;;
 101 killcmd)
 102         mode=cmd
 103         ;;
 104 killexe)
 105         mode=exe
 106         ;;
 107 *)
 108         errx -2 "Call by either of these names: killp, killpgrp, killcmd, killexe"
 109         ;;
 110 esac
 111
 112
 113 opt_verbose=''
 114 opt_quiet=''
 115 opt_dryrun=''
 116 opt_match_command_args=''
 117 opt_match_full_path=''
 118 Signal=TERM
 119 Interval=0.2
 120 declare -A processes_left
 121 declare -a grep_opts=()
 122 declare -a args=()
 123
 124 # explode "-abc" arguments into separate "-a -b -c" pieces.
 125 for arg in "$@"
 126 do
 127         case "$arg" in
 128         -[!-]*)
 129                 pos=1
 130                 while [ -n "${arg:$pos:1}" ]
 131                 do
 132                         args+=("-${arg:$pos:1}")
 133                         let pos++
 134                 done
 135                 ;;
 136         *)
 137                 args+=("$arg")
 138                 ;;
 139         esac
 140 done
 141
 142 for arg in "${args[@]}"
 143 do
 144         case "$arg" in
 145         signal=*|sig=*|s=*|--signal=*|--sig=*|-s=*)
 146                 Signal=${arg#*=}
 147                 ;;
 148         interval=*|--interval=*)
 149                 Interval=${arg#*=}
 150                 ;;
 151         --help)
 152                 pod2text "$0"
 153                 exit 0
 154                 ;;
 155         -E|--extended-regexp|-F|--fixed-strings|-G|--basic-regexp|-P|--perl-regexp|-i|--ignore-case|-w|--word-regexp|-x|--line-regexp)
 156                 grep_opts+=("$arg")
 157                 ;;
 158         -a)
 159                 if [ "$mode" != cmd ]
 160                 then
 161                         pod2text "$0" >&2
 162                         exit 2
 163                 fi
 164                 opt_match_command_args=1
 165                 ;;
 166         -f)
 167                 if [ "$mode" != cmd -a "$mode" != exe ]
 168                 then
 169                         pod2text "$0" >&2
 170                         exit 2
 171                 fi
 172                 opt_match_full_path=1
 173                 ;;
 174         -q|--quiet)
 175                 opt_quiet=1
 176                 ;;
 177         -v|--verbose)
 178                 opt_verbose=1
 179                 ;;
 180         -n|--dryrun|--dry-run)
 181                 opt_dryrun=1
 182                 ;;
 183         -*|*=*)
 184                 pod2text "$0" >&2
 185                 exit 2
 186                 ;;
 187         *)
 188                 processes_left[$arg]=''
 189                 ;;
 190         esac
 191 done
 192
 193
 194 find_processes()
 195 {
 196         # fill global $processes and $ps_title arrays
 197
 198         declare -g -a processes
 199
 200         case "$mode" in
 201         cmd)
 202                 proclist=`ps -e o pid:1=,cmd= ww`
 203                 ;;
 204         exe)
 205                 proclist=`rcmod any=0 find /proc -mindepth 2 -maxdepth 2 -name exe -printf '%h %l\n' 2>/dev/null | cut -b7-`
 206                 ;;
 207         esac
 208
 209         proclist_to_search=$proclist
 210
 211         if [ "$mode" = cmd -a ! "$opt_match_command_args" ]
 212         then
 213                 # discard the command arguments (keep the pid and the first word)
 214                 proclist_to_search=`cut -f 1,2 -d ' ' <<< "$proclist_to_search"`
 215         fi
 216         if [ \( "$mode" = cmd -o "$mode" = exe \) -a ! "$opt_match_full_path" ]
 217         then
 218                 # discard everything except pid (everything before space) and the basename (everything after the last slash)
 219                 proclist_to_search=`sed -e 's, .*/\([^/]\+\)$, \1,' <<< "$proclist_to_search"`
 220         fi
 221
 222
 223         case "$mode" in
 224         pid|pgid)
 225                 processes=("${!processes_left[@]}")
 226                 ;;
 227         cmd|exe)
 228                 grep_params=()
 229                 for cmd in "${!processes_left[@]}"
 230                 do
 231                         grep_params+=(-e "$cmd")
 232                 done
 233                 lines=`echo "$proclist_to_search" | cut -d' ' -f2- | rcmod 1=0 grep --line-number "${grep_opts[@]}" "${grep_params[@]}" | cut -d: -f1`
 234
 235                 declare -g -A ps_title=()
 236                 while read -r pid title
 237                 do
 238                         if [ $pid = $$ -o $pid = "$PPID" ]; then continue; fi  # except ourself and parent
 239                         ps_title[$pid]=$title
 240                 done < <(echo "$proclist" | lines $lines)
 241
 242                 processes=("${!ps_title[@]}")
 243                 ;;
 244         esac
 245 }
 246
 247
 248
 249 # don't use bash built-in kill command
 250 enable -n kill
 251
 252
 253 attempt=0
 254
 255 while [ ${#processes_left[@]} -gt 0 ]
 256 do
 257         if [ $attempt -gt 0 -a \( $mode = cmd -o $mode = exe \) ]
 258         then
 259                 # leave some time for the processes to exit.
 260                 # it's usually faster to wait here a bit,
 261                 # than rush to find processes, then wait a longer time if the processes
 262                 # were still there which we killed.
 263                 sleep 0.1
 264         fi
 265
 266         find_processes
 267
 268         if [ ${#processes[@]} = 0 ]
 269         then
 270                 # did not find any more of the named commands
 271                 if [ $attempt = 0 ]
 272                 then
 273                         warnx "no such process found"
 274                 fi
 275                 break
 276         fi
 277
 278
 279         if [ $attempt -gt 0 ]
 280         then
 281                 sleep $Interval
 282
 283                 find_processes
 284
 285                 if [ ${#processes[@]} = 0 ]
 286                 then
 287                         # did not find more of the named commands
 288                         break
 289                 fi
 290         fi
 291
 292
 293         for proc in "${processes[@]}"
 294         do
 295                 case "$mode" in
 296                 pid|cmd|exe)
 297                         id=$proc
 298                         what="process"
 299                         ;;
 300                 pgid)
 301                         id=-$proc
 302                         what="process group"
 303                         ;;
 304                 esac
 305
 306                 if kill -- -0 "$id"
 307                 then
 308                         if [ ! $opt_quiet ]
 309                         then
 310                                 if [ "$opt_verbose" -o $attempt -gt 0 ]
 311                                 then
 312                                         msg_end=''
 313                                         case "$mode" in
 314                                         cmd|exe)
 315                                                 msg_end=": ${ps_title[$proc]}"
 316                                                 ;;
 317                                         esac
 318                                         warnx "attempt $attempt: $id$msg_end"
 319                                 fi
 320                         fi
 321
 322                         if [ ! $opt_dryrun ]
 323                         then
 324                                 kill -- -$Signal "$id" || true
 325                         fi
 326                 else
 327                         case "$mode" in
 328                         pid|pgid)
 329                                 warnx "$what $proc terminated"
 330                                 unset processes_left[$proc]
 331                                 ;;
 332                         cmd|exe)
 333                                 warnx "$what $proc terminated: ${ps_title[$proc]}"
 334                                 ;;
 335                         esac
 336                 fi
 337         done
 338
 339
 340         attempt=$[attempt + 1]
 341 done