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

Driver: fortcout: Simplify fort_callout_install()

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-02-24 12:40:35 +03:00
parent a15f126b59
commit ecf822dedd
1 changed files with 31 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/driver/fortcout.c
View File

@ -613,7 +613,7 @@ inline static NTSTATUS fort_callout_register(
return STATUS_SUCCESS;
}
FORT_API NTSTATUS fort_callout_install(PDEVICE_OBJECT device)
static NTSTATUS fort_callout_install_ale(PDEVICE_OBJECT device)
{
const FWPS_CALLOUT0 callouts[] = {
/* IPv4 connect callout */
@ -640,6 +640,15 @@ FORT_API NTSTATUS fort_callout_install(PDEVICE_OBJECT device)
.classifyFn = &fort_callout_accept_v6,
.notifyFn = &fort_callout_notify,
},
};
return fort_callout_register(
device, callouts, /*count=*/sizeof(callouts) / sizeof(callouts[0]));
}
static NTSTATUS fort_callout_install_stream(PDEVICE_OBJECT device)
{
const FWPS_CALLOUT0 callouts[] = {
/* IPv4 stream callout */
{
.calloutKey = FORT_GUID_CALLOUT_STREAM_V4,
@ -672,6 +681,15 @@ FORT_API NTSTATUS fort_callout_install(PDEVICE_OBJECT device)
.flowDeleteFn = &fort_callout_flow_delete,
.flags = FWP_CALLOUT_FLAG_CONDITIONAL_ON_FLOW,
},
};
return fort_callout_register(
device, callouts, /*count=*/sizeof(callouts) / sizeof(callouts[0]));
}
static NTSTATUS fort_callout_install_transport(PDEVICE_OBJECT device)
{
const FWPS_CALLOUT0 callouts[] = {
/* IPv4 inbound transport callout */
{
.calloutKey = FORT_GUID_CALLOUT_IN_TRANSPORT_V4,
@ -710,6 +728,18 @@ FORT_API NTSTATUS fort_callout_install(PDEVICE_OBJECT device)
device, callouts, /*count=*/sizeof(callouts) / sizeof(callouts[0]));
}
FORT_API NTSTATUS fort_callout_install(PDEVICE_OBJECT device)
{
NTSTATUS status;
if (!NT_SUCCESS(status = fort_callout_install_ale(device))
|| !NT_SUCCESS(status = fort_callout_install_stream(device))
|| !NT_SUCCESS(status = fort_callout_install_transport(device))) {
return status;
}
return STATUS_SUCCESS;
}
FORT_API void fort_callout_remove(void)
{
PFORT_STAT stat = &fort_device()->stat;