Shell指令碼實現自動檢測修改最快的Ubuntu軟體源_linux shell

每次裝好Ubuntu，對於大多數使用者來說，首先要做的事就是手動修改/etc/apt/sources.list檔案，將裡面的官方軟體源地址更換為自己學校或者公司的軟體源。當我們更換一個工作環境後，可能伴隨著又要替換舊的軟體源地址。

筆者覺得這樣每次手動更改軟體源是一件及其麻煩重複的勞動，於是編寫了一個自動更新最快軟體源的指令碼，從此一勞永逸。

原理

最直觀的想法就是：對各個軟體源進行測速，選出最快的那個，之後將其替換為新的軟體源。

那麼如何對各個軟體源測速呢？有兩種方法：

一、用ping命令 測量其平均回應時間 選出回應時間最短的那個

二、用wget命令 測量下載一個檔案的總時間 選出耗時最少的那個

那麼這兩種方法有什麼區別呢？我們該用哪個呢？

前者選出的是回應時間最優的，後者選出的是下載速度最快的。我們都知道軟體源的作用是供用戶端下載更新軟體，所以當然是後者的方法更為準確，但筆者最終選擇了前者作為測速方案，因為前者的使用者體驗更好且代碼簡單易懂。設想，如果我們採用後者，那麼需要從每個軟體源下載一個檔案，並且這個檔案不能太小，否則無法區分他們的速度，那麼一個顯而易見的情況是指令碼需要運行較長的時間。

雖然存在某些軟體源可能回應時間很短，而下載速度卻很慢的情況，但經過筆者的多次實驗，發現這樣的情況並不常見。

實現

首先測試使用者網路狀態

利用

複製代碼 代碼如下:

local speed=`ping -W1 -c1 www.baidu.com 2> /dev/null | grep "^rtt" |  cut -d '/' -f5`

取出其平均回應時間 如果speed == “” 則說明網路不通，提示使用者，且退出程式。 否則，說明網路正常，繼續執行。

檢測軟體源列表檔案是否存在

複製代碼 代碼如下:

test -f $SOURCES_MIRRORS_FILE

若不存在，提示使用者，且退出程式。

對每個軟體源地址進行測速

在測速之前清空上次啟動並執行測速結果檔案，之後將每個軟體源的測速結果(源地址 平均回應時間)寫入測速結果檔案

對測速結果進行排序

複製代碼 代碼如下:

sort -k 2 -n -o $MIRRORS_SPEED_FILE $MIRRORS_SPEED_FILE

對每行記錄 按照平均回應時間升序排列

選出最快的軟體源

複製代碼 代碼如下:

head -n 1 $MIRRORS_SPEED_FILE | cut -d ' ' -f1 `

通過取已排序列表中的第一條，選出最快的軟體源

詢問使用者是否要使用該軟體源

使用者確認後，先對使用者之前的軟體源進行備份，然後再替換。

getfastmirror.sh指令碼原始碼:

複製代碼 代碼如下:

#!/bin/bash

#Program:
#    This program gets the fastest ubuntu software sources from SOURCES_MIRRORS_FILE
#    and backup && update /etc/apt/sources.list

#Author:  KJlmfe    www.freepanda.me

#History:
#    2012/12/6    KJlmfe    First release

VERSION="precise"  # precise is code of Ubuntu 12.04 if your ubuntu is not 12.04 please change
TEST_NETCONNECT_HOST="www.baidu.com"
SOURCES_MIRRORS_FILE="sources_mirrors.list"   
MIRRORS_SPEED_FILE="mirrors_speed.list"

function get_ping_speed()    #return average ping $1 time
{
    local speed=`ping -W1 -c1 $1 2> /dev/null | grep "^rtt" |  cut -d '/' -f5`
    echo $speed
}

function test_mirror_speed()    #
{
    rm $MIRRORS_SPEED_FILE 2> /dev/null; touch $MIRRORS_SPEED_FILE
   
     cat $SOURCES_MIRRORS_FILE | while read mirror
    do
        if [ "$mirror" != "" ]; then
            echo -e "Ping $mirror c"
            local mirror_host=`echo $mirror | cut -d '/' -f3`    #change mirror_url to host
   
            local speed=$(get_ping_speed $mirror_host)
   
            if [ "$speed" != "" ]; then
                echo "Time is $speed"
                echo "$mirror $speed" >> $MIRRORS_SPEED_FILE
            else
                echo "Connected failed."
            fi
        fi
    done
}

function get_fast_mirror()
{
    sort -k 2 -n -o $MIRRORS_SPEED_FILE $MIRRORS_SPEED_FILE
    local fast_mirror=`head -n 1 $MIRRORS_SPEED_FILE | cut -d ' ' -f1`
    echo $fast_mirror
}

function backup_sources()
{
    echo -e "Backup your sources.list.n"
    sudo mv /etc/apt/sources.list /etc/apt/sources.list.`date +%F-%R:%S`
}

function update_sources()
{
    local COMP="main restricted universe multiverse"
    local mirror="$1"
    local tmp=$(mktemp)

    echo "deb $mirror $VERSION $COMP" >> $tmp
    echo "deb $mirror $VERSION-updates $COMP" >> $tmp
    echo "deb $mirror $VERSION-backports $COMP" >> $tmp
    echo "deb $mirror $VERSION-security $COMP" >> $tmp
    echo "deb $mirror $VERSION-proposed $COMP" >> $tmp

    echo "deb-src $mirror $VERSION $COMP" >> $tmp
    echo "deb-src $mirror $VERSION-updates $COMP" >> $tmp
    echo "deb-src $mirror $VERSION-backports $COMP" >> $tmp
    echo "deb-src $mirror $VERSION-security $COMP" >> $tmp
    echo "deb-src $mirror $VERSION-proposed $COMP" >> $tmp

    sudo mv "$tmp" /etc/apt/sources.list
    echo -e "Your sources has been updated, and maybe you want to run "sudo apt-get update" now.n";
}

echo -e "nTesting the network connection.nPlease wait...   c"

if [ "$(get_ping_speed $TEST_NETCONNECT_HOST)" == "" ]; then
    echo -e "Network is bad.nPlease check your network."; exit 1
else
    echo -e "Network is good.n"
    test -f $SOURCES_MIRRORS_FILE

    if [ "$?" != "0" ]; then 
        echo -e "$SOURCES_MIRRORS_FILE is not exist.n"; exit 2
    else
        test_mirror_speed
        fast_mirror=$(get_fast_mirror)

        if [ "$fast_mirror" == "" ]; then
            echo -e "Can't find the fastest software sources. Please check your $SOURCES_MIRRORS_FILEn"
            exit 0
        fi

        echo -e "n$fast_mirror is the fastest software sources. Do you want to use it? [y/n] c"   
        read choice

        if [ "$choice" != "y" ]; then
            exit 0
        fi

        backup_sources
        update_sources $fast_mirror
    fi
fi

exit 0

sources_mirrors.list源碼：

複製代碼 代碼如下:

http://cn.archive.ubuntu.com/ubuntu/
http://run.hit.edu.cn/ubuntu/
http://mirrors.sohu.com/ubuntu/
http://mirrors.163.com/ubuntu/
http://mirrors.tuna.tsinghua.edu.cn/ubuntu/
http://mirrors.ustc.edu.cn/ubuntu/
http://mirrors.yun-idc.com/ubuntu/
http://ubuntu.cn99.com/ubuntu/