javasrc2cpg - add YAML files as configuration files (#4007)

joernio · Dec 21, 2023 · 38ba540 · 38ba540
1 parent bd76501
commit 38ba540
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/...tends/javasrc2cpg/src/main/scala/io/joern/javasrc2cpg/passes/ConfigFileCreationPass.scala b/...tends/javasrc2cpg/src/main/scala/io/joern/javasrc2cpg/passes/ConfigFileCreationPass.scala
@@ -33,7 +33,10 @@ class ConfigFileCreationPass(cpg: Cpg) extends XConfigFileCreationPass(cpg) {
     pathEndFilter("build.gradle"),
     pathEndFilter("build.gradle.kts"),
     // ANDROID
-    pathEndFilter("AndroidManifest.xml")
+    pathEndFilter("AndroidManifest.xml"),
+    // SPRING
+    extensionFilter(".yaml"),
+    extensionFilter(".yml")
   )
 
   private def mybatisFilter(file: File): Boolean = {

diff --git a/joern-cli/frontends/javasrc2cpg/src/test/resources/config_tests/basic.yaml b/joern-cli/frontends/javasrc2cpg/src/test/resources/config_tests/basic.yaml
@@ -0,0 +1 @@
+key: value
diff --git a/joern-cli/frontends/javasrc2cpg/src/test/resources/config_tests/basic.yml b/joern-cli/frontends/javasrc2cpg/src/test/resources/config_tests/basic.yml
@@ -0,0 +1 @@
+key: value
diff --git a/.../javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/passes/ConfigFileCreationPassTests.scala b/.../javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/passes/ConfigFileCreationPassTests.scala
@@ -36,7 +36,9 @@ class ConfigFileCreationPassTests extends JavaSrcCode2CpgFixture {
       Paths.get(absoluteConfigDir, "struts.xml").toString,
       Paths.get(absoluteConfigDir, "web.xml").toString,
       Paths.get(absoluteConfigDir, "build.gradle").toString,
-      Paths.get(absoluteConfigDir, "build.gradle.kts").toString
+      Paths.get(absoluteConfigDir, "build.gradle.kts").toString,
+      Paths.get(absoluteConfigDir, "basic.yaml").toString,
+      Paths.get(absoluteConfigDir, "basic.yml").toString
     )
   }