mirrors/fort
1
0
Fork 0
mirror of https://github.com/tnodir/fort synced 2024-11-15 09:59:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Driver: fortstat: Improve fort_stat_close_flows()

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-04-04 07:54:53 +03:00
parent 80f001cca0
commit 03ff18fb54
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/driver/fortstat.c
View File

@ -399,10 +399,13 @@ FORT_API void fort_stat_close_flows(PFORT_STAT stat)
KLOCK_QUEUE_HANDLE lock_queue;
KeAcquireInStackQueuedSpinLock(&stat->lock, &lock_queue);
{
InterlockedAdd(&stat->flow_closing_count, (LONG) stat->flows_map.count);
const UCHAR old_stat_flags = fort_stat_flags_set(stat, FORT_STAT_LOG, FALSE);
fort_stat_flags_set(stat, FORT_STAT_LOG, FALSE);
fort_stat_flags_set(stat, FORT_STAT_CLOSED, TRUE);
if ((old_stat_flags & FORT_STAT_CLOSED) == 0) {
fort_stat_flags_set(stat, FORT_STAT_CLOSED, TRUE);
InterlockedAdd(&stat->flow_closing_count, (LONG) stat->flows_map.count);
}
}
KeReleaseInStackQueuedSpinLock(&lock_queue);
@ -415,7 +418,7 @@ FORT_API void fort_stat_close_flows(PFORT_STAT stat)
/* Wait for asynchronously deleting flows */
LARGE_INTEGER delay;
delay.QuadPart = -150 * 1000 * 10; /* sleep 150000us (150ms) */
delay.QuadPart = -50 * 1000 * 10; /* sleep 50000us (50ms) */
KeDelayExecutionThread(KernelMode, FALSE, &delay);
}