add timing support

Author: RadWolfie

README.md

@@ -21,6 +21,7 @@ The following list of options can be used inside the config.txt file:
 - `tests-exclude` = `<decimal (support up to 378) or case insensitive API name>[,...]`
 - `disable-video` = `<boolean>`[^1]
 - `disable-verbose` = `<boolean>`[^1]
+- `disable-timing` = `<boolean>`[^1]
 
 [^1]: boolean value can be 1 or 0
 

src/main.cpp

@@ -28,6 +28,7 @@ extern "C" {
 // defined in util/output.h file, used privately here only
 extern "C" BOOL output_video;
 extern "C" BOOL output_verbose;
+extern "C" BOOL output_timing;
 // Initialize the actual default values here if the config file is either successfully loaded before reading inputs or it failed to load.
 static void init_default_values()
 {
@@ -191,6 +192,9 @@ int load_conf_file(const char *file_path)
         if (strcmp("disable-verbose", current_key) == 0) {
             output_verbose = !strtoul(strtok(NULL, NEWLINE_DELIMITER), NULL, 10);
         }
+        if (strcmp("disable-timing", current_key) == 0) {
+            output_timing = !strtoul(strtok(NULL, NEWLINE_DELIMITER), NULL, 10);
+        }
     }
 
     free(buffer);
@@ -219,10 +223,25 @@ static void run_tests()
             print("%zu test(s) will be excluded.", tests_to_run_excluded.count());
         }
     }
+    ULONGLONG frequency;
+    if (output_timing) {
+        frequency = KeQueryPerformanceFrequency();
+    }
+    ULONGLONG tick_start, tick_end;
     print("-------------------------------------------------------------");
     for (int i = 0; i < kernel_api_tests_size; i++) {
         if (tests_to_run.test(i)) {
+            if (output_timing) {
+                tick_start = KeQueryPerformanceCounter();
+            }
             kernel_api_tests[i].func(i + 1, kernel_api_tests[i].name);
+            if (output_timing) {
+                tick_end = KeQueryPerformanceCounter();
+                ULONGLONG total_ms = ((tick_end - tick_start) * 1000) / frequency;
+                ULONGLONG total_seconds = total_ms / 1000;
+                ULONGLONG remainder_ms = total_ms % 1000;
+                print("%03u - %s: Test completed in %llu.%03llu seconds", i + 1, kernel_api_tests[i].name, total_seconds, remainder_ms);
+            }
         }
     }
     print("------------------------ End of Tests -----------------------");

src/util/output.c

@@ -11,6 +11,7 @@
 
 BOOL output_video = FALSE; // NOTE: Must be set to a default of FALSE until config file is loaded and before video initialization.
 BOOL output_verbose = TRUE;
+BOOL output_timing = TRUE;
 
 static HANDLE output_filehandle = INVALID_HANDLE_VALUE;