@@ -180,20 +180,20 @@ class RACFED : public PassInfoMixin<RACFED> {
180180 GlobalVariable *RuntimeSigGV, Type *IntType);
181181
182182 // --- UPDATE BRANCH SIGNATURE BEFORE JUMP ---
183+ void checkBranches (Module &Md, GlobalVariable *RuntimeSigGV, Type *IntType,
184+ BasicBlock &BB);
183185
184- void checkBranches (Module &Md, GlobalVariable *RuntimeSigGV, Type *IntType,
185- IRBuilder<> B, BasicBlock &BB);
186-
187- void splitBBsAtCalls (Module &Md);
188- // int countOriginalInstructions(BasicBlock &BB);
189- CallBase *isCallBB (BasicBlock &BB);
190- void initializeEntryBlocksMap (Module &Md);
191- Value *getCondition (Instruction &I);
192- // void createCFGVerificationBB(BasicBlock &BB,
193- // std::unordered_map<BasicBlock *, int> &RandomNumberBBs,
194- // std::unordered_map<BasicBlock *, int> &SubRanPrevVals,
195- // Value &RuntimeSig, Value &RetSig,
196- // BasicBlock &ErrBB);
186+ // --- UPDATE RETURN VALUE AND CHECK ---
187+ void checkReturnValue (Module &Md, Function &Fn,
188+ GlobalVariable *RuntimeSigGV,
189+ Type *I32, Type *I64, BasicBlock &BB, std::unordered_map<std::string, std::vector<StringRef>> &lMap);
190+
191+
192+
193+ // void splitBBsAtCalls(Module &Md);
194+ // CallBase *isCallBB(BasicBlock &BB);
195+ // void initializeEntryBlocksMap(Module &Md);
196+ // Value *getCondition(Instruction &I);
197197
198198public:
199199 PreservedAnalyses run (Module &Md, ModuleAnalysisManager &);
0 commit comments