driver: move init-only functions into INIT segment

Reference: https://docs.microsoft.com/en-us/windows-hardware/drivers/kernel/writing-a-driverentry-routine Signed-off-by: Simon Rozman <[email protected]>
WireGuard · May 5, 2021 · 4eb306a · 4eb306a
1 parent aba41fc
commit 4eb306a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/driver/wintun.c b/driver/wintun.c
@@ -838,6 +838,12 @@ TunForceHandlesClosed(_Inout_ DEVICE_OBJECT *DeviceObject)
     return DidClose;
 }
 
+_Must_inspect_result_
+static NTSTATUS TunInitializeDispatchSecurityDescriptor(VOID);
+#ifdef ALLOC_PRAGMA
+#    pragma alloc_text(INIT, TunInitializeDispatchSecurityDescriptor)
+#endif
+_Use_decl_annotations_
 static NTSTATUS TunInitializeDispatchSecurityDescriptor(VOID)
 {
     NTSTATUS Status;
@@ -1451,6 +1457,9 @@ TunUnload(PDRIVER_OBJECT DriverObject)
 }
 
 DRIVER_INITIALIZE DriverEntry;
+#ifdef ALLOC_PRAGMA
+#    pragma alloc_text(INIT, DriverEntry)
+#endif
 _Use_decl_annotations_
 NTSTATUS
 DriverEntry(DRIVER_OBJECT *DriverObject, UNICODE_STRING *RegistryPath)