通常我們在做核心程式設計的時候，會用到核心的資料結構，比如說textsearch提供了幾種演算法用於支付串查詢。在用於正式的專案前，我們會希望考察下他的用法以及想體驗下。最通常的做法是自己寫個module,寫個makefile,編譯，執行，然後去dmesg裡面看printk的結果。這個過程沒啥問題，就是太羅嗦。好了，現在我們有更方便的方法了：systemtap.

Systemtap是個指令碼，先翻譯成c kernel模組程式碼，然後編譯，插入到核心執行，同時提供最基本的核心和應用模組的通訊管道，在應用模組這裡收集資訊。 它還支援guru模式，讓使用者直接插入c程式碼。 這樣我們就可以利用stap的這一特性來做我們的實驗。

底下我們來演示下：

# cat ts.stp
%{
#include <linux/textsearch.h>
%}
function test_textsearch:long (algo:string) %{ /* pure */ /* unprivileged */
struct ts_config* ts;
struct ts_state state;

ts = textsearch_prepare(THIS->algo, "world", 5, GFP_KERNEL, TS_AUTOLOAD);
if(IS_ERR(ts)) goto err;
_stp_printf("%s\n", "prepare ok!");
if(UINT_MAX==textsearch_find_continuous(ts, &state, "hello world", 11)) goto err;
_stp_printf("%s\n", "find ok!");
textsearch_destroy(ts);
THIS->__retvalue = 0;
return;
err:
_stp_printf("%s\n", "init error!");
THIS->__retvalue = -1;
return;
%}

probe begin
{
printf("result:%d\n", test_textsearch(@1));
exit();
}

%%請注意我們上面用到了 _stp_printf，代替printk, 這個是stap的runtime的基礎服務。

# stap -gu ts.stp bm %%我們用的是BM演算法
prepare ok!
find ok!
result:0

cool吧！ 把c語言當指令碼編寫啦。

但是這個stap有個缺點，就是標頭檔案必須是核心include路徑裡面的，如果我們用到了其他標頭檔案，編譯的時候就出錯。

# touch xyz.h #建個空的標頭檔案
# diff -u ts.stp ts1.stp
--- ts.stp      2010-11-08 12:04:54.000000000 +0800
+++ ts1.stp     2010-11-10 14:19:47.000000000 +0800
@@ -1,5 +1,6 @@
 %{
 #include <linux/textsearch.h>
+#include "xyz.h"
 %}
 
# stap -gu ts1.stp bm
/tmp/stapKuLo9R/stap_90d03f83d4d4a8efa7b721a6ff0093fe_2503.c:169:17: error: xyz.h: No such file or directory
make[1]: *** [/tmp/stapKuLo9R/stap_90d03f83d4d4a8efa7b721a6ff0093fe_2503.o] Error 1
make: *** [_module_/tmp/stapKuLo9R] Error 2
Pass 4: compilation failed.  Try again with another '--vp 0001' option.

這個問題挺煩人的，但是我們有探索精神，不怕的，我們來crack下吧

# diff systemtap-1.3/buildrun.cxx systemtap-1.3.patched/buildrun.cxx 
235a236,238
> 
>   o << "EXTRA_CFLAGS += $(STAP_CFLAGS)" << endl;
> 
237a241
>
# ./configure --prefix=/usr && make && make install

到現在為止 stap是crack過的. 我們再來實驗下吧!首先把我們的標頭檔案所在的路徑通知到systemtap，它會在讓編譯模組的時候加上我們的標頭檔案路徑。

# export STAP_CFLAGS="-I`pwd`"
# printenv STAP_CFLAGS
-I/home/wentong

# stap -gu ts1.stp bm
prepare ok!
find ok!
result:0

哈哈，成功了。

大家可以這樣的方式用stap寫很多好玩的東西，祝玩的開心。

Post Footer automatically generated by wp-posturl plugin for wordpress.

No related posts.