4

Как мне реализовать скрипт pstrace в bash, который изменяет интерфейс

[sudo] strace -c -p [PID]

в

[sudo] pstrace -c -p [PROCESS-NAME]

похоже на то, как

killall [PROCESS-NAME]

используется. С завершением и всем.

2 ответа2

2

Твой это:

ps auxw | grep [PROCESS-NAME] | awk '{print"-p " $2}' | xargs strace
1

Обманчиво сложные требования :-)

В двух частях, во-первых, сценарий оболочки pstrace для strace , он использует pgrep для операции name-to-PID.

#!/bin/bash

IFS=$' \t\n'

# process the arguments to find "-p procname", only support one instance though
for ((nn=1; nn<=$#; nn++)); do
    if [ "${!nn}" = "-p" ]; then
        :
    elif [ "$prev" = "-p" ]; then
        pname="${!nn}"
    else
        args+=( "${!nn}" )    # just copy 
    fi
    prev="${!nn}"
done

pids=()
if [ -n "$pname" ]; then
    # skip this shell's PID, which pgrep -f will match
    # note the use of exec to avoid picking up a matching subshell too
    # uncomment  && printf for pid/pname list
    while read pp pname; do 
         [ "$pp" != "$$" ] && pids+=($pp) # && printf "%6i %s\n" "$pp" "$pname"
    done < <(exec pgrep -l -f "${pname}")
fi

npids=${#pids[*]}

if [ $npids -eq 0 ]; then
    echo "No PIDs to trace."; exit 2
elif [ $npids -eq 1 ]; then
    args=( "${args[@]}" -p ${pids[0]} )
elif [ $npids -le 32 ]; then
    read -p "$npids PIDS found, enter Y to proceed: " yy
    [ "$yy" != "Y" ] && echo "Cancelled..." && exit 1
    args=( "${args[@]}" ${pids[@]/#/-p } ) 
else 
    echo "Too many PIDs to trace: $npids (max 32)."; exit 2
fi 

strace "${args[@]}"

Во второй части я буду использовать программируемое завершение bash для завершения процессов по имени, поместите это в ваш ~/.bash_profile или аналогичный:

# process-name patterns to ignore
PROCIGNORE=( "^\[", "^-bash" )

_c8n_listprocs ()
{
    local cur prv ignore IFS nn mm
    prv=${COMP_WORDS[COMP_CWORD-1]}
    cur=${COMP_WORDS[COMP_CWORD]}

    case "$prv" in
        '-p') 
              IFS=$'\n' COMPREPLY=( $(ps axwwo "args") ) IFS=$' \t\n'
              COMPREPLY=(${COMPREPLY[*]// */})  # remove arguments

              ignore="0" # ps header
              for ((nn=1; nn<${#COMPREPLY[*]}; nn++)); do
                  # filter by (partially) typed name in cur
                  # use  " =~ ^$cur " for prefix match, without ^ it's substr match
                  [[ -n "$cur"  &&  ! "${COMPREPLY[$nn]}" =~ $cur ]] && {
                      ignore="$nn $ignore"
                  } || { 
                      # skip names matching PROCIGNORE[]
                      for ((mm=0; mm<${#PROCIGNORE[*]}; mm++)); do
                          [[ "${COMPREPLY[$nn]}" =~ ${PROCIGNORE[$mm]} ]] && 
                              ignore="$nn $ignore"
                      done
                  }
              done
              # remove unwanted, in reverse index order
              for nn in $ignore; do unset COMPREPLY[$nn]; done

              ;;
        *)    COMPREPLY=()
              ;;
    esac
}
complete -F _c8n_listprocs pstrace

Протестировано и используется на Linux с bash-3.x и bash-4.x. ps могут нуждаться в настройке на платформах, отличных от Linux, также должны поддерживать truss с изменением в одну строку.

Ограничения включают в себя:

  • нет правильного экранирования потока, подобного процессу ядра [names] , это приведет к тому, что pgrep (скорее всего) не сделает то, что вы хотите
  • несоответствие процессов с пробелом в имени (вместо « args » используется « comm », чтобы можно было использовать /paths , где это возможно)

Всё ещё ищете ответ? Посмотрите другие вопросы с метками .