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

#include

#include

#include

#include

using namespace SVF;

using namespace llvm;

using namespace std;

$//$ Local set to store paths for testing verification

std::set collectedPaths;

$//$ Get sources function names read from checker$\_$source$\_$api collected from a text file

std::set &TaintGraphTraversal::identifySources$()$

$\{$

$$ static std::set sources;

$$ for $($const CallICFGNode $*$cs : pag$->$getCallSiteSet$())$

$\{$

$$ const SVFFunction $*$fun $=$ SVFUtil::getCallee$($cs$->$getCallSite$())$;

$$ if $($checker$\_$source$\_$api.find$($fun$->$getName$())!=$ checker$\_$source$\_$api.end$())$

$\{$

$$ sources.insert$($cs$)$;

$\}$

$\}$

$$ return sources;

$\}$

$//$ Get sinks function names read from checker$\_$sink$\_$api collected from a text file

std::set &TaintGraphTraversal::identifySinks$()$

$\{$

$$ static std::set sinks;

$$ for $($const CallICFGNode $*$cs : pag$->$getCallSiteSet$())$

$\{$

$$ const SVFFunction $*$fun $=$ SVFUtil::getCallee$($cs$->$getCallSite$())$;

$$ if $($checker$\_$sink$\_$api.find$($fun$->$getName$())!=$ checker$\_$sink$\_$api.end$())$

$\{$

$$ sinks$.$insert$($cs$)$;

$\}$

$\}$

$$ return sinks;

$\}$

$//$ Start taint checking

void TaintGraphTraversal::taintChecking$()$

$\{$

$//$ Configure sources and sinks for taint analysis

$$ readSrcSnkFromFile$("./$Assignment$-4/$SrcSnk$.$txt$")$;

$$ ander $=$ new AndersenPTA$($pag$)$;

$$ ander$->$analyze$()$;

$$ for $($const CallICFGNode $*$src : identifySources$())$

$\{$

$$ for $($const CallICFGNode $*$snk : identifySinks$())$

$\{$

$$ if $($aliasCheck$($src$,$ snk$))$

$$ reachability$($src$,$ snk$)$;

$\}$

$\}$

$//$ Output collected paths for testing verification

$$ cout $<<$ "Collected paths:" $<<$ endl;

$$ for $($const auto &path : collectedPaths$)$

$\{$

$$ cout $<<$ path $<<$ endl;

$\}$

$\}$

$//$ Collect and format ICFG path details

void TaintGraphTraversal::collectICFGPath$($std::vector &path$)$

$\{$

$$ std::stringstream pathStr;

$$ pathStr $<<$ "START";

$$ for $($size$\_$t i $=0$; i $<$ path.size$()$; $++$i$)$

$\{$

$$ pathStr $<<"->"<<$ path$[$i$]$;

$\}$

$$ pathStr $<<"->$END";

$//$ Store the formatted path in the local collectedPaths set

$$ collectedPaths.insert$($pathStr$.$str$())$;

$//$ Debug print to check path being added

$$ cout $<<$ "Adding path: $"<<$ pathStr.str$()<<$ endl;

$\}$

$//$ Parse the $`$SrcSnk$.$txt$`$ file for sources and sinks

void TaintGraphTraversal::readSrcSnkFromFile$($const string &filename$)$

$\{$

$$ std::ifstream file$($filename$)$;

$$ if $(!$file.is$\_$open$())$

$\{$

$$ return;

$\}$

$$ std::string line;

$//$ Read sources

$$ if $($getline$($file$,$ line$))$

$\{$

$$ std::istringstream ss$($line$.$substr$(1,$ line.size$()-2))$; $//$ Remove $\{\}$

$$ std::string api;

$$ while $($ss $>>$ api$)$

$\{$

$$ checker$\_$source$\_$api.insert$($api$)$;

$\}$

$\}$

$//$ Read sinks

$$ if $($getline$($file$,$ line$))$

$\{$

$$ std::istringstream ss$($line$.$substr$(1,$ line.size$()-2))$; $//$ Remove $\{\}$

$$ std::string api;

$$ while $($ss $>>$ api$)$

$\{$

$$ checker$\_$sink$\_$api.insert$($api$)$;

$\}$

$\}$

$$ file.close$()$;

$\}$

$//$ Perform alias checking with dyn$\_$cast for safety

bool TaintGraphTraversal::aliasCheck$($const CallICFGNode $*$src$,$ const CallICFGNode $*$snk$)$

$\{$

$$ const RetICFGNode $*$retNode $=$ SVFUtil::dyn$\_$cast$($src$)$;

$$ const CallICFGNode $*$callNode $=$ SVFUtil::dyn$\_$cast$($snk$)$;

$$ if $(!$retNode $||!$callNode$)$

$\{$

$//$ Return false if casting fails

$$ return false;

$\}$

$$ const SVFVar $*$srcRet $=$ retNode$->$getActualRet$()$;

$$ const auto &snkParams $=$ callNode$->$getActualParms$()$;

$$ for $($const auto $*$param : snkParams$)$

$\{$

$$ if $($ander$->$alias$($srcRet$->$getId$(),$ param$->$getId$()))$

$\{$

$$ return true;

$\}$

$\}$

$$ return false;

$\}$

Why does it not work and pass my test files: