diff --git a/mingw-build.sh b/mingw-build.sh index 454c060..931a56a 100644 --- a/mingw-build.sh +++ b/mingw-build.sh @@ -39,13 +39,15 @@ do fi echo "BUILD WDDK-$CPU" CC="$ENV-gcc" + STRIP="$ENV-strip" if [ -x "`which $CC`" ] then echo "\tmake install/MINGW/$CPU..." mkdir -p "install/MINGW/$CPU" echo "\tbuild install/MINGW/$CPU/WinDivert.dll..." $CC -Wall -O2 -Iinclude/ -c dll/divert.c -o dll/divert.o - $CC -Wall -shared -o "install/MINGW/$CPU/WinDivert.dll" dll/divert.o + $CC -Wall -shared -o "install/MINGW/$CPU/WinDivert.dll" dll/divert.o + $STRIP --strip-debug "install/MINGW/$CPU/WinDivert.dll" echo "\tbuild install/MINGW/$CPU/netdump.exe..." $CC -s -O2 -Iinclude/ examples/netdump/netdump.c \ -o "install/MINGW/$CPU/netdump.exe" -lWinDivert -lws2_32 \ diff --git a/sys/divert.c b/sys/divert.c index e69514d..4cbbf3f 100644 --- a/sys/divert.c +++ b/sys/divert.c @@ -556,13 +556,17 @@ extern NTSTATUS DriverEntry(IN PDRIVER_OBJECT driver_obj, WdfControlFinishInitializing(device); // Create the packet injection handles. - status = FwpsInjectionHandleCreate0(AF_INET, 0, &inject_handle); + status = FwpsInjectionHandleCreate0(AF_INET, + FWPS_INJECTION_TYPE_NETWORK | FWPS_INJECTION_TYPE_FORWARD, + &inject_handle); if (!NT_SUCCESS(status)) { DEBUG_ERROR("failed to create WFP packet injection handle", status); return status; } - status = FwpsInjectionHandleCreate0(AF_INET6, 0, &injectv6_handle); + status = FwpsInjectionHandleCreate0(AF_INET6, + FWPS_INJECTION_TYPE_NETWORK | FWPS_INJECTION_TYPE_FORWARD, + &injectv6_handle); if (!NT_SUCCESS(status)) { DEBUG_ERROR("failed to create WFP ipv6 packet injection handle",