day030-Shell自动化编程-函数

孙克旭_2025-06-10 8:17

文章目录

  • [1. 循环控制语句](#1. 循环控制语句)
    • [1.1 案例-通过交互方式获取用户需要检查的ip/域名，用户输入q则退出](#1.1 案例-通过交互方式获取用户需要检查的ip/域名，用户输入q则退出)
  • [2. 函数](#2. 函数)
    • [2.1 用函数改造上面脚本](#2.1 用函数改造上面脚本)
    • [2.2 案例-通过ip.txt文件，检查常用端口是否开启](#2.2 案例-通过ip.txt文件，检查常用端口是否开启)
    • [2.3 函数与参数](#2.3 函数与参数)
  • [3. 颜色](#3. 颜色)
    • [3.1 案例-用户自定义函数库，自定义颜色](#3.1 案例-用户自定义函数库，自定义颜色)
  • [4. 思维导图](#4. 思维导图)

1. 循环控制语句

1.1 案例-通过交互方式获取用户需要检查的ip/域名，用户输入q则退出

shell 复制代码 
[root@oldboy99-Kylin /server/scripts/function_scripts]# cat check_ip.sh
#!/bin/bash
##############################################################
# File Name: check_ip.sh
# Version: V1.0
# Author: SunKexu
# Organization: www.oldboyedu.com
# Description:检查ip/域名
##############################################################
export LANG=en_US.UTF-8

# vars
tmp_file=`mktemp`
# command
# while 读取ip
while true
do
	read -p '请输入ip或url，结束请输入q：' ip
	[ -z "$ip" ] && {
		echo "输出错误请重新输入"
		continue
	} 
	[ "$ip" = "q" ] && {
		echo "输入结束，进行后续检查操作"
		break
	}
	echo $ip >> $tmp_file
done 
# 检查ip
while read line
do
	# ping 
	ping -c 1 -W 1 $line >/dev/null 2>&1
	[ $? -eq 0 ] && echo "$line 可以ping通" || echo "$line 不能ping通"
	# wget
	wget -S -T 1 -t 1 $line >/dev/null 2>&1
	[ $? -eq 0 ] && echo "$line 可以访问" || echo "$line 不能访问"
done <$tmp_file
[root@oldboy99-Kylin /server/scripts/function_scripts]# bash check_ip.sh
请输入ip或url，结束请输入q：baidu.com
请输入ip或url，结束请输入q：jd.com
请输入ip或url，结束请输入q：127.0.0.1
请输入ip或url，结束请输入q：q
输入结束，进行后续检查操作
baidu.com 可以ping通
baidu.com 可以访问
jd.com 可以ping通
jd.com 不能访问
127.0.0.1 可以ping通
127.0.0.1 不能访问

2. 函数

2.1 用函数改造上面脚本

shell 复制代码 
[root@oldboy99-Kylin /server/scripts/function_scripts]# cat check_ip2.sh
#!/bin/bash
##############################################################
# File Name: check_ip.sh
# Version: V1.0
# Author: SunKexu
# Organization: www.oldboyedu.com
# Description:检查ip/域名
##############################################################
export LANG=en_US.UTF-8

# vars
tmp_file=`mktemp`
# command
# while 读取ip
function input(){
	while true
	do
		read -p '请输入ip或url，结束请输入q：' ip
		[ -z "$ip" ] && {
			echo "输出错误请重新输入"
			continue
		} 
		[ "$ip" = "q" ] && {
			echo "输入结束，进行后续检查操作"
			break
		}
		echo $ip >> $tmp_file
	done 
	return $?
}
# 检查ip
function check_ip(){
	ping -c 1 -W 1 $line >/dev/null 2>&1
	[ $? -eq 0 ] && echo "$line 可以ping通" || echo "$line 不能ping通"
	return $?
}
# 检查wget
function check_wget(){
	wget -S -T 1 -t 1 $line >/dev/null 2>&1
	[ $? -eq 0 ] && echo "$line 可以访问" || echo "$line 不能访问"
	return $?
}
# 读取文件
function read_file(){
	while read line 
	do
		check_ip
		check_wget
	done < $tmp_file
	return $?
}
# main函数
function main(){
	input
	read_file
	return $?
}
main
[root@oldboy99-Kylin /server/scripts/function_scripts]# bash check_ip2.sh
请输入ip或url，结束请输入q：  
输出错误请重新输入
请输入ip或url，结束请输入q：baidu.com
请输入ip或url，结束请输入q：jd.com
请输入ip或url，结束请输入q：q
输入结束，进行后续检查操作
baidu.com 可以ping通
baidu.com 可以访问
jd.com 可以ping通
jd.com 可以访问

2.2 案例-通过ip.txt文件，检查常用端口是否开启

shell 复制代码 
[root@oldboy99-Kylin /server/scripts/function_scripts]# cat check_ip_port.sh
#!/bin/bash
##############################################################
# File Name: check_ip_port.sh
# Version: V1.0
# Author: SunKexu
# Organization: www.oldboyedu.com
# Description:检查ip的一些端口是否开启
##############################################################
export LANG=en_US.UTF-8

# vars
file="/server/scripts/log/ip.txt"
ports="22,80,443,3306,8080"
source /etc/os-release
# 检查nmap是否存在，不存在则按安装
function check_cmd(){
	which nmap >/dev/null 2>&1 ||{
		case "$ID" in 
			kylin|centos|rocky)
				yum clean all
				yum install -y nmap
				;;
			ubuntu|debian)
				apt install -y nmap
				;;
			*)
				echo "$ID未识别的系统请手动按照软件包：nmap，然后重新测试"
				exit 1
		esac
	}
	return $?
}
# 检查nmap
function check_nmap(){
	nmap -p $ports $line |grep '^[0-9]' |while read port state name
	do
		if [ "$state" = "open" ];then
			echo "服务器：$line ,端口：$port ,端口是否开启：是"
		else
			echo "服务器：$line ,端口：$port ,端口是否开启：否/屏蔽"
		fi
	done
	return $?
}
# 读取文件
function read_file(){
	# 检查ping
	while read line 
	do
		ping -c 1 -W 1 $line > /dev/null 2>&1 || {
			echo "$line 无法访问"
			continue
		}
		check_nmap
	done<$file
	return $?
}
# main
function main(){
	check_cmd
	read_file
	return $?
}

# command
main
[root@oldboy99-Kylin /server/scripts/function_scripts]# bash check_ip_port.sh
服务器：10.0.0.200 ,端口：22/tcp ,端口是否开启：否/屏蔽
服务器：10.0.0.200 ,端口：80/tcp ,端口是否开启：是
服务器：10.0.0.200 ,端口：443/tcp ,端口是否开启：否/屏蔽
服务器：10.0.0.200 ,端口：3306/tcp ,端口是否开启：否/屏蔽
服务器：10.0.0.200 ,端口：8080/tcp ,端口是否开启：否/屏蔽
服务器：baidu.com ,端口：22/tcp ,端口是否开启：否/屏蔽
服务器：baidu.com ,端口：80/tcp ,端口是否开启：是
服务器：baidu.com ,端口：443/tcp ,端口是否开启：是
服务器：baidu.com ,端口：3306/tcp ,端口是否开启：否/屏蔽
服务器：baidu.com ,端口：8080/tcp ,端口是否开启：否/屏蔽
服务器：jd.com ,端口：22/tcp ,端口是否开启：否/屏蔽
服务器：jd.com ,端口：80/tcp ,端口是否开启：是
服务器：jd.com ,端口：443/tcp ,端口是否开启：是
服务器：jd.com ,端口：3306/tcp ,端口是否开启：否/屏蔽
服务器：jd.com ,端口：8080/tcp ,端口是否开启：否/屏蔽
服务器：csdn.net ,端口：22/tcp ,端口是否开启：否/屏蔽
服务器：csdn.net ,端口：80/tcp ,端口是否开启：是
服务器：csdn.net ,端口：443/tcp ,端口是否开启：是
服务器：csdn.net ,端口：3306/tcp ,端口是否开启：否/屏蔽
服务器：csdn.net ,端口：8080/tcp ,端口是否开启：否/屏蔽
服务器：520skx.com ,端口：22/tcp ,端口是否开启：是
服务器：520skx.com ,端口：80/tcp ,端口是否开启：是
服务器：520skx.com ,端口：443/tcp ,端口是否开启：否/屏蔽
服务器：520skx.com ,端口：3306/tcp ,端口是否开启：否/屏蔽
服务器：520skx.com ,端口：8080/tcp ,端口是否开启：否/屏蔽
服务器：www.520skx.com ,端口：22/tcp ,端口是否开启：是
服务器：www.520skx.com ,端口：80/tcp ,端口是否开启：是
服务器：www.520skx.com ,端口：443/tcp ,端口是否开启：否/屏蔽
服务器：www.520skx.com ,端口：3306/tcp ,端口是否开启：否/屏蔽
服务器：www.520skx.com ,端口：8080/tcp ,端口是否开启：否/屏蔽
Failed to resolve "孙克旭.com".
WARNING: No targets were specified, so 0 hosts scanned.
123waeffawe.com 无法访问

2.3 函数与参数

特殊变量 脚本中的含义 函数中的含义
$n 脚本的第n个参数 函数的第n个参数
$0 脚本名字 脚本名字
$# 脚本参数个数 函数参数个数
$* 脚本所有参数 函数所有参数
$@ 脚本所有参数 函数所有参数
$? 上一个命令的返回值 上一个命令的返回值

3. 颜色

  • \E 或\033 表示要开启这种功能。

  • 1;31m

  • \E[0m 颜色设置结束。

3.1 案例-用户自定义函数库，自定义颜色

shell 复制代码 
[root@oldboy99-Kylin /server/scripts/function_scripts]# cat color_func.sh 
#!/bin/bash
##############################################################
# File Name: color_func.sh
# Version: V1.0
# Author: SunKexu
# Organization: www.oldboyedu.com
# Description:
##############################################################

#1.红色函数
function redecho() {
 str="$@"
 echo -e "\E[5;31m${str}\E[0m"
 }

 #2.红色函数
function greenecho() {
 str="$@"
 echo -e "\E[1;32m${str}\E[0m"
 }
 #3.红色函数
function yellowecho() {
 str="$@"
 echo -e "\E[1;33m${str}\E[0m"
 }
 #4.红色函数
function blueecho() {
 str="$@"
 echo -e "\E[1;34m${str}\E[0m"
 }

4. 思维导图

【金山文档】 思维导图 https://www.kdocs.cn/l/co3I7PtpTYQX

相关推荐
不爱学英文的码字机器41 分钟前
持续交付的进化：从DevOps到AI驱动的IT新动能
运维·人工智能·devops
什么半岛铁盒43 分钟前
【Linux系统】Linux环境变量：系统配置的隐形指挥官
linux
Lw老王要学习1 小时前
Linux容器篇、第一章_02Rocky9.5 系统下 Docker 的持久化操作与 Dockerfile 指令详解
linux·运维·docker·容器·云计算
橙子小哥的代码世界2 小时前
【大模型RAG】Docker 一键部署 Milvus 完整攻略
linux·docker·大模型·milvus·向量数据库·rag
斯普信专业组2 小时前
Kafka主题运维全指南：从基础配置到故障处理
运维·分布式·kafka
中国软件测试质量协会2 小时前
UI自动化测试：现状，效果和最佳实践
ui·自动化·ui自动化·webui·界面自动化
倔强的石头1062 小时前
【Linux指南】用户与系统基础操作
linux·运维·服务器
云上艺旅3 小时前
centos升级内核
linux·运维·centos
kaikaile19953 小时前
centos开启samba服务
linux·运维·centos
云上艺旅3 小时前
centos部署k8s v1.33版本
linux·云原生·kubernetes·centos
热门推荐
01【图像处理与机器视觉】XJTU期末考点02从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑03KGG转MP3工具|非KGM文件|解密音频04Coze扣子平台完整体验和实践（附国内和国际版对比）05YOLOv8入门 | 重要性能衡量指标、训练结果评价及分析及影响mAP的因素【发论文关注的指标】06【SpeedAI科研小助手】2分钟极速解决知网维普重复率、AIGC率过高，一键全文降！文件格式不变，公式都保留的！07海康Visionmaster-常见问题排查方法-启动阶段08DeepSeek各版本说明与优缺点分析09VMware虚拟机安装Win7专业版保姆级教程（附镜像包）10零代码入门 | Coze——让大模型接入自己的数据库