經(jīng)常要在bash腳本里面或者直接對(duì)腳本本身加上sudo運(yùn)行命令,但是這引發(fā)了一系列的問(wèn)題。
比如用sudo的時(shí)候,腳本里的~或$HOME指代用戶文件夾的這個(gè)變量,到底是應(yīng)該指向我真正的用戶文件夾如/home/pi呢,還是指向了超級(jí)管理員的用戶文件夾/root/呢?
實(shí)際上它指向了/root/文件夾,這是我們絕對(duì)不想要的。但是很多命令如安裝個(gè)程序,都不得不用sudo,那怎么辦?
首先要說(shuō)下經(jīng)驗(yàn):命令行的權(quán)限執(zhí)行,從表現(xiàn)上來(lái)看,可以分為以下5種情況:
- admin-manual: 普通用戶手敲命令
- sudo-manual: 手敲命令加sudo
- admin-bash: 以普通用戶執(zhí)行bash腳本
- sudo-bash: 以sudo執(zhí)行bash腳本
- root-any: 以root用戶登錄
很多變量、環(huán)境變量在這4中情況下,會(huì)經(jīng)常出現(xiàn)混亂!(混亂指的是我們自己,不是電腦)
另外,說(shuō)個(gè)小技巧。
我們都直到~變量是指向當(dāng)前用戶目錄,實(shí)際上~abc格式的變量可以指向指定用戶的用戶目錄,如~pi會(huì)指向/home/pi,或~ubuntu指向/home/ubuntu.
理清一下思路:
在正常執(zhí)行腳本如./test.sh時(shí)是沒(méi)有任何問(wèn)題的,即使腳本里面出現(xiàn)了sudo如sudo apt-get update這樣也是沒(méi)有問(wèn)題的。
也就是說(shuō),就只有對(duì)整個(gè)腳本執(zhí)行sudo的情況下如sudo ./test.sh,才會(huì)出現(xiàn)嚴(yán)重問(wèn)題的!
那么假設(shè)我的真實(shí)用戶是pi,而HOME目錄在/home/pi,現(xiàn)在我要在sudo ./test.sh這樣的執(zhí)行方式下找出正確的解決方案。
以下為腳本中的各種語(yǔ)句和變量以及顯示結(jié)果:
# (不推薦?。?
$ whoami
>>> root
# 不同于whoami,能夠指出當(dāng)前有哪些用戶登錄電腦,包括本機(jī)登錄和ssh登錄的所有人
$ who am i
>>> 有些機(jī)器上顯示為空
>>> Mac上顯示: pi ttys001 Nov 26 16:57
# 等同于whoami (不推薦!)
$ echo $USER
>>> root
# 用戶主目錄位置 (不靠譜不推薦?。?
echo $HOME
>>> /root
$ 用戶主目錄位置,等同于$HOME (不推薦?。?
$ echo ~
>>> /root
# 直接使用環(huán)境變量LOGNAME
$ echo $LOGNAME
>>> root
# 顯式調(diào)用環(huán)境變量LOGNAME
$ printenv LOGNAME
>>> root
# SUDO_USER是root的ENV中的環(huán)境變量,
# 同時(shí)普通用戶的env是沒(méi)有的,只有root用戶才能顯示出來(lái)
$ sudo echo $SUDO_USER
>>> pi
# 顯示調(diào)用環(huán)境變量SUDO_USER (不推薦?。?
# 從結(jié)果中可以看到,即使是sudo身份執(zhí)行的腳本,腳本里面是否加sudo也會(huì)不同!
$ printenv SUDO_USER
>>> pi
$ sudo printenv SUDO_USER
>>> root
從上面測(cè)試中可以看出,如果我們是用sudo執(zhí)行bash腳本的話,很多變量都是“不靠譜”的。
Stackoverflow中,比較一致性的傾向就是使用$SUDO_USER這個(gè)環(huán)境變量。而測(cè)試中也的確,它是最“穩(wěn)定的”,即在不同的權(quán)限、OS系統(tǒng)下,都能始終如一(只限有sudo的系統(tǒng))。
那么現(xiàn)在我們有了用戶名,就可以用~pi這樣的命令獲取主目錄/home/pi了,但是!
這時(shí)候問(wèn)題又出現(xiàn)了:手敲時(shí)候,我們可以獲得~pi的正確地址,但是腳本中卻不識(shí)別~pi是個(gè)什么東西,頂多是個(gè)字符串,沒(méi)法像變量一樣。
那既然是這樣,我們就不能用~abc方法了,改用雖然老套但是絕對(duì)不混亂的方法:
從/etc/passwd中直接看。
手動(dòng)的話可以直接打開(kāi)passwd查看,腳本里面就比較麻煩,最方便的是用系統(tǒng)命令getent即Get Entries命令,獲得指定用戶的信息:
$ getent passwd pi
>>> pi:x:1000:1000:,,,:/home/pi:/bin/bash
那么,剩下的是有把其中的/home/pi取出來(lái)了,我們用cut就輕松取出。
所以全部過(guò)程如下:
me=$SUDO_USER
myhome=`getent passwd $me | cut -d: -f 6`
順利得到/home/pi!
再進(jìn)一步,如果腳本沒(méi)有以sudo方式運(yùn)行呢?這時(shí)候root用戶和普通用戶的環(huán)境變量下都是沒(méi)有SUDO_USER這個(gè)變量的。那么就需要加一步判斷了:
me=${SUDO_USER:-$LOGNAME}
myhome=`getent passwd $me | cut -d: -f 6`
即如果SUDO_USER為空,則正常使用$LOGNAME獲取當(dāng)前用戶。為什么不用$USER而是用$LOGNAME呢?因?yàn)閁SER不是每個(gè)系統(tǒng)都有,但是LOGNAME是*nix系統(tǒng)下都會(huì)有的。
更新
由于部分OS不能正確獲取LOGNAME,所以統(tǒng)一采用uid的方式獲取用戶路徑:
HOUSE=`getent passwd ${SUDO_UID:-$(id -u)} | cut -d: -f 6`
再更新
MacOS沒(méi)有/etc/passwd,也不支持getent passwd <UID>方式獲取用戶信息,但是sudo下也能保持$USER和$HOME變量?jī)?nèi)容不變。
所以更改為下:
HOUSE=${$(`getent passwd ${SUDO_UID:-$(id -u)} | cut -d: -f 6`):-$HOME}
即如果getent方式無(wú)法獲取內(nèi)容,則直接取$HOME的值。
再再更新
因?yàn)閎ash不支持以上嵌套的三元運(yùn)算表達(dá)式,所以要拆開(kāi):
HOUSE="`cat /etc/passwd |grep ${SUDO_UID:-$(id -u)} | cut -d: -f 6`"
HOUSE=${HOUSE:-$HOME}
再再再更新
如果是root的話,grep uid的時(shí)候會(huì)匹配到passwd中所有含0的行,所以要改進(jìn)為以下:
HOUSE="`cat /etc/passwd |grep ^${SUDO_USER:-$(id -un)}: | cut -d: -f 6`"
HOUSE=${HOUSE:-$HOME}
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。