linux利用谷歌BBR算法提高上网速度

「爱情、让人受尽委屈。」 2022-05-24 11:38 210阅读 0赞

如题,算法一两句话也说不清楚,但是可以通过下面的脚本更新linux内核,然后提高上网速度

#!/usr/bin/env bash
    #
    # Auto install latest kernel for TCP BBR
    #
    # System Required: CentOS 6+, Debian7+, Ubuntu12+
    #
    # Copyright (C) 2016-2018 Teddysun <i@teddysun.com>
    #
    # URL: https://teddysun.com/489.html
    #
    
    red='\033[0;31m'
    green='\033[0;32m'
    yellow='\033[0;33m'
    plain='\033[0m'
    
    cur_dir=$(pwd)
    
    [[ $EUID -ne 0 ]] && echo -e "${red}Error:${plain} This script must be run as root!" && exit 1
    
    [[ -d "/proc/vz" ]] && echo -e "${red}Error:${plain} Your VPS is based on OpenVZ, which is not supported." && exit 1
    
    if [ -f /etc/redhat-release ]; then
        release="centos"
    elif cat /etc/issue | grep -Eqi "debian"; then
        release="debian"
    elif cat /etc/issue | grep -Eqi "ubuntu"; then
        release="ubuntu"
    elif cat /etc/issue | grep -Eqi "centos|red hat|redhat"; then
        release="centos"
    elif cat /proc/version | grep -Eqi "debian"; then
        release="debian"
    elif cat /proc/version | grep -Eqi "ubuntu"; then
        release="ubuntu"
    elif cat /proc/version | grep -Eqi "centos|red hat|redhat"; then
        release="centos"
    else
        release=""
    fi
    
    is_digit(){
        local input=${1}
        if [[ "$input" =~ ^[0-9]+$ ]]; then
            return 0
        else
            return 1
        fi
    }
    
    get_valid_valname(){
        local val=${1}
        local new_val=$(eval echo $val | sed 's/[-.]/_/g')
        echo ${new_val}
    }
    
    get_hint(){
        local val=${1}
        local new_val=$(get_valid_valname $val)
        eval echo "\$hint_${new_val}"
    }
    
    #Display Memu
    display_menu(){
        local soft=${1}
        local default=${2}
        eval local arr=(\${${soft}_arr[@]})
        local default_prompt
        if [[ "$default" != "" ]]; then
            if [[ "$default" == "last" ]]; then
                default=${#arr[@]}
            fi
            default_prompt="(default ${arr[$default-1]})"
        fi
        local pick
        local hint
        local vname
        local prompt="which ${soft} you'd select ${default_prompt}: "
    
        while :
        do
            echo -e "\n------------ ${soft} setting ------------\n"
            for ((i=1;i<=${#arr[@]};i++ )); do
                vname="$(get_valid_valname ${arr[$i-1]})"
                hint="$(get_hint $vname)"
                [[ "$hint" == "" ]] && hint="${arr[$i-1]}"
                echo -e "${green}${i}${plain}) $hint"
            done
            echo
            read -p "${prompt}" pick
            if [[ "$pick" == "" && "$default" != "" ]]; then
                pick=${default}
                break
            fi
    
            if ! is_digit "$pick"; then
                prompt="Input error, please input a number"
                continue
            fi
    
            if [[ "$pick" -lt 1 || "$pick" -gt ${#arr[@]} ]]; then
                prompt="Input error, please input a number between 1 and ${#arr[@]}: "
                continue
            fi
    
            break
        done
    
        eval ${soft}=${arr[$pick-1]}
        vname="$(get_valid_valname ${arr[$pick-1]})"
        hint="$(get_hint $vname)"
        [[ "$hint" == "" ]] && hint="${arr[$pick-1]}"
        echo -e "\nyour selection: $hint\n"
    }
    
    version_ge(){
        test "$(echo "$@" | tr " " "\n" | sort -rV | head -n 1)" == "$1"
    }
    
    get_latest_version() {
        latest_version=($(wget -qO- http://kernel.ubuntu.com/~kernel-ppa/mainline/ | awk -F'\"v' '/v[4-9]./{print $2}' | cut -d/ -f1 | grep -v - | sort -V))
    
        [ ${#latest_version[@]} -eq 0 ] && echo -e "${red}Error:${plain} Get latest kernel version failed." && exit 1
    
        kernel_arr=()
        for i in ${latest_version[@]}; do
            if version_ge $i 4.14; then
                kernel_arr+=($i);
            fi
        done
    
        display_menu kernel last
    
        if [[ `getconf WORD_BIT` == "32" && `getconf LONG_BIT` == "64" ]]; then
            deb_name=$(wget -qO- http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/ | grep "linux-image" | grep "generic" | awk -F'\">' '/amd64.deb/{print $2}' | cut -d'<' -f1 | head -1)
            deb_kernel_url="http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/${deb_name}"
            deb_kernel_name="linux-image-${kernel}-amd64.deb"
            modules_deb_name=$(wget -qO- http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/ | grep "linux-modules" | grep "generic" | awk -F'\">' '/amd64.deb/{print $2}' | cut -d'<' -f1 | head -1)
            deb_kernel_modules_url="http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/${modules_deb_name}"
            deb_kernel_modules_name="linux-modules-${kernel}-amd64.deb"
        else
            deb_name=$(wget -qO- http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/ | grep "linux-image" | grep "generic" | awk -F'\">' '/i386.deb/{print $2}' | cut -d'<' -f1 | head -1)
            deb_kernel_url="http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/${deb_name}"
            deb_kernel_name="linux-image-${kernel}-i386.deb"
            modules_deb_name=$(wget -qO- http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/ | grep "linux-modules" | grep "generic" | awk -F'\">' '/i386.deb/{print $2}' | cut -d'<' -f1 | head -1)
            deb_kernel_modules_url="http://kernel.ubuntu.com/~kernel-ppa/mainline/v${kernel}/${modules_deb_name}"
            deb_kernel_modules_name="linux-modules-${kernel}-i386.deb"
        fi
    
        [ -z ${deb_name} ] && echo -e "${red}Error:${plain} Getting Linux kernel binary package name failed, maybe kernel build failed. Please choose other one and try again." && exit 1
    }
    
    get_opsy() {
        [ -f /etc/redhat-release ] && awk '{print ($1,$3~/^[0-9]/?$3:$4)}' /etc/redhat-release && return
        [ -f /etc/os-release ] && awk -F'[= "]' '/PRETTY_NAME/{print $3,$4,$5}' /etc/os-release && return
        [ -f /etc/lsb-release ] && awk -F'[="]+' '/DESCRIPTION/{print $2}' /etc/lsb-release && return
    }
    
    opsy=$( get_opsy )
    arch=$( uname -m )
    lbit=$( getconf LONG_BIT )
    kern=$( uname -r )
    
    get_char() {
        SAVEDSTTY=`stty -g`
        stty -echo
        stty cbreak
        dd if=/dev/tty bs=1 count=1 2> /dev/null
        stty -raw
        stty echo
        stty $SAVEDSTTY
    }
    
    getversion() {
        if [[ -s /etc/redhat-release ]]; then
            grep -oE  "[0-9.]+" /etc/redhat-release
        else
            grep -oE  "[0-9.]+" /etc/issue
        fi
    }
    
    centosversion() {
        if [ x"${release}" == x"centos" ]; then
            local code=$1
            local version="$(getversion)"
            local main_ver=${version%%.*}
            if [ "$main_ver" == "$code" ]; then
                return 0
            else
                return 1
            fi
        else
            return 1
        fi
    }
    
    check_bbr_status() {
        local param=$(sysctl net.ipv4.tcp_congestion_control | awk '{print $3}')
        if [[ x"${param}" == x"bbr" ]]; then
            return 0
        else
            return 1
        fi
    }
    
    check_kernel_version() {
        local kernel_version=$(uname -r | cut -d- -f1)
        if version_ge ${kernel_version} 4.9; then
            return 0
        else
            return 1
        fi
    }
    
    install_elrepo() {
    
        if centosversion 5; then
            echo -e "${red}Error:${plain} not supported CentOS 5."
            exit 1
        fi
    
        rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
    
        if centosversion 6; then
            rpm -Uvh http://www.elrepo.org/elrepo-release-6-8.el6.elrepo.noarch.rpm
        elif centosversion 7; then
            rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-3.el7.elrepo.noarch.rpm
        fi
    
        if [ ! -f /etc/yum.repos.d/elrepo.repo ]; then
            echo -e "${red}Error:${plain} Install elrepo failed, please check it."
            exit 1
        fi
    }
    
    sysctl_config() {
        sed -i '/net.core.default_qdisc/d' /etc/sysctl.conf
        sed -i '/net.ipv4.tcp_congestion_control/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 >/dev/null 2>&1
    }
    
    install_config() {
        if [[ x"${release}" == x"centos" ]]; then
            if centosversion 6; then
                if [ ! -f "/boot/grub/grub.conf" ]; then
                    echo -e "${red}Error:${plain} /boot/grub/grub.conf not found, please check it."
                    exit 1
                fi
                sed -i 's/^default=.*/default=0/g' /boot/grub/grub.conf
            elif centosversion 7; then
                if [ ! -f "/boot/grub2/grub.cfg" ]; then
                    echo -e "${red}Error:${plain} /boot/grub2/grub.cfg not found, please check it."
                    exit 1
                fi
                grub2-set-default 0
            fi
        elif [[ x"${release}" == x"debian" || x"${release}" == x"ubuntu" ]]; then
            /usr/sbin/update-grub
        fi
    }
    
    reboot_os() {
        echo
        echo -e "${green}Info:${plain} The system needs to reboot."
        read -p "Do you want to restart system? [y/n]" is_reboot
        if [[ ${is_reboot} == "y" || ${is_reboot} == "Y" ]]; then
            reboot
        else
            echo -e "${green}Info:${plain} Reboot has been canceled..."
            exit 0
        fi
    }
    
    install_bbr() {
        check_bbr_status
        if [ $? -eq 0 ]; then
            echo
            echo -e "${green}Info:${plain} TCP BBR has been installed. nothing to do..."
            exit 0
        fi
        check_kernel_version
        if [ $? -eq 0 ]; then
            echo
            echo -e "${green}Info:${plain} Your kernel version is greater than 4.9, directly setting TCP BBR..."
            sysctl_config
            echo -e "${green}Info:${plain} Setting TCP BBR completed..."
            exit 0
        fi
    
        if [[ x"${release}" == x"centos" ]]; then
            install_elrepo
            yum --enablerepo=elrepo-kernel -y install kernel-ml kernel-ml-devel
            if [ $? -ne 0 ]; then
                echo -e "${red}Error:${plain} Install latest kernel failed, please check it."
                exit 1
            fi
        elif [[ x"${release}" == x"debian" || x"${release}" == x"ubuntu" ]]; then
            [[ ! -e "/usr/bin/wget" ]] && apt-get -y update && apt-get -y install wget
            echo -e "${green}Info:${plain} Getting latest kernel version..."
            get_latest_version
            if [ -n ${modules_deb_name} ]; then
                wget -c -t3 -T60 -O ${deb_kernel_modules_name} ${deb_kernel_modules_url}
                if [ $? -ne 0 ]; then
                    echo -e "${red}Error:${plain} Download ${deb_kernel_modules_name} failed, please check it."
                    exit 1
                fi
            fi
            wget -c -t3 -T60 -O ${deb_kernel_name} ${deb_kernel_url}
            if [ $? -ne 0 ]; then
                echo -e "${red}Error:${plain} Download ${deb_kernel_name} failed, please check it."
                exit 1
            fi
            [ -f ${deb_kernel_modules_name} ] && dpkg -i ${deb_kernel_modules_name}
            dpkg -i ${deb_kernel_name}
            rm -f ${deb_kernel_name} ${deb_kernel_modules_name}
        else
            echo -e "${red}Error:${plain} OS is not be supported, please change to CentOS/Debian/Ubuntu and try again."
            exit 1
        fi
    
        install_config
        sysctl_config
        reboot_os
    }
    
    
    clear
    echo "---------- System Information ----------"
    echo " OS : $opsy"
    echo " Arch : $arch ($lbit Bit)"
    echo " Kernel : $kern"
    echo "----------------------------------------"
    echo " Auto install latest kernel for TCP BBR"
    echo
    echo " URL: https://teddysun.com/489.html"
    echo "----------------------------------------"
    echo
    echo "Press any key to start...or Press Ctrl+C to cancel"
    char=`get_char`
    
    install_bbr 2>&1 | tee ${cur_dir}/install_bbr.log