Compare commits
No commits in common. "697e67c40658fed7038d7fbb55a9a0cac084d5b1" and "30a9fa9c0cfdbbc99777a410e8eee0bffc77124c" have entirely different histories.
697e67c406
...
30a9fa9c0c
311
cpulimit-all.sh
311
cpulimit-all.sh
|
|
@ -1,311 +0,0 @@
|
||||||
#!/bin/sh
|
|
||||||
set -e
|
|
||||||
|
|
||||||
kill_child_jobs() {
|
|
||||||
# From https://stackoverflow.com/a/23336595
|
|
||||||
# Kills all child processes, not just jobs.
|
|
||||||
pkill -P $$
|
|
||||||
}
|
|
||||||
cleanup() {
|
|
||||||
kill_child_jobs
|
|
||||||
}
|
|
||||||
|
|
||||||
# From https://unix.stackexchange.com/a/240736
|
|
||||||
for sig in INT QUIT HUP TERM; do
|
|
||||||
trap "
|
|
||||||
cleanup
|
|
||||||
trap - $sig EXIT
|
|
||||||
kill -s $sig "'"$$"' "$sig"
|
|
||||||
done
|
|
||||||
trap cleanup EXIT
|
|
||||||
|
|
||||||
|
|
||||||
verbose=y
|
|
||||||
cpulimit_args="--lazy"
|
|
||||||
limit=""
|
|
||||||
pids=""
|
|
||||||
exes=""
|
|
||||||
paths=""
|
|
||||||
max_processes=100
|
|
||||||
max_depth=3
|
|
||||||
watch_interval=2
|
|
||||||
subprocess_watch_interval=0.5
|
|
||||||
while [ $# -gt 0 ]
|
|
||||||
do
|
|
||||||
opt="$1"
|
|
||||||
case "$opt" in
|
|
||||||
*=*)
|
|
||||||
arg="${1#*=}"
|
|
||||||
opt="${1%%=*}"
|
|
||||||
argshift=1
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
if [ "$#" -ge 2 ]
|
|
||||||
then
|
|
||||||
arg="$2"
|
|
||||||
else
|
|
||||||
arg=""
|
|
||||||
fi
|
|
||||||
argshift=2
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
|
|
||||||
case "$opt" in
|
|
||||||
--max-depth)
|
|
||||||
case $arg in
|
|
||||||
''|*[!0-9]*)
|
|
||||||
echo "Invalid max depth: $arg, must be a non-negative integer."
|
|
||||||
exit 5
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
max_depth=$arg
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
;;
|
|
||||||
--max-processes)
|
|
||||||
case $arg in
|
|
||||||
''|*[!0-9]*)
|
|
||||||
echo "Invalid max processes: $arg, must be a positive integer."
|
|
||||||
exit 5
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
max_processes=$arg
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
;;
|
|
||||||
--watch-interval)
|
|
||||||
watch_interval="$arg"
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
--subprocess-watch-interval)
|
|
||||||
subprocess_watch_interval="$arg"
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
-p|--pid)
|
|
||||||
if [ -z "$pids" ]
|
|
||||||
then
|
|
||||||
pids="$arg"
|
|
||||||
else
|
|
||||||
pids="$pids
|
|
||||||
$arg"
|
|
||||||
fi
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
-e|--exe)
|
|
||||||
if [ -z "$exes" ]
|
|
||||||
then
|
|
||||||
exes="$arg"
|
|
||||||
else
|
|
||||||
exes="$exes
|
|
||||||
$arg"
|
|
||||||
fi
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
-P|--path)
|
|
||||||
if [ -z "$paths" ]
|
|
||||||
then
|
|
||||||
paths="$arg"
|
|
||||||
else
|
|
||||||
paths="$paths
|
|
||||||
$arg"
|
|
||||||
fi
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
-l|--limit)
|
|
||||||
limit="$arg"
|
|
||||||
cpulimit_args="$cpulimit_args --limit=$arg"
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
-c|--cpu)
|
|
||||||
cpulimit_args="$cpulimit_args --cpu=$arg"
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
-v|--verbose)
|
|
||||||
cpulimit_args="$cpulimit_args --verbose"
|
|
||||||
shift
|
|
||||||
;;
|
|
||||||
-q|--quiet)
|
|
||||||
verbose=""
|
|
||||||
cpulimit_args="$cpulimit_args --quiet"
|
|
||||||
shift
|
|
||||||
;;
|
|
||||||
-k|--kill)
|
|
||||||
cpulimit_args="$cpulimit_args --kill"
|
|
||||||
shift
|
|
||||||
;;
|
|
||||||
-r|--restore)
|
|
||||||
cpulimit_args="$cpulimit_args --restore"
|
|
||||||
shift
|
|
||||||
;;
|
|
||||||
-s|--signal)
|
|
||||||
cpulimit_args="$cpulimit_args --signal=$arg"
|
|
||||||
shift $argshift
|
|
||||||
;;
|
|
||||||
--)
|
|
||||||
shift
|
|
||||||
break
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
echo "Programming error: unexpected argument \"$1\" (opt=\"$opt\", arg=\"$arg\")"
|
|
||||||
exit 3
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ -z "$limit" ]
|
|
||||||
then
|
|
||||||
echo "Must specify a CPU percentage to limit to (-l/--limit)."
|
|
||||||
exit 4
|
|
||||||
fi
|
|
||||||
if [ -z "$pids" ] && [ -z "$exes" ] && [ -z "$paths" ] && [ "$#" -eq 0 ]
|
|
||||||
then
|
|
||||||
echo "Must specify at least one PID, executable file, path, or command line to limit."
|
|
||||||
exit 5
|
|
||||||
fi
|
|
||||||
|
|
||||||
watched_pids=""
|
|
||||||
limit_pid() {
|
|
||||||
if echo "$watched_pids" | grep --silent -F "$1"
|
|
||||||
then
|
|
||||||
return
|
|
||||||
fi
|
|
||||||
if [ "$(echo "$watched_pids" | wc -l)" -ge "$max_processes" ]
|
|
||||||
then
|
|
||||||
return
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -n "$verbose" ]
|
|
||||||
then
|
|
||||||
if [ -n "$3" ]
|
|
||||||
then
|
|
||||||
echo "Limiting $3: "
|
|
||||||
fi
|
|
||||||
echo "cpulimit --pid=$1 $cpulimit_args"
|
|
||||||
fi
|
|
||||||
# shellcheck disable=SC2086 # $cpulimit_args really is the intended args.
|
|
||||||
cpulimit --pid="$1" $cpulimit_args &
|
|
||||||
cpulimit_pid=$!
|
|
||||||
new_watched="$1:$2:$cpulimit_pid"
|
|
||||||
if [ -z "$watched_pids" ]
|
|
||||||
then
|
|
||||||
watched_pids="$new_watched"
|
|
||||||
else
|
|
||||||
watched_pids="$watched_pids
|
|
||||||
$new_watched"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
limit_pids() {
|
|
||||||
pids="$1"
|
|
||||||
depth="$2"
|
|
||||||
while read -r pid
|
|
||||||
do
|
|
||||||
# From https://stackoverflow.com/a/3951175
|
|
||||||
case $pid in
|
|
||||||
''|*[!0-9]*)
|
|
||||||
# PID is not a number
|
|
||||||
;;
|
|
||||||
*)
|
|
||||||
limit_pid "$pid" "$depth" "$3"
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
done <<EOF
|
|
||||||
$pids
|
|
||||||
EOF
|
|
||||||
}
|
|
||||||
|
|
||||||
limit_by_executable() {
|
|
||||||
if [ -n "$exes" ]
|
|
||||||
then
|
|
||||||
while read -r exe
|
|
||||||
do
|
|
||||||
limit_pids "$(pgrep -x "$exe")" 0 "$exe"
|
|
||||||
done <<EOF
|
|
||||||
$exes
|
|
||||||
EOF
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -n "$paths" ]
|
|
||||||
then
|
|
||||||
while read -r path
|
|
||||||
do
|
|
||||||
limit_pids "$(pgrep -xf "$path")" 0 "$path"
|
|
||||||
done <<EOF
|
|
||||||
$paths
|
|
||||||
EOF
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
limit_by_subprocess() {
|
|
||||||
if [ -z "$watched_pids" ]
|
|
||||||
then
|
|
||||||
return
|
|
||||||
fi
|
|
||||||
|
|
||||||
while read -r watched
|
|
||||||
do
|
|
||||||
depth="$(echo "$watched" | cut -d: -f2)"
|
|
||||||
if [ "$max_depth" -gt "$depth" ]
|
|
||||||
then
|
|
||||||
# Make sure the parent is still alive.
|
|
||||||
if ps -p "$(echo "$watched" | cut -d: -f3)" >/dev/null
|
|
||||||
then
|
|
||||||
ppid="$(echo "$watched" | cut -d: -f1)"
|
|
||||||
limit_pids "$(pgrep -P "$ppid")" "$((depth + 1))" "Child of $ppid"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
done <<EOF
|
|
||||||
$watched_pids
|
|
||||||
EOF
|
|
||||||
}
|
|
||||||
|
|
||||||
clean_dead_cpulimit() {
|
|
||||||
if [ -z "$watched_pids" ]
|
|
||||||
then
|
|
||||||
return
|
|
||||||
fi
|
|
||||||
|
|
||||||
tmp="$(echo "$watched_pids" | while read -r watched
|
|
||||||
do
|
|
||||||
if ps -p "$(echo "$watched" | cut -d: -f3)" >/dev/null
|
|
||||||
then
|
|
||||||
echo "$watched"
|
|
||||||
fi
|
|
||||||
done)"
|
|
||||||
watched_pids="$tmp"
|
|
||||||
}
|
|
||||||
|
|
||||||
if [ "$#" -gt 0 ]
|
|
||||||
then
|
|
||||||
"$@" &
|
|
||||||
limit_pid "$!" 0 "program run on command line: $*"
|
|
||||||
fi
|
|
||||||
|
|
||||||
limit_pids "$pids" 0
|
|
||||||
while true
|
|
||||||
do
|
|
||||||
clean_dead_cpulimit
|
|
||||||
if [ -z "$exes" ] && [ -z "$paths" ] && [ -z "$watched_pids" ]
|
|
||||||
then
|
|
||||||
# If there's nothing left to wait for, then exit.
|
|
||||||
exit
|
|
||||||
fi
|
|
||||||
limit_by_executable
|
|
||||||
num_watched_before="$(echo "$watched_pids" | wc -l)"
|
|
||||||
limit_by_subprocess
|
|
||||||
num_watched_after="$(echo "$watched_pids" | wc -l)"
|
|
||||||
if [ "$num_watched_before" -eq "$num_watched_after" ]
|
|
||||||
then
|
|
||||||
if [ "$watch_interval" = "0" ]
|
|
||||||
then
|
|
||||||
wait
|
|
||||||
else
|
|
||||||
sleep "$watch_interval"
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
sleep "$subprocess_watch_interval"
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
Loading…
Reference in New Issue
Block a user