PenTest/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2025-12-25 17:09:02 +00:00
Code Issues Projects Releases Wiki Activity

Simplify check for immediate return from nsock

Browse Source
This commit is contained in:
dmiller
2025-04-11 14:57:55 +00:00
parent 5b52e7a3f2
commit 06d3c923bb
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
nse_nsock.cc
View File

@@ -66,6 +66,8 @@ typedef struct nse_nsock_udata
} nse_nsock_udata;
static const char *NU_ACTION_IMMEDIATE = "returned immediately";
static int gc_pool (lua_State *L)
{
nsock_pool *nsp = (nsock_pool *) lua_touserdata(L, 1);
@@ -361,7 +363,7 @@ static void callback (nsock_pool nsp, nsock_event nse, void *ud)
// l_connect to return an error instead of yielding.
// http://seclists.org/nmap-dev/2016/q1/201
trace(nse_iod(nse), nu->action, nu->direction);
nu->action = "ERROR";
nu->action = NU_ACTION_IMMEDIATE;
return;
}
switch (nse_type(nse)) {
@@ -574,9 +576,9 @@ static int connect (lua_State *L, int status, lua_KContext ctx)
if (dest != NULL)
freeaddrinfo(dest);
if (!strncmp(nu->action, "ERROR", 5)) {
if (nu->action == NU_ACTION_IMMEDIATE) {
// Immediate error
return nseU_safeerror(L, "Nsock connect failed immediately");
return nseU_safeerror(L, nse_status2str(NSE_STATUS_ERROR));
}
return yield(L, nu, "CONNECT", TO, 0, NULL);
}