Ubuntu-bbr.sh - XuanLove.Download

#!/usr/bin/env bash
PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
export PATH

#=================================================
#	System Required: Debian/Ubuntu
#	Description: TCP-BBR
#	Version: 1.0.2
#	Author: Toyo
#	Blog: https://www.xuanlove.net/
#=================================================

deb_ver_new="linux-image-4.9.0-040900-generic"
deb_ver_new_url="http://www.xuanlove.download/kernel/Ubuntu/linux-image-4.9.0-040900-generic_4.9.0-040900.201612111631"

#检查系统
check_sys(){
	if [[ -f /etc/redhat-release ]]; then
		release="centos"
	elif cat /etc/issue | grep -q -E -i "debian"; then
		release="debian"
	elif cat /etc/issue | grep -q -E -i "ubuntu"; then
		release="ubuntu"
	elif cat /etc/issue | grep -q -E -i "centos|red hat|redhat"; then
		release="centos"
	elif cat /proc/version | grep -q -E -i "debian"; then
		release="debian"
	elif cat /proc/version | grep -q -E -i "ubuntu"; then
		release="ubuntu"
	elif cat /proc/version | grep -q -E -i "centos|red hat|redhat"; then
		release="centos"
    fi
	bit=`uname -m`
}
#检查内核是否满足
check_deb_off(){
	deb_ver=`dpkg -l|grep linux-image | awk '{print $2}' | grep 'linux-image-4.9.0'`
	if [ "${deb_ver}" != "" ]; then
		if [ "${deb_ver}" = "${deb_ver_new}" ]; then
			echo -e "\033[42;37m [信息] \033[0m 检测到 内核版本 已满足要求，继续..."
		else
			echo -e "\033[42;37m [错误] \033[0m 检测到 内核版本 不是最新版本，建议重新使用 bash bbr.sh 来升级内核 !"
		fi
	else
		echo -e "\033[41;37m [错误] \033[0m 检测到 内核版本 不支持开启BBR，请使用 bash bbr.sh install 来安装 !"
		exit 1
	fi
}
# 删除其余内核
del_deb(){
	deb_total=`dpkg -l|grep linux-image | awk '{print $2}' | grep -v "${deb_ver_new}" | wc -l`
	if [ "${deb_total}" > "1" ]; then
		echo -e "\033[42;37m [信息] \033[0m 检测到 ${deb_total} 个其余内核，开始卸载..."
		for((integer = 1; integer <= ${deb_total}; integer++))
		do
			deb_del=`dpkg -l|grep linux-image | awk '{print $2}' | grep -v "${deb_ver_new}" | head -${integer}`
			echo -e "\033[42;37m [信息] \033[0m 开始卸载 ${deb_del} 内核..."
			apt-get purge -y ${deb_del}
			echo -e "\033[42;37m [信息] \033[0m 卸载 ${deb_del} 内核卸载完成，继续..."
		done
		deb_total=`dpkg -l|grep linux-image | awk '{print $2}' | grep -v "${deb_ver_new}" | wc -l`
		
		if [ "${deb_total}" = "0" ]; then
			echo -e "\033[42;37m [信息] \033[0m 内核卸载完毕，继续..."
		else
			echo -e "\033[41;37m [错误] \033[0m 内核卸载异常，请检查 !"
			exit 1
		fi
	else
		echo -e "\033[41;37m [错误] \033[0m 检测到 内核 数量不正确，请检查 !"
		exit 1
	fi
}
del_deb_over(){
	del_deb
	update-grub
	read -p "需要重启VPS后，才能安装完成BBR，是否现在重启 ? [Y/n] :" yn
	[ -z "${yn}" ] && yn="y"
		if [[ $yn == [Yy] ]]; then
		echo -e "\033[41;37m [信息] \033[0m VPS 重启中..."
		reboot
		fi
}
# 安装BBR
installbbr(){
# 判断是否安装BBR
	deb_ver=`dpkg -l|grep linux-image | awk '{print $2}' | grep 'linux-image-4.9.0'`
	if [ "${deb_ver}" != "" ]; then	
		if [ "${deb_ver}" = "${deb_ver_new}" ]; then
			echo -e "\033[42;37m [信息] \033[0m 检测到 内核版本 已满足要求，无需继续安装 !"
			deb_total=`dpkg -l|grep linux-image | awk '{print $2}' | grep -v "${deb_ver_new}" | wc -l`
			if [ "${deb_total}" != "0" ]; then
				echo -e "\033[41;37m [错误] \033[0m 检测到内核数量异常，存在多余内核，开始删除..."
				del_deb_over
			else
				exit 1
			fi
		else
			echo -e "\033[42;37m [错误] \033[0m 检测到 内核版本 不是最新版本，升级内核..."
		fi
	else
		echo -e "\033[41;37m [错误] \033[0m 检测到 内核版本 不支持开启BBR，开始安装..."
	fi
	
	check_sys
# 系统判断
	if [[ ${release} != "debian" ]]; then
		if [[ ${release} != "ubuntu" ]]; then
			echo -e "\033[41;37m [错误] \033[0m 本脚本不支持当前系统 !"
			exit 1
		fi
	fi
	
#修改DNS为8.8.8.8
	echo "nameserver 8.8.8.8" > /etc/resolv.conf
	echo "nameserver 8.8.4.4" >> /etc/resolv.conf
# 判断系统位数
	if [ ${bit} == "x86_64" ]; then
		echo -e "\033[42;37m [信息] \033[0m 检测到系统是 x86_64 位数，开始下载内核文件..."
		wget -N -O linux-image-4.9.0-amd64.deb "${deb_ver_new_url}_amd64.deb"
		# 判断是否下载成功
		if [ -s "linux-image-4.9.0-amd64.deb" ]; then
			echo -e "\033[42;37m [信息] \033[0m 内核文件下载成功，开始安装内核..."
		else
			echo -e "\033[41;37m [错误] \033[0m 内核文件下载失败，请检查 !"
			exit 1
		fi
		dpkg -i linux-image-4.9.0-amd64.deb
	elif [ ${bit} == "i386" ]; then
		echo -e "\033[42;37m [信息] \033[0m 检测到系统是 i386 位数，开始下载内核文件..."
		wget -N -O linux-image-4.9.0-i386.deb "${deb_ver_new_url}_i386.deb"
		# 判断是否下载成功
		if [ -s "linux-image-4.9.0-i386.deb" ]; then
			echo -e "\033[42;37m [信息] \033[0m 内核文件下载成功，开始安装内核..."
		else
			echo -e "\033[41;37m [错误] \033[0m 内核文件下载失败，请检查 !"
			exit 1
		fi
		dpkg -i linux-image-4.9.0-i386.deb
	else
		echo -e "\033[41;37m [错误] \033[0m 不支持 ${bit} !"
		exit 1
	fi
	#判断内核是否安装成功
	deb_ver=`dpkg -l | grep linux-image | awk '{print $2}' | grep "${deb_ver_new}"`
	if [ "${deb_ver}" != "" ]; then
		echo -e "\033[42;37m [信息] \033[0m 检测到 内核 已安装成功，开始卸载其余内核..."
	else
		echo -e "\033[41;37m [错误] \033[0m 检测到 内核版本 安装失败，请检查 !"
		exit 1
	fi
	
	del_deb_over
}
bbrstatus(){
	check_bbr_status_on=`sysctl net.ipv4.tcp_available_congestion_control | awk '{print $3}'`
	if [ "${check_bbr_status_on}" = "bbr" ]; then
		echo -e "\033[42;37m [信息] \033[0m 检测到 BBR 已开启 !"
		# 检查是否启动BBR
		check_bbr_status_off=`lsmod | grep bbr`
		if [ "${check_bbr_status_off}" = "" ]; then
			echo -e "\033[41;37m [错误] \033[0m 检测到 BBR 已开启但未正常启动，请检查 !"
		else
			echo -e "\033[42;37m [信息] \033[0m 检测到 BBR 已开启并已正常启动 !"
		fi
		exit 1
	fi
}
# 开启BBR
startbbr(){
# 检查是否安装
	check_deb_off
# 检查是否开启BBR
	bbrstatus
	# 如果原来添加的有全部删除
	sed -i '/net\.core\.default_qdisc=fq/d' /etc/sysctl.conf
	sed -i '/net\.ipv4\.tcp_congestion_control=bbr/d' /etc/sysctl.conf
	
	echo "net.core.default_qdisc=fq" >> /etc/sysctl.conf
	echo "net.ipv4.tcp_congestion_control=bbr" >> /etc/sysctl.conf
	sysctl -p
	sleep 1s
	bbrstatus
	echo -e "\033[41;37m [错误] \033[0m BBR 启动失败，请检查 !"
}
# 关闭BBR
stopbbr(){
# 检查是否安装
	check_deb_off
	sed -i '/net\.core\.default_qdisc=fq/d' /etc/sysctl.conf
	sed -i '/net\.ipv4\.tcp_congestion_control=bbr/d' /etc/sysctl.conf
	sysctl -p
	sleep 1s
	
	read -p "需要重启VPS后，才能彻底停止BBR，是否现在重启 ? [Y/n] :" yn
	[ -z "${yn}" ] && yn="y"
		if [[ $yn == [Yy] ]]; then
		echo -e "\033[41;37m [信息] \033[0m VPS 重启中..."
		reboot
		fi
}
# 查看BBR状态
statusbbr(){
# 检查是否安装
	check_deb_off
	bbrstatus
	echo -e "\033[41;37m [错误] \033[0m BBR 未开启 !"
}

action=$1
[ -z $1 ] && action=install
case "$action" in
	install|start|stop|status)
	${action}bbr
	;;
	*)
	echo "输入错误 !"
	echo "用法: { install | start | stop | status }"
	;;
esac
XuanLove.Download

/ BBR / Ubuntu-bbr.sh
