Classes in this File | Line Coverage | Branch Coverage | Complexity | ||||
AuthConfig |
|
| 1.5;1.5 |
1 | /* | |
2 | * Copyright 2003-2008 the original author or authors. | |
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 | * You are receiving this code free of charge, which represents many hours of | |
17 | * effort from other individuals and corporations. As a responsible member | |
18 | * of the community, you are asked (but not required) to donate any | |
19 | * enhancements or improvements back to the community under a similar open | |
20 | * source license. Thank you. -TMN | |
21 | */ | |
22 | package groovyx.net.http; | |
23 | ||
24 | import java.io.IOException; | |
25 | import java.io.InputStream; | |
26 | import java.net.MalformedURLException; | |
27 | import java.net.URL; | |
28 | import java.security.GeneralSecurityException; | |
29 | import java.security.KeyStore; | |
30 | ||
31 | import org.apache.http.auth.AuthScope; | |
32 | import org.apache.http.auth.UsernamePasswordCredentials; | |
33 | import org.apache.http.conn.scheme.Scheme; | |
34 | import org.apache.http.conn.ssl.SSLSocketFactory; | |
35 | ||
36 | /** | |
37 | * Encapsulates all configuration related to HTTP authentication methods. | |
38 | * @see HTTPBuilder#getAuth() | |
39 | * | |
40 | * @author <a href='mailto:tnichols@enernoc.com'>Tom Nichols</a> | |
41 | */ | |
42 | public class AuthConfig { | |
43 | protected HTTPBuilder builder; | |
44 | 36 | public AuthConfig( HTTPBuilder builder ) { |
45 | 36 | this.builder = builder; |
46 | 36 | } |
47 | ||
48 | /** | |
49 | * Set authentication credentials to be used for the current | |
50 | * {@link HTTPBuilder#getURL() default host}. This method name is a bit of | |
51 | * a misnomer, since these credentials will actually work for "digest" | |
52 | * authentication as well. | |
53 | * @param user | |
54 | * @param pass | |
55 | */ | |
56 | public void basic( String user, String pass ) { | |
57 | 12 | URL url = (URL)builder.getURL(); |
58 | 12 | if ( url == null ) throw new IllegalStateException( "a default URL must be set" ); |
59 | 12 | this.basic( url.getHost(), url.getPort(), user, pass ); |
60 | 12 | } |
61 | ||
62 | /** | |
63 | * Set authentication credentials to be used for the given host and port. | |
64 | * @param host | |
65 | * @param port | |
66 | * @param user | |
67 | * @param pass | |
68 | */ | |
69 | public void basic( String host, int port, String user, String pass ) { | |
70 | 12 | builder.getClient().getCredentialsProvider().setCredentials( |
71 | new AuthScope( host, port ), | |
72 | new UsernamePasswordCredentials( user, pass ) | |
73 | ); | |
74 | 12 | } |
75 | ||
76 | /** | |
77 | * Sets a certificate to be used for SSL authentication. | |
78 | * @param certURL URL to a JKS keystore where the certificate is stored | |
79 | * @param password password to decrypt the keystore | |
80 | */ | |
81 | public void certificate( String certURL, String password ) | |
82 | throws GeneralSecurityException, MalformedURLException, IOException { | |
83 | ||
84 | 0 | KeyStore keyStore = KeyStore.getInstance( KeyStore.getDefaultType() ); |
85 | 0 | InputStream jksStream = new URL(certURL).openStream(); |
86 | try { | |
87 | 0 | keyStore.load( jksStream, password.toCharArray() ); |
88 | 0 | } finally { jksStream.close(); } |
89 | ||
90 | 0 | SSLSocketFactory ssl = new SSLSocketFactory(keyStore, password); |
91 | 0 | ssl.setHostnameVerifier( SSLSocketFactory.STRICT_HOSTNAME_VERIFIER ); |
92 | ||
93 | 0 | builder.getClient().getConnectionManager().getSchemeRegistry() |
94 | .register( new Scheme("https", ssl, 443) ); | |
95 | 0 | } |
96 | } |