Fix the BUG that functions in intrin.h cannot be used even if it is i… (

#33)

* Fix the BUG that functions in intrin.h cannot be used even if it is introduced

* add demonstrate

cmake/FindWdk.cmake

@@ -96,6 +96,7 @@ set(WDK_COMPILE_FLAGS
     "/kernel"  # create kernel mode binary
     "/FIwarning.h" # disable warnings in WDK headers
     "/FI${WDK_ADDITIONAL_FLAGS_FILE}" # include file to disable RTC
+	"/Oi" # enable intrinsic functions so that you can use functions like _disable or _enable
     )
 
 set(WDK_COMPILE_DEFINITIONS "WINNT=1")

samples/CMakeLists.txt

@@ -16,4 +16,5 @@ add_subdirectory(MinifilterCppDriver)
 add_subdirectory(WdmCppDriver)
 add_subdirectory(WdmCppLib)
 add_subdirectory(WdmDriver)
-add_subdirectory(WdmLib)
+add_subdirectory(WdmLib)
+add_subdirectory(WdmIntrinsicFunctions)

samples/WdmIntrinsicFunctions/CMakeLists.txt

@@ -0,0 +1,4 @@
+wdk_add_driver(WdmIntrinsicFunctions
+    Main.cpp
+    )
+target_link_libraries(WdmIntrinsicFunctions WdmCppLib)

samples/WdmIntrinsicFunctions/Main.cpp

@@ -0,0 +1,25 @@
+#include <wdm.h>
+#include <intrin.h>
+
+DRIVER_UNLOAD driverUnload;
+VOID driverUnload(_In_ PDRIVER_OBJECT driverObject)
+{
+    UNREFERENCED_PARAMETER(driverObject);
+
+    DbgPrint("Driver unloaded\n");
+}
+
+EXTERN_C NTSTATUS DriverEntry(_In_ PDRIVER_OBJECT driverObject, _In_ PUNICODE_STRING registryPath)
+{
+    UNREFERENCED_PARAMETER(registryPath);
+
+    DbgPrint("Driver loaded\n");
+
+    _disable(); //use intrinsic function
+    ULONG64 cr3 = __readcr3(); //use intrinsic function
+    _enable(); //use intrinsic function
+
+    DbgPrint("CR3: %p\n", cr3);
+    driverObject->DriverUnload = driverUnload;
+    return STATUS_SUCCESS;
+}