在linux上寫程序、做網管的人,或多或少都會幾種腳本。腳本語言靈活的變量類型、強大的正則表達式處理能力,再加上linux系統(tǒng)本身的管道、重定向以及豐富的命令行工具,讓你編程起來游刃有余。
而C語言固然有種種優(yōu)勢,但不可否認,很多場合下,用腳本語言更為方便,比如我們將舉例說明的對配置文件的處理。
先看看我們示例程序的任務:
假設我們有一個用c寫的程序,它有一個配置文件 user.conf,保存了一些用戶信息,user.conf定義如下:
1)、以 # 開頭的行為注釋行,不做處理
2)、允許空行
3)、如果不是1和2,那么就是有效的數據,格式如下
# user.conf: configure file for user
# username age sex country
tom 20 male us
chen 22 female cn
每一列分為4個字段,字段之間用一個或多個空白字符(空格或者制表符)隔開,字段依次是 姓名、年齡、性別、國家
我們的c程序要完成對 user.conf的添加、刪除、編輯、查詢
這樣一個簡單的任務,用c處理起來不算復雜,不過也是要花點功夫的,而如果用腳本語言來做,卻很簡單,能不能在c中調用腳本來完成任務了?
Awk是linux上一種腳本語言,它的長處在于處理有一定格式規(guī)則的文件,例如咱們的user.conf。關于 awk 的資料有很多,oreilly公司出了專門的 awk 編程的書籍,網上也是可以下載到的。你也可以直接 man awk看看。
我們先看看如何用 shell 結合 awk來完成上述任務:
1) 添加一條記錄
例如,要添加 jack 18 male us 這樣一條記錄,可以簡單的用重定向功能
Echo –e “jack 18 male us” >> user.conf
現在,這條記錄被添加到 user.conf末尾了。
2) 刪除一條記錄
例如,現在要刪除用戶 chen 的信息
cat user.conf | awk ‘!/^chen[[:blank:]] / {print}" > tmp.conf; mv –f tmp.conf user.conf
3)、編輯一條記錄
現在,想把 tom的性別改為 female
cat user.conf | awk ‘{if($0 ~ /^tom[[:blank:]] /) print $1 $2 female $3; else print}"
通過 system()這個函數,我們就可以在 c 中調用以上腳本,完成任務了。
但是,system() 用起來還是覺得不爽,它的不足是只能執(zhí)行腳本,卻無法獲得腳本的輸出數據,而這通常是我們進一步處理的數據來源。(在shell和perl中,可以通過反引號( `` )來取得命令的輸出結果)。一個解決辦法是把輸出結果重定向到一個臨時文件中,然后在c中讀取文件,獲取數據,最后當然還要刪除這個文件。不過,這個方法總是讓人覺得有一點點不爽,如果能直接把腳本執(zhí)行中輸出的數據輸到我們的緩沖區(qū)來就更好了。
我寫了個小函數,叫 my_system(),通過管道以及重定向,實現了以上想法。函數原型如下:
int my_system(const char* pCmd, char* pResult, int size);
輸出數據被保存到 pResult所指向的緩沖區(qū)中,緩沖區(qū)大小為 size,最多可以保存 size-1的數據。
函數的實現放在本文的最后
有了這個函數以后,在 c中調用腳本就更方便了,我們可以通過它來實現對 user.conf的查詢。
4)、查詢一個記錄
例如,我們要獲取 tom 的性別
可以用腳本這樣來實現:
cat user.conf | awk ‘/^tom[[:blank:]] / {print $3}"
腳本的執(zhí)行結果是 tom的性別 male被輸出到屏幕上
在我們的 c程序中,如此調用 my_system(),
char buf[101];
my_system(“cat user.conf | awk ‘/^tom[[:blank:]] / {print $3}"”, buf, 101);
調用完以后,buf中的數據就是 “male”了,怎么樣,還算方便吧?
以上只是用結合腳本完成了一個比較簡單的任務,所以我沒有把這些腳本單獨形成腳本文件。如果你善于使用 perl、shell、awk,那么可以寫出更強大的腳本文件來處理更復雜的問題,然后通過類似 my_system( )的方法,在 c/c 等其它語言中取得腳本的輸出結果,實現有趣的“混合編程”。
希望你能從中得到樂趣!
#include
#include
#include
#include
#include
static int my_system(const char* pCmd, char* pResult, int size)
{
int fd[2];
int pid;
int count;
int left;
char* p = 0;
int maxlen = size – 1;
memset(pResult, 0, size);
if(pipe(fd))
{
printf("pipe error\n");
return –1;
}
if((pid = fork()) == 0)
{// chile process
int fd2[2];
if(pipe(fd2))
{
printf("pipe2 error\n");
return –1;
}
close(1);
dup2(fd2[1],1);
close(fd[0]);
close(fd2[1]);
system(pCmd);
read(fd2[0], pResult, maxlen);
pResult[strlen(pResult)-1] = 0;
write(fd[1], pResult, strlen(pResult));
close(fd2[0]);
exit(0);
}
// parent process
close(fd[1]);
p = pResult;
left = maxlen;
while((count = read(fd[0], p, left))) {
p = count;
left -= count;
if(left == 0)
break;
}
close(fd[0]);
return 0;
}
int main(void)
{
char result[1025];
my_system("/sbin/ifconfig", result, 1025);
printf("the result is\n\n%s\n", result);
return 0;
}
維珍