add param types in svffuntype (#1673)

* add param types in svffuntype * add param types in svffuntype
SVF-tools · Mar 3, 2025 · a3b6a23 · a3b6a23
1 parent fa081a8
commit a3b6a23
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/svf-llvm/lib/LLVMModule.cpp b/svf-llvm/lib/LLVMModule.cpp
@@ -1198,8 +1198,13 @@ SVFType* LLVMModuleSet::addSVFTypeInfo(const Type* T)
         svfIntT->setSignAndWidth(intT->getSignBit() ? -signWidth : signWidth);
         svftype = svfIntT;
     }
-    else if (const FunctionType* ft = SVFUtil::dyn_cast<FunctionType>(T))
-        svftype = new SVFFunctionType(getSVFType(ft->getReturnType()));
+    else if (const FunctionType* ft = SVFUtil::dyn_cast<FunctionType>(T)) {
+        std::vector<const SVFType*> paramTypes;
+        for (const auto& t: ft->params()) {
+            paramTypes.push_back(getSVFType(t));
+        }
+        svftype = new SVFFunctionType(getSVFType(ft->getReturnType()), paramTypes);
+    }
     else if (const StructType* st = SVFUtil::dyn_cast<StructType>(T))
     {
         auto svfst = new SVFStructType(byteSize);

diff --git a/svf/include/SVFIR/SVFType.h b/svf/include/SVFIR/SVFType.h
@@ -323,12 +323,14 @@ class SVFFunctionType : public SVFType
 
 private:
     const SVFType* retTy;
+    std::vector<const SVFType*> params;
 
 public:
-    SVFFunctionType(const SVFType* rt)
-        : SVFType(false, SVFFunctionTy, 1), retTy(rt)
+    SVFFunctionType(const SVFType* rt, const std::vector<const SVFType*>& p)
+        : SVFType(false, SVFFunctionTy, 1), retTy(rt), params(p)
     {
     }
+
     static inline bool classof(const SVFType* node)
     {
         return node->getKind() == SVFFunctionTy;
@@ -338,6 +340,10 @@ class SVFFunctionType : public SVFType
         return retTy;
     }
 
+    const std::vector<const SVFType*>& getParamTypes() const {
+        return params;
+    }
+
     void print(std::ostream& os) const override;
 };