-#ifndef PNPTEST_H
-#define PNPTEST_H
+#ifndef KMTEST_H
+#define KMTEST_H
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
+#include "ntddk.h"
+
/*
Some macros, structs, and vars are based or inspired from the great
extern tls_data glob_data;
VOID StartTest();
-VOID FinishTest(LPSTR TestName);
+VOID FinishTest(HANDLE KeyHandle, LPWSTR TestName);
void kmtest_set_location(const char* file, int line);
#ifdef __GNUC__
#endif /* __GNUC__ */
-#define ok_(file, line) (kmtest_set_location(file, line), 0) ? 0 : kmtest_ok
+#define ok_(file, line) (kmtest_set_location(file, line), 0) ? (void)0 : kmtest_ok
#define ok ok_(__FILE__, __LINE__)
-#endif /* PNPTEST_H */
+PDEVICE_OBJECT AttachDeviceObject;
+PDEVICE_OBJECT MainDeviceObject;
+PDRIVER_OBJECT ThisDriverObject;
+
+#endif /* KMTEST_H */