| _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=$(for I in /proc/[0-9]*; do echo ${I##"/proc/"}; done) |
| 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 |