[TSAR, Analysis] Add pass for Restrict Arguments analysis.

Author: VolandTymim

include/tsar/Analysis/Memory/Passes.h

@@ -191,6 +191,12 @@ ImmutablePass *createGlobalDefinedMemoryStorage();
 /// analysis.
 void initializeGlobalDefinedMemoryWrapperPass(PassRegistry &Registry);
 
+/// Initialize a pass to perform iterprocedural analysis of restrict arguments
+void initializeRestrictionArgumentsPassPass(PassRegistry& Registry);
+
+/// Create a pass to perform iterprocedural analysis of restrict arguments
+ModulePass* createRestrictionArgumentsPass();
+
 /// Create analysis server.
 ModulePass *createDIMemoryAnalysisServer();
 

include/tsar/Analysis/Memory/RestrictionArgumentsPass.h

@@ -0,0 +1,93 @@
+
+#ifndef SAPFOR_RESTRICTIONARGUMENTSPASS_H
+#define SAPFOR_RESTRICTIONARGUMENTSPASS_H
+
+#include <llvm/Pass.h>
+#include <bcl/utility.h>
+#include "tsar/Analysis/Memory/Passes.h"
+#include <llvm/IR/Instructions.h>
+#include <llvm/IR/Type.h>
+
+namespace llvm {
+
+    class RestrictionArgumentsPass : public ModulePass, private bcl::Uncopyable {
+    public:
+        static char ID;
+
+        RestrictionArgumentsPass() : ModulePass(ID) {
+            initializeRestrictionArgumentsPassPass(*PassRegistry::getPassRegistry());
+        }
+
+        bool runOnModule(llvm::Module& M) override;
+
+        void getAnalysisUsage(AnalysisUsage& AU) const override;
+    };
+
+}
+
+namespace tsar {
+
+    typedef llvm::DenseSet<llvm::Value*> MemorySources;
+
+    struct FunctionCallWithMemorySources {
+        llvm::CallInst* mCallI;
+        llvm::DenseMap<int, MemorySources> mArgumentMemorySources;
+
+        FunctionCallWithMemorySources() :
+            mCallI(nullptr),
+            mArgumentMemorySources(llvm::DenseMap<int, MemorySources>()) {}
+
+        explicit FunctionCallWithMemorySources(llvm::CallInst* CallI) :
+            mCallI(CallI),
+            mArgumentMemorySources(llvm::DenseMap<int, MemorySources>()) {}
+    };
+
+    struct ValueWithMemorySources {
+        llvm::Value* mV;
+        MemorySources mMemorySources;
+
+        explicit ValueWithMemorySources(llvm::Value* V) :
+            mV(V) {}
+
+        ValueWithMemorySources(llvm::Value* V, llvm::Value* MemorySource) :
+            mV(V) {
+            mMemorySources.insert(MemorySource);
+        }
+
+        ValueWithMemorySources(llvm::Value* V, MemorySources Sources) :
+            mV(V) {
+            mMemorySources.insert(Sources.begin(), Sources.end());
+        }
+    };
+
+    struct FunctionResultArgumentsMemoryDependencies {
+        llvm::DenseSet<int> mInfluencingArgumentsIndexes;
+        // Indicates is returned value always points to the unique memory
+        bool mIsRestrict;
+
+        FunctionResultArgumentsMemoryDependencies() :
+            mIsRestrict(false) {}
+    };
+
+    struct FunctionArgumentRestrictCalls {
+        int mArgumentIndex;
+        llvm::DenseSet<llvm::CallInst*> mRestrictCalls;
+        bool mAllFunctionCallsRestrict;
+
+        FunctionArgumentRestrictCalls(int index) :
+            mArgumentIndex(index),
+            mAllFunctionCallsRestrict(false) {}
+    };
+
+    struct FunctionCallsInfo {
+        llvm::Function* mFunction;
+        llvm::DenseMap<int, FunctionArgumentRestrictCalls> mRestrictCallsByArguments;
+    };
+
+    //typedef llvm::DenseSet<int> FunctionResultArgumentsMemoryDependencies;
+
+}
+
+
+
+#endif //SAPFOR_RESTRICTIONARGUMENTSPASS_H

lib/Analysis/Memory/CMakeLists.txt

@@ -4,8 +4,8 @@ set(ANALYSIS_SOURCES Passes.cpp DependenceAnalysis.cpp PrivateAnalysis.cpp
   EstimateMemory.cpp DefinedMemory.cpp LiveMemory.cpp AliasTreePrinter.cpp
   DIAliasTreePrinter.cpp DIMemoryLocation.cpp DFMemoryLocation.cpp
   Delinearization.cpp ServerUtils.cpp ClonedDIMemoryMatcher.cpp
-  GlobalLiveMemory.cpp GlobalDefinedMemory.cpp DIClientServerInfo.cpp
-  DIMemoryAnalysisServer.cpp DIArrayAccess.cpp)
+  GlobalLiveMemory.cpp GlobalDefinedMemory.cpp RestrictionArgumentsPass.cpp 
+  DIClientServerInfo.cpp DIMemoryAnalysisServer.cpp DIArrayAccess.cpp)
 
 if(MSVC_IDE)
   file(GLOB_RECURSE ANALYSIS_HEADERS RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

lib/Analysis/Memory/Passes.cpp

@@ -46,4 +46,5 @@ void llvm::initializeMemoryAnalysis(PassRegistry &Registry) {
   initializeGlobalDefinedMemoryPass(Registry);
   initializeGlobalLiveMemoryPass(Registry);
   initializeDIArrayAccessWrapperPass(Registry);
+  initializeRestrictionArgumentsPassPass(Registry);
 }