From 170713f571cf875d22b42f9ba58261741cddb495 Mon Sep 17 00:00:00 2001
From: a1xd <68629610+a1xd@users.noreply.github.com>
Date: Mon, 6 Sep 2021 19:14:05 -0400
Subject: move safeboot check into DriverEntry

---
 driver/driver.cpp | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

(limited to 'driver')

diff --git a/driver/driver.cpp b/driver/driver.cpp
index 2a4f460..07b947f 100644
--- a/driver/driver.cpp
+++ b/driver/driver.cpp
@@ -319,8 +319,6 @@ RawaccelInit(WDFDRIVER driver)
 {
     NTSTATUS status;
 
-    if (*InitSafeBootMode > 0) return;
-
     status = CreateControlDevice(driver);
 
     if (!NT_SUCCESS(status)) {
@@ -472,7 +470,7 @@ Routine Description:
 
 
     if (NT_SUCCESS(status)) {
-        RawaccelInit(driver);
+        if (*InitSafeBootMode == 0) RawaccelInit(driver);
     }
     else {
         DebugPrint(("WdfDriverCreate failed with status 0x%x\n", status));
-- 
cgit v1.2.3