Find if gep operand is a array type

include/InstModel/LFCPAInstModel/LFCPAInstModel.h

@@ -6,6 +6,7 @@
 #include "Token/Token.h"
 
 #include "vector"
+#include "iostream"
 #include "llvm/IR/Argument.h"
 #include "llvm/IR/Function.h"
 #include "llvm/IR/GlobalVariable.h"
@@ -42,6 +43,8 @@ class LFCPAInstModel : public InstModel {
 
   bool isInstSkip(llvm::Instruction *);
   template <typename GOP> bool isStructFieldPointerTy(GOP *);
+  template <typename GOP> bool isGEPOperandArrayTy(GOP *, int);
+  template <typename GOP> std::vector<int> getGEPArrayIndex(GOP *);
   Token *extractDummy(std::string);
 };
 } // namespace spatial

lib/InstModel/LFCPAInstModel/LFCPAInstModel.cpp

@@ -407,19 +407,69 @@ LFCPAInstModel::handleGEPUtil<llvm::GEPOperator>(llvm::GEPOperator *G,
                                                  Token *Ptr);
 
 template <typename GOP> bool LFCPAInstModel::isStructFieldPointerTy(GOP *G) {
-  long int cntOp = G->getNumOperands();
-  llvm::Type *BaseTy = G->getOperand(0)->getType()->getContainedType(0);
-  llvm::ConstantInt *CI =
-      llvm::dyn_cast<llvm::ConstantInt>(G->getOperand(cntOp - 1));
-  auto conVal = CI->getSExtValue();
-  return (BaseTy->getStructElementType(conVal)->isPointerTy());
+  llvm::Type *StructType = G->getOperand(0)->getType();
+  if(StructType->getNumContainedTypes() > 0)
+      StructType = StructType->getContainedType(0);
+  for (int i = 2; i < G->getNumOperands(); i++) {
+    while(StructType->isArrayTy()){
+        StructType = StructType->getArrayElementType();
+        i = i + 1;
+    }
+    if(i >= G->getNumOperands()) break;
+    llvm::Value *IdxV = G->getOperand(i);
+    llvm::ConstantInt *Idx = llvm::cast<llvm::ConstantInt>(IdxV); 
+    StructType = StructType->getStructElementType(Idx->getSExtValue());
+  }
+  return StructType->isPointerTy();
 }
 
 template bool LFCPAInstModel::isStructFieldPointerTy<llvm::GetElementPtrInst>(
     llvm::GetElementPtrInst *G);
 template bool
 LFCPAInstModel::isStructFieldPointerTy<llvm::GEPOperator>(llvm::GEPOperator *G);
 
+template <typename GOP> bool LFCPAInstModel::isGEPOperandArrayTy(GOP *G, int Idx) {
+  llvm::Type *StructType = G->getOperand(0)->getType();
+  if(StructType->getNumContainedTypes() > 0)
+      StructType = StructType->getContainedType(0);
+  int OpNum = G->getNumOperands();
+  if(Idx > 0)
+      OpNum = Idx;
+  for (int i = 2; i <= OpNum; i++) {
+    while(StructType->isArrayTy()){
+        StructType = StructType->getArrayElementType();
+        i = i + 1;
+    }
+    if(i >= OpNum) break;
+    llvm::Value *IdxV = G->getOperand(i);
+    llvm::ConstantInt *Idx = llvm::cast<llvm::ConstantInt>(IdxV); 
+    StructType = StructType->getStructElementType(Idx->getSExtValue());
+  }
+  return StructType->isArrayTy();
+}
+
+template bool LFCPAInstModel::isGEPOperandArrayTy<llvm::GetElementPtrInst>(
+    llvm::GetElementPtrInst *G, int Idx = -1);
+template bool
+LFCPAInstModel::isGEPOperandArrayTy<llvm::GEPOperator>(llvm::GEPOperator *G, int Idx = -1);
+
+template <typename GOP> std::vector<int> LFCPAInstModel::getGEPArrayIndex(GOP *G) {
+  std::vector<int> Idx;
+  for (int i = 2; i < G->getNumOperands(); i++) {
+      if(isGEPOperandArrayTy(G, i - 1)){
+          Idx.push_back(i);
+      }
+  }
+  std::cout << std::endl;
+  return Idx;
+}
+
+template std::vector<int> LFCPAInstModel::getGEPArrayIndex<llvm::GetElementPtrInst>(
+    llvm::GetElementPtrInst *G);
+template std::vector<int>
+LFCPAInstModel::getGEPArrayIndex<llvm::GEPOperator>(llvm::GEPOperator *G);
+
+
 Token *LFCPAInstModel::extractDummy(std::string S) {
   return (this->getTokenWrapper()->getToken(S, nullptr));
 }