heckler: update sev-step user lib build

ahoi-attacks · Apr 19, 2024 · 7f938e1 · 7f938e1
1 parent 043adae
commit 7f938e1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/userspace/sev-step/README.md b/userspace/sev-step/README.md
@@ -9,6 +9,12 @@ make
 
 Ensure to have built heckler host kernel first.
 
+```
+# Export kernel header files
+cd ../../linux/host
+make headers
+```
+
 ### Credits
 This directory contains code snippets of sev-step, released under GPLv2.
 https://github.com/sev-step/sev-step-userland/
diff --git a/userspace/sev-step/sev-step-lib/sev_step_api.h b/userspace/sev-step/sev-step-lib/sev_step_api.h
@@ -12,12 +12,20 @@
 #define _SEV_STEP_API_H
 
 #include "linux/kvm.h"
-#include "linux/sev-step/sev-step.h"
+#include "linux/heckler/sev-step.h"
 
 #include <stdint.h>
 #include <stdbool.h>
 
 
+enum kvm_page_track_mode {
+	KVM_PAGE_TRACK_WRITE,
+	KVM_PAGE_TRACK_EXEC,       // heckler: mark as nx
+	KVM_PAGE_TRACK_RESET_EXEC, // mark as !nx (not used for page track)
+	KVM_PAGE_TRACK_MAX,
+};
+
+
 #include "sev_step_error_codes.h"
 
 char* descriptive_perf_name(perf_ctl_config_t cfg);