实现U盘升级的shell脚本

已于 2024-07-20 09:04:38 修改
阅读量336 收藏 3
点赞数 4
文章标签： linux c语言 eclipse ssh
于 2024-07-20 09:03:14 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_45183402/article/details/140565134
版权

这个脚本是放到应用程序里去使用的，所以main函数里主要是三个功能：（1）杀死正在运行的应用进程（2）把新的应用程序覆盖旧的（3）重启应用程序。但实现着三个功能需要一些函数去辅助，以下先讲讲其中一些函数，再把完整的代码给大家。

目录

进程是否存在函数

比较文件函数

升级函数

杀死进程函数

完整代码

实现效果

进程是否存在函数

# 进程是否存在
# 参数：进程名字
# 返回值：RET_EP 1 存在，0不存在
ExistProcess ( )
{
	#echo "ps -ef | grep $1 | grep -v grep"
	CHECK=`ps -ef | grep $1 | grep -v grep`  #列出所有的进程信息、再从列出项中找到包含进程名字的、grep -v 排除掉包含grep的
	                                         #因为本条指令就是包含进程名的，但我们要查找的是进程本身
	if [ -z "$CHECK" ];then                  #检查变量$CHECK是否为空，-z测试字符串是否为空
		RET_EP=0
	else
		RET_EP=1
	fi
	#echo $CHECK
}

大家可以先挨着输入一下这些命令，就知道每条命令具体再处理什么。可以用echo打印到出口去看看，我这里是注释掉了。

比较文件函数

我们升级应用程序一定会有一个比较的过程，如果不一样则升级，一样就无需升级。

# 两个文件是否相同
# 参数1：文件1路径
# 参数2：文件2路径
# 返回值：RET_IS_SAME_FILE，1 相同  0 不相同
IsSameFile ( )
{
	LOG "==========in IsSameFile=========="
	LOG "param1 is $1"
	LOG "param2 is $2"
	if [[ $# -lt 2 ]];then                  #检查传递给函数的参数数量是否少于2个，$# ——添加到Shell的参数个数 -lt比较运算符小于，小于2说明给的参数不够
		RET_IS_SAME_FILE=0
	fi

	FILE_DATE1=`ls -e "$1" | cut -c30-69`   #列出参数1路径下应用程序的信息，截取第30到69的字符（代码大小，最后修改日期时间）
	FILE_DATE2=`ls -e "$2" | cut -c30-69`   #列出参数2路径下应用程序的信息，截取第30到69的字符（代码大小，最后修改日期时间）
	#DATE_DIFF=$((FILE_DATE1 - FILE_DATE2))
	LOG "$FILE_DATE1"
	LOG "$FILE_DATE2"

	if [[ "$FILE_DATE1" == "$FILE_DATE2" ]];then   #比较文件大小及最后修改时间，一样则返回1，不一样则返回0
		RET_IS_SAME_FILE=1
	else
		RET_IS_SAME_FILE=0
	fi
}

LOG是我们的一个日志函数，下面完整的代码里会有，我们这里是通过比较文件的修改时间，来判断需不需要升级的。

升级函数

其实升级，就是一个文件拷贝的过程。

# 更新文件
# 参数1 新文件路径
# 参数2 目标文件路径
# 返回值 RET_UPDATE： 0-更新失败  1-成功	2-无需更新
UpdateFile ( )
{
	IsSameFile $1 $2                                     #调用比较程序函数，一样则不需要更新
	#LOG "$1 $2 is same : $RET_IS_SAME_FILE"
	if [[ $RET_IS_SAME_FILE == 1 ]];then
		LOG "$1 文件一致，无需更新！"
		RET_UPDATE=2
	else                                                 #不一样则将参数一路径下的文件拷贝到参数二路径下文件
		cp -p "$1" "$2"
		if [[ $? == 0 ]];then                            #用来检查上一个命令执行的退出状态码是否为0
			LOG "更新文件, 成功: $2"
			RET_UPDATE=1
		else
			LOG "更新文件, 失败: $2"
			RET_UPDATE=0
		fi
	fi
}

杀死进程函数

# 杀死进程
# 参数 进程名字
KillProcess () {
echo $1
  Linux_new_id=`ps -ef | grep $1 | grep -v grep | awk '{print $1}'`
echo $Linux_new_id
 
for id in $Linux_new_id
do
    kill $id  
    echo "killed $id"  
done

}

在我们升级前需要先kill掉进程，所以这个函数就是实现，先找的这个进程本身，并将它杀死。

完整代码

完整代码里还包含了判断升级状态，和创建文件目录，打印日志到指定路径的小功能，更加完善。大家可以参考。大家如果测试时对路径做相应的修改就可以了。加入我们运行这hello1程序，在U盘里放入hello2,然后插入即可看到实现的效果。

#!/usr/bin/env bash

HARDDISK1_PATH='/opt'     #放应用程序和回滚版本应用程序（其实总共就是这两个目录）
HARDDISK12_PATH='/HardDisk12'   #放日志记录
# DRU_200C主控程序目录
PROGRAM_PATH='/opt'       #就是第一个目录，只不过给个定义
LOG_PATH='/HardDisk12/USB_log'  #日志的具体目录

PROGRAM_NAME='hello'            #程序名称




# 升级结果缓存
# 0 : 升级失败
# 1 : 升级成功
# 2 : 无需升级
TEMP_RET_UPDATE_DRU_200C_ZK=2      #初始升级状态为无需升级

# 功能：DRU_200C主控升级状态处理
# 输入参数：升级状态
# 注：
# 若升级成功，且当前总的升级状态不是“升级失败”，则更新为“升级成功”；
# 若升级失败，则更新总的升级状态为“升级失败”
# 否则，不处理
DRU_200CUpdateStatusHandler ()
{
	if [[ $1 == 1 ]];then                            # 若升级成功，且当前总的升级状态不是“升级失败”，则更新为“升级成功”；
	if [[ $TEMP_RET_UPDATE_DRU_200C_ZK != 0 ]];then
		TEMP_RET_UPDATE_DRU_200C_ZK=1
	fi
	elif [[ $1 == 0 ]];then                          # 若升级失败，则更新总的升级状态为“升级失败”
		TEMP_RET_UPDATE_DRU_200C_ZK=0
	fi
}

# 如果没有这两个目录就创建 HardDisk1 and HardDisk12 
if [[ ! -d "$HARDDISK1_PATH" ]];then
  mkdir "$HARDDISK1_PATH"
fi
if [[ ! -d "$HARDDISK12_PATH" ]];then
  mkdir "$HARDDISK12_PATH"
fi

# 如果没有就创建  myapp-log directory
if [[ ! -d "$LOG_PATH" ]];then
  mkdir "$LOG_PATH"
fi

#log函数，记录日志，保存在LOG_PATH='/HardDisk12/log_sys'里
LOG ( )
{
	LOG_DATE=`date '+%Y-%m-%d'`                                        #以日期命名log名称
	echo `date '+%Y-%m-%d %H:%M:%S'` $1 >> "$LOG_PATH/$LOG_DATE.txt"   #打印日期时间到日志里
	echo `date '+%Y-%m-%d %H:%M:%S'` $1                                #打印日期时间到终端
}

##########################################添加myapp版本号#################################################
                              LOG "myapp version: V2.0.0 20211102"
##########################################################################################################

# 进程是否存在
# 参数：进程名字
# 返回值：RET_EP 1 存在，0不存在
ExistProcess ( )
{
	#echo "ps -ef | grep $1 | grep -v grep"
	CHECK=`ps -ef | grep $1 | grep -v grep`  #列出所有的进程信息、再从列出项中找到包含进程名字的、grep -v 排除掉包含grep的
	                                         #因为本条指令就是包含进程名的，但我们要查找的是进程本身
	if [ -z "$CHECK" ];then                  #检查变量$CHECK是否为空，-z测试字符串是否为空
		RET_EP=0
	else
		RET_EP=1
	fi
	#echo $CHECK
}

# 两个文件是否相同
# 参数1：文件1路径
# 参数2：文件2路径
# 返回值：RET_IS_SAME_FILE，1 相同  0 不相同
IsSameFile ( )
{
	LOG "==========in IsSameFile=========="
	LOG "param1 is $1"
	LOG "param2 is $2"
	if [[ $# -lt 2 ]];then                  #检查传递给函数的参数数量是否少于2个，$# ——添加到Shell的参数个数 -lt比较运算符小于，小于2说明给的参数不够
		RET_IS_SAME_FILE=0
	fi

	FILE_DATE1=`ls -e "$1" | cut -c30-69`   #列出参数1路径下应用程序的信息，截取第30到69的字符（代码大小，最后修改日期时间）
	FILE_DATE2=`ls -e "$2" | cut -c30-69`   #列出参数2路径下应用程序的信息，截取第30到69的字符（代码大小，最后修改日期时间）
	#DATE_DIFF=$((FILE_DATE1 - FILE_DATE2))
	LOG "$FILE_DATE1"
	LOG "$FILE_DATE2"

	if [[ "$FILE_DATE1" == "$FILE_DATE2" ]];then   #比较文件大小及最后修改时间，一样则返回1，不一样则返回0
		RET_IS_SAME_FILE=1
	else
		RET_IS_SAME_FILE=0
	fi
}


# 更新文件
# 参数1 新文件路径
# 参数2 目标文件路径
# 返回值 RET_UPDATE： 0-更新失败  1-成功	2-无需更新
UpdateFile ( )
{
	IsSameFile $1 $2                                     #调用比较程序函数，一样则不需要更新
	#LOG "$1 $2 is same : $RET_IS_SAME_FILE"
	if [[ $RET_IS_SAME_FILE == 1 ]];then
		LOG "$1 文件一致，无需更新！"
		RET_UPDATE=2
	else                                                 #不一样则将参数一路径下的文件拷贝到参数二路径下文件
		cp -p "$1" "$2"
		if [[ $? == 0 ]];then                            #用来检查上一个命令执行的退出状态码是否为0
			LOG "更新文件, 成功: $2"
			RET_UPDATE=1
		else
			LOG "更新文件, 失败: $2"
			RET_UPDATE=0
		fi
	fi
}

# 杀死进程
# 参数 进程名字
KillProcess () {
echo $1
  Linux_new_id=`ps -ef | grep $1 | grep -v grep | awk '{print $1}'`
echo $Linux_new_id
 
for id in $Linux_new_id
do
    kill $id  
    echo "killed $id"  
done

}
	
main(){
		GREPRET=`grep 'sd[a-z][0-9]\? /media' /proc/mounts` #在/proc/mounts文件中查找挂载在/media目录下的块设备（通常代表U盘或硬盘分区），
															#sd[a-z][0-9]\?匹配如sda1、sdb等常见的块设备名称。
		UDISK_PATH=$(echo $GREPRET | { read UP1 UP2 UP3; echo $UP2; })
		echo $UDISK_PATH
		if [[ -e "$UDISK_PATH/$PROGRAM_NAME" ]];then
			IsSameFile "$UDISK_PATH/$PROGRAM_NAME" "$PROGRAM_PATH/$PROGRAM_NAME"
			if [ $RET_IS_SAME_FILE == 1 ];then
				LOG "DRU_200C_ZK Program need not update"
				exit 1
			else
				LOG "Check if the DRU_200C_ZK process exists?"
				ExistProcess "$PROGRAM_NAME"
				if [ $RET_EP == 1 ];then
					LOG "Check ok, quit DRU_200C_ZK program"
					KillProcess $PROGRAM_NAME
				fi
				UpdateFile "$UDISK_PATH/$PROGRAM_NAME" "$PROGRAM_PATH/$PROGRAM_NAME"
				DRU_200CUpdateStatusHandler $RET_UPDATE
				
				if [[ $TEMP_RET_UPDATE_DRU_200C_ZK == 1 ]];then
					LOG "Update DRU_200C_ZK Program, success"
					echo -e -n "\x00" > "$FILE_DOWNLOADS_VERSION"
					cd "$PROGRAM_PATH"
					"./$PROGRAM_NAME" & 					#非阻塞运行应用程序
					sleep 10s
					exit 1
				elif [[ $TEMP_RET_UPDATE_DRU_200C_ZK == 2 ]];then
					LOG "DRU_200C_ZK Program need not update"  #前面存在IsSameFile判断，实际上不存在该分支，只作为逻辑出口
					
				else
					LOG "Update DRU_200C_ZK Program, fail: copy fail"
					exit 0
				fi
			fi
		else
			LOG "DRU_200C_ZK upgrade program does not exist"
			exit 1
		fi

	
#	sleep 10s
}

main

实现效果

刚开始它会检测到版本不一样实现升级，当升级完后再次重启发现U盘里和运行的是一个版本后则无需升级了。

m0_45183402
关注
UEFI-SHELL 环境 U盘制作
weixin_49423593的博客
06-30 1461
访问：https://github.com/pbatard/UEFI-Shell/releases。然后用rufus 做启动盘（Rufus：https://rufus.ie/en/）
EFI shell U盘制作 刷BIOS
07-24
制作EFI SHELL U盘的工具 适用于英特尔平台 AMD未测试
【Linux】将U盘中的程序更新到开发板中 shell 脚本
-凌凌漆-的博客
05-31 303
【代码】【Linux】将U盘中的程序更新到开发板中 shell 脚本。
放在U盘根目录就可运行的EFI shell
02-05
下载压缩包解压后，把 EFI 目录放在U盘的根目录，启动电脑，从U盘启动，就可以进入EFI shell了，不需要安装。
使用Shell脚本实现USB设备的挂载与文件复制
高性价比服务器就选：蓝易云
04-19 322
因此，在使用该脚本之前，请确保你已经理解了每一行代码是如何工作的，并且已经做好了相应准备工作以处理可能出现的各种情况。这个脚本首先定义了USB设备的设备文件路径和挂载点。接着，它检查设备文件是否存在，并尝试将其挂载到指定的挂载点。在这个脚本中，我们将会检测USB设备是否已经连接到系统上，并且将其挂载到指定目录。在Linux系统中，我们可以使用Shell脚本来实现USB设备的挂载以及文件的复制。如果USB设备成功地被挂载了，脚本将会尝试从该USB设备复制一个指定的文件到另一个目录。
设计完成一个简单的shell脚本程序！
qq_43527345的博客
03-20 5246
shell程序后缀.sh 执行两种方法 1.sh xx.sh 2.命令1：chmod +x xx.sh 命令2：./xx.sh 参考链接：https://blog.csdn.net/weixin_41143631/article/details/93633664 感谢大佬 设计完成一个简单的shell脚本程序！ 编写一个shell程序，使用一个菜单界面，方便Linux用户对U盘的加载，卸载操作。程...
使用shell脚本实现USB设备的加载与文件复制
01-09
在Linux操作系统中，如果插入一个USB设备，需要用mount挂载命令才能实现这个设备的加载，下面写一个USB设备挂载与文件复制的Shell程序，程序需求： 　1、运行时，提示用户输入“y”或者“Y”，确定是否挂载USB设备...
Linux系统下执行U盘中的脚本文件
01-20
在本文中，我们将深入探讨如何在Linux环境下处理U盘中的脚本文件，特别是遇到权限问题以及如何使用`chmod`命令和`sh`命令来解决这些问题。 首先，当我们在Linux中创建了一个脚本文件，比如`script.sh`，我们通常会...
编写Shell脚本
03-04
在IT行业中，Shell脚本是Linux和Unix操作系统中不可或缺的一部分，尤其对于系统管理员和开发者而言，熟练掌握Shell脚本编写技巧至关重要。标题“编写Shell脚本”表明我们要深入探讨这一主题，而描述则强调了Shell...
禁止或容许使用U盘脚本
07-26
接下来，我们将详细探讨这两种脚本以及如何通过它们来实现U盘的禁止和启用功能。 1. **批处理文件（BAT）**：批处理文件是基于DOS命令的简单脚本，用于自动化一系列操作系统命令。在禁止或允许U盘使用时，批处理...
详解Linux实现U盘自动挂载（图文教程）
09-15
本文将深入讲解如何在Linux中实现U盘的自动挂载，包括涉及的核心概念和技术细节。 1. 设备节点的创建： 当U盘插入Linux系统时，系统会识别出新的设备，并通过`device_create()`函数创建相应的设备节点，如`/dev/...
linux下centos7 ubuntu利用shell进行U盘挂载步骤
xie686106的博客
10-07 2380
linux下centos7 ubuntu利用shell进行U盘挂载步骤
U盘shell启动盘制作
热门推荐
碧海的专栏
02-25 1万+
写这篇文章是因为，虽然这种功能不是很常用，但需要时，查资料是真折腾。故记在这里，以备需要时查阅。 以下制作步骤，已经过验证。   制作UEFI USB启动盘.  分两种情况，如果目标平台是UEFI平台，按如下步骤来做：  1.  格式化U盘为FAT(FAT,FAT16, FAT32)格式  2.  在U盘上建立目录 efi\boot  3.  将efi的应用程序 copy到 ef
Linux系统管理易错点总结（一）——使用shell命令挂载U盘的问题
艾醒的博客
12-08 1181
盘点linux操作系统上挂载U盘出现的问题以及解决方法
Linux —— 网络基础（一）
学习C++的小陈同学
09-22 1004
本篇只是简单的建立一个关于网络的基本概念和框架，有很多概念上不太完整，只是自己为了自己在初入学习网络时的简单粗俗理解，方便后续深入的学习网络相关的知识，如果有需要纠错的地方，可以评论一起讨论。
【Linux实践】实验四：Shell实用功能及文件权限
Fulling的博客
09-25 715
实验内容 1、使用命令“cat /etc/named.conf”设置为别名named，然后再取消别名。 2、使用echo命令和输出重定向创建文本文件/root/nn，内容是hello，然后再使用追加重定向输入内容为word。 3、使用管道方式分页显示/etc/passwd的内容。 4、分别用文字设定法和数字设定法，对/root/ab文件设置权限，所有者为读取、写入和执行权限，同组用户为读取和写入权限，其他用户没有任何权限。 5、将文件/root/ab所有者更改为用户zhangsan。 6、将目录/root/
Linux 安装redis主从模式+哨兵模式3台节点
最新发布
一千个读者就有一千个哈姆雷特
09-27 190
Linux 安装redis主从模式+哨兵模式3台节点
shell脚本u盘管理
10-24
下面是一个简单的Shell脚本示例，可以实现U盘的挂载和文件复制功能： ``` #!/bin/bash echo "是否挂载USB设备？(y/n)" read answer if [ "$answer" == "y" ] || [ "$answer" == "Y" ]; then sudo mount /dev/sdb1 ...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值