跳到主要內容

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參數去下多個處理動作包含:
  1. 刪除以AutomaticLogin開頭的列。
  2. 將[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最花時間有兩個地方:
  1. 我想做的事情不曉得是否有對應的語法,必須花時間查
  2. 嘗試特殊字元與變數的標示方式
我想高手們一定可以寫出更簡潔的語法,但對我來說,寫出我看得懂的語法比較重要。另外提供給大家下載的source code中,也包含一些我撰寫時使用的測試資料,大家可以自行修改處理路徑去看看script對應的行為。

留言

這個網誌中的熱門文章

解決RobotFramework從3.1.2升級到3.2.2之後,Choose File突然會整個Hand住的問題

考慮到自動測試環境的維護,我們很久以前就使用java去執行robot framework。前陣子開始處理從3.1.2升級到3.2.2的事情,主要先把明確的runtime語法錯誤與deprecate item處理好,這部分內容可以參考: link 。 直到最近才發現,透過SeleniumLibrary執行Choose File去上傳檔案的動作,會導致測試案例timeout。本篇文章主要分享心路歷程與解決方法,我也送了一條issue給robot framework: link 。 我的環境如下: RobotFramework: 3.2.2 Selenium: 3.141.0 SeleniumLibrary: 3.3.1 Remote Selenium Version: selenium-server-standalone-3.141.59 首先並非所有Choose File的動作都會hang住,有些測試案例是可以執行的,但是上傳一個作業系統ISO檔案一定會發生問題。後來我透過wireshark去比對新舊版本的上傳動作,因為我使用 Remote Selenium ,所以Selenium會先把檔案透過REST API發送到Remote Selenium Server上。從下圖我們可以發現,在3.2.2的最後一個TCP封包,比3.1.2大概少了500個bytes。 於是就開始了我trace code之路。包含SeleniumLibrary產生要送給Remote Selenium Server的request內容,還有HTTP Content-Length的計算,我都確認過沒有問題。 最後發現問題是出在socket API的使用上,就是下圖的這支code: 最後發現可能因為開始使用nio的方式送資料,但沒處理到尚未送完的資料內容,而導致發生問題。加一個loop去做計算就可以解決了。 最後我有把解法提供給robot framework官方,在他們出新的版本之前,我是將改完的_socket.py放在我們自己的Lib底下,好讓我們測試可以正常進行。(shutil.py應該也是為了解某個bug而產生的樣子..)

Show NIC selection when setting the network command with the device option

 Problem  在answer file中設定網卡名稱後,安裝時會停在以下畫面: 所使用的command參數如下: network --onboot = yes --bootproto =dhcp --ipv6 =auto --device =eth1 Diagnostic Result 這樣的參數,以前試驗過是可以安裝完成的。因此在發生這個問題後,我檢查了它的debug console: 從console得知,eth1可能是沒有連接網路線或者是網路太慢而導致的問題。後來和Ivy再三確認,有問題的是有接網路線的網卡,且問題是發生在activate階段: Solution 我想既然有retry應該就有次數或者timeout限制,因此發現在Anaconda的說明文件中( link ),有提到dhcptimeout這個boot參數。看了一些人的使用範例,應該是可以直接串在isolinux.cfg中,如下: default linux ksdevice = link ip =dhcp ks =cdrom: / ks.cfg dhcptimeout = 90 然而我在RHEL/CentOS 6.7與6.8試驗後都無效。 因此我就拿了顯示的錯誤字串,問問Google大師,想找一下Anaconda source code來看一下。最後找到別人根據Anaconda code修改的版本: link ,關鍵在於setupIfaceStruct函式中的setupIfaceStruct與readNetConfig: setupIfaceStruct: 會在dhcp時設定dhcptimeout。 readNetConfig: 在writeEnabledNetInfo將timeout寫入dhclient config中;在wait_for_iface_activation內會根據timeout做retry。 再來從log與code可以得知,它讀取的檔案是answer file而不是boot command line。因此我接下來的測試,就是在answer file的network command上加入dhcptimeout: network --onboot = yes --bootproto =dhcp --ipv6 =auto --device =eth1 --...

第一次寫MIB就上手

SNMP(Simple Network Management Protocol)是用來管理網路設備的一種Protocol,我對它的認識也是從工作接觸開始。雖說是管理網路設備,但是主機、電源供應器、RAID等也都可以透過它來做管理。如果你做了一個應用程式,當然所有的操作也都可以透過SNMP來完成,不過可能會很痛苦。前陣子遇到一個學弟,它告訴我說:「我可能不會想寫程式。」為什麼? 因為這是他痛苦的根源。 在這篇文章中,不是要告訴你SNMP是什麼,會看這篇文章的大哥們,應該已經對SNMP有些認識了。 是的!主題是MIB(Management information base)! 對於一個3th-party的SNMP oid,有MIB可以幫助你去了解它所提供的資訊是什麼,且可以對它做什麼操作。最近我運氣很好剛好做到關於修改MIB的工作,也讓我順便了解一下它的語法,接下來我要交給大家MIB的基礎認識。 smidump 我並非使用什麼高強的Editor去編寫MIB,我僅透過Nodepad++編輯和smidump編譯而已。smidump是Kay教我使用的一個將MIB module轉成樹狀結構或oid列表的工具,唯一的缺點是不會告訴你哪一行打錯。當然有錢直接買編輯樹狀結構的工具就可以不需要了解語法了! 安裝 在Ubuntu上可先輸入smidump確認是否安裝,如果沒安裝可透過apt-get install libsmi2ldbl安裝。(CentOS可以透過yum install libsmi) root@tonylin:~/multi-boot-server# smidump The program 'smidump' is currently not installed. You can install it by typing: apt-get install libsmi2ldbl 使用 透過下面兩行指令,就可以將mib file產生出對應的tree與oid列表的檔案。也可以透過這個結果確認MIB是不是你想要的。 smidump -f tree example1.mib > xtree.txt smidump -f identifiers example1.mib > xiden.txt 如果有參考其它檔案要加上p的參數: smidum...