blah

Cisco-Talos · Apr 29, 2024 · 2352730 · 2352730
1 parent 3418d67
commit 2352730
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/libclambcc/ClamBCRemovePointerPHIs.cpp b/libclambcc/ClamBCRemovePointerPHIs.cpp
@@ -363,7 +363,8 @@ class ClamBCRemovePointerPHIs : public PassInfoMixin<ClamBCRemovePointerPHIs>
 
             Type * vt = Type::getVoidTy(pMod->getContext());
             Type * pt = Type::getInt8Ty(pMod->getContext())->getPointerTo();
-            pRet = FunctionType::get(vt, {pt->getPointerTo(), pt}, false);
+            Type * it = Type::getInt32Ty(pMod->getContext());
+            pRet = FunctionType::get(vt, {pt->getPointerTo(), it, pt, it}, false);
 
         }
 
@@ -379,7 +380,7 @@ return pRet;
         pRet->addFnAttr(Attribute::NoInline);
 
         BasicBlock * pEntry = BasicBlock::Create(pMod->getContext(), "entry", pRet, nullptr);
-            new StoreInst(pRet->getArg(1), pRet->getArg(0), pEntry);
+            new StoreInst(pRet->getArg(2), pRet->getArg(0), pEntry);
             ReturnInst::Create(pMod->getContext(), nullptr, pEntry);
 
             DEBUG_VALUE(pRet);
@@ -390,7 +391,8 @@ return pRet;
 
     void savePointer(Value * src, Value * dst, Instruction * insPt){
         Function * saveFunction = getSaveFunction();
-        CallInst::Create(getSaveFunctionType(), saveFunction, {dst, src}, "", insPt);
+        ConstantInt * pci = ConstantInt::get(Type::getInt32Ty(pMod->getContext()), 8);
+        CallInst::Create(getSaveFunctionType(), saveFunction, {dst, pci, src, pci}, "", insPt);
         //CallInst::Create(saveFunction->getType(), saveFunction, {dst, src}, "ClamBCRemovePointerPHIs_", insPt);
     }
 

diff --git a/libclambcc/ClamBCWriter.cpp b/libclambcc/ClamBCWriter.cpp
@@ -855,6 +855,10 @@ class ClamBCWriter : public PassInfoMixin<ClamBCWriter>, public InstVisitor<Clam
 
         pMod = F.getParent();
 
+        DEBUG_NONPOINTER("Decide whether or not i need this");
+        F.addFnAttr(Attribute::OptimizeNone);
+        F.addFnAttr(Attribute::NoInline);
+
         BBMap.clear();
         dbgInfo.clear();
         anyDbg = false;