Linux之编程进阶（函数、trap信号捕捉、数组、字符串处理...）

翻译

阳光橙92 2019-04-13 09:45:59

如何定义函数(/etc/init.d下有个系统内定functions)

function: function name { COMMANDS ; } or name () { COMMANDS ; }

Define shell function.

Create a shell function named NAME.  When invoked as a simple command,
NAME runs COMMANDs in the calling shell's context.  When NAME is invoked,
the arguments are passed to the function as $1...$n, and the function's
name is in $FUNCNAME.

Exit Status:
Returns success unless NAME is readonly.

语法一： f_name （）{ ...函数体... } 语法二： function f_name { ...函数体... } 语法三： function f_name （） { ...函数体... }

定义函数和调用函数 单独写一个脚本functions,内部定义需要用到的函数，用的时候直接在新的shell中source functions。

示例 functions func1 () { echo func1 local name=mage----------》函数内的变量尽量定义成本地变量，防止脚本调用该函数时，脚本内有变量名称和函数内的变量名称一样，造成变量污染 return 10---------》返回调用此函数的脚本中，其后可带NUM，用于$?的判断。在return下面的命令不予执行。 } func2 () { echo func2 }

diaoyonn.sh #!/bin/bash source functions name=wang echo diaoyong.sh $name func1 echo diaoyong.sh $name

/etc/init.d/functions内部有很多函数，用的时候需要重新source该文件，其中有个action函数，是启动或关闭服务右边的**[ ok ]**显示。

信号捕捉trap

进程收到系统发出的指定信号后，将执行自定义指令，而不会执行原操作

trap '触发指令' 信号

忽略信号的操作

trap '' 信号

恢复原信号的操作

trap '-' 信号

列出自定义信号操作

trap -p

当脚本退出时，执行finish函数

trap finish EXIT

示例：捕捉kill里的信号 int=ctrl+c

#!/bin/bash
trap 'echo “signal:SIGINT"' int
trap -p
for((i=0;i<=10;i++))
do
sleep 1
echo $i
done
trap '' int
trap -p
for((i=11;i<=20;i++))
do
sleep 1
echo $i
done
trap '-' int
trap -p
for((i=21;i<=30;i++))
do
sleep 1
echo $i
done

数组

**数组：**存储多个元素的连续的内存空间，相当于多个变量的集合 数组名和索引 索引：编号从0开始，属于数值索引注意：索引可支持使用自定义的格式，而不仅是数值格式，即为关联索引，bash4.0版本之后开始支持 bash的数组支持稀疏格式（索引不连续） 声明数组：

declare -a ARRAY_NAME 定义数组为数字标识从0开始，支持跳跃默认不定义为数字标识 declare -A ARRAY_NAME 关联数组定义数组为关联数组自定义标识注意：两者不可相互转换

数组赋值 数组元素的赋值**（数组支持{}展开，如name=({1..10})**）引用时可以echo ${name=[0]},打印数组的第一个变量值 (1) 一次只赋值一个元素

ARRAY_NAME[INDEX]=VALUE weekdays[0]="Sunday" weekdays[4]="Thursday"

(2) 一次赋值全部元素

ARRAY_NAME=("VAL1" "VAL2" "VAL3" ...) name=("zhangsna" "lisi" "wangwu")

(3) 只赋值特定元素

ARRAY_NAME=([0]="VAL1" [3]="VAL2" ...) name=([0]="zhangsna" [2]="lisi" [4]="wangwu")

(4) 交互式数组值对赋值(如不定义此项，则表示用数字标识为角标，否则使用abc)

read -a ARRAY
declare -A name
name=([a]="zhangsna" [c]="lisi" [d]="wangwu")

显示所有数组：

declare -a

引用数组 引用数组元素

${ARRAY_NAME[INDEX]} 注意：省略[INDEX]表示引用下标为0的元素

引用数组所有元素

${ARRAY_NAME[*]} ${ARRAY_NAME[@]}

数组的长度(数组中元素的个数)

${#ARRAY_NAME[*]} ${#ARRAY_NAME[@]}

删除数组中的某元素：导致稀疏格式

unset ARRAY[INDEX]

删除整个数组

unset ARRAY

数组数据处理 引用数组中的元素：数组切片：

${ARRAY[@]:offset:number} offset 要跳过的元素个数 number 要取出的元素个数

取偏移量之后的所有元素

${ARRAY[@]:offset}

向数组中追加元素：

ARRAY[${#ARRAY[*]}]=value name[${#name[@]}]=7

关联数组：

declare -A ARRAY_NAME ARRAY_NAME=([idx_name1]='val1' [idx_name2]='val2‘...) 注意：关联数组必须先声明再调用

字符串切片 **${#var}:**返回字符串变量var的长度

**${var:offset}:**返回字符串变量var中从第offset个字符后（不包括第offset个字符）的字符开始，到最后的部分，offset的取值在0 到 ${#var}-1 之间(bash4.2后，允许为负值)

**${var:offset:number}：**返回字符串变量var中从第offset个字符后（不包括第offset个字符）的字符开始，长度为number的部分

**${var: -length}：**取字符串的最右侧几个字符 注意：冒号后必须有一空白字符

**${var:offset:-length}：**从最左侧跳过offset字符，一直向右取到距离最右侧lengh个字符之前的内容

**${var: -length:-offset}：**先从最右侧向左取到length个字符开始，再向右取到距离最右侧offset个字符之间的内容注意：-length前空格

字符串处理

基于模式取子串 ${var#*word}：其中word可以是指定的任意字符功能：自左而右，查找var变量所存储的字符串中，第一次出现的word, 删除字符串开头至第一次出现word字符串（含）之间的所有字符

**${var##*word}：**同上，贪婪模式，不同的是，删除的是字符串开头至最后一次由word指定的字符之间的所有内容

示例：

file=“var/log/messages” ${file#*/}: log/messages ${file##*/}: messages

**${var%word*}：**其中word可以是指定的任意字符功能：自右而左，查找var变量所存储的字符串中，第一次出现的word, 删除字符串最后一个字符向左至第一次出现word字符串（含）之间的所有字符

file="/var/log/messages" ${file%/*}: /var/log

**${var%%word*}：**同上，只不过删除字符串最右侧的字符向左至最后一次出现word字符之间的所有字符

示例：

url=http://www.magedu.com:80 ${url##*:} 80 ${url%%:*} http

查找替换

${var/pattern/substr}：查找var所表示的字符串中，第一次被pattern所匹配到的字符串，以substr替换之 ${var//pattern/substr}: 查找var所表示的字符串中，所有能被pattern所匹配到的字符串，以substr替换之 ${var/#pattern/substr}：查找var所表示的字符串中，行首被pattern所匹配到的字符串，以substr替换之 ${var/%pattern/substr}：查找var所表示的字符串中，行尾被pattern所匹配到的字符串，以substr替换之

查找并删除

${var/pattern}：删除var表示的字符串中第一次被pattern匹配到的字符串 ${var//pattern}：删除var表示的字符串中所有被pattern匹配到的字符串 ${var/#pattern}：删除var表示的字符串中所有以pattern为行首匹配到的字符串 ${var/%pattern}：删除var所表示的字符串中所有以pattern为行尾所匹配到的字符串

字符大小写转换 ${var^^}：把var中的所有小写字母转换为大写 ${var,,}：把var中的所有大写字母转换为小写

变量赋值

高级变量用法-有类型变量 Shell变量一般是无类型的，但是bash Shell提供了declare和typeset两个命令用于指定变量的类型，两个命令是等价的 declare [选项] 变量名

-r 声明或显示只读变量 -i 将变量定义为整型数 -a 将变量定义为数组 -A 将变量定义为关联数组 -f 显示已定义的所有函数名及其内容 -F 仅显示已定义的所有函数名 -x 声明或显示环境变量和函数 -l 声明变量为小写字母 declare –l var=UPPER -u 声明变量为大写字母 declare –u var=lower

eval命令 eval命令将会首先扫描命令行进行所有的置换，然后再执行该命令。该命令适用于那些一次扫描无法实现其功能的变量.该命令对变量进行两次扫描 示例：

[root@server ~]# CMD=whoami
[root@server ~]# echo $CMD
whoami
[root@server ~]# eval $CMD
root
[root@server ~]# n=10
[root@server ~]# echo {0..$n}
{0..10}
[root@server ~]# eval echo {0..$n}
0 1 2 3 4 5 6 7 8 9 10

间接变量引用 如果第一个变量的值是第二个变量的名字，从第一个变量引用第二个变量的值就称为间接变量引用 variable1的值是variable2，而variable2又是变量名，variable2的值为value，间接变量引用是指通过variable1获得变量值value的行为

variable1=variable2 variable2=value

bash Shell提供了两种格式实现间接变量引用

eval tempvar=$$variable1 tempvar=${!variable1}

示例：

[root@server ~]# N=NAME [root@server ~]# NAME=wangxiaochun [root@server ~]# N1=${!N} [root@server ~]# echo $N1 wangxiaochun [root@server ~]# eval N2=$$N [root@server ~]# echo $N2 wangxiaochun

创建临时文件 mktemp命令：创建并显示临时文件，可避免冲突

mktemp [OPTION]... [TEMPLATE] TEMPLATE: filenameXXX X至少要出现三个

OPTION：

-d: 创建临时目录 -p DIR或--tmpdir=DIR：指明临时文件所存放目录位置

示例：

mktemp /tmp/testXXX tmpdir=`mktemp –d /tmp/testdirXXX` mktemp --tmpdir=/testdir testXXXXXX

安装复制文件 install命令：

install [OPTION]... [-T] SOURCE DEST 单文件 install [OPTION]... SOURCE... DIRECTORY install [OPTION]... -t DIRECTORY SOURCE... install [OPTION]... -d DIRECTORY...创建空目录 选项： -m MODE，默认755 -o OWNER -g GROUP

示例：

install -m 700 -o wang -g admins srcfile desfile install –m 770 –d /testdir/installdir

expect介绍 expect 是由Don Libes基于Tcl（ Tool Command Language ）语言开发的，主要应用于自动化交互式操作的场景，借助 expect 处理交互的命令，可以将交互过程如：ssh登录，ftp登录等写在一个脚本上，使之自动化完成。尤其适用于需要对多台服务器执行相同操作的环境中，可以大大提高系统管理人员的工作效率 expect命令

expect 语法：

expect [选项] [ -c cmds ] [ [ -[f|b] ] cmdfile ] [ args ]

选项

-c：从命令行执行expect脚本，默认expect是交互地执行的

示例：expect -c 'expect "\n" {send "pressed enter\n"}

-d：可以输出输出调试信息

示例：expect -d ssh.exp

expect中相关命令 spawn 启动新的进程 send 用于向进程发送字符串 expect 从进程接收字符串 interact 允许用户交互 exp_continue 匹配多个字符串在执行动作后加此命令

expect expect最常用的语法(tcl语言:模式-动作)

单一分支模式语法：

expect “hi” {send “You said hi\n"} 匹配到hi后，会输出“you said hi”，并换行

多分支模式语法：

expect "hi" { send "You said hi\n" }
"hehe" { send "Hehe yourself\n" }
"bye" { send "Good bye\n" }

匹配hi,hello,bye任意字符串时，执行相应输出。等同如下：

expect { "hi" { send "You said hi\n"} "hehe" { send "Hehe yourself\n"} "bye" { send " Good bye\n"} }

示例

#!/usr/bin/expect
spawn scp /etc/fstab 192.168.8.100:/app
expect {
"yes/no" { send "yes\n";exp_continue }
"password" { send "magedu\n" }
}
expect eof

#!/usr/bin/expect
spawn ssh 192.168.8.100
expect {
"yes/no" { send "yes\n";exp_continue }
"password" { send "magedu\n" }
}
interact
#expect eof

示例:变量

#!/usr/bin/expect
set ip 192.168.8.100
set user root
set password magedu
set timeout 10
spawn ssh $user@$ip
expect {
"yes/no" { send "yes\n";exp_continue }
"password" { send "$password\n" }
}
interact

#!/usr/bin/expect
set ip [lindex $argv 0]
set user [lindex $argv 1]
set password [lindex $argv 2]
spawn ssh $user@$ip
expect {
"yes/no" { send "yes\n";exp_continue }
"password" { send "$password\n" }
}
interact
#./ssh3.exp 192.168.8.100 root magedu

示例：执行多个命令

#!/usr/bin/expect
set ip [lindex $argv 0]
set user [lindex $argv 1]
set password [lindex $argv 2]
set timeout 10
spawn ssh $user@$ip
expect {
"yes/no" { send "yes\n";exp_continue }
"password" { send "$password\n" }
}
expect "]#" { send "useradd haha\n" }
expect "]#" { send "echo magedu |passwd --stdin haha\n" }
send "exit\n"
expect eof
#./ssh4.exp 192.168.8.100 root magedu

示例：shell脚本调用expect

#!/bin/bash
ip=$1
user=$2
password=$3
expect <<EOF
set timeout 20
spawn ssh $user@$ip
expect {
"yes/no" { send "yes\n";exp_continue }
"password" { send "$password\n" }
}
expect "]#" { send "useradd hehe\n" }
expect "]#" { send "echo magedu |passwd --stdin hehe\n" }
expect "]#" { send "exit\n" }
expect eof
EOF
#./ssh5.sh 192.168.8.100 root magedu