wslite.http.HTTPConnectionFactory.groovy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of groovy-wslite Show documentation
Lightweight SOAP and REST webservice clients for Groovy
The newest version!
/* Copyright 2011 the original author or authors.
 *
 * Licensed 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 wslite.http

import javax.net.ssl.*
import java.security.KeyStore
import java.security.SecureRandom

class HTTPConnectionFactory {

    def getConnection(URL url, Proxy proxy=Proxy.NO_PROXY) {
        return url.openConnection(proxy)
    }

    def getConnectionTrustAllSSLCerts(URL url, Proxy proxy=Proxy.NO_PROXY) {
        def trustingTrustManager = [
                getAcceptedIssuers: {},
                checkClientTrusted: { arg0, arg1 -> },
                checkServerTrusted: {arg0, arg1 -> }
        ] as X509TrustManager
        SSLContext sc = SSLContext.getInstance('SSL')
        sc.init(null, [trustingTrustManager] as TrustManager[], null)
        def conn = getConnection(url, proxy)
        conn.setSSLSocketFactory(sc.getSocketFactory())
        conn.setHostnameVerifier({arg0, arg1 -> return true} as HostnameVerifier)
        return conn
    }

    def getConnectionUsingTrustStore(URL url, String trustStoreFile, String trustStorePassword,
                                     Proxy proxy=Proxy.NO_PROXY) {
        InputStream tsFile = new FileInputStream(new File(trustStoreFile))
        char[] tsPassword = trustStorePassword?.getChars()

        def keyStore = KeyStore.getInstance(KeyStore.getDefaultType())
        keyStore.load(tsFile, tsPassword)

        def kmf = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm())
        kmf.init(keyStore, tsPassword)

        def tmf = TrustManagerFactory.getInstance(TrustManagerFactory.getDefaultAlgorithm())
        tmf.init(keyStore)

        def sc = SSLContext.getInstance('SSL')
        sc.init(kmf.getKeyManagers(), tmf.getTrustManagers(), new SecureRandom())

        def conn = getConnection(url, proxy)
        conn.setSSLSocketFactory(sc.getSocketFactory())

        return conn
    }

}