1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27 package org.apache.http.impl.nio.client;
28
29 import org.apache.http.HttpConnection;
30 import org.apache.http.annotation.Contract;
31 import org.apache.http.annotation.ThreadingBehavior;
32 import org.apache.http.auth.AuthScheme;
33 import org.apache.http.auth.AuthState;
34 import org.apache.http.auth.Credentials;
35 import org.apache.http.client.UserTokenHandler;
36 import org.apache.http.client.protocol.HttpClientContext;
37 import org.apache.http.nio.conn.ManagedNHttpClientConnection;
38 import org.apache.http.protocol.HttpContext;
39
40 import javax.net.ssl.SSLSession;
41 import java.security.Principal;
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59 @Contract(threading = ThreadingBehavior.IMMUTABLE)
60 public class DefaultAsyncUserTokenHandler implements UserTokenHandler {
61
62 public static final DefaultAsyncUserTokenHandlererTokenHandler.html#DefaultAsyncUserTokenHandler">DefaultAsyncUserTokenHandler INSTANCE = new DefaultAsyncUserTokenHandler();
63
64 @Override
65 public Object getUserToken(final HttpContext context) {
66
67 final HttpClientContext clientContext = HttpClientContext.adapt(context);
68
69 Principal userPrincipal = null;
70
71 final AuthState targetAuthState = clientContext.getTargetAuthState();
72 if (targetAuthState != null) {
73 userPrincipal = getAuthPrincipal(targetAuthState);
74 if (userPrincipal == null) {
75 final AuthState proxyAuthState = clientContext.getProxyAuthState();
76 userPrincipal = getAuthPrincipal(proxyAuthState);
77 }
78 }
79
80 if (userPrincipal == null) {
81 final HttpConnection conn = clientContext.getConnection();
82 if (conn.isOpen() && conn instanceof ManagedNHttpClientConnection) {
83 final SSLSession sslsession = ((ManagedNHttpClientConnection) conn).getSSLSession();
84 if (sslsession != null) {
85 userPrincipal = sslsession.getLocalPrincipal();
86 }
87 }
88 }
89
90 return userPrincipal;
91 }
92
93 private static Principal getAuthPrincipal(final AuthState authState) {
94 final AuthScheme scheme = authState.getAuthScheme();
95 if (scheme != null && scheme.isComplete() && scheme.isConnectionBased()) {
96 final Credentials creds = authState.getCredentials();
97 if (creds != null) {
98 return creds.getUserPrincipal();
99 }
100 }
101 return null;
102 }
103
104 }