Fix time error bug when installing arkdrv driver.

BlackINT3 · Dec 22, 2020 · cf3d548 · cf3d548
1 parent 5a4f7b4
commit cf3d548
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/OpenArk/kernel/driver/driver.h b/src/OpenArk/kernel/driver/driver.h
@@ -77,10 +77,21 @@ std::wstring ParseDriverPath(UCHAR *symlnk);
 #define RECOVER_SIGN_TIME() \
 		SYSTEMTIME saved = { 0 };\
 		GetLocalTime(&saved);\
+		FILETIME ft;\
+		SystemTimeToFileTime(&saved, &ft);\
 		auto ticks = GetTickCount();\
 		SYSTEMTIME cur = { 0 };\
 		cur.wYear = 2012;\
 		cur.wMonth = 1;\
 		cur.wDay = 1;\
 		SetLocalTime(&cur);\
-		ON_SCOPE_EXIT([&] {	saved.wSecond += (GetTickCount() - ticks)/1000; SetLocalTime(&saved); });
+		ON_SCOPE_EXIT([&] {	\
+			LARGE_INTEGER li;\
+			li.LowPart = ft.dwLowDateTime;\
+			li.HighPart = ft.dwHighDateTime;\
+			li.QuadPart += (GetTickCount() - ticks) * 10000; \
+			ft.dwHighDateTime = li.HighPart; \
+			ft.dwLowDateTime = li.LowPart; \
+			FileTimeToSystemTime(&ft, &saved); \
+			SetLocalTime(&saved); \
+		});