SAMLSSOServiceClient.java

/*
*  Copyright (c) 2005-2010, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
*
*  WSO2 Inc. licenses this file to you under the Apache License,
*  Version 2.0 (the "License"); you may not use this file except
*  in compliance with the License.
*  You may obtain a copy of the License at
*
*    http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied.  See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.wso2.carbon.identity.authenticator.saml2.sso.ui.client;

import org.apache.axis2.AxisFault;
import org.apache.axis2.client.Options;
import org.apache.axis2.client.ServiceClient;
import org.apache.axis2.context.ConfigurationContext;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.wso2.carbon.identity.authenticator.saml2.sso.common.SAML2SSOUIAuthenticatorException;
import org.wso2.carbon.identity.sso.saml.stub.IdentitySAMLSSOServiceStub;
import org.wso2.carbon.identity.sso.saml.stub.types.SAMLSSOAuthnReqDTO;
import org.wso2.carbon.identity.sso.saml.stub.types.SAMLSSOReqValidationResponseDTO;
import org.wso2.carbon.identity.sso.saml.stub.types.SAMLSSORespDTO;

public class SAMLSSOServiceClient {

    private static Log log = LogFactory.getLog(SAMLSSOServiceClient.class);
    private IdentitySAMLSSOServiceStub stub;

    public SAMLSSOServiceClient(String backendServerURL, ConfigurationContext configCtx) throws AxisFault {
        String serviceURL = backendServerURL + "IdentitySAMLSSOService";
        stub = new IdentitySAMLSSOServiceStub(configCtx, serviceURL);
        ServiceClient client = stub._getServiceClient();
        Options option = client.getOptions();
        option.setManageSession(true);
    }

    /**
     * @param sessionId
     * @param rpSessionId
     * @param authnMode
     * @return
     * @throws SAML2SSOUIAuthenticatorException
     */
    public SAMLSSOReqValidationResponseDTO validate(String samlReq, String queryString,
                                                    String sessionId, String rpSessionId,
                                                    String authnMode, boolean isPost)
            throws SAML2SSOUIAuthenticatorException {
        try {
            return stub.validateSPInitSSORequest(samlReq, queryString, sessionId, rpSessionId, authnMode, isPost);
        } catch (Exception e) {
            log.error("Error validating the Authentication Request", e);
            throw new SAML2SSOUIAuthenticatorException("Error in parsing authentication request", e);
        }
    }

    public SAMLSSORespDTO authenticate(SAMLSSOAuthnReqDTO authnReqDTO, String sessionId) throws SAML2SSOUIAuthenticatorException {
        try {
            // TODO FIX THIS - ADDED ONLY TO BUILD THE COMPONENT
            return stub.authenticate(authnReqDTO, sessionId, false, null, null);
        } catch (Exception e) {
            log.error("Error authenticating the user.", e);
            throw new SAML2SSOUIAuthenticatorException("Authentication Failure", e);
        }
    }

    public SAMLSSOReqValidationResponseDTO doSingleLogout(String sessionId) throws SAML2SSOUIAuthenticatorException {
        try {
            return stub.doSingleLogout(sessionId);
        } catch (Exception ex) {
            log.error("Error performing single logout.", ex);
            throw new SAML2SSOUIAuthenticatorException("Error performing Single Logout", ex);
        }
    }
}