WIP

Author: NaC-L

lifter/lifter.cpp

@@ -5,6 +5,7 @@
 #include "includes.h"
 #include "lifterClass.h"
 #include "nt/nt_headers.hpp"
+#include "test_instructions.hpp"
 #include "utils.h"
 #include <fstream>
 #include <iostream>

lifter/lifterClass.h

@@ -269,7 +269,10 @@ class lifterClass {
   llvm::Value* TEB;
   llvm::Function* fnc;
 
-  lifterClass(llvm::IRBuilder<>& irbuilder) : builder(irbuilder){};
+  lifterClass(llvm::IRBuilder<>& irbuilder, uint64_t runtime_addr = 0)
+      : builder(irbuilder) {
+    InitRegisters(irbuilder.GetInsertBlock()->getParent(), runtime_addr);
+  };
 
   lifterClass(const lifterClass& other)
       : builder(other.builder), // Reference copied directly