Apache/httpcomponents-client
1
0
Fork 0
mirror of https://github.com/apache/httpcomponents-client.git synced 2025-02-28 05:39:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Corrected resolution of the target host in DefaultUserTokenHandler

Browse Source
This commit is contained in:
Oleg Kalnichevski 2021-09-26 15:51:23 +02:00
parent fb0c073783
commit 4dd7cefbde
4 changed files with 46 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
httpclient5/src/main/java/org/apache/hc/client5/http/UserTokenHandler.java
View File

@ -29,6 +29,7 @@
import org.apache.hc.core5.annotation.Contract;
import org.apache.hc.core5.annotation.ThreadingBehavior;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.protocol.HttpContext;
/**
@ -60,4 +61,22 @@ public interface UserTokenHandler {
*/
Object getUserToken(HttpRoute route, HttpContext context);
/**
* The token object returned by this method is expected to uniquely
* identify the current user if the context is user specific or to be
* {@code null} if it is not.
*
* @param route HTTP route
* @param request HTTP request
* @param context the execution context
*
* @return user token that uniquely identifies the user or
* {@code null} if the context is not user specific.
*
* @since 5.2
*/
default Object getUserToken(HttpRoute route, HttpRequest request, HttpContext context) {
return getUserToken(route, context);
}
}

38
httpclient5/src/main/java/org/apache/hc/client5/http/impl/DefaultUserTokenHandler.java
View File

@ -37,6 +37,8 @@
import org.apache.hc.client5.http.protocol.HttpClientContext;
import org.apache.hc.core5.annotation.Contract;
import org.apache.hc.core5.annotation.ThreadingBehavior;
import org.apache.hc.core5.http.HttpHost;
import org.apache.hc.core5.http.HttpRequest;
import org.apache.hc.core5.http.protocol.HttpContext;
/**
@ -61,28 +63,38 @@ public class DefaultUserTokenHandler implements UserTokenHandler {
@Override
public Object getUserToken(final HttpRoute route, final HttpContext context) {
return getUserToken(route, null, context);
}
@Override
public Object getUserToken(final HttpRoute route, final HttpRequest request, final HttpContext context) {
final HttpClientContext clientContext = HttpClientContext.adapt(context);
Principal userPrincipal = null;
final HttpHost target = request != null ? new HttpHost(request.getScheme(), request.getAuthority()) : route.getTargetHost();
final AuthExchange targetAuthExchange = clientContext.getAuthExchange(route.getTargetHost());
final AuthExchange targetAuthExchange = clientContext.getAuthExchange(target);
if (targetAuthExchange != null) {
userPrincipal = getAuthPrincipal(targetAuthExchange);
if (userPrincipal == null && route.getProxyHost() != null) {
final AuthExchange proxyAuthExchange = clientContext.getAuthExchange(route.getProxyHost());
userPrincipal = getAuthPrincipal(proxyAuthExchange);
final Principal authPrincipal = getAuthPrincipal(targetAuthExchange);
if (authPrincipal != null) {
return authPrincipal;
}
}
if (userPrincipal == null) {
final SSLSession sslSession = clientContext.getSSLSession();
if (sslSession != null) {
userPrincipal = sslSession.getLocalPrincipal();
final HttpHost proxy = route.getProxyHost();
if (proxy != null) {
final AuthExchange proxyAuthExchange = clientContext.getAuthExchange(proxy);
if (proxyAuthExchange != null) {
final Principal authPrincipal = getAuthPrincipal(proxyAuthExchange);
if (authPrincipal != null) {
return authPrincipal;
}
}
}
return userPrincipal;
final SSLSession sslSession = clientContext.getSSLSession();
if (sslSession != null) {
return sslSession.getLocalPrincipal();
}
return null;
}
private static Principal getAuthPrincipal(final AuthExchange authExchange) {

2
httpclient5/src/main/java/org/apache/hc/client5/http/impl/async/HttpAsyncMainClientExec.java
View File

@ -196,7 +196,7 @@ public void consumeResponse(
final TimeValue keepAliveDuration = keepAliveStrategy.getKeepAliveDuration(response, clientContext);
Object userToken = clientContext.getUserToken();
if (userToken == null) {
userToken = userTokenHandler.getUserToken(route, clientContext);
userToken = userTokenHandler.getUserToken(route, request, clientContext);
clientContext.setAttribute(HttpClientContext.USER_TOKEN, userToken);
}
execRuntime.markConnectionReusable(userToken, keepAliveDuration);

2
httpclient5/src/main/java/org/apache/hc/client5/http/impl/classic/MainClientExec.java
View File

@ -106,7 +106,7 @@ public ClassicHttpResponse execute(
Object userToken = context.getUserToken();
if (userToken == null) {
userToken = userTokenHandler.getUserToken(route, context);
userToken = userTokenHandler.getUserToken(route, request, context);
context.setAttribute(HttpClientContext.USER_TOKEN, userToken);
}