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

Driver: fortstat: Simplify fort_flow_add()

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-01-08 18:26:49 +03:00
parent df0efe6d42
commit 723c92df1c
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/driver/fortstat.c
View File

@ -317,6 +317,22 @@ inline static UCHAR fort_stat_group_speed_limit(PFORT_CONF_GROUP conf_group, UCH
return (((conf_group->limit_io_bits) >> (group_index * 2)) & 3); return (((conf_group->limit_io_bits) >> (group_index * 2)) & 3);
} }
inline static NTSTATUS fort_flow_add_new(PFORT_STAT stat, PFORT_FLOW *flow, UINT64 flow_id,
tommy_key_t flow_hash, BOOL isIPv6, BOOL is_tcp, BOOL inbound, BOOL is_reauth)
{
if (is_reauth) {
/* Remove existing flow context after reauth. to be able to associate a flow-context */
if (!fort_flow_context_remove_id(stat, flow_id, isIPv6, is_tcp))
return FORT_STATUS_FLOW_BLOCK;
}
*flow = fort_flow_new(stat, flow_id, flow_hash, isIPv6, is_tcp, inbound);
if (*flow == NULL)
return STATUS_INSUFFICIENT_RESOURCES;
return STATUS_SUCCESS;
}
static NTSTATUS fort_flow_add(PFORT_STAT stat, UINT64 flow_id, UCHAR group_index, UINT16 proc_index, static NTSTATUS fort_flow_add(PFORT_STAT stat, UINT64 flow_id, UCHAR group_index, UINT16 proc_index,
BOOL isIPv6, BOOL is_tcp, BOOL inbound, BOOL is_reauth) BOOL isIPv6, BOOL is_tcp, BOOL inbound, BOOL is_reauth)
{ {
@ -325,15 +341,11 @@ static NTSTATUS fort_flow_add(PFORT_STAT stat, UINT64 flow_id, UCHAR group_index
BOOL is_new_flow = FALSE; BOOL is_new_flow = FALSE;
if (flow == NULL) { if (flow == NULL) {
if (is_reauth) { const NTSTATUS status = fort_flow_add_new(
/* Remove existing flow context after reauth. to be able to associate a flow-context */ stat, &flow, flow_id, flow_hash, isIPv6, is_tcp, inbound, is_reauth);
if (!fort_flow_context_remove_id(stat, flow_id, isIPv6, is_tcp))
return FORT_STATUS_FLOW_BLOCK;
}
flow = fort_flow_new(stat, flow_id, flow_hash, isIPv6, is_tcp, inbound); if (!NT_SUCCESS(status))
if (flow == NULL) return status;
return STATUS_INSUFFICIENT_RESOURCES;
is_new_flow = TRUE; is_new_flow = TRUE;
} else { } else {