Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ISPN-14196 Correctly parse serialization allow list in YAML and JSON #10370

Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,15 @@ public <T> AllowListConfigurationBuilder addClass(String clazz) {
return this;
}

/**
* Helper method that allows for registration of classes to the {@link ClassAllowList}.
*/
public <T> AllowListConfigurationBuilder addClasses(String... classes) {
List<String> classNames = Arrays.asList(classes);
this.classes.addAll(classNames);
return this;
}

/**
* Helper method that allows for registration of classes to the {@link ClassAllowList}.
*/
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,8 @@ public enum Attribute {
INVALIDATION_THRESHOLD,
CONTEXT_INITIALIZERS,
GROUPER,
ACCURATE_SIZE;
ACCURATE_SIZE,
REGEX;

private final String name;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,22 @@ private void parseSerializationContextInitializer(final ConfigurationReader read
}

private void parseAllowList(final ConfigurationReader reader, final AllowListConfigurationBuilder builder, Element outerElement) {
while (reader.inTag(outerElement)) {
for(int i = 0; i < reader.getAttributeCount(); i++) { // JSON/YAML
String[] values = reader.getListAttributeValue(i);
Attribute attribute = Attribute.forName(reader.getAttributeName(i));
switch (attribute) {
case CLASS:
builder.addClasses(values);
break;
case REGEX:
builder.addRegexps(values);
break;
default: {
throw ParseUtils.unexpectedElement(reader);
}
}
}
while (reader.inTag(outerElement)) { // XML
Element element = Element.forName(reader.getLocalName());
switch (element) {
case CLASS: {
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
package org.infinispan.configuration;

import static org.testng.AssertJUnit.assertEquals;

import java.io.IOException;
import java.util.List;
import java.util.Set;

import org.infinispan.configuration.global.GlobalConfiguration;
import org.infinispan.configuration.parsing.ConfigurationBuilderHolder;
import org.infinispan.configuration.parsing.ParserRegistry;
import org.infinispan.test.AbstractInfinispanTest;
import org.testng.annotations.Test;

@Test(groups = "unit", testName = "configuration.JsonParsingTest")
public class JsonParsingTest extends AbstractInfinispanTest {
public void testSerializationAllowList() throws IOException {
ParserRegistry parserRegistry = new ParserRegistry(Thread.currentThread().getContextClassLoader(), true, System.getProperties());
ConfigurationBuilderHolder holder = parserRegistry.parseFile("configs/serialization-test.json");
GlobalConfiguration globalConfiguration = holder.getGlobalConfigurationBuilder().build();
Set<String> classes = globalConfiguration.serialization().allowList().getClasses();
assertEquals(3, classes.size());
List<String> regexps = globalConfiguration.serialization().allowList().getRegexps();
assertEquals(2, regexps.size());
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
package org.infinispan.configuration;

import static org.testng.AssertJUnit.assertEquals;

import java.io.IOException;
import java.util.List;
import java.util.Set;

import org.infinispan.configuration.global.GlobalConfiguration;
import org.infinispan.configuration.parsing.ConfigurationBuilderHolder;
import org.infinispan.configuration.parsing.ParserRegistry;
import org.infinispan.test.AbstractInfinispanTest;
import org.testng.annotations.Test;

@Test(groups = "unit", testName = "configuration.YamlParsingTest")
public class YamlParsingTest extends AbstractInfinispanTest {
public void testSerializationAllowList() throws IOException {
ParserRegistry parserRegistry = new ParserRegistry(Thread.currentThread().getContextClassLoader(), true, System.getProperties());
ConfigurationBuilderHolder holder = parserRegistry.parseFile("configs/serialization-test.yaml");
GlobalConfiguration globalConfiguration = holder.getGlobalConfigurationBuilder().build();
Set<String> classes = globalConfiguration.serialization().allowList().getClasses();
assertEquals(3, classes.size());
List<String> regexps = globalConfiguration.serialization().allowList().getRegexps();
assertEquals(2, regexps.size());
}
}
19 changes: 19 additions & 0 deletions core/src/test/resources/configs/serialization-test.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
{
"infinispan": {
"cache-container": {
"serialization": {
"allow-list": {
"class": [
"org.infinispan.test.data.Rock",
"org.infinispan.test.data.Paper",
"org.infinispan.test.data.Scissors"
],
"regex": [
"org.infinispan.test.model.*",
"org.infinispan.test.types.*"
]
}
}
}
}
}
11 changes: 11 additions & 0 deletions core/src/test/resources/configs/serialization-test.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
infinispan:
cacheContainer:
serialization:
allowList:
class:
- "org.infinispan.test.data.Rock"
- "org.infinispan.test.data.Paper"
- "org.infinispan.test.data.Scissors"
regex:
- "org.infinispan.test.model.*"
- "org.infinispan.test.types.*"