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.hc.client5.http.impl;
28
29 import java.security.Principal;
30
31 import javax.net.ssl.SSLSession;
32
33 import org.apache.hc.client5.http.HttpRoute;
34 import org.apache.hc.client5.http.UserTokenHandler;
35 import org.apache.hc.client5.http.auth.AuthExchange;
36 import org.apache.hc.client5.http.auth.AuthScheme;
37 import org.apache.hc.client5.http.protocol.HttpClientContext;
38 import org.apache.hc.core5.annotation.Contract;
39 import org.apache.hc.core5.annotation.ThreadingBehavior;
40 import org.apache.hc.core5.http.HttpHost;
41 import org.apache.hc.core5.http.HttpRequest;
42 import org.apache.hc.core5.http.protocol.HttpContext;
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59 @Contract(threading = ThreadingBehavior.STATELESS)
60 public class DefaultUserTokenHandler implements UserTokenHandler {
61
62
63
64
65 public static final DefaultUserTokenHandler INSTANCE = new DefaultUserTokenHandler();
66
67 @Override
68 public Object getUserToken(final HttpRoute route, final HttpContext context) {
69 return getUserToken(route, null, context);
70 }
71
72 @Override
73 public Object getUserToken(final HttpRoute route, final HttpRequest request, final HttpContext context) {
74
75 final HttpClientContext clientContext = HttpClientContext.cast(context);
76
77 final HttpHost target = request != null ? new HttpHost(request.getScheme(), request.getAuthority()) : route.getTargetHost();
78
79 final AuthExchange targetAuthExchange = clientContext.getAuthExchange(target);
80 if (targetAuthExchange != null) {
81 final Principal authPrincipal = getAuthPrincipal(targetAuthExchange);
82 if (authPrincipal != null) {
83 return authPrincipal;
84 }
85 }
86 final HttpHost proxy = route.getProxyHost();
87 if (proxy != null) {
88 final AuthExchange proxyAuthExchange = clientContext.getAuthExchange(proxy);
89 if (proxyAuthExchange != null) {
90 final Principal authPrincipal = getAuthPrincipal(proxyAuthExchange);
91 if (authPrincipal != null) {
92 return authPrincipal;
93 }
94 }
95 }
96 final SSLSession sslSession = clientContext.getSSLSession();
97 if (sslSession != null) {
98 return sslSession.getLocalPrincipal();
99 }
100 return null;
101 }
102
103 private static Principal getAuthPrincipal(final AuthExchange authExchange) {
104 final AuthScheme scheme = authExchange.getAuthScheme();
105 if (scheme != null && scheme.isConnectionBased()) {
106 return scheme.getPrincipal();
107 }
108 return null;
109 }
110
111 }