Examples with AttributeQueryType org.keycloak.dom.saml.v2.protocol.AttributeQueryType used on opensource projects

Example 1 with AttributeQueryType

use of org.keycloak.dom.saml.v2.protocol.AttributeQueryType in project keycloak by keycloak.

the class SAMLAttributeQueryParserTest method testSaml20AttributeQueryWithExtension.

@Test(timeout = 2000)
public void testSaml20AttributeQueryWithExtension() throws Exception {
    try (InputStream is = SAMLAttributeQueryParserTest.class.getResourceAsStream("saml20-attributequery-with-extension.xml")) {
        Object parsedObject = parser.parse(is);
        assertThat(parsedObject, instanceOf(AttributeQueryType.class));
        AttributeQueryType query = (AttributeQueryType) parsedObject;
        assertThat(query.getSignature(), nullValue());
        assertThat(query.getConsent(), nullValue());
        assertThat(query.getIssuer(), not(nullValue()));
        assertThat(query.getIssuer().getValue(), is("https://sp/"));
        NameIDType nameId = (NameIDType) query.getSubject().getSubType().getBaseID();
        assertThat(nameId.getValue(), is("CN=trscavo@uiuc.edu,OU=User,O=NCSA-TEST,C=US"));
        assertThat(query.getExtensions(), not(nullValue()));
        assertThat(query.getExtensions().getAny().size(), is(1));
        assertThat(query.getExtensions().getAny().get(0), instanceOf(Element.class));
        Element el = (Element) query.getExtensions().getAny().get(0);
        assertThat(el.getLocalName(), is("KeyInfo"));
        assertThat(el.getNamespaceURI(), is("urn:keycloak:ext:key:1.0"));
        assertThat(el.getAttribute("MessageSigningKeyId"), is("FJ86GcF3jTbNLOco4NvZkUCIUmfYCqoqtOQeMfbhNlE"));
    }
}

Example 2 with AttributeQueryType

use of org.keycloak.dom.saml.v2.protocol.AttributeQueryType in project keycloak by keycloak.

the class SAMLRequestWriter method write.

public void write(AttributeQueryType request) throws ProcessingException {
    StaxUtil.writeStartElement(writer, PROTOCOL_PREFIX, JBossSAMLConstants.ATTRIBUTE_QUERY.get(), PROTOCOL_NSURI.get());
    StaxUtil.writeNameSpace(writer, PROTOCOL_PREFIX, PROTOCOL_NSURI.get());
    StaxUtil.writeNameSpace(writer, ASSERTION_PREFIX, ASSERTION_NSURI.get());
    StaxUtil.writeDefaultNameSpace(writer, ASSERTION_NSURI.get());
    // Attributes
    StaxUtil.writeAttribute(writer, JBossSAMLConstants.ID.get(), request.getID());
    StaxUtil.writeAttribute(writer, JBossSAMLConstants.VERSION.get(), request.getVersion());
    StaxUtil.writeAttribute(writer, JBossSAMLConstants.ISSUE_INSTANT.get(), request.getIssueInstant().toString());
    URI destination = request.getDestination();
    if (destination != null)
        StaxUtil.writeAttribute(writer, JBossSAMLConstants.DESTINATION.get(), destination.toASCIIString());
    String consent = request.getConsent();
    if (StringUtil.isNotNull(consent))
        StaxUtil.writeAttribute(writer, JBossSAMLConstants.CONSENT.get(), consent);
    NameIDType issuer = request.getIssuer();
    if (issuer != null) {
        write(issuer, new QName(ASSERTION_NSURI.get(), JBossSAMLConstants.ISSUER.get(), ASSERTION_PREFIX));
    }
    Element sig = request.getSignature();
    if (sig != null) {
        StaxUtil.writeDOMElement(writer, sig);
    }
    ExtensionsType extensions = request.getExtensions();
    if (extensions != null && !extensions.getAny().isEmpty()) {
        write(extensions);
    }
    SubjectType subject = request.getSubject();
    if (subject != null) {
        write(subject);
    }
    List<AttributeType> attributes = request.getAttribute();
    for (AttributeType attr : attributes) {
        write(attr);
    }
    StaxUtil.writeEndElement(writer);
    StaxUtil.flush(writer);
}

Example 3 with AttributeQueryType

use of org.keycloak.dom.saml.v2.protocol.AttributeQueryType in project keycloak by keycloak.

the class SamlDocumentStepBuilder method saml2Object2String.

public static String saml2Object2String(final SAML2Object transformed) {
    try {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        XMLStreamWriter xmlStreamWriter = StaxUtil.getXMLStreamWriter(bos);
        if (transformed instanceof AuthnRequestType) {
            new SAMLRequestWriter(xmlStreamWriter).write((AuthnRequestType) transformed);
        } else if (transformed instanceof LogoutRequestType) {
            new SAMLRequestWriter(xmlStreamWriter).write((LogoutRequestType) transformed);
        } else if (transformed instanceof ArtifactResolveType) {
            new SAMLRequestWriter(xmlStreamWriter).write((ArtifactResolveType) transformed);
        } else if (transformed instanceof AttributeQueryType) {
            new SAMLRequestWriter(xmlStreamWriter).write((AttributeQueryType) transformed);
        } else if (transformed instanceof ResponseType) {
            new SAMLResponseWriter(xmlStreamWriter).write((ResponseType) transformed);
        } else if (transformed instanceof ArtifactResponseType) {
            new SAMLResponseWriter(xmlStreamWriter).write((ArtifactResponseType) transformed);
        } else if (transformed instanceof StatusResponseType) {
            new SAMLResponseWriter(xmlStreamWriter).write((StatusResponseType) transformed, SAMLProtocolQNames.LOGOUT_RESPONSE.getQName("samlp"));
        } else {
            Assert.assertNotNull("Unknown type: <null>", transformed);
            Assert.fail("Unknown type: " + transformed.getClass().getName());
        }
        return new String(bos.toByteArray(), GeneralConstants.SAML_CHARSET);
    } catch (ProcessingException ex) {
        throw new RuntimeException(ex);
    }
}

Example 4 with AttributeQueryType

use of org.keycloak.dom.saml.v2.protocol.AttributeQueryType in project keycloak by keycloak.

the class SAMLAttributeQueryParserTest method testSaml20AttributeQuery.

@Test(timeout = 2000000)
public void testSaml20AttributeQuery() throws Exception {
    try (InputStream is = SAMLAttributeQueryParserTest.class.getResourceAsStream("saml20-attributequery.xml")) {
        Object parsedObject = parser.parse(is);
        assertThat(parsedObject, instanceOf(AttributeQueryType.class));
        AttributeQueryType query = (AttributeQueryType) parsedObject;
        assertThat(query.getSignature(), nullValue());
        assertThat(query.getConsent(), nullValue());
        assertThat(query.getIssuer(), not(nullValue()));
        assertThat(query.getIssuer().getValue(), is("https://sp/"));
        NameIDType nameId = (NameIDType) query.getSubject().getSubType().getBaseID();
        assertThat(nameId.getValue(), is("CN=trscavo@uiuc.edu,OU=User,O=NCSA-TEST,C=US"));
    }
}

Example 5 with AttributeQueryType

use of org.keycloak.dom.saml.v2.protocol.AttributeQueryType in project keycloak by keycloak.

the class SAMLAttributeQueryParser method instantiateElement.

@Override
protected AttributeQueryType instantiateElement(XMLEventReader xmlEventReader, StartElement element) throws ParsingException {
    SAMLParserUtil.validateAttributeValue(element, SAMLProtocolQNames.ATTR_VERSION, VERSION_2_0);
    String id = StaxParserUtil.getRequiredAttributeValue(element, SAMLProtocolQNames.ATTR_ID);
    XMLGregorianCalendar issueInstant = XMLTimeUtil.parse(StaxParserUtil.getRequiredAttributeValue(element, SAMLProtocolQNames.ATTR_ISSUE_INSTANT));
    AttributeQueryType authnRequest = new AttributeQueryType(id, issueInstant);
    super.parseBaseAttributes(element, authnRequest);
    return authnRequest;
}