Accept default handling for MemberAccess and IndexAccess on assoc key (…

…joernio#5100)
Privado-Inc · Nov 12, 2024 · df29596 · df29596
1 parent b75354e
commit df29596
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 1 deletion.
diff --git a/...ubysrc2cpg/src/main/scala/io/joern/rubysrc2cpg/astcreation/AstForExpressionsCreator.scala b/...ubysrc2cpg/src/main/scala/io/joern/rubysrc2cpg/astcreation/AstForExpressionsCreator.scala
@@ -1004,7 +1004,7 @@ trait AstForExpressionsCreator(implicit withSchemaValidation: ValidationMode) {
       case keyIdentifier: SimpleIdentifier => setArgumentName(value, keyIdentifier.text)
       case symbol @ StaticLiteral(typ) if typ == getBuiltInType(Defines.Symbol) =>
         setArgumentName(value, symbol.text.stripPrefix(":"))
-      case _: (LiteralExpr | RubyCall | ProcOrLambdaExpr) => astForExpression(assoc)
+      case _: (LiteralExpr | RubyCall | ProcOrLambdaExpr | MemberAccess | IndexAccess) => astForExpression(assoc)
       case x =>
         logger.warn(s"Not explicitly handled argument association key of type ${x.getClass.getSimpleName}")
         astForExpression(assoc)

diff --git a/joern-cli/frontends/rubysrc2cpg/src/test/scala/io/joern/rubysrc2cpg/querying/CallTests.scala b/joern-cli/frontends/rubysrc2cpg/src/test/scala/io/joern/rubysrc2cpg/querying/CallTests.scala
@@ -523,4 +523,48 @@ class CallTests extends RubyCode2CpgFixture(withPostProcessing = true) {
       case xs => fail(s"Expected 6 parameters for call, got [${xs.code.mkString(", ")}]")
     }
   }
+
+  "Call with association IndexAccess key" in {
+    val cpg = code("""
+        |foo(bar[:baz] => nil)
+        |""".stripMargin)
+
+    inside(cpg.call.name("foo").argument.l) {
+      case _ :: (assocParam: Call) :: Nil =>
+        assocParam.methodFullName shouldBe RubyOperators.association
+        assocParam.code shouldBe "bar[:baz] => nil"
+
+        inside(assocParam.argument.l) {
+          case (lhs: Call) :: (rhs: Literal) :: Nil =>
+            lhs.methodFullName shouldBe Operators.indexAccess
+            lhs.code shouldBe "bar[:baz]"
+
+            rhs.code shouldBe "nil"
+          case xs => fail(s"Expected lhs and rhs for association, got [${xs.code.mkString(",")}]")
+        }
+      case xs => fail(s"Expected two params, got [${xs.code.mkString(",")}]")
+    }
+  }
+
+  "Call with association MemberAccess key" in {
+    val cpg = code("""
+        |foo(bar.baz => nil)
+        |""".stripMargin)
+
+    inside(cpg.call.name("foo").argument.l) {
+      case _ :: (assocParam: Call) :: Nil =>
+        assocParam.methodFullName shouldBe RubyOperators.association
+        assocParam.code shouldBe "bar.baz => nil"
+
+        inside(assocParam.argument.l) {
+          case (lhs: Call) :: (rhs: Literal) :: Nil =>
+            lhs.methodFullName shouldBe Operators.fieldAccess
+            lhs.code shouldBe "bar.baz"
+
+            rhs.code shouldBe "nil"
+          case xs => fail(s"Expected lhs and rhs for association, got [${xs.code.mkString(",")}]")
+        }
+      case xs => fail(s"Expected two params, got [${xs.code.mkString(",")}]")
+    }
+  }
 }