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

相关推荐
##echo28 分钟前
嵌入式Linux裸机开发笔记9(IMX6ULL)GPIO 中断实验（1）
linux·c语言·笔记·单片机·嵌入式硬件
Reggie_L35 分钟前
RabbiteMQ安装-ubuntu
linux·ubuntu·ruby
AIGC_北苏1 小时前
让UV管理一切！！！
linux·人工智能·uv
YCY^v^3 小时前
centos 7 开启80，443端口,怎么弄？
linux·运维·centos
北南京海3 小时前
[Linux]进程地址空间
linux·运维·服务器
yzx9910134 小时前
服务器生成图片
运维·服务器
db_murphy5 小时前
Oracle数据块8KB、OS默认认块管理4KB，是否需调整大小为一致？
linux
小阳睡不醒8 小时前
小白成长之路-部署Zabbix7（二）
android·运维
mCell8 小时前
从删库到跑路？这50个Linux命令能保你职业生涯
linux·windows·macos
杰克逊的日记8 小时前
GPU运维常见问题处理
linux·运维·gpu
热门推荐
01Qwen3-Coder 快速上手教程 | Qwen Code + Claude Code02vue数据变化但页面不变03全球最强模型Grok4，国内已可免费使用！（附教程）04KGG转MP3工具|非KGM文件|解密音频05扣子开源本地部署教程 丨Coze智能体小白喂饭级指南06sqli-labs 靶场 less-8、9、10 第八关到第十关详解：布尔注入，时间注入07干翻 Typora！MilkUp：完全免费的桌面端 Markdown 编辑器！08Coze 开源了，送上保姆级私有化部署方案【建议收藏】09【2025.7.18】更新vscode后所有.vue文件template标签后报红的临时解决办法，Vue - Official 插件3.0.2导致10Claude Code用不了？来试下Qwen3-Coder加持的Qwen Code吧