Bunch o' bug fixes, all IdP configuration files now load.
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / idp / profile / saml2 / HTTPSOAPAttributeQuery.java
1 /*
2  * Copyright [2007] [University Corporation for Advanced Internet Development, Inc.]
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package edu.internet2.middleware.shibboleth.idp.profile.saml2;
18
19 import javax.servlet.ServletRequest;
20 import javax.servlet.ServletResponse;
21
22 import org.opensaml.common.binding.decoding.MessageDecoder;
23 import org.opensaml.common.binding.encoding.MessageEncoder;
24
25 import edu.internet2.middleware.shibboleth.common.profile.ProfileException;
26
27 /**
28  * SAML 2.0 SOAP Attribute Query profile handler.
29  */
30 public class HTTPSOAPAttributeQuery extends AbstractAttributeQuery {
31
32     /** SAML binding URI. */
33     public static final String BINDING = "urn:oasis:names:tc:SAML:2.0:bindings:SOAP";
34
35     /** Constructor. */
36     public HTTPSOAPAttributeQuery() {
37         super();
38     }
39
40     /** {@inheritDoc} */
41     @SuppressWarnings("unchecked")
42     protected void getMessageDecoder(AttributeQueryRequestContext requestContext) throws ProfileException {
43         MessageDecoder<ServletRequest> decoder = getMessageDecoderFactory().getMessageDecoder(BINDING);
44         if (decoder == null) {
45             throw new ProfileException("No request decoder was registered for binding type: " + BINDING);
46         }
47
48         super.populateMessageDecoder(decoder);
49         decoder.setRequest(requestContext.getProfileRequest().getRawRequest());
50         requestContext.setMessageDecoder(decoder);
51     }
52
53     /** {@inheritDoc} */
54     @SuppressWarnings("unchecked")
55     protected void getMessageEncoder(AttributeQueryRequestContext requestContext) throws ProfileException {
56
57         MessageEncoder<ServletResponse> encoder = getMessageEncoderFactory().getMessageEncoder(BINDING);
58         if (encoder == null) {
59             throw new ProfileException("No response encoder was registered for binding type: " + BINDING);
60         }
61
62         super.populateMessageEncoder(encoder);
63         encoder.setResponse(requestContext.getProfileResponse().getRawResponse());
64         encoder.setSamlMessage(requestContext.getAttributeQueryResponse());
65         requestContext.setMessageEncoder(encoder);
66     }
67 }