函数和别名乍一看很相似,不过两者在行为上还是略有不同。最大的差异在于函数参数可以在函数体中任意位置上使用,而别名只能将参数放在命令尾部。

1.12.1 实战演练

函数的定义包括function命令、函数名、开/闭括号以及包含在一对花括号中的函数体。

函数可以这样定义

  1. function fname()
  2. {
  3. statements;
  4. }

或者

  1. fname()
  2. {
  3. statements;
  4. }

甚至是这样(对于简单的函数):

  1. name() { statement; }

只需使用函数名就可以调用函数

  1. $ fname ; #执行函数

函数参数可以按位置访问

函数参数可以按位置访问,$1是第一个参数,$2是第二个参数,以此类推:

  1. fname arg1 arg2 ; #传递参数

以下是函数fname的定义。在函数fname中,包含了各种访问函数参数的方法。

  1. fname()
  2. {
  3. echo $1, $2; #访问参数1和参数2
  4. echo "$@"; #以列表的方式一次性打印所有参数
  5. echo "$*"; #类似于$@,但是所有参数被视为单个实体
  6. return 0; #返回值
  7. }

传入脚本的参数可以通过下列形式访问:

  • $0是脚本名称。
  • $1是第一个参数。
  • $2是第二个参数。
  • $n是第n个参数。
  • $@“被扩展成”$1“”$2“”$3“等。
  • $*“被扩展成”$1c$2c$3“,其中cIFS的第一个字符。
  • $@“要比”$*“用得多。由于”$*“将所有的参数当作单个字符串,因此它很少被使用。

比较别名与函数

  • 下面的这个别名通过将ls的输出传入grep来显示文件子集。别名的参数添加到命令的尾部,因此lsg txt就被扩展成了ls | grep txt

    1. $ alias lsg='ls | grep'
    2. $ lsg txt
    1. [root@dev vitest]# alias lsg='ls | grep'
    2. [root@dev vitest]# lsg txt
    3. host.txt
    4. [root@dev vitest]#
  • 如果想获得/sbin/ifconfig文件中设备对应的IP地址,可以尝试这样做:

    1. $ alias wontWork='/sbin/ifconfig | grep'
    2. $ wontWork eth0
    1. [root@dev vitest]# alias wontWork='/sbin/ifconfig | grep'
    2. [root@dev vitest]# wontWork eth0
    3. eth0: flags=4163<UP,BROADCAST,RUNNING,MULTICAST> mtu 1500
    4. [root@dev vitest]#
  • grep命令找到的是字符串eth0,而不是IP地址。如果我们使用函数来实现的话,可以将设备名作为参数传入ifconfig,不再交给grep

    1. $ function getIP() { /sbin/ifconfig $1 | grep 'inet '; }
    2. $ getIP eth0
    3. inet 172.18.101.55 netmask 255.255.240.0 broadcast 172.18.111.255
    1. [root@dev vitest]# function getIP() { /sbin/ifconfig $1 | grep 'inet '; }
    2. [root@dev vitest]# getIP eth0
    3. inet 172.18.101.55 netmask 255.255.240.0 broadcast 172.18.111.255
    4. [root@dev vitest]#

    1.12.2 补充内容

    让我们再研究一些Bash函数的技巧。

递归函数

Bash中,函数同样支持递归调用(可以调用自身的函数)。例如,F() { echo $1; F hello;leep 1; }

Fork炸弹

递归函数是能够调用自身的函数。这种函数必须有退出条件,否则就会不断地生成自身,直到系统耗尽所有的资源或是崩溃。

  1. :(){ :|:& };:

这个函数会一直地生成新的进程,最终形成拒绝服务攻击。

函数调用前的&将子进程放入后台。这段危险的代码能够不停地衍生出进程,因而被称为Fork炸弹。

上面这段代码要理解起来可不容易。请参阅维基百科 https://en.wikipedia.org/wiki/Fork_bomb,那里给出了有关Fork炸弹的更多细节以及解释。

可以通过修改配置文件/etc/security/limits.conf中的nproc来限制可生成的最大进程数,进而阻止这种攻击。

下面的语句将所有用户可生成的进程数限制为100

  1. hard nproc 100

导出函数

函数也能像环境变量一样用export导出,如此一来,函数的作用域就可以扩展到子进程中:

  1. export -f fname
  2. $ function getIP() { /sbin/ifconfig $1 | grep 'inet '; }
  3. $ echo "getIP eth0" >test.sh
  4. $ sh test.sh
  5. sh: getIP: No such file or directory
  6. $ export -f getIP
  7. $ sh test.sh
  8. inet addr: 192.168.1.2 Bcast: 192.168.255.255 Mask:255.255.0.0
  1. [root@dev vitest]# function getIP() { /sbin/ifconfig $1 | grep 'inet '; }
  2. [root@dev vitest]# ls
  3. host.txt man_db.conf
  4. [root@dev vitest]# echo "getIP eth0" >test.sh
  5. [root@dev vitest]# ls
  6. host.txt man_db.conf test.sh
  7. [root@dev vitest]# bash test.sh
  8. test.sh: line 1: getIP: command not found
  9. [root@dev vitest]# export -f getIP
  10. [root@dev vitest]# bash test.sh
  11. inet 172.18.101.55 netmask 255.255.240.0 broadcast 172.18.111.255
  12. [root@dev vitest]#

读取命令返回值(状态)

命令的返回值被保存在变量$?中。

  1. cmd;
  2. echo $?;

返回值被称为退出状态。它可用于确定命令执行成功与否。如果命令成功退出,那么退出状态为0,否则为非0

下面的脚本可以报告命令是否成功结束:

  1. #!/bin/bash
  2. #文件名: success_test.sh
  3. #对命令行参数求值,比如success_test.sh ‘ls | grep txt’
  4. eval $@
  5. if [ $? -eq 0 ];
  6. then
  7. echo "$CMD executed successfully"
  8. else
  9. echo "$CMD terminated unsuccessfully"
  10. fi

向命令传递参数

大多数应用都能够接受不同格式的参数。假设-p-v是可用选项,-k N是另一个可以接受数字的选项,同时该命令还要求使用一个文件名作为参数。那么,它有如下几种执行方式:

  • command -p -v -k 1 file
  • command -pv -k 1 file
  • command -vpk 1 file
  • command file -pvk 1

在脚本中,命令行参数可以依据其在命令行中的位置来访问。第一个参数是$1,第二个参数是$2,以此类推。

下面的语句可以显示出前3个命令行参数:

  1. echo $1 $2 $3

更为常见的处理方式是迭代所有的命令行参数。shift命令可以将参数依次向左移动一个位置,让脚本能够使用$1来访问到每一个参数。下面的代码显示出了所有的命令行参数:

  1. $ cat showArgs.sh
  2. for i in `seq 1 $#`
  3. do
  4. echo $i is $1
  5. shift
  6. done
  7. $ sh showArgs.sh a b c
  1. [root@dev workspace]# vi showArgs.sh
  2. [root@dev workspace]# cat showArgs.sh
  3. for i in `seq 1 $#`
  4. do
  5. echo $i is $1
  6. shift
  7. done
  8. [root@dev workspace]# sh showArgs.sh a b c
  9. 1 is a
  10. 2 is b
  11. 3 is c
  12. [root@dev workspace]#