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

IOCP engine: further checks against duplicating eovs

Browse Source
This commit is contained in:
dmiller
2024-10-08 21:14:42 +00:00
parent 30bc184e0f
commit f765af4070
1 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
nsock/src/engine_iocp.c
View File

@@ -275,20 +275,33 @@ int iocp_iod_modify(struct npool *nsp, struct niod *iod, struct nevent *nse, int
new_events |= ev_set;
new_events &= ~ev_clr;
if (new_events == iod->watched_events)
return 1; /* nothing to do */
iod->watched_events = new_events;
switch (nse->type) {
case NSE_TYPE_CONNECT:
case NSE_TYPE_CONNECT_SSL:
break;
case NSE_TYPE_READ:
case NSE_TYPE_PCAP_READ:
assert((ev_set | ev_clr) == EV_READ);
break;
case NSE_TYPE_WRITE:
assert((ev_set | ev_clr) == EV_WRITE);
break;
default:
fatal("Invalid NSE type %d for iocp_iod_modify", nse->type);
break;
}
if (ev_set != EV_NONE) {
// XXX: Pretty sure this is only for the SSL connect case.
// We may be able to make this more efficient?
if (!nse->eov)
initiate_overlapped_event(nsp, nse);
}
else if (ev_clr != EV_NONE)
terminate_overlapped_event(nsp, nse);
if (new_events == iod->watched_events)
return 1; /* nothing to do */
iod->watched_events = new_events;
return 1;
}