[WFCORE-6503]: Adding configuration for YAML file size.

Signed-off-by: Emmanuel Hugonnet <[email protected]>
wildfly · Mar 21, 2024 · fac75cb · fac75cb
1 parent ed19e2f
commit fac75cb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...er/src/main/java/org/jboss/as/controller/persistence/yaml/YamlConfigurationExtension.java b/...er/src/main/java/org/jboss/as/controller/persistence/yaml/YamlConfigurationExtension.java
@@ -75,6 +75,7 @@
 public class YamlConfigurationExtension implements ConfigurationExtension {
 
     private static final String CONFIGURATION_ROOT_KEY = "wildfly-configuration";
+    private static final String YAML_CODEPOINT_LIMIT = "org.wildfly.configuration.extension.yaml.codepoint.limit";
 
     private static final String YAML_CONFIG = "--yaml";
     private static final String SHORT_YAML_CONFIG = "-y";
@@ -104,7 +105,12 @@ private void load() {
         for (Path file : files) {
             if (file != null && Files.exists(file) && Files.isRegularFile(file)) {
                 Map<String, Object> yamlConfig = Collections.emptyMap();
-                Yaml yaml = new Yaml(new OperationConstructor(new LoaderOptions()));
+                LoaderOptions loadingConfig = new LoaderOptions();
+                //Default to 3MB
+                loadingConfig.setCodePointLimit(
+                        Integer.parseInt(
+                                WildFlySecurityManager.getPropertyPrivileged(YAML_CODEPOINT_LIMIT, "3145728")));
+                Yaml yaml = new Yaml(new OperationConstructor(loadingConfig));
                 try (InputStream inputStream = Files.newInputStream(file)) {
                     yamlConfig = yaml.load(inputStream);
                 } catch (IOException ioex) {