#include "Assignment$-4.$h$"$

#include $"$SVF$-$LLVM$/$LLVMUtil$.$h$"$

#include $"$SVF$-$LLVM$/$SVFIRBuilder$.$h$"$

#include "Graphs$/$CallGraph$.$h$"$

#include "Util$/$Options$.$h$"$

#include "Util$/$CommandLine$.$h$"$

using namespace std;

void Test$1()$

$\{$

$$ cout $<<"$

running test$1$: $"<<$ endl;

$$ SVFModule $*$svfModule $=$ LLVMModuleSet::getLLVMModuleSet$()->$buildSVFModule$(\{"./$Assignment$-4/$testcase$/$bc$/$test$1.$ll$"\})$;

$///$ Build Program Assignment Graph $($SVFIR$)$

$$ SVF::SVFIRBuilder builder$($svfModule$)$;

$$ SVF::SVFIR $*$pag $=$ builder.build$()$;

$$ TaintGraphTraversal $*$taint $=$ new TaintGraphTraversal$($pag$)$;

$$ taint$->$taintChecking$()$;

$$ set expected $=\{"$START$->5->1->2->3->6->7->8->9->$END"$\}$;

$$ assert$($taint$->$getPaths$()==$ expected && $"$

wrong paths generated $-$ test$1$ failed $!")$;

$$ cout $<<"$

test$1$ passed $!"<<$ endl;

$$ SVF::LLVMModuleSet::releaseLLVMModuleSet$()$;

$$ SVF::SVFIR::releaseSVFIR$()$;

$\}$

void Test$2()$

$\{$

$$ cout $<<"$

running test$2$ :$"<<$ endl;

$$ SVFModule $*$svfModule $=$ LLVMModuleSet::getLLVMModuleSet$()->$buildSVFModule$(\{"./$Assignment$-4/$testcase$/$bc$/$test$2.$ll$"\})$;

$///$ Build Program Assignment Graph $($SVFIR$)$

$$ SVF::SVFIRBuilder builder$($svfModule$)$;

$$ SVF::SVFIR $*$pag $=$ builder.build$()$;

$$ TaintGraphTraversal $*$taint $=$ new TaintGraphTraversal$($pag$)$;

$$ taint$->$taintChecking$()$;

$$ assert$($taint$->$getPaths$().$size$()==0$ && $"$

should not exist tainted path $-$ test$2$ failed $!")$;

$$ cout $<<"$

test$2$ passed $!"<<$ endl;

$$ SVF::LLVMModuleSet::releaseLLVMModuleSet$()$;

$$ SVF::SVFIR::releaseSVFIR$()$;

$\}$

void Test$3()$

$\{$

$$ cout $<<"$

running test$3$ :$"<<$ endl;

$$ SVFModule $*$svfModule $=$ LLVMModuleSet::getLLVMModuleSet$()->$buildSVFModule$(\{"./$Assignment$-4/$testcase$/$bc$/$test$3.$ll$"\})$;

$///$ Build Program Assignment Graph $($SVFIR$)$

$$ SVF::SVFIRBuilder builder$($svfModule$)$;

$$ SVF::SVFIR $*$pag $=$ builder.build$()$;

$$ TaintGraphTraversal $*$taint $=$ new TaintGraphTraversal$($pag$)$;

$$ taint$->$taintChecking$()$;

$$ assert$($taint$->$getPaths$().$size$()==0$ && $"$

should not exist tainted path $-$ test$3$ failed $!")$;

$$ cout $<<"$

test$3$ passed $!"<<$ endl;

$$ SVF::SVFIR::releaseSVFIR$()$;

$$ SVF::LLVMModuleSet::releaseLLVMModuleSet$()$;

$\}$

void Test$4()$

$\{$

$$ cout $<<"$

running test$4$ :$"<<$ endl;

$$ SVFModule $*$svfModule $=$ LLVMModuleSet::getLLVMModuleSet$()->$buildSVFModule$(\{"./$Assignment$-4/$testcase$/$bc$/$test$4.$ll$"\})$;

$///$ Build Program Assignment Graph $($SVFIR$)$

$$ SVF::SVFIRBuilder builder$($svfModule$)$;

$$ SVF::SVFIR $*$pag $=$ builder.build$()$;

$$ TaintGraphTraversal $*$taint $=$ new TaintGraphTraversal$($pag$)$;

$$ taint$->$taintChecking$()$;

$$ set expected $=\{"$START$->5->1->2->3->6->7->8->9->10->12->14->$END"$\}$;

$$ assert$($taint$->$getPaths$()==$ expected && $"$

wrong paths generated $-$ test$4$ failed $!")$;

$$ cout $<<"$

test$4$ passed $!"<<$ endl;

$$ SVF::LLVMModuleSet::releaseLLVMModuleSet$()$;

$$ SVF::SVFIR::releaseSVFIR$()$;

$\}$

int main$($int argc, char $**$argv$)$

$\{$

$$ int arg$\_$num $=0$;

$$ int extraArgc $=1$;

$$ char $**$arg$\_$value $=$ new char $*[$argc $+$ extraArgc$]$;

$$ for $($; arg$\_$num $<$ argc; $++$arg$\_$num$)$

$\{$

$$ arg$\_$value$[$arg$\_$num$]=$ argv$[$arg$\_$num$]$;

$\}$

$//$ You may comment it to see the details of the analysis

$$ arg$\_$value$[$arg$\_$num$++]=($char $*)"-$stat$=$false";

$$ std::vector moduleNameVec;

$$ moduleNameVec $=$ OptionBase::parseOptions$($

$$ arg$\_$num, arg$\_$value, "Teaching$-$Software$-$Analysis Assignment $4","[$options$]")$;

$$ Test$1()$;

$$ Test$2()$;

$$ Test$3()$;

$$ Test$4()$;

$$ return $0$;

$\}$

These are the test files