Linux安全与高级应用（六）Linux Shell脚本编程的高级应用：条件测试与if语句的妙用

👍 个人网站：【 洛秋小站】【洛秋资源小站】

Linux Shell脚本编程的高级应用：条件测试与if语句的妙用

在Linux的世界中，Shell脚本作为系统管理员和开发者的重要工具，其强大之处在于可以自动化地处理各种任务。在众多Shell脚本功能中，条件测试和if语句是最为关键的部分之一。本文将深入探讨Shell脚本中的条件测试和if语句的高级应用，以帮助读者更好地理解和使用这些工具。

一、条件测试操作详解

条件测试是Shell脚本中判断逻辑的基础。它通过测试表达式的真伪来决定脚本的执行路径。常见的条件测试操作包括字符串比较、整数比较、文件测试和逻辑测试等。

1. 字符串比较

字符串比较主要用于判断两个字符串是否相等或不等。常用的操作符包括：

• =：判断两个字符串是否相同。
• !=：判断两个字符串是否不同。
• -z：判断字符串是否为空。

例如：

[root@localhost ~]# echo $LANG 
zh_CN.UTF-8
[root@localhost ~]# [ $LANG !=  "en.US" ] && echo  "Not en.US"
Not en.US

2. 整数比较

整数比较用于判断两个整数之间的关系。常用操作符有：

• -eq：等于
• -ne：不等于
• -gt：大于
• -lt：小于
• -ge：大于或等于
• -le：小于或等于

例如：

[root@localhost ~]# [ $(who | wc -l) -gt 5 ] && echo "Too many."
Too many.

3. 文件测试

文件测试用于检查文件或目录的存在性及其属性。常用操作符包括：

• -d：测试是否为目录
• -e：测试目录或文件是否存在
• -f：测试是否为文件
• -r：测试当前用户是否有权限读取
• -w：测试当前用户是否有权限写入
• -x：测试当前用户是否有权限执行

例如：

[root@localhost ~]# [ -d /media/cdrom ] && echo "YES"
YES

4. 逻辑测试

逻辑测试用于组合多个条件表达式，形成更复杂的判断逻辑。常用操作符包括：

• -a 或 &&：逻辑与
• -o 或 ||：逻辑或
• !：逻辑非

例如：

[root@localhost ~]# echo $USER
root
[root@localhost ~]# [ $USER != "teacher" ]  &&  echo "Not teacher"
Not teacher

二、if语句的结构与应用

if语句是条件测试的实际应用，是Shell脚本中实现逻辑判断和流程控制的核心语句。根据条件测试的结果，if语句可以决定是否执行某段代码。if语句主要有单分支结构、双分支结构和多分支结构三种形式。

1. 单分支结构

单分支结构用于在条件成立时执行一段代码，而在条件不成立时不执行任何操作。其基本格式为：

if 条件测试操作
then
    命令序列
fi

例如，判断挂载点目录是否存在，如果不存在则自动创建：

#!/bin/bash
MOUNT_DIR="/media/cdrom/"
if [ ! -d $MOUNT_DIR ]
then
    mkdir -p $MOUNT_DIR
fi

2. 双分支结构

双分支结构用于在条件成立时执行一段代码，而在条件不成立时执行另一段代码。其基本格式为：

if 条件测试操作
then
    命令序列1
else
    命令序列2
fi

例如，判断目标主机是否存活，并显示检测结果：

#!/bin/bash
ping -c 3 -i 0.2 -W 3 $1 &> /dev/null
if [ $? -eq 0 ]
then
    echo "Host $1 is up."
else
    echo "Host $1 is down."
fi

3. 多分支结构

多分支结构用于在多个条件之间进行选择。其基本格式为：

if 条件测试操作1
then
    命令序列1
elif 条件测试操作2 
then
    命令序列2
else
    命令序列3
fi

例如，判断分数范围并分出优秀、合格和不合格三个档次：

#!/bin/bash
read -p "请输入您的分数（0-100）：" GRADE
if [ $GRADE -ge 85 ] && [ $GRADE -le 100 ] ; then
    echo "$GRADE 分！优秀"
elif [ $GRADE -ge 70 ] && [ $GRADE -le 84 ] ; then
    echo "$GRADE 分，合格"
else
    echo "$GRADE 分？不合格"
fi

三、实际应用案例

通过以上对条件测试和if语句的介绍，我们可以编写更复杂和实用的Shell脚本。下面是一个监控脚本的例子，该脚本用于监控CPU使用率、内存使用率和根分区的占用率，当检测到异常时发送告警邮件。

1. 需求描述

编写监控脚本sysmon.sh，在以下情况时告警：

• 磁盘占用率超过90%
• CPU使用率超过80%
• 内存使用率超过90%

脚本每半小时检查一次，并通过邮件发送告警。

2. 实现思路

• 使用df、mpstat、free等命令提取各种监控指标
• 将各指标与正常值进行比较，保存异常情况
• 检查异常记录，若存在则发送告警邮件
• 设置crontab任务，定期调用sysmon.sh脚本

3. 代码实现

以下是sysmon.sh脚本的具体实现：

#!/bin/bash

# 获取当前的时间戳
timestamp=$(date +"%Y-%m-%d %H:%M:%S")

# 定义告警函数
send_alert() {
    local message=$1
    echo "$message" | mail -s "System Monitor Alert" admin@example.com
}

# 检查磁盘占用率
disk_usage=$(df / | grep / | awk '{ print $5 }' | sed 's/%//g')
if [ $disk_usage -gt 90 ]; then
    send_alert "[$timestamp] Disk usage is above 90%: ${disk_usage}%"
fi

# 检查CPU使用率
cpu_usage=$(mpstat 1 1 | grep "Average" | awk '{print 100 - $NF}')
if [ $(echo "$cpu_usage > 80" | bc) -ne 0 ]; then
    send_alert "[$timestamp] CPU usage is above 80%: ${cpu_usage}%"
fi

# 检查内存使用率
memory_usage=$(free | grep Mem | awk '{print $3/$2 * 100.0}')
if [ $(echo "$memory_usage > 90" | bc) -ne 0 ]; then
    send_alert "[$timestamp] Memory usage is above 90%: ${memory_usage}%"
fi

4. 设置定时任务

使用crontab设置每半小时执行一次sysmon.sh脚本：

*/30 * * * * /path/to/sysmon.sh

四、总结

Shell脚本中的条件测试和if语句是实现自动化任务的关键工具。通过灵活运用这些工具，我们可以编写高效、可靠的脚本来处理复杂的任务。在实际应用中，结合系统的具体需求，编写适合的Shell脚本，可以极大地提高工作效率和系统稳定性。希望本文对您理解和使用Shell脚本中的条件测试和if语句有所帮助。

👉 最后，愿大家都可以解决工作中和生活中遇到的难题，剑锋所指,所向披靡～