#!/bin/sh

NAME=$1
ExactSearches=$2
KilAll=$3
ForceKillProcess=$4
ProcessLive=$5


#判斷參數(shù)是否為空，空則提示其使用方法
if [ -z "$NAME" ];then
    echo "Usage:"
    echo "示例：./killprocess.sh QQ N Y Y N"
    echo "第一個參數(shù)為：進程名稱，注意嚴格區(qū)分大小寫"
    echo "第二個參數(shù)為：精確搜索（Y）/模糊搜索（N）,默認精確搜索"
    echo "第三個參數(shù)為：結(jié)束所有匹配到的進程（Y）/結(jié)束匹配到的第一個進程（N）,默認全部結(jié)束"
    echo "第四個參數(shù)為：強制結(jié)束進程（Y）/非強制結(jié)束進程（N）,默認強制結(jié)束進程"
    echo "第五個參數(shù)為：顯示匹配到進程存活情況（Y）/不顯示匹配到進程存活情況（N）,默認顯示存活情況"
    echo "注意：腳本結(jié)束會生成processResult.txt顯示詳情"
exit
fi

case "$1" in
-h|--help)
    echo "Usage:"
    echo "示例：./killprocess.sh QQ N Y Y N"
    echo "第一個參數(shù)為：進程名稱，注意嚴格區(qū)分大小寫"
    echo "第二個參數(shù)為：精確搜索（Y）/模糊搜索（N）,默認精確搜索"
    echo "第三個參數(shù)為：結(jié)束所有匹配到的進程（Y）/結(jié)束匹配到的第一個進程（N）,默認全部結(jié)束"
    echo "第四個參數(shù)為：強制結(jié)束進程（Y）/非強制結(jié)束進程（N）,默認強制結(jié)束進程"
    echo "第五個參數(shù)為：顯示匹配到進程存活情況（Y）/不顯示匹配到進程存活情況（N）,默認顯示存活情況"
    echo "注意：腳本結(jié)束會生成processResult.txt顯示詳情"
exit
;;
date)
    echo `date +%Y-%m-%d---%H:%M:%S`
exit
esac


if [ -e "tempProcessList.txt" ]; then
    rm tempProcessList.txt
fi

if [ -e "processResult.txt" ]; then
    rm processResult.txt
fi

#傳入pid殺死進程
killProcessFun(){
if test "N" = "$ForceKillProcess"
then
    kill $1 2>/dev/null
else
    kill -9 $1 2>/dev/null
fi
}


echo "進程名稱：$NAME"
echo "進程名稱：$NAME" >> processResult.txt

if test "N" = "$KilAll"
then
    echo "結(jié)束進程方式：非強制結(jié)束" >> processResult.txt
    echo "結(jié)束進程方式：非強制結(jié)束"
else
    echo "結(jié)束進程方式：強制結(jié)束" >> processResult.txt
    echo "結(jié)束進程方式：強制結(jié)束"
fi

#通過輸入?yún)?shù)篩選出ID
if test "N" = "$ExactSearches"
then
    echo "搜索方式：模糊搜索"
    echo "搜索方式：模糊搜索" >> processResult.txt
    #ID=`ps -ef | grep "$NAME" | grep -v "$0" | grep -v "grep" | awk '{print $2}'`
    ID=`pgrep "$NAME"`
else
    echo "搜索方式：精確搜索"
    echo "搜索方式：精確搜索" >> processResult.txt
    #ID=`ps -ef | grep "\<$NAME\>" | grep -v "$0" | grep -v "grep" | awk '{print $2}'`
    ID=`pgrep -x "$NAME"`
fi

echo "相關(guān)進程pid:"
echo "相關(guān)進程pid:" >> processResult.txt
echo $ID
echo $ID >> processResult.txt

echo ">>>--------------搜索結(jié)果詳情------------------\n" >> processResult.txt
echo ">>>--------------搜索結(jié)果詳情------------------\n"

#輸出相關(guān)進程信息
for id in $ID
do
    detail=`ps -ef | grep "$id" | grep -v "grep"| awk '{print $0}'`
    echo "pid:$id" >> processResult.txt
    echo "pid:$id"
    echo "detail:$detail\n" >> processResult.txt
    echo "detail:$detail\n"
    echo "$id|||$detail" >> tempProcessList.txt
done
    echo "-----------------搜索結(jié)果詳情---------------<<<" >> processResult.txt
    echo "-----------------搜索結(jié)果詳情---------------<<<"


echo "\\n\\n\\n>>>--------------處理目標進程------------------\n" >> processResult.txt
echo "\\n\\n\\n>>>--------------處理目標進程------------------\n"

#遍歷ID結(jié)束相關(guān)進程
for id in $ID
do
    detail=`ps -ef | grep "$id" | grep -v "grep"| awk '{print $0}'`

    #如果結(jié)束全部
    if test "N" = "$KilAll"
    then

        echo "killing_pid:$id"
        echo "killing_pid:$id" >> processResult.txt
        echo "detail:$detail"
        echo "detail:$detail" >> processResult.txt
        killProcessFun $id

    break
    else
        echo "killing_pid:$id" >> processResult.txt
        echo "killing_pid:$id"

        while read LINE
        do
            str=$LINE

            if test "$id" = "${str%%|||*}"
            then
                echo "detail:${LINE#*|||}\n" >> processResult.txt
                echo "detail:${LINE#*|||}\n"
            fi

        done < tempProcessList.txt
        killProcessFun $id
    fi
done

echo "-----------------處理目標進程---------------<<<" >> processResult.txt
echo "-----------------處理目標進程---------------<<<"



if test "N" = "$processLive"
then
echo " "
else
echo "\\n\\n\\n>>>--------------目標進程處理結(jié)果------------------\n" >> processResult.txt
echo "\\n\\n\\n>>>--------------目標進程處理結(jié)果------------------\n"

for id in $ID
do
detail=`ps -ef | grep "$id" | grep -v "grep"| awk '{print $0}'`
isLive='kill -0 $id'

    if kill -0 $id 2>/dev/null; then
        echo "pid（存活）:$id" >> processResult.txt
        echo "pid（存活）:$id"
        echo "detail:$detail\n" >> processResult.txt
        echo "detail:$detail\n"

    else
        echo "pid（死亡）:$id" >> processResult.txt
        echo "pid（死亡）:$id"
        while read LINE
        do
            str=$LINE
            if test "$id" = "${str%%|||*}"
            then
                echo "detail:${LINE#*|||}\n" >> processResult.txt
                echo "detail:${LINE#*|||}\n"
            break
            fi
        done < tempProcessList.txt
    fi

done

echo "-----------------目標進程處理結(jié)果---------------<<<" >> processResult.txt
echo "-----------------目標進程處理結(jié)果---------------<<<"

fi

if [ -e "tempProcessList.txt" ]; then
    rm tempProcessList.txt
fi