javasrc2cpg - workaround for failing to resolve a method's external r…

…eturn type when it contains type arguments (#4044)
joernio · Jan 13, 2024 · f14eea0 · f14eea0
1 parent 21558c7
commit f14eea0
Show file tree

Hide file tree

Showing 2 changed files with 84 additions and 0 deletions.
diff --git a/...g/src/main/scala/io/joern/javasrc2cpg/astcreation/declarations/AstForMethodsCreator.scala b/...g/src/main/scala/io/joern/javasrc2cpg/astcreation/declarations/AstForMethodsCreator.scala
@@ -40,6 +40,7 @@ import io.shiftleft.codepropertygraph.generated.Operators
 import io.shiftleft.codepropertygraph.generated.DispatchTypes
 import io.shiftleft.codepropertygraph.generated.EdgeTypes
 import com.github.javaparser.ast.Node
+import com.github.javaparser.symbolsolver.javaparsermodel.declarations.JavaParserParameterDeclaration
 import io.joern.javasrc2cpg.astcreation.declarations.AstForMethodsCreator.PartialConstructorDeclaration
 
 private[declarations] trait AstForMethodsCreator { this: AstCreator =>
@@ -54,6 +55,9 @@ private[declarations] trait AstForMethodsCreator { this: AstCreator =>
     val returnTypeFullName = expectedReturnType
       .flatMap(typeInfoCalc.fullName)
       .orElse(scope.lookupType(simpleMethodReturnType))
+      .orElse(
+        Try(methodDeclaration.getType.asClassOrInterfaceType).toOption.flatMap(t => scope.lookupType(t.getNameAsString))
+      )
       .orElse(typeParameters.find(_.name == simpleMethodReturnType).map(_.typeFullName))
 
     scope.pushMethodScope(
@@ -260,6 +264,14 @@ private[declarations] trait AstForMethodsCreator { this: AstCreator =>
         .map { param =>
           Try(param.getType).toOption
             .flatMap(paramType => typeInfoCalc.fullName(paramType, typeParamValues))
+            // In a scenario where we have an import of an external type e.g. `import foo.bar.Baz` and
+            // this parameter's type is e.g. `Baz<String>`, the lookup will fail. However, if we lookup
+            // for `Baz` instead (i.e. without type arguments), then the lookup will succeed.
+            .orElse(
+              Try(
+                param.asInstanceOf[JavaParserParameterDeclaration].getWrappedNode.getType.asClassOrInterfaceType
+              ).toOption.flatMap(t => scope.lookupType(t.getNameAsString))
+            )
         }
 
     toOptionList(parameterTypes)

diff --git a/...-cli/frontends/javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/querying/MethodTests.scala b/...-cli/frontends/javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/querying/MethodTests.scala
@@ -103,3 +103,75 @@ class MethodTests3 extends JavaSrcCode2CpgFixture {
     cpg.method("virtualMethod").isVirtual.fullName.head shouldBe "Foo.virtualMethod:void(java.lang.Integer)"
   }
 }
+
+class MethodTests4 extends JavaSrcCode2CpgFixture {
+
+  "List<String> in the method return type" should {
+    val cpg = code("""
+        |import java.util.*;
+        |class Foo {
+        | List<String> run() {
+        |   return null;
+        | }
+        |}
+        |""".stripMargin)
+
+    "have correct signature and full name" in {
+      val List(method) = cpg.method("run").l
+      method.signature shouldBe "java.util.List()"
+      method.fullName shouldBe "Foo.run:java.util.List()"
+    }
+  }
+
+  "Baz<String> in the method return type" should {
+    val cpg = code("""
+        |import foo.bar.Baz;
+        |class Foo {
+        | Baz<String> run() {
+        |   return null;
+        | }
+        |}
+        |""".stripMargin)
+
+    "have correct signature and full name" in {
+      val List(method) = cpg.method("run").l
+      method.signature shouldBe "foo.bar.Baz()"
+      method.fullName shouldBe "Foo.run:foo.bar.Baz()"
+    }
+  }
+
+  "Identity method for Baz<String>" should {
+    val cpg = code("""
+        |import foo.bar.Baz;
+        |class Foo {
+        | Baz<String> run(Baz<String> x) {
+        |   return x;
+        | }
+        |}
+        |""".stripMargin)
+
+    "have correct signature and full name" in {
+      val List(method) = cpg.method("run").l
+      method.signature shouldBe "foo.bar.Baz(foo.bar.Baz)"
+      method.fullName shouldBe "Foo.run:foo.bar.Baz(foo.bar.Baz)"
+    }
+  }
+
+  "Generic identity method for Baz<T>" should {
+    val cpg = code("""
+        |import foo.bar.Baz;
+        |class Foo {
+        | <T> Baz<T> run(Baz<T> x) {
+        |   return x;
+        | }
+        |}
+        |""".stripMargin)
+
+    "have correct signature and full name" in {
+      val List(method) = cpg.method("run").l
+      method.signature shouldBe "foo.bar.Baz(foo.bar.Baz)"
+      method.fullName shouldBe "Foo.run:foo.bar.Baz(foo.bar.Baz)"
+    }
+  }
+
+}