48 lines
1.2 KiB
Plaintext
48 lines
1.2 KiB
Plaintext
_taskset_module()
|
|
{
|
|
local cur prev OPTS
|
|
COMPREPLY=()
|
|
cur="${COMP_WORDS[COMP_CWORD]}"
|
|
prev="${COMP_WORDS[COMP_CWORD-1]}"
|
|
case $prev in
|
|
'-c'|'--cpu-list')
|
|
local prefix realcur CPULIST_ALL CPULIST
|
|
realcur="${cur##*,}"
|
|
prefix="${cur%$realcur}"
|
|
CPULIST_ALL=$(sed 's/^/{/; s/-/../g; s/,/} {/g; s/$/}/' /sys/devices/system/cpu/online)
|
|
for WORD in $(eval echo $CPULIST_ALL); do
|
|
if ! [[ $prefix == *"$WORD"* ]]; then
|
|
CPULIST="$WORD ${CPULIST:-""}"
|
|
fi
|
|
done
|
|
compopt -o nospace
|
|
COMPREPLY=( $(compgen -P "$prefix" -W "$CPULIST" -S ',' -- $realcur) )
|
|
return 0
|
|
;;
|
|
'-p'|'--pid')
|
|
local PIDS
|
|
# FIXME: the pid argument is ambiguous. When
|
|
# setting an affinity the optarg has to be cpu
|
|
# mask. The following is good only for getting
|
|
# affinity.
|
|
PIDS=$(cd /proc && echo [0-9]*)
|
|
COMPREPLY=( $(compgen -W "$PIDS" -- $cur) )
|
|
return 0
|
|
;;
|
|
'-h'|'--help'|'-V'|'--version')
|
|
return 0
|
|
;;
|
|
esac
|
|
case $cur in
|
|
-*)
|
|
OPTS="--all-tasks --pid --cpu-list --help --version"
|
|
COMPREPLY=( $(compgen -W "${OPTS[*]}" -- $cur) )
|
|
return 0
|
|
;;
|
|
esac
|
|
compopt -o bashdefault
|
|
COMPREPLY=( $(compgen -c -- $cur) )
|
|
return 0
|
|
}
|
|
complete -F _taskset_module taskset
|