ShellScript-Linux AutoLogin

Problem

為了利於測試，先前做過各個不同作業系統的自動登入(link)。隨著測試機逐日增加，為了節省時間與練習shellscript，因此決定寫一個script將設置自動登入的動作給簡單化。但面對不同的Linux，會有不同的自動登入方式，我也很直接的針對不同Distribution使用不同的做法。目前我所支援的作業系統為SLES11、RHEL5、RHEL6、CentOS5、CentOS6。

How to?

確認Linux Distribution

首先我們可以透過/etc/issue去判斷作業系統版本。以下是我收集的資訊:

Welcome to SUSE Linux Enterprise Server 10 SP2
Welcome to SUSE Linux Enterprise Server 11 SP2
Red Hat Enterprise Linux Server release 6.0
Red Hat Enterprise Linux Server release 5.6
CentOS release 5.6
CentOS Linux release 6.0

作法是透過grep指令去判斷/etc/issue檔案是否有出現對應的OS Pattern。SLES判斷是否有SUSE；CentOS/RHEL則是判斷開頭字串與主要版本號；其它的顯示不支援。找到對應的Distribution後就呼叫對應的function。這裡設置OS_INFO_FILE變數為$1與將實際操作寫為各個function，主要是為了便於測試，不然系統檔改壞就要DRBL了。

OS_INFO_FILE=$1
if [ "$1" == "" ]; then
	OS_INFO_FILE=/etc/issue
fi
 
if grep SUSE $OS_INFO_FILE > /dev/null; then
	setupAL_SLES
elif grep -e "\(CentOS\)\|\(Red Hat\).*5\.*" $OS_INFO_FILE > /dev/null; then
	setupAL_RHEL5
elif grep -e "\(CentOS\)\|\(Red Hat\).*6\.*" $OS_INFO_FILE > /dev/null; then
	setupAL_RHEL6
else 
	echo Not support on \"`cat "$OS_INFO_FILE"`\". Please setup autologin by yourself.
fi

SUSE

SUSE只要將displaymanager檔案中的DISPLAYMANAGER_AUTOLOGIN屬性設定自動登入的使用者即可。sed的-i參數可以將修改內容直接反映到處理的檔案。但我曾在網上看到有人說某些Linux不支援，如果不支援可以改採暫存檔案的方式去處理。另外大家可以發現，在做修改之前都會去產生一個.bak檔案，這是為了避免造成不可挽回的錯誤!

function setupAL_SLES()
{
	config_path=/etc/sysconfig/displaymanager
	cp $config_path $config_path.bak
	sed -i 's/DISPLAYMANAGER_AUTOLOGIN=".*"/DISPLAYMANAGER_AUTOLOGIN="root"/g' $config_path
}

CentOS6/RHEL6

CentOS6/RHEL6要在custom.conf的[daemon]區塊中加入對應的設定。考量先前可能已有設定過之情況，我透過sed的-e參數去下多個處理動作包含:

刪除以AutomaticLogin開頭的列。
將[daemon]取代為[daemon]加必要屬性。這個做法可以讓我不需考慮字串插入位置的問題。

function setupAL_RHEL6()
{
	config_path=/etc/gdm/custom.conf
	cp $config_path $config_path.bak
	sed -i -e '/^AutomaticLogin.*=.*/d' -e 's/\[daemon\]/\[daemon\]\nAutomaticLoginEnable=True\nAutomaticLogin=root/g' $config_path
}

CentOS5/RHEL5

其實CentOS5/RHEL5也可以使用CentOS6/RHEL6的方法，但如果是root就不行了。因此我還是只能採取較複雜的做法，要處理的檔案包含/etc/inittab與/root/.bash_profile。inittab部分比較簡單，我只需要取代default的run-level與tty1的參數即可，我使用兩個command替我做這件事情。.bash_profile的部分也必須考量先前設置過的情況，因此我透過sed將做自動登入的那三行script刪除，刪除方式是找$DISPLAY為開始而為fi結束，最後再把這三行script重新寫入。會另外做刪除startx是因為舊的測試環境沒做tty1的判斷，我必須去處理先前版本做法的情況。

function setupAL_RHEL5()
{
	inittab_path=/etc/inittab
	bash_profile_path=/root/.bash_profile	
	cp $inittab_path $inittab_path.bak
	cp $bash_profile_path $bash_profile_path.bak
	sed -i -e 's/id:5:init/id:3:init/g' -e 's/\/sbin\/mingetty tty1/\/sbin\/mingetty --autologin root tty1/g' $inittab_path
	sed -i -e '/$DISPLAY/,/fi/d' -e '/startx/d' $bash_profile_path
	start_x_str='\nif [ -z "$DISPLAY" ] && [ "$(tty)" = "/dev/tty1" ]; then\n       startx\nfi'
	echo -e $start_x_str >> $bash_profile_path
}

Summary

由於對sed指令還不夠熟悉，我撰寫這些script最花時間有兩個地方:

我想做的事情不曉得是否有對應的語法，必須花時間查
嘗試特殊字元與變數的標示方式

我想高手們一定可以寫出更簡潔的語法，但對我來說，寫出我看得懂的語法比較重要。另外提供給大家下載的source code中，也包含一些我撰寫時使用的測試資料，大家可以自行修改處理路徑去看看script對應的行為。

阿兩的筆記本 Ryoutsu's Notebook

搜尋此網誌