linux - Bash - 如何在这种情况下调用函数

Question

你好，我想知道将这段代码分解成函数的最佳方法是什么

 case $# in
    1) ports='1-1023'
       host=$1 ;;
    2) ports=$1
       host=$2 ;;
    *) echo 'Usage: portscan [port|range] host'
       exit 1 ;;
esac
# check port range
if [ "$(echo $ports | grep '^[1-9][0-9]*-[1-9][0-9]*$')" != "" ]; then
    firstport=$(echo $ports | cut -d- -f1)
    lastport=$(echo $ports | cut -d- -f2)
elif [ "$(echo $ports | grep '^[1-9][0-9]*$')" != "" ]; then
    firstport=$ports
    lastport=$ports
else
    echo "$ports is an invalid port(s) value"
    exit 2
fi
# check firstport > lastport
if [ $firstport -gt $lastport ]; then
    echo $firstport is larger than $lastport
    exit 3
fi

** 像这样在 main 中调用它们****

# Check parameters and exit 1 if problem
check_parms $#
# Check port range and exit 2 if problem
check_ports $ports
# Check port order and exit 3 if problem
check_order $firstport $lastport

Answer 1

当您调用 bash 函数时，它也有自己的 $*、$#、$1、.... 来反射(reflect)该调用。所以你可以构建一堆小函数，比如

function check_parms() {
    local num=$# <----------- I changed this line
    case $num)
    ... etc
}

然后，

function check_ports() {
    local ports=$1
    if ...
}

等等

我希望这能给你一些重构的想法。如果您正在考虑 sourceing bash 脚本，一个好主意可能是 unset 您在脚本末尾定义的函数。

here 上的一个不错的教程

在主 block 中，像调用任何其他 bash 函数一样调用它。例如:

echo "hello"
check_parms $*
check_ports $ports
echo "..."