PenTest/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2025-12-15 20:29:03 +00:00
Code Issues Projects Releases Wiki Activity

Better response to misuse of stdnse.base|new_thread when not executing an

Browse Source 
action function (current == nil).
This commit is contained in:
batrick
2012-09-23 21:31:02 +00:00
parent 6b86535138
commit 4fe62b040c
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
nse_main.lua
View File

@@ -790,7 +790,7 @@ local function run (threads_iter, hosts)
return NSE_YIELD_VALUE; -- return NSE_YIELD_VALUE
end
_R[BASE] = function ()
return current.co;
return current and current.co;
end
-- _R[WAITING_TO_RUNNING] is called by nse_restore in nse_main.cc
_R[WAITING_TO_RUNNING] = function (co, ...)
@@ -823,6 +823,9 @@ local function run (threads_iter, hosts)
end
rawset(stdnse, "new_thread", function (main, ...)
assert(type(main) == "function", "function expected");
if current == nil then
error "stdnse.new_thread can only be run from an active script"
end
local co = create(function(...) main(...) end); -- do not return results
print_debug(2, "%s spawning new thread (%s).",
current.parent.info, tostring(co));
@@ -852,7 +855,7 @@ local function run (threads_iter, hosts)
return co, info;
end);
rawset(stdnse, "base", function ()
return current.co;
return current and current.co;
end);
while threads_iter and num_threads < CONCURRENCY_LIMIT do
@@ -1285,6 +1288,10 @@ local function main (hosts, scantype)
print_verbose(1, "Script Post-scanning.");
end
-- These functions do not exist until we are executing action functions.
rawset(stdnse, "new_thread", nil)
rawset(stdnse, "base", nil)
for runlevel, scripts in ipairs(runlevels) do
-- This iterator is passed to the run function. It returns one new script
-- thread on demand until exhausted.