add "isInline" method for "CallMethods" (#5032)

* minor changes: code format * add "isInline" method for "CallMethods" * add a method "isInline" for CallTraversal
joernio · Oct 28, 2024 · 9ee0a39 · 9ee0a39
1 parent b200a36
commit 9ee0a39
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/semanticcpg/src/main/scala/io/shiftleft/semanticcpg/language/nodemethods/CallMethods.scala b/semanticcpg/src/main/scala/io/shiftleft/semanticcpg/language/nodemethods/CallMethods.scala
@@ -8,10 +8,13 @@ import io.shiftleft.semanticcpg.language.*
 class CallMethods(val node: Call) extends AnyVal with NodeExtension with HasLocation {
 
   def isStatic: Boolean =
-    node.dispatchType == "STATIC_DISPATCH"
+    node.dispatchType == DispatchTypes.STATIC_DISPATCH
 
   def isDynamic: Boolean =
-    node.dispatchType == "DYNAMIC_DISPATCH"
+    node.dispatchType == DispatchTypes.DYNAMIC_DISPATCH
+
+  def isInline: Boolean =
+    node.dispatchType == DispatchTypes.INLINED
 
   def receiver: Iterator[Expression] =
     node.receiverOut.collectAll[Expression]

diff --git a/...pg/src/main/scala/io/shiftleft/semanticcpg/language/types/expressions/CallTraversal.scala b/...pg/src/main/scala/io/shiftleft/semanticcpg/language/types/expressions/CallTraversal.scala
@@ -16,6 +16,10 @@ class CallTraversal(val traversal: Iterator[Call]) extends AnyVal {
   def isDynamic: Iterator[Call] =
     traversal.filter(_.isDynamic)
 
+  /** Only dispatched calls inline */
+  def isInline: Iterator[Call] =
+    traversal.filter(_.isInline)
+
   /** Only assignment calls
     */
   def isAssignment: Iterator[Assignment] =