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

Driver: fortstat: Simplify fort_flow_delete()

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-03-21 08:33:22 +03:00
parent 9ef48719ac
commit c5d78becce
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/driver/fortstat.c
View File

@ -527,22 +527,27 @@ FORT_API NTSTATUS fort_flow_associate(PFORT_STAT stat, UINT64 flow_id, UINT32 pr
return status; return status;
} }
static BOOL fort_flow_delete_closing(PFORT_STAT stat)
{
if ((fort_stat_flags(stat) & FORT_STAT_CLOSED) != 0) {
InterlockedDecrement(&stat->flow_closing_count);
return TRUE;
}
return FALSE;
}
FORT_API void fort_flow_delete(PFORT_STAT stat, UINT64 flowContext) FORT_API void fort_flow_delete(PFORT_STAT stat, UINT64 flowContext)
{ {
PFORT_FLOW flow = (PFORT_FLOW) flowContext; PFORT_FLOW flow = (PFORT_FLOW) flowContext;
if ((fort_stat_flags(stat) & FORT_STAT_CLOSED) != 0) { if (fort_flow_delete_closing(stat))
InterlockedDecrement(&stat->flow_closing_count);
return; return;
}
KLOCK_QUEUE_HANDLE lock_queue; KLOCK_QUEUE_HANDLE lock_queue;
KeAcquireInStackQueuedSpinLock(&stat->lock, &lock_queue); KeAcquireInStackQueuedSpinLock(&stat->lock, &lock_queue);
/* Double check for locked flows */ /* Double check for locked flows */
if ((fort_stat_flags(stat) & FORT_STAT_CLOSED) != 0) { if (!fort_flow_delete_closing(stat)) {
InterlockedDecrement(&stat->flow_closing_count);
} else {
fort_flow_free(stat, flow); fort_flow_free(stat, flow);
} }