实战：shell脚本练习

发布于：2024-07-20 ⋅ 阅读:(142) ⋅ 点赞:(0)

高效编写Bash脚本的技巧

总结了10个实用技巧，帮助提高脚本的效率和可靠性，具体包括：

  1. 多写注释：在脚本中添加注释，以帮助理解脚本的不同部分。

  2. 当运行失败时使脚本退出：使用set -o errexit或set -e，在命令失败时退出脚本。

  3. 当Bash用未声明变量时使脚本退出：使用set -o nounset或set -u，避免使用未声明的变量。

  4. 使用双引号来引用变量：防止由于空格或通配符导致的不必要匹配。

  5. 在脚本中使用函数：通过函数使代码模块化，提高可读性和可重用性。

  6. 字符串比较时用=而不是==：在Bash中，=和==是等价的，但推荐使用=。

  7. 用$(command)而不是反引号command进行命令代换：推荐使用$(command)，因为它更清晰。

  8. 用readonly声明静态变量：声明变量为只读，防止其值被修改。

  9. 环境变量用大写字母命名，自定义变量用小写：遵循命名约定，避免变量名冲突。

  10. 总是对长脚本进行调试：在脚本执行前进行调试，以便于修正错误。

相关例子
 

  1. 多写注释

    • 举例：在脚本中添加注释，例如：

      
# This script updates the system packages
update_system() {
  sudo apt-get update
  sudo apt-get upgrade -y
}

    • 好处：增加代码的可读性，便于维护和理解。

    • 问题：没有注释的脚本难以理解，特别是在团队协作中，其他成员可能难以快速上手。

  2. 当运行失败时使脚本退出

    • 举例：使用set -e来确保脚本在命令失败时立即退出：

      • 
set -e ./some_command || true  # 这个命令失败不会退出脚本

    • 好处：防止错误的累积和传播，提高脚本的健壮性。

    • 问题：不立即退出可能导致后续命令基于错误的前提执行，造成更严重的错误。

  3. 当Bash用未声明变量时使脚本退出

    • 举例：使用set -u来避免使用未声明的变量：

    • set -u
echo "$undeclared_var"  # 这将导致脚本退出

      好处：避免使用未初始化的变量，减少潜在的bug。

    • 问题：使用未声明的变量可能导致不可预测的行为或错误的输出。

  4. 使用双引号来引用变量

    • 举例：使用双引号来防止变量展开或通配符匹配：

    • names="TecMint FOSSMint"
echo "$names"  # 正确："TecMint FOSSMint"
echo $names   # 错误：可能分割成两个单词

      好处：确保变量按照预期的方式被引用，防止意外的单词分割。

    • 问题：不使用双引号可能导致变量值中的空格或特殊字符被错误解释。

  5. 在脚本中使用函数

    • 举例：将重复的代码块封装成函数：

      check_root() {
  if [ "$(id -u)" != "0" ]; then
    echo "This script must be run as root" >&2
    exit 1
  fi
}

      好处：提高代码的模块化和重用性，简化脚本结构。

    • 问题：不使用函数可能导致代码重复，难以维护和扩展。

  6. 字符串比较时用=而不是==

    • 举例：使用单等号进行字符串比较：

      if [ "$value1" = "$value2" ]; then
  echo "Values are equal"
fi

      好处：简化语法，避免混淆。

    • 问题：使用==在某些shell中可能不被识别，导致语法错误。

    • 哈哈，虽然简单场景用 [ ] 和 = 更保证可移植性和安全性。  但是这个我还是更喜欢用 [[  ]] 和 == ；

  7. 用$(command)而不是反引号进行命令代换

    • 举例：使用$()进行命令替换：

      
user=$(whoami)  # 正确
user=`whoami`   # 错误：不建议使用

      好处：提高代码的可读性和一致性。

    • 问题：反引号在复杂表达式中可能导致错误，且不如$()易读。

  8. 用readonly声明静态变量

    • 举例：使用readonly来防止变量被修改：

      readonly config_path="/etc/config"

      好处：确保配置值在脚本中保持不变，防止意外的修改。

    • 问题：不声明为只读可能导致变量在脚本中被错误地修改。

  9. 环境变量用大写字母命名，自定义变量用小写

    • 举例：遵循命名约定：

      export PATH="/usr/bin:/bin"
local script_path="$(dirname "$0")"

      好处：避免与系统环境变量冲突，提高变量名的可读性。

    • 问题：不遵循命名约定可能导致变量名冲突或混淆。

  10. 总是对长脚本进行调试

    • 举例：使用bash -n script.sh来检查脚本中的语法错误：

      bash -n my_script.sh

      好处：提前发现潜在的错误，避免在生产环境中出现问题。

    • 问题：不进行调试可能导致脚本在实际运行时出现未预料的错误，影响系统的稳定性。

遵循这些技巧可以编写出更加健壮、可维护和易于理解的Bash脚本。

用nice指定shell脚本在运行过程中的优先级

    在Linux系统中，脚本、程序、命令等可能同一时刻需要运行，而CPU在一个时刻只能运行其中的一个，这个时候就牵扯到先后的问题，换言之，也就是优先级。

    系统中，进程的优先级一般包括实时优先级和静态优先级，在任何时刻，实时进程的优先级高于普通进程，实时进程的优先级范围0~99，普通进程的优先级范围100~139,因此，数值越小，优先级越高。

 1 查看Linux中进程的优先级

[root@achao ~]# ps -lF S   UID    PID   PPID  C PRI  NI ADDR SZ WCHAN  TTY          TIME CMD4 S     0   2935   2928  0  80   0 - 29180 do_wai pts/0    00:00:00 bash0 R     0   3071   2935  0  80   0 - 38328 -      pts/0    00:00:00 ps

    ps命令能够显示当前系统中正在运行的进程，并且能够显示进程的详细信息，其中，PRI和NI是和进程优先级相关的字段。PRI的默认取值范围是0~99，进程的优先级涉及到进程调度相关的知识，此处不做展开。本篇旨在使用。

 2 使用nice指定优先级

        NICE值是进程优先级的修正值，NICE和优先级的关系如下：

    PRI值=PRI值（旧值）+NICE值

进程最终的优先级是PRI和NICE值的和，NICE值可以是正数，也可以是负数，当NICE值为正时，会使得进程优先级降低，反之，可以提高进程优先级。

    NICE取值范围-20~19。

    设置进程NICE值使用nice命令，介绍如下

[root@achao ~]# nice --helpUsage: nice [OPTION] [COMMAND [ARG]...]Run COMMAND with an adjusted niceness, which affects process scheduling.With no COMMAND, print the current niceness.  Niceness values range from-20 (most favorable to the process) to 19 (least favorable to the process).
Mandatory arguments to long options are mandatory for short options too.  -n, --adjustment=N   add integer N to the niceness (default 10)      --help     display this help and exit      --version  output version information and exit

 3 正常执行shell脚本，并查看

[root@achao libai]# lsdemo.sh[root@achao libai]# cat demo.sh #!/bin/bashwhile((1))do  sleep 1done[root@achao libai]# ./demo.sh &[1] 3376[root@achao libai]# ps -lF S   UID    PID   PPID  C PRI  NI ADDR SZ WCHAN  TTY          TIME CMD4 S     0   2935   2928  0  80   0 - 29213 do_wai pts/0    00:00:00 bash0 S     0   3376   2935  0  80   0 - 28320 do_wai pts/0    00:00:00 demo.sh0 S     0   3386   3376  0  80   0 - 27014 hrtime pts/0    00:00:00 sleep0 R     0   3387   2935  0  80   0 - 38328 -      pts/0    00:00:00 ps

    不指定nice值执行shell脚本PID值是3376

 4 使用nice指定优先级执行shell脚本

[root@achao libai]# nice -10 ./demo.sh &[2] 3573[root@achao libai]# ps -elf | grep demo.sh | grep -v grep 0 S root       3376   2935  0  80   0 - 28320 do_wai 01:39 pts/0    00:00:00 /bin/bash ./demo.sh0 S root       3573   2935  0  90  10 - 28320 do_wai 01:42 pts/0    00:00:00 /bin/bash ./demo.sh

    我们发现NI和PRI的值发生了变化，因此使用nice命令可以直接对进程执行的优先级产生影响，该影响在进行进程调度的时候，会发生很大的作用，不知道宝子们在生产环境中是否遇见过呢？阿超第一次遇见的时候是在shell脚本中的，当时也是一脸懵，只能悄悄查资料学习理解。    

 5 renice的语法格式

        renice的范围是[-20~19]

[root@achao libai]# renice --help
Usage: renice [-n] <priority> [-p|--pid] <pid>... renice [-n] <priority>  -g|--pgrp <pgid>... renice [-n] <priority>  -u|--user <user>...
Options: -g, --pgrp <id>        interpret argument as process group ID -n, --priority <num>   specify the nice increment value -p, --pid <id>         interpret argument as process ID (default) -u, --user <name|id>   interpret argument as username or user ID -h, --help             display help text and exit -V, --version          display version information and exit

     在运行的过程中，如果需要提高或者降低优先级，也是有办法的，这个时候就用renice命令就可以完成。

 6 使用renice重置优先级

使用kill -9结束3376进程[root@achao libai]# kill -9 3376[1]-  Killed                  ./demo.sh[root@achao libai]# ps -elf | grep demo.sh | grep -v grep 0 S root       3573   2935  0  90  10 - 28320 do_wai 01:42 pts/0    00:00:00 /bin/bash ./demo.sh[root@achao libai]# renice 15 -p 35733573 (process ID) old priority 10, new priority 15[root@achao libai]# ps -elf | grep demo.sh | grep -v grep 0 S root       3573   2935  0  95  15 - 28320 do_wai 01:42 pts/0    00:00:00 /bin/bash ./demo.sh

       注意：renice命令也是只能通过影响进程的NICE值来改变进程的优先级，而不能直接改变进程的优先级。

一键生成SSL证书：详解Shell脚本实现及OpenSSL应用

OpenSSL简介

OpenSSL是一个强大的开源工具包，用于实现SSL和TLS协议以及提供加密功能。它广泛应用于Web服务器，用于加密HTTPS通信。在本脚本中，我们将使用OpenSSL来生成自签名的CA（证书颁发机构）证书以及服务器证书。

脚本分析

接下来，我们将逐行分析这个Shell脚本，了解它是如何工作的。

CERT_INFO=(      [00]="/O=heaven/CN=ca.god.com"      [01]="cakey.pem"      [02]="cacert.pem"      [03]=2048      [04]=3650      [05]=0      [10]="/C=CN/ST=hubei/L=wuhan/O=Central.Hospital/CN=master.liwenliang.org"      [11]="master.key"      [12]="master.crt"      [13]=2048      [14]=365      [15]=1      [16]="master.csr"      [20]="/C=CN/ST=hubei/L=wuhan/O=Central.Hospital/CN=slave.liwenliang.org"      [21]="slave.key"      [22]="slave.crt"      [23]=2048      [24]=365      [25]=2      [26]="slave.csr"  )

这段定义了一个名为CERT_INFO的数组，包含了生成证书所需的各种信息，包括证书主题、密钥文件名、证书文件名、密钥长度、有效期、序列号以及CSR（证书签名请求）文件名。

COLOR="echo -e \\E[1;32m"  END="\\E[0m"  DIR=/data  cd $DIR

这几行定义了用于输出彩色文本的命令，并设置了工作目录为/data。

for i in {0..2};do      if [ $i -eq 0 ] ;then          openssl req  -x509 -newkey rsa:${CERT_INFO[${i}3]} -subj ${CERT_INFO[${i}0]} \              -set_serial ${CERT_INFO[${i}5]} -keyout ${CERT_INFO[${i}1]} -nodes -days ${CERT_INFO[${i}4]} \              -out ${CERT_INFO[${i}2]} &>/dev/null      else          openssl req -newkey rsa:${CERT_INFO[${i}3]} -nodes -subj ${CERT_INFO[${i}0]} \              -keyout ${CERT_INFO[${i}1]}   -out ${CERT_INFO[${i}6]} &>/dev/null          openssl x509 -req -in ${CERT_INFO[${i}6]}  -CA ${CERT_INFO[02]} -CAkey ${CERT_INFO[01]}  \              -set_serial ${CERT_INFO[${i}5]}  -days ${CERT_INFO[${i}4]} -out ${CERT_INFO[${i}2]} &>/dev/null      fi      $COLOR"**************************************生成证书信息 **************************************"$END      openssl x509 -in ${CERT_INFO[${i}2]} -noout -subject -dates -serial      echo  done

这是脚本的核心部分，通过一个循环，分别为CA和两个服务器生成证书。

  • 对于CA（$i -eq 0），使用openssl req -x509命令直接生成自签名的X.509证书。

  • 对于服务器证书，首先使用openssl req生成CSR，然后使用openssl x509根据CSR和CA证书生成最终的服务器证书。

chmod 600 *.key

这行代码将所有密钥文件的权限设置为600，确保只有文件所有者可以读写密钥文件，增加安全性。

action "证书生成完成"

这行假设有一个名为action的函数，用于显示操作结果。虽然脚本中没有定义这个函数，但你可以根据需要自行实现，例如通过简单的echo命令来输出完成信息。

使用脚本

  1. 保存脚本：将上述脚本保存为generate_certs.sh。

  2. 赋予权限：运行chmod +x generate_certs.sh，使脚本具有执行权限。

  3. 执行脚本：运行./generate_certs.sh，脚本将自动生成所需的证书和密钥。

注意事项

  • 确保在运行脚本之前，你已经安装了OpenSSL。

  • 根据你的实际需求，可能需要调整证书的有效期、密钥长度等参数。

  • 在生产环境中，建议使用受信任的CA颁发的证书，而不是自签名证书。

结语

通过这个简单的Shell脚本，你可以轻松地为你的服务器生成SSL证书，无论是用于开发环境还是测试环境。当然，对于生产环境，建议使用更严格的证书管理策略，并确保定期更新证书，以保障通信安全。希望这篇文章对你有所帮助，如果你有任何问题或建议，欢迎在评论区留言。

18 个一线工作中常用 Shell 脚本

  • 1、检测两台服务器指定目录下的文件一致性

  • 2、定时清空文件内容，定时记录文件大小

  • 3、检测网卡流量，并按规定格式记录在日志中

  • 4、计算文档每行出现的数字个数，并计算整个文档的数字总数

  • 5、从 FTP 服务器下载文件

  • 6、连续输入5个100以内的数字，统计和、最小和最大

  • 7、监测 Nginx 访问日志 502 情况，并做相应动作

  • 8、将结果分别赋值给变量

  • 9、批量修改文件名

  • 10、统计当前目录中以 .html 结尾的文件总大

  • 11、扫描主机端口状态

  • 12、输入数字运行相应命令

  • 13、Expect 实现 SSH 免交互执行命令

  • 14、监控 httpd 的进程数，根据监控情况做相应处理

  • 15、批量修改服务器用户密码

  • 16、iptables 自动屏蔽访问网站频繁的IP

  • 17、根据web访问日志，封禁请求量异常的IP，如IP在半小时后恢复正常，则解除封禁

  • 18、判断用户输入的是否为IP地址

1、检测两台服务器指定目录下的文件一致性

#!/bin/bash
#####################################
#检测两台服务器指定目录下的文件一致性
#####################################
#通过对比两台服务器上文件的md5值，达到检测一致性的目的
dir=/data/web
b_ip=192.168.88.10
#将指定目录下的文件全部遍历出来并作为md5sum命令的参数，进而得到所有文件的md5值，并写入到指定文件中
find $dir -type f|xargs md5sum > /tmp/md5_a.txt
ssh $b_ip "find $dir -type f|xargs md5sum > /tmp/md5_b.txt"
scp $b_ip:/tmp/md5_b.txt /tmp
#将文件名作为遍历对象进行一一比对
for f in `awk '{print 2} /tmp/md5_a.txt'`
do
    #以a机器为标准，当b机器不存在遍历对象中的文件时直接输出不存在的结果
    if grep -qw "$f" /tmp/md5_b.txt
    then
        md5_a=`grep -w "$f" /tmp/md5_a.txt|awk '{print 1}'`
        md5_b=`grep -w "$f" /tmp/md5_b.txt|awk '{print 1}'`
        #当文件存在时，如果md5值不一致则输出文件改变的结果
        if [ $md5_a != $md5_b ]
        then
            echo "$f changed."
        fi
    else
        echo "$f deleted."
    fi
done

2、定时清空文件内容，定时记录文件大小

#!/bin/bash
################################################################
#每小时执行一次脚本（任务计划），当时间为0点或12点时，将目标目录下的所有文件内
#容清空，但不删除文件，其他时间则只统计各个文件的大小，一个文件一行，输出到以时#间和日期命名的文件中，需要考虑目标目录下二级、三级等子目录的文件
################################################################
logfile=/tmp/`date +%H-%F`.log
n=`date +%H`
if [ $n -eq 00 ] || [ $n -eq 12 ]
then
    #通过for循环，以find命令作为遍历条件，将目标目录下的所有文件进行遍历并做相应操作
    for i in `find /data/log/ -type f`
    do
        true > $i
    done
else
    for i in `find /data/log/ -type f`
    do
        du -sh $i >> $logfile
    done
fi

3、检测网卡流量，并按规定格式记录在日志中

#!/bin/bash
#######################################################
#检测网卡流量，并按规定格式记录在日志中
#规定一分钟记录一次
#日志格式如下所示:
#2019-08-12 20:40
#ens33 input: 1234bps
#ens33 output: 1235bps
######################################################3
while :
do
    #设置语言为英文，保障输出结果是英文，否则会出现bug
    LANG=en
    logfile=/tmp/`date +%d`.log
    #将下面执行的命令结果输出重定向到logfile日志中
    exec >> $logfile
    date +"%F %H:%M"
    #sar命令统计的流量单位为kb/s，日志格式为bps，因此要*1000*8
    sar -n DEV 1 59|grep Average|grep ens33|awk '{print $2,"\t","input:","\t",$5*1000*8,"bps","\n",$2,"\t","output:","\t",$6*1000*8,"bps"}'
    echo "####################"
    #因为执行sar命令需要59秒，因此不需要sleep
done

4、计算文档每行出现的数字个数，并计算整个文档的数字总数

#!/bin/bash
#########################################################
#计算文档每行出现的数字个数，并计算整个文档的数字总数
########################################################
#使用awk只输出文档行数（截取第一段）
n=`wc -l a.txt|awk '{print $1}'`
sum=0
#文档中每一行可能存在空格，因此不能直接用文档内容进行遍历
for i in `seq 1 $n`
do
    #输出的行用变量表示时，需要用双引号
    line=`sed -n "$i"p a.txt`
    #wc -L选项，统计最长行的长度
    n_n=`echo $line|sed s'/[^0-9]//'g|wc -L`
    echo $n_n
    sum=$[$sum+$n_n]
done
echo "sum:$sum"

杀死所有脚本

#!/bin/bash
if [ $# -ne 1 ]; then
    echo "Usage: $0 filename"
fi
dir=$(dirname $1)
file=$(basename $1)
ftp -n -v << EOF   # -n 自动登录
open 192.168.1.10  # ftp服务器
user admin password
binary   # 设置ftp传输模式为二进制，避免MD5值不同或.tar.gz压缩包格式错误
cd $dir
get "$file"
EOF

5、从 FTP 服务器下载文件

#!/bin/bash  
if [ $# -ne 1 ]; then  
    echo "Usage: $0 filename"  
fi  
dir=$(dirname $1)  
file=$(basename $1)  
ftp -n -v << EOF   # -n 自动登录  
open 192.168.1.10  # ftp服务器  
user admin password  
binary   # 设置ftp传输模式为二进制，避免MD5值不同或.tar.gz压缩包格式错误  
cd $dir  
get "$file"  
EOF

6、连续输入5个100以内的数字，统计和、最小和最大

#!/bin/bash
COUNT=1
SUM=0
MIN=0
MAX=100
while [ $COUNT -le 5 ]; do
    read -p "请输入1-10个整数：" INT
    if [[ ! $INT =~ ^[0-9]+$ ]]; then
        echo "输入必须是整数！"
        exit 1
        elif [[ $INT -gt 100 ]]; then
        echo "输入必须是100以内！"
        exit 1
    fi
    SUM=$(($SUM+$INT))
    [ $MIN -lt $INT ] && MIN=$INT
    [ $MAX -gt $INT ] && MAX=$INT
    let COUNT++
done
echo "SUM: $SUM"
echo "MIN: $MIN"
echo "MAX: $MAX"

用户猜数字

#!/bin/bash  # 脚本生成一个 100 以内的随机数,提示用户猜数字,根据用户的输入,提示用户猜对了,
# 猜小了或猜大了,直至用户猜对脚本结束。
# RANDOM 为系统自带的系统变量,值为 0‐32767的随机数
# 使用取余算法将随机数变为 1‐100 的随机数num=$[RANDOM%100+1]echo "$num"
# 使用 read 提示用户猜数字
# 使用 if 判断用户猜数字的大小关系:‐eq(等于),‐ne(不等于),‐gt(大于),‐ge(大于等于),
# ‐lt(小于),‐le(小于等于)

while :
do
    read -p "计算机生成了一个 1‐100 的随机数,你猜: " cai
    if [ $cai -eq $num ]
    then
        echo "恭喜,猜对了"
        exit
    elif [ $cai -gt $num ]
    then
        echo "Oops,猜大了"
    else
        echo "Oops,猜小了"
    fi
done

7、监测 Nginx 访问日志 502 情况，并做相应动作

假设服务器环境为 lnmp，近期访问经常出现 502 现象，且 502 错误在重启 php-fpm 服务后消失，因此需要编写监控脚本，一旦出现 502，则自动重启 php-fpm 服务。

#场景：
#1.访问日志文件的路径：/data/log/access.log
#2.脚本死循环，每10秒检测一次，10秒的日志条数为300条，出现502的比例不低于10%（30条）则需要重启php-fpm服务
#3.重启命令为：/etc/init.d/php-fpm restart
#!/bin/bash
###########################################################
#监测Nginx访问日志502情况，并做相应动作
###########################################################
log=/data/log/access.log
N=30 #设定阈值
while :
do
    #查看访问日志的最新300条，并统计502的次数
    err=`tail -n 300 $log |grep -c '502" '`
    if [ $err -ge $N ]
    then
        /etc/init.d/php-fpm restart 2> /dev/null
        #设定60s延迟防止脚本bug导致无限重启php-fpm服务
        sleep 60
    fi
    sleep 10
done

8、将结果分别赋值给变量

应用场景：希望将执行结果或者位置参数赋值给变量，以便后续使用。

方法1：

for i in $(echo "4 5 6"); do
    eval a$i=$i
done
echo $a4 $a5 $a6

方法2：将位置参数192.168.1.1{1,2}拆分为到每个变量

num=0
for i in $(eval echo $*);do   #eval将{1,2}分解为1 2
    let num+=1
    eval node${num}="$i"
done
echo $node1 $node2 $node3
# bash a.sh 192.168.1.1{1,2}
192.168.1.11 192.168.1.12

方法3：

arr=(4 5 6)
INDEX1=$(echo ${arr[0]})
INDEX2=$(echo ${arr[1]})
INDEX3=$(echo ${arr[2]})

9、批量修改文件名

示例：

# touch article_{1..3}.html  
# lsarticle_1.html  article_2.html  article_3.html

方法1：

for file in $(ls *html); do  
    mv $file bbs_${file#*_}  
    # mv $file $(echo $file |sed -r 's/.*(_.*)/bbs\1/')  
    # mv $file $(echo $file |echo bbs_$(cut -d_ -f2)

方法2：

for file in $(find . -maxdepth 1 -name "*html"); do  
     mv $file bbs_${file#*_}done

方法3：

# rename article bbs *.html

把一个文档前五行中包含字母的行删掉，同时删除6到10行包含的所有字母

1）准备测试文件，文件名为2.txt

第1行1234567不包含字母  
第2行56789BBBBBB  
第3行67890CCCCCCCC  
第4行78asdfDDDDDDDDD  
第5行123456EEEEEEEE  
第6行1234567ASDF  
第7行56789ASDF  
第8行67890ASDF  
第9行78asdfADSF  
第10行123456AAAA  
第11行67890ASDF  
第12行78asdfADSF  
第13行123456AAAA

2）脚本如下：

#!/bin/bash  
###############################################################  
把一个文档前五行中包含字母的行删掉，同时删除6到10行包含的所有字母  
##############################################################  
sed -n '1,5'p 2.txt |sed '/[a-zA-Z]/'d  
sed -n '6,10'p 2.txt |sed s'/[a-zA-Z]//'g  
sed -n '11,$'p 2.txt  
#最终结果只是在屏幕上打印结果，如果想直接更改文件，可将输出结果写入临时文件中，再替换2.txt或者使用-i选项

10、统计当前目录中以 .html 结尾的文件总大

方法1：

# find . -name "*.html" -exec du -k {} \; |awk '{sum+=$1}END{print sum}'

方法2：

for size in $(ls -l *.html |awk '{print $5}'); do  
    sum=$(($sum+$size))  
done  
echo $sum  

11、扫描主机端口状态

#!/bin/bash
HOST=$1
PORT="22 25 80 8080"
for PORT in $PORT; do
    if echo &>/dev/null > /dev/tcp/$HOST/$PORT; then
        echo "$PORT open"
    else
        echo "$PORT close"
    fi
done

用 shell 打印示例语句中字母数小于 6 的单词

#示例语句：
#Bash also interprets a number of multi-character options.
#!/bin/bash
##############################################################
#shell打印示例语句中字母数小于6的单词
##############################################################
for s in Bash also interprets a number of multi-character options.
do
    n=`echo $s|wc -c`
    if [ $n -lt 6 ]
    then
        echo $s
    fi
done

12、输入数字运行相应命令

#!/bin/bash
##############################################################
#输入数字运行相应命令
##############################################################
echo "*cmd menu* 1-date 2-ls 3-who 4-pwd 0-exit "
while :
do
    #捕获用户键入值
    read -p "please input number :" n
    n1=`echo $n|sed s'/[0-9]//'g`
    #空输入检测
    if [ -z "$n" ]
    then
        continue
    fi
    #非数字输入检测
    if [ -n "$n1" ]
    then
        exit 0
    fi
    break
done
case $n in
    1)
        date
    ;;
    2)
        ls
    ;;
    3)
        who
    ;;
    4)
        pwd
    ;;
    0)
        break
    ;;
    #输入数字非1-4的提示
    *)
        echo "please input number is [1-4]"
esac

13、Expect 实现 SSH 免交互执行命令

Expect是一个自动交互式应用程序的工具，如telnet，ftp，passwd等。

需先安装expect软件包。

方法1：EOF 标准输出作为 expect 标准输入

#!/bin/bash
USER=root
PASS=123.com
IP=192.168.1.120
expect << EOFset timeout 30spawn ssh $USER@$IP   expect {    "(yes/no)" {send "yes\r"; exp_continue}    "password:" {send "$PASS\r"}
}
expect "$USER@*"  {send "$1\r"}
expect "$USER@*"  {send "exit\r"}
expect eof
EOF

方法2：

#!/bin/bash
USER=root
PASS=123.com
IP=192.168.1.120
expect -c "
    spawn ssh $USER@$IP
    expect {
        \"(yes/no)\" {send \"yes\r\"; exp_continue}
        \"password:\" {send \"$PASS\r\"; exp_continue}
        \"$USER@*\" {send \"df -h\r exit\r\"; exp_continue}
}"

方法3：将expect脚本独立出来

登录脚本：

# cat login.exp
#!/usr/bin/expect
set ip [lindex $argv 0]
set user [lindex $argv 1]
set passwd [lindex $argv 2]
set cmd [lindex $argv 3]
if { $argc != 4 } {
    puts "Usage: expect login.exp ip user passwd"
    exit 1
}
set timeout 30
spawn ssh $user@$ip
expect {
    "(yes/no)" {send "yes\r"; exp_continue}
    "password:" {send "$passwd\r"}
}
expect "$user@*"  {send "$cmd\r"}
expect "$user@*"  {send "exit\r"}
expect eof

执行命令脚本：写个循环可以批量操作多台服务器

#!/bin/bash
HOST_INFO=user_info.txt
for ip in $(awk '{print $1}' $HOST_INFO)
do
    user=$(awk -v I="$ip" 'I==$1{print $2}' $HOST_INFO)
    pass=$(awk -v I="$ip" 'I==$1{print $3}' $HOST_INFO)
    expect login.exp $ip $user $pass $1
done

Linux主机SSH连接信息：

# cat user_info.txt  
192.168.1.120 root 123456

创建10个用户，并分别设置密码，密码要求10位且包含大小写字母以及数字，最后需要把每个用户的密码存在指定文件中 。

#!/bin/bash
##############################################################
#创建10个用户，并分别设置密码，密码要求10位且包含大小写字母以及数字
#最后需要把每个用户的密码存在指定文件中
#前提条件：安装mkpasswd命令
##############################################################
#生成10个用户的序列（00-09）
for u in `seq -w 0 09`
do
    #创建用户
    useradd user_$u
    #生成密码
    p=`mkpasswd -s 0 -l 10`
    #从标准输入中读取密码进行修改（不安全）
    echo $p|passwd --stdin user_$u
    #常规修改密码
    echo -e "$p\n$p"|passwd user_$u
    #将创建的用户及对应的密码记录到日志文件中
    echo "user_$u $p" >> /tmp/userpassword
done

14、监控 httpd 的进程数，根据监控情况做相应处理

#!/bin/bash
###############################################################################################################################
#需求：
#1.每隔10s监控httpd的进程数，若进程数大于等于500，则自动重启Apache服务，并检测服务是否重启成功
#2.若未成功则需要再次启动，若重启5次依旧没有成功，则向管理员发送告警邮件，并退出检测
#3.如果启动成功，则等待1分钟后再次检测httpd进程数，若进程数正常，则恢复正常检测（10s一次），否则放弃重启并向管理员发送告警邮件，并退出检测
###############################################################################################################################
#计数器函数
check_service()
{
    j=0
    for i in `seq 1 5`
    do
        #重启Apache的命令
        /usr/local/apache2/bin/apachectl restart 2> /var/log/httpderr.log
        #判断服务是否重启成功
        if [ $? -eq 0 ]
        then
            break
        else
            j=$[$j+1]
        fi
        #判断服务是否已尝试重启5次
        if [ $j -eq 5 ]
        then
            mail.py
            exit
        fi
    done
}
while :
do
    n=`pgrep -l httpd|wc -l`
    #判断httpd服务进程数是否超过500
    if [ $n -gt 500 ]
    then
        /usr/local/apache2/bin/apachectl restart
        if [ $? -ne 0 ]
        then
            check_service
        else
            sleep 60
            n2=`pgrep -l httpd|wc -l`
            #判断重启后是否依旧超过500
            if [ $n2 -gt 500 ]
            then
                mail.py
                exit
            fi
        fi
    fi
    #每隔10s检测一次
    sleep 10
done

15、批量修改服务器用户密码

Linux主机SSH连接信息：旧密码

# cat old_pass.txt   
192.168.18.217  root    123456     22  
192.168.18.218  root    123456     22

内容格式：IP User Password Port

SSH远程修改密码脚本：新密码随机生成

#!/bin/bash  
OLD_INFO=old_pass.txt  
NEW_INFO=new_pass.txt  
for IP in $(awk '/^[^#]/{print $1}' $OLD_INFO); do  
    USER=$(awk -v I=$IP 'I==$1{print $2}' $OLD_INFO)  
    PASS=$(awk -v I=$IP 'I==$1{print $3}' $OLD_INFO)  
    PORT=$(awk -v I=$IP 'I==$1{print $4}' $OLD_INFO)  
    NEW_PASS=$(mkpasswd -l 8)  # 随机密码  
    echo "$IP   $USER   $NEW_PASS   $PORT" >> $NEW_INFO  
    expect -c "  
    spawn ssh -p$PORT $USER@$IP  
    set timeout 2  
    expect {  
        \"(yes/no)\" {send \"yes\r\";exp_continue}  
        \"password:\" {send \"$PASS\r\";exp_continue}  
        \"$USER@*\" {send \"echo \'$NEW_PASS\' |passwd --stdin $USER\r exit\r\";exp_continue}  
    }"  
done  

生成新密码文件：

# cat new_pass.txt   
192.168.18.217  root    n8wX3mU%      22  
192.168.18.218  root    c87;ZnnL      22

16、iptables 自动屏蔽访问网站频繁的IP

场景：恶意访问,安全防范

1）屏蔽每分钟访问超过200的IP

方法1：根据访问日志（Nginx为例）

#!/bin/bash
DATE=$(date +%d/%b/%Y:%H:%M)
ABNORMAL_IP=$(tail -n5000 access.log |grep $DATE |awk '{a[$1]++}END{for(i in a)if(a[i]>100)print i}')
#先tail防止文件过大，读取慢，数字可调整每分钟最大的访问量。awk不能直接过滤日志，因为包含特殊字符。
for IP in $ABNORMAL_IP; do
    if [ $(iptables -vnL |grep -c "$IP") -eq 0 ];
    then
        iptables -I INPUT -s $IP -j DROP
    fi
done

方法2：通过TCP建立的连接

#!/bin/bash
ABNORMAL_IP=$(netstat -an |awk '$4~/:80$/ && $6~/ESTABLISHED/{gsub(/:[0-9]+/,"",$5);{a[$5]++}}END{for(i in a)if(a[i]>100)print i}')
#gsub是将第五列（客户端IP）的冒号和端口去掉
for IP in $ABNORMAL_IP; do
    if [ $(iptables -vnL |grep -c "$IP") -eq 0 ]; then
        iptables -I INPUT -s $IP -j DROP
    fi
done

2）屏蔽每分钟SSH尝试登录超过10次的IP

方法1：通过lastb获取登录状态:

#!/bin/bash
DATE=$(date +"%a %b %e %H:%M") #星期月天时分  %e单数字时显示7，而%d显示07
ABNORMAL_IP=$(lastb |grep "$DATE" |awk '{a[$3]++}END{for(i in a)if(a[i]>10)print i}')
for IP in $ABNORMAL_IP; do
    if [ $(iptables -vnL |grep -c "$IP") -eq 0 ];
    then
        iptables -I INPUT -s $IP -j DROP
    fi
done

方法2：通过日志获取登录状态

#!/bin/bash  
DATE=$(date +"%b %d %H")  
ABNORMAL_IP="$(tail -n10000 /var/log/auth.log |grep "$DATE" |awk '/Failed/{a[$(NF-3)]++}END{for(i in a)if(a[i]>5)print i}')"  
for IP in $ABNORMAL_IP; do  
    if [ $(iptables -vnL |grep -c "$IP") -eq 0 ]; then  
        iptables -A INPUT -s $IP -j DROP          
        echo "$(date +"%F %T") - iptables -A INPUT -s $IP -j DROP" >>~/ssh-login-limit.log      
    fi  
done  

17、根据web访问日志，封禁请求量异常的IP，如IP在半小时后恢复正常，则解除封禁

#!/bin/bash
####################################################################################
#根据web访问日志，封禁请求量异常的IP，如IP在半小时后恢复正常，则解除封禁
####################################################################################
logfile=/data/log/access.log
#显示一分钟前的小时和分钟
d1=`date -d "-1 minute" +%H%M`
d2=`date +%M`
ipt=/sbin/iptables
ips=/tmp/ips.txt
block()
{
    #将一分钟前的日志全部过滤出来并提取IP以及统计访问次数
    grep '$d1:' $logfile|awk '{print $1}'|sort -n|uniq -c|sort -n > $ips
    #利用for循环将次数超过100的IP依次遍历出来并予以封禁
    for i in `awk '$1>100 {print $2}' $ips`
    do
        $ipt -I INPUT -p tcp --dport 80 -s $i -j REJECT
        echo "`date +%F-%T` $i" >> /tmp/badip.log
    done
}
unblock()
{
    #将封禁后所产生的pkts数量小于10的IP依次遍历予以解封
    for a in `$ipt -nvL INPUT --line-numbers |grep '0.0.0.0/0'|awk '$2<10 {print $1}'|sort -nr`
    do
        $ipt -D INPUT $a
    done
    $ipt -Z
}
#当时间在00分以及30分时执行解封函数
if [ $d2 -eq "00" ] || [ $d2 -eq "30" ]
then
    #要先解再封，因为刚刚封禁时产生的pkts数量很少
    unblock
    block
else
    block
fi

18、判断用户输入的是否为IP地址

方法1:

#!/bin/bash  
function check_ip(){  
    IP=$1  
    VALID_CHECK=$(echo $IP|awk -F. '$1< =255&&$2<=255&&$3<=255&&$4<=255{print "yes"}')  
    if echo $IP|grep -E "^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$">/dev/null; then  
        if [ $VALID_CHECK == "yes" ]; then  
            echo "$IP available."  
        else  
            echo "$IP not available!"  
        fi  
    else  
        echo "Format error!"  
    fi  
}  
check_ip 192.168.1.1  
check_ip 256.1.1.1

方法2：

#!/bin/bash  
function check_ip(){  
    IP=$1  
    if [[ $IP =~ ^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$ ]]; then  
        FIELD1=$(echo $IP|cut -d. -f1)  
        FIELD2=$(echo $IP|cut -d. -f2)  
        FIELD3=$(echo $IP|cut -d. -f3)  
        FIELD4=$(echo $IP|cut -d. -f4)  
        if [ $FIELD1 -le 255 -a $FIELD2 -le 255 -a $FIELD3 -le 255 -a $FIELD4 -le 255 ]; then  
            echo "$IP available."  
        else  
            echo "$IP not available!"  
        fi  
    else  
        echo "Format error!"  
    fi  
}  
check_ip 192.168.1.1  
check_ip 256.1.1.1

增加版：

加个死循环，如果IP可用就退出，不可用提示继续输入，并使用awk判断。

#!/bin/bash  
function check_ip(){  
    local IP=$1  
    VALID_CHECK=$(echo $IP|awk -F. '$1< =255&&$2<=255&&$3<=255&&$4<=255{print "yes"}')  
    if echo $IP|grep -E "^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$" >/dev/null; then  
        if [ $VALID_CHECK == "yes" ]; then  
            return 0  
        else  
            echo "$IP not available!"  
            return 1  
        fi  
    else  
        echo "Format error! Please input again."  
        return 1  
    fi  
}  
while true; do  
    read -p "Please enter IP: " IP  
    check_ip $IP  
    [ $? -eq 0 ] && break || continue  
done  

网站公告

今日签到

点亮在社区的每一天
去签到

热门文章

最新发布