private void SetupContext(HttpClientContext context)
{
if (context.GetAttribute(HttpClientContext.TargetAuthState) == null)
{
context.SetAttribute(HttpClientContext.TargetAuthState, new AuthState());
}
if (context.GetAttribute(HttpClientContext.ProxyAuthState) == null)
{
context.SetAttribute(HttpClientContext.ProxyAuthState, new AuthState());
}
if (context.GetAttribute(HttpClientContext.AuthschemeRegistry) == null)
{
context.SetAttribute(HttpClientContext.AuthschemeRegistry, this.authSchemeRegistry
);
}
if (context.GetAttribute(HttpClientContext.CookiespecRegistry) == null)
{
context.SetAttribute(HttpClientContext.CookiespecRegistry, this.cookieSpecRegistry
);
}
if (context.GetAttribute(HttpClientContext.CookieStore) == null)
{
context.SetAttribute(HttpClientContext.CookieStore, this.cookieStore);
}
if (context.GetAttribute(HttpClientContext.CredsProvider) == null)
{
context.SetAttribute(HttpClientContext.CredsProvider, this.credentialsProvider);
}
if (context.GetAttribute(HttpClientContext.RequestConfig) == null)
{
context.SetAttribute(HttpClientContext.RequestConfig, this.defaultConfig);
}
}
/// <exception cref="System.IO.IOException"></exception>
/// <exception cref="Apache.Http.HttpException"></exception>
public virtual CloseableHttpResponse Execute(HttpRoute route, HttpRequestWrapper
request, HttpClientContext context, HttpExecutionAware execAware)
{
Args.NotNull(route, "HTTP route");
Args.NotNull(request, "HTTP request");
Args.NotNull(context, "HTTP context");
IHttpRequest original = request.GetOriginal();
URI uri = null;
if (original is IHttpUriRequest)
{
uri = ((IHttpUriRequest)original).GetURI();
}
else
{
string uriString = original.GetRequestLine().GetUri();
try
{
uri = URI.Create(uriString);
}
catch (ArgumentException ex)
{
if (this.log.IsDebugEnabled())
{
this.log.Debug("Unable to parse '" + uriString + "' as a valid URI; " + "request URI and Host header may be inconsistent"
, ex);
}
}
}
request.SetURI(uri);
// Re-write request URI if needed
RewriteRequestURI(request, route);
HttpParams @params = request.GetParams();
HttpHost virtualHost = (HttpHost)@params.GetParameter(ClientPNames.VirtualHost);
// HTTPCLIENT-1092 - add the port if necessary
if (virtualHost != null && virtualHost.GetPort() == -1)
{
int port = route.GetTargetHost().GetPort();
if (port != -1)
{
virtualHost = new HttpHost(virtualHost.GetHostName(), port, virtualHost.GetSchemeName
());
}
if (this.log.IsDebugEnabled())
{
this.log.Debug("Using virtual host" + virtualHost);
}
}
HttpHost target = null;
if (virtualHost != null)
{
target = virtualHost;
}
else
{
if (uri != null && uri.IsAbsolute() && uri.GetHost() != null)
{
target = new HttpHost(uri.GetHost(), uri.GetPort(), uri.GetScheme());
}
}
if (target == null)
{
target = route.GetTargetHost();
}
// Get user info from the URI
if (uri != null)
{
string userinfo = uri.GetUserInfo();
if (userinfo != null)
{
CredentialsProvider credsProvider = context.GetCredentialsProvider();
if (credsProvider == null)
{
credsProvider = new BasicCredentialsProvider();
context.SetCredentialsProvider(credsProvider);
}
credsProvider.SetCredentials(new AuthScope(target), new UsernamePasswordCredentials
(userinfo));
}
}
// Run request protocol interceptors
context.SetAttribute(HttpClientContext.HttpTargetHost, target);
context.SetAttribute(HttpClientContext.HttpRoute, route);
context.SetAttribute(HttpClientContext.HttpRequest, request);
this.httpProcessor.Process(request, context);
CloseableHttpResponse response = this.requestExecutor.Execute(route, request, context
, execAware);
try
{
// Run response protocol interceptors
context.SetAttribute(HttpClientContext.HttpResponse, response);
this.httpProcessor.Process(response, context);
return(response);
}
catch (RuntimeException ex)
{
response.Close();
throw;
}
catch (IOException ex)
{
response.Close();
throw;
}
catch (HttpException ex)
{
response.Close();
throw;
}
}
/// <exception cref="System.IO.IOException"></exception>
/// <exception cref="Apache.Http.HttpException"></exception>
public virtual CloseableHttpResponse Execute(HttpRoute route, HttpRequestWrapper
request, HttpClientContext context, HttpExecutionAware execAware)
{
Args.NotNull(route, "HTTP route");
Args.NotNull(request, "HTTP request");
Args.NotNull(context, "HTTP context");
ConnectionRequest connRequest = connManager.RequestConnection(route, null);
if (execAware != null)
{
if (execAware.IsAborted())
{
connRequest.Cancel();
throw new RequestAbortedException("Request aborted");
}
else
{
execAware.SetCancellable(connRequest);
}
}
RequestConfig config = context.GetRequestConfig();
HttpClientConnection managedConn;
try
{
int timeout = config.GetConnectionRequestTimeout();
managedConn = connRequest.Get(timeout > 0 ? timeout : 0, TimeUnit.Milliseconds);
}
catch (Exception interrupted)
{
Sharpen.Thread.CurrentThread().Interrupt();
throw new RequestAbortedException("Request aborted", interrupted);
}
catch (ExecutionException ex)
{
Exception cause = ex.InnerException;
if (cause == null)
{
cause = ex;
}
throw new RequestAbortedException("Request execution failed", cause);
}
ConnectionHolder releaseTrigger = new ConnectionHolder(log, connManager, managedConn
);
try
{
if (execAware != null)
{
if (execAware.IsAborted())
{
releaseTrigger.Close();
throw new RequestAbortedException("Request aborted");
}
else
{
execAware.SetCancellable(releaseTrigger);
}
}
if (!managedConn.IsOpen())
{
int timeout = config.GetConnectTimeout();
this.connManager.Connect(managedConn, route, timeout > 0 ? timeout : 0, context);
this.connManager.RouteComplete(managedConn, route, context);
}
int timeout_1 = config.GetSocketTimeout();
if (timeout_1 >= 0)
{
managedConn.SetSocketTimeout(timeout_1);
}
HttpHost target = null;
IHttpRequest original = request.GetOriginal();
if (original is IHttpUriRequest)
{
URI uri = ((IHttpUriRequest)original).GetURI();
if (uri.IsAbsolute())
{
target = new HttpHost(uri.GetHost(), uri.GetPort(), uri.GetScheme());
}
}
if (target == null)
{
target = route.GetTargetHost();
}
context.SetAttribute(HttpClientContext.HttpTargetHost, target);
context.SetAttribute(HttpClientContext.HttpRequest, request);
context.SetAttribute(HttpClientContext.HttpConnection, managedConn);
context.SetAttribute(HttpClientContext.HttpRoute, route);
httpProcessor.Process(request, context);
HttpResponse response = requestExecutor.Execute(request, managedConn, context);
httpProcessor.Process(response, context);
// The connection is in or can be brought to a re-usable state.
if (reuseStrategy.KeepAlive(response, context))
{
// Set the idle duration of this connection
long duration = keepAliveStrategy.GetKeepAliveDuration(response, context);
releaseTrigger.SetValidFor(duration, TimeUnit.Milliseconds);
releaseTrigger.MarkReusable();
}
else
{
releaseTrigger.MarkNonReusable();
}
// check for entity, release connection if possible
HttpEntity entity = response.GetEntity();
if (entity == null || !entity.IsStreaming())
{
// connection not needed and (assumed to be) in re-usable state
releaseTrigger.ReleaseConnection();
return(Proxies.EnhanceResponse(response, null));
}
else
{
return(Proxies.EnhanceResponse(response, releaseTrigger));
}
}
catch (ConnectionShutdownException ex)
{
ThreadInterruptedException ioex = new ThreadInterruptedException("Connection has been shut down"
);
Sharpen.Extensions.InitCause(ioex, ex);
throw ioex;
}
catch (HttpException ex)
{
releaseTrigger.AbortConnection();
throw;
}
catch (IOException ex)
{
releaseTrigger.AbortConnection();
throw;
}
catch (RuntimeException ex)
{
releaseTrigger.AbortConnection();
throw;
}
}
/// <exception cref="System.IO.IOException"></exception>
/// <exception cref="Apache.Http.HttpException"></exception>
public virtual CloseableHttpResponse Execute(HttpRoute route, HttpRequestWrapper
request, HttpClientContext context, HttpExecutionAware execAware)
{
Args.NotNull(route, "HTTP route");
Args.NotNull(request, "HTTP request");
Args.NotNull(context, "HTTP context");
AuthState targetAuthState = context.GetTargetAuthState();
if (targetAuthState == null)
{
targetAuthState = new AuthState();
context.SetAttribute(HttpClientContext.TargetAuthState, targetAuthState);
}
AuthState proxyAuthState = context.GetProxyAuthState();
if (proxyAuthState == null)
{
proxyAuthState = new AuthState();
context.SetAttribute(HttpClientContext.ProxyAuthState, proxyAuthState);
}
if (request is HttpEntityEnclosingRequest)
{
Proxies.EnhanceEntity((HttpEntityEnclosingRequest)request);
}
object userToken = context.GetUserToken();
ConnectionRequest connRequest = connManager.RequestConnection(route, userToken);
if (execAware != null)
{
if (execAware.IsAborted())
{
connRequest.Cancel();
throw new RequestAbortedException("Request aborted");
}
else
{
execAware.SetCancellable(connRequest);
}
}
RequestConfig config = context.GetRequestConfig();
HttpClientConnection managedConn;
try
{
int timeout = config.GetConnectionRequestTimeout();
managedConn = connRequest.Get(timeout > 0 ? timeout : 0, TimeUnit.Milliseconds);
}
catch (Exception interrupted)
{
Sharpen.Thread.CurrentThread().Interrupt();
throw new RequestAbortedException("Request aborted", interrupted);
}
catch (ExecutionException ex)
{
Exception cause = ex.InnerException;
if (cause == null)
{
cause = ex;
}
throw new RequestAbortedException("Request execution failed", cause);
}
context.SetAttribute(HttpClientContext.HttpConnection, managedConn);
if (config.IsStaleConnectionCheckEnabled())
{
// validate connection
if (managedConn.IsOpen())
{
this.log.Debug("Stale connection check");
if (managedConn.IsStale())
{
this.log.Debug("Stale connection detected");
managedConn.Close();
}
}
}
ConnectionHolder connHolder = new ConnectionHolder(this.log, this.connManager, managedConn
);
try
{
if (execAware != null)
{
execAware.SetCancellable(connHolder);
}
HttpResponse response;
for (int execCount = 1; ; execCount++)
{
if (execCount > 1 && !Proxies.IsRepeatable(request))
{
throw new NonRepeatableRequestException("Cannot retry request " + "with a non-repeatable request entity."
);
}
if (execAware != null && execAware.IsAborted())
{
throw new RequestAbortedException("Request aborted");
}
if (!managedConn.IsOpen())
{
this.log.Debug("Opening connection " + route);
try
{
EstablishRoute(proxyAuthState, managedConn, route, request, context);
}
catch (TunnelRefusedException ex)
{
if (this.log.IsDebugEnabled())
{
this.log.Debug(ex.Message);
}
response = ex.GetResponse();
break;
}
}
int timeout = config.GetSocketTimeout();
if (timeout >= 0)
{
managedConn.SetSocketTimeout(timeout);
}
if (execAware != null && execAware.IsAborted())
{
throw new RequestAbortedException("Request aborted");
}
if (this.log.IsDebugEnabled())
{
this.log.Debug("Executing request " + request.GetRequestLine());
}
if (!request.ContainsHeader(AUTH.WwwAuthResp))
{
if (this.log.IsDebugEnabled())
{
this.log.Debug("Target auth state: " + targetAuthState.GetState());
}
this.authenticator.GenerateAuthResponse(request, targetAuthState, context);
}
if (!request.ContainsHeader(AUTH.ProxyAuthResp) && !route.IsTunnelled())
{
if (this.log.IsDebugEnabled())
{
this.log.Debug("Proxy auth state: " + proxyAuthState.GetState());
}
this.authenticator.GenerateAuthResponse(request, proxyAuthState, context);
}
response = requestExecutor.Execute(request, managedConn, context);
// The connection is in or can be brought to a re-usable state.
if (reuseStrategy.KeepAlive(response, context))
{
// Set the idle duration of this connection
long duration = keepAliveStrategy.GetKeepAliveDuration(response, context);
if (this.log.IsDebugEnabled())
{
string s;
if (duration > 0)
{
s = "for " + duration + " " + TimeUnit.Milliseconds;
}
else
{
s = "indefinitely";
}
this.log.Debug("Connection can be kept alive " + s);
}
connHolder.SetValidFor(duration, TimeUnit.Milliseconds);
connHolder.MarkReusable();
}
else
{
connHolder.MarkNonReusable();
}
if (NeedAuthentication(targetAuthState, proxyAuthState, route, response, context))
{
// Make sure the response body is fully consumed, if present
HttpEntity entity = response.GetEntity();
if (connHolder.IsReusable())
{
EntityUtils.Consume(entity);
}
else
{
managedConn.Close();
if (proxyAuthState.GetState() == AuthProtocolState.Success && proxyAuthState.GetAuthScheme
() != null && proxyAuthState.GetAuthScheme().IsConnectionBased())
{
this.log.Debug("Resetting proxy auth state");
proxyAuthState.Reset();
}
if (targetAuthState.GetState() == AuthProtocolState.Success && targetAuthState.GetAuthScheme
() != null && targetAuthState.GetAuthScheme().IsConnectionBased())
{
this.log.Debug("Resetting target auth state");
targetAuthState.Reset();
}
}
// discard previous auth headers
IHttpRequest original = request.GetOriginal();
if (!original.ContainsHeader(AUTH.WwwAuthResp))
{
request.RemoveHeaders(AUTH.WwwAuthResp);
}
if (!original.ContainsHeader(AUTH.ProxyAuthResp))
{
request.RemoveHeaders(AUTH.ProxyAuthResp);
}
}
else
{
break;
}
}
if (userToken == null)
{
userToken = userTokenHandler.GetUserToken(context);
context.SetAttribute(HttpClientContext.UserToken, userToken);
}
if (userToken != null)
{
connHolder.SetState(userToken);
}
// check for entity, release connection if possible
HttpEntity entity_1 = response.GetEntity();
if (entity_1 == null || !entity_1.IsStreaming())
{
// connection not needed and (assumed to be) in re-usable state
connHolder.ReleaseConnection();
return(Proxies.EnhanceResponse(response, null));
}
else
{
return(Proxies.EnhanceResponse(response, connHolder));
}
}
catch (ConnectionShutdownException ex)
{
ThreadInterruptedException ioex = new ThreadInterruptedException("Connection has been shut down"
);
Sharpen.Extensions.InitCause(ioex, ex);
throw ioex;
}
catch (HttpException ex)
{
connHolder.AbortConnection();
throw;
}
catch (IOException ex)
{
connHolder.AbortConnection();
throw;
}
catch (RuntimeException ex)
{
connHolder.AbortConnection();
throw;
}
}