libre.sh/keycloak-scim
3
0
Fork 0
Code Issues 15 Pull requests Releases Packages Activity Actions

Support 0/1 values for XML boolean attributes

Browse source 
Closes #10802
This commit is contained in:
Francis PEROT 2022-03-25 12:03:16 +01:00 committed by Hynek Mlnařík
parent 9e57f836f2
commit 7555063ed9
2 changed files with 51 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
saml-core/src/main/java/org/keycloak/saml/common/util/StaxParserUtil.java
View file

@ -370,7 +370,7 @@ public class StaxParserUtil {
public static Boolean getBooleanAttributeValue(StartElement startElement, HasQName attrName) { public static Boolean getBooleanAttributeValue(StartElement startElement, HasQName attrName) {
Attribute attr = startElement.getAttributeByName(attrName.getQName()); Attribute attr = startElement.getAttributeByName(attrName.getQName());
String value = getAttributeValue(attr); String value = getAttributeValue(attr);
return value == null ? null : Boolean.valueOf(value); return toBoolean(value);
} }
/** /**
@ -384,7 +384,11 @@ public class StaxParserUtil {
public static Boolean getBooleanAttributeValueRP(StartElement startElement, HasQName attrName) { public static Boolean getBooleanAttributeValueRP(StartElement startElement, HasQName attrName) {
Attribute attr = startElement.getAttributeByName(attrName.getQName()); Attribute attr = startElement.getAttributeByName(attrName.getQName());
String value = getAttributeValueRP(attr); String value = getAttributeValueRP(attr);
return value == null ? null : Boolean.valueOf(value); return toBoolean(value);
}
private static Boolean toBoolean(String value) {
return value==null ? null : Boolean.valueOf(value) || "1".equals(value);
} }
/** /**

45
saml-core/src/test/java/org/keycloak/saml/common/util/StaxParserUtilTest.java
View file

@ -17,8 +17,13 @@
package org.keycloak.saml.common.util; package org.keycloak.saml.common.util;
import org.keycloak.saml.common.exceptions.ParsingException; import org.keycloak.saml.common.exceptions.ParsingException;
import org.keycloak.saml.processing.core.parsers.saml.protocol.SAMLProtocolQNames;
import org.keycloak.saml.processing.core.parsers.util.HasQName;
import java.nio.charset.Charset; import java.nio.charset.Charset;
import java.util.NoSuchElementException; import java.util.NoSuchElementException;
import java.util.function.BiFunction;
import javax.xml.stream.XMLEventReader; import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLStreamException; import javax.xml.stream.XMLStreamException;
import javax.xml.stream.events.Characters; import javax.xml.stream.events.Characters;
@ -35,7 +40,7 @@ import org.junit.Test;
import org.junit.rules.ExpectedException; import org.junit.rules.ExpectedException;
import org.w3c.dom.Element; import org.w3c.dom.Element;
import org.w3c.dom.Text; import org.w3c.dom.Text;
import static org.junit.Assert.assertThat; import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.CoreMatchers.*; import static org.hamcrest.CoreMatchers.*;
/** /**
@ -244,4 +249,42 @@ public class StaxParserUtilTest {
} }
} }
@Test
public void testGetBooleanAttributeValue() throws XMLStreamException, ParsingException {
testGetBooleanAttributeValue(new BiFunction<StartElement, HasQName, Boolean>() {
@Override
public Boolean apply(StartElement t, HasQName u) {
return StaxParserUtil.getBooleanAttributeValue(t, u);
}
});
}
@Test
public void testGetBooleanAttributeValueRP() throws XMLStreamException, ParsingException {
testGetBooleanAttributeValue(new BiFunction<StartElement, HasQName, Boolean>() {
@Override
public Boolean apply(StartElement t, HasQName u) {
return StaxParserUtil.getBooleanAttributeValueRP(t, u);
}
});
}
private void testGetBooleanAttributeValue(BiFunction<StartElement, HasQName, Boolean> predicate) throws XMLStreamException, ParsingException {
testGetBooleanAttributeValue("<a AllowCreate=\"false\">text</a>", predicate, false);
testGetBooleanAttributeValue("<a AllowCreate=\"true\">text</a>", predicate, true);
testGetBooleanAttributeValue("<a AllowCreate=\"0\">text</a>", predicate, false);
testGetBooleanAttributeValue("<a AllowCreate=\"1\">text</a>", predicate, true);
testGetBooleanAttributeValue("<a AllowCreate=\"invalid\">text</a>", predicate, false);
testGetBooleanAttributeValue("<a>text</a>", predicate, null);
}
private void testGetBooleanAttributeValue(String xml, BiFunction<StartElement, HasQName, Boolean> predicate, Boolean expectedResult) throws XMLStreamException, ParsingException {
XMLEventReader reader = StaxParserUtil.getXMLEventReader(IOUtils.toInputStream(xml, Charset.defaultCharset()));
assertThat(reader.nextEvent(), instanceOf(StartDocument.class));
StartElement domElement = StaxParserUtil.getNextStartElement(reader);
Boolean bool = predicate.apply(domElement, SAMLProtocolQNames.ATTR_ALLOW_CREATE);
assertThat(bool, is(expectedResult));
}
} }