天天看點

local lua 多個_檢測Lua腳本中的死循環

論壇上有人問,是以把以前做的東西拿出來秀一下。

Lua是一門小巧精緻的語言,特别适用于嵌入其它的程式為它們提供腳本支援。不過腳本通常是使用者編寫的,很有可能出現死循環,雖說這是使用者的問題,但卻會造成我們的宿主程式死掉。是以檢測使用者腳本中的死循環并中止這段腳本的運作就顯得非常重要了。

可是,一個現實的問題是死循環并不好檢測,一些隐藏較深的死循環連人都很難找出來,更不用說讓機器去找了。是以實際采用的方案多是檢測腳本的執行時間,如果超過一定的限度,就認為裡面有死循環,我下面的例子也是用的這種方法。

以下是幾個相關的全局變量(我是喜歡把C++當C用的程式員,C++的忠實粉絲請忍耐一下:))的定義。

1 lua_State*g_lua=NULL;//lua腳本引擎2 volatileunsigned g_begin=0;//腳本開始執行的時間3 volatilelongg_counter=0;//腳本執行計數, 用于判斷執行逾時4 volatilelongg_check=0;//進行逾時檢查時的執行計數

run_user_script用來執行使用者腳本,它首先通過GetTickCount把目前的時間記錄到g_begin中去。然後将g_counter加一,在執行完使用者腳本後再将其加一,這樣就可以保證執行使用者腳本時它是個奇數,而不執行時是偶數,檢測腳本逾時的代碼可以籍此來判斷目前是否在執行使用者腳本。還要注意調用使用者腳本要使用lua_pcall而不是lua_call,因為我們中止腳本的執行會産生一個Lua中的“錯誤”,在C/C++中它是一個異常,隻有用lua_pcall才能保證這個錯誤被Lua腳本引擎正确處理。

1 intrun_user_script(intnargs,intnresults,interrfunc )2 {3 g_begin=GetTickCount();4 _InterlockedIncrement(&g_counter );5 interr=lua_pcall( g_lua, nargs, nresults, errfunc );6 _InterlockedIncrement(&g_counter );7 returnerr;8 }

下面的check_script_timeout用來檢測腳本逾時,需要在另外一個線程中周期性的調用,原因我想就不用解釋了吧。它首先檢查是否在執行使用者腳本,或者是否已經讓目前執行的使用者腳本中止過。然後看這段腳本執行了多長時間,超過限度就把目前腳本計數記錄到g_check中去,并通過lua_sethook設定一個鈎子函數timeout_break,這個鈎子函數會在使用者腳本執行時被調用。

1 voidcheck_script_timeout()2 {3 longcounter=g_counter;4 5 //沒有執行使用者腳本, 不檢查逾時6 if( (counter&0x00000001)==0)7 return;8 9 //已經讓目前執行的使用者腳本中止了10 if( g_check==counter )11 return;12 13 //如果執行時間超過了設定的逾時時間(這裡是1秒), 終止它14 if( GetTickCount()-g_begin>1000)15 {16 g_check=counter;17 intmask=LUA_MASKCALL|LUA_MASKRET|LUA_MASKLINE|LUA_MASKCOUNT;18 lua_sethook( g_lua, timeout_break, mask,1);19 }20 }

最後就是那個鈎子函數了,它首先把鈎子去掉,因為這個鈎子隻要執行一次就行了。由于設定鈎子和執行鈎子是在不同的線程中,并且鈎子從設定到執行需要一定的時間,是以它要通過對比g_check和g_counter來判斷是否還在運作判斷逾時所執行的那段腳本,不是就什麼也不做,是就通過luaL_error産生一個錯誤,并中止腳本的執行,而這個錯誤最終會被run_user_script中的lua_pcall捕獲。

1 voidtimeout_break( lua_State*L, lua_Debug*ar )2 {3 lua_sethook( L, NULL,0,0);4 //鈎子從設定到執行, 需要一段時間, 是以要檢測是否仍在執行那個逾時的腳本5 if( g_check==g_counter )6 luaL_error( L,"script timeout.");7 }

上面的檢測使用了兩個線程,其實在一個線程中也可以做到,并且更簡單。但那樣會導緻鈎子函數頻繁執行,影響效率,如果對性能沒什麼要求的話,也可以采用。