michael@0: /* michael@0: * ==================================================================== michael@0: * Licensed to the Apache Software Foundation (ASF) under one michael@0: * or more contributor license agreements. See the NOTICE file michael@0: * distributed with this work for additional information michael@0: * regarding copyright ownership. The ASF licenses this file michael@0: * to you under the Apache License, Version 2.0 (the michael@0: * "License"); you may not use this file except in compliance michael@0: * with the License. You may obtain a copy of the License at michael@0: * michael@0: * http://www.apache.org/licenses/LICENSE-2.0 michael@0: * michael@0: * Unless required by applicable law or agreed to in writing, michael@0: * software distributed under the License is distributed on an michael@0: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY michael@0: * KIND, either express or implied. See the License for the michael@0: * specific language governing permissions and limitations michael@0: * under the License. michael@0: * ==================================================================== michael@0: * michael@0: * This software consists of voluntary contributions made by many michael@0: * individuals on behalf of the Apache Software Foundation. For more michael@0: * information on the Apache Software Foundation, please see michael@0: * . michael@0: * michael@0: */ michael@0: michael@0: package ch.boye.httpclientandroidlib.client.protocol; michael@0: michael@0: import java.io.IOException; michael@0: michael@0: import ch.boye.httpclientandroidlib.androidextra.HttpClientAndroidLog; michael@0: /* LogFactory removed by HttpClient for Android script. */ michael@0: import ch.boye.httpclientandroidlib.HttpException; michael@0: import ch.boye.httpclientandroidlib.HttpHost; michael@0: import ch.boye.httpclientandroidlib.HttpRequest; michael@0: import ch.boye.httpclientandroidlib.HttpRequestInterceptor; michael@0: import ch.boye.httpclientandroidlib.annotation.Immutable; michael@0: import ch.boye.httpclientandroidlib.auth.AuthScheme; michael@0: import ch.boye.httpclientandroidlib.auth.AuthScope; michael@0: import ch.boye.httpclientandroidlib.auth.AuthState; michael@0: import ch.boye.httpclientandroidlib.auth.Credentials; michael@0: import ch.boye.httpclientandroidlib.client.AuthCache; michael@0: import ch.boye.httpclientandroidlib.client.CredentialsProvider; michael@0: import ch.boye.httpclientandroidlib.protocol.ExecutionContext; michael@0: import ch.boye.httpclientandroidlib.protocol.HttpContext; michael@0: michael@0: /** michael@0: * Request interceptor that can preemptively authenticate against known hosts, michael@0: * if there is a cached {@link AuthScheme} instance in the local michael@0: * {@link AuthCache} associated with the given target or proxy host. michael@0: * michael@0: * @since 4.1 michael@0: */ michael@0: @Immutable michael@0: public class RequestAuthCache implements HttpRequestInterceptor { michael@0: michael@0: public HttpClientAndroidLog log = new HttpClientAndroidLog(getClass()); michael@0: michael@0: public RequestAuthCache() { michael@0: super(); michael@0: } michael@0: michael@0: public void process(final HttpRequest request, final HttpContext context) michael@0: throws HttpException, IOException { michael@0: if (request == null) { michael@0: throw new IllegalArgumentException("HTTP request may not be null"); michael@0: } michael@0: if (context == null) { michael@0: throw new IllegalArgumentException("HTTP context may not be null"); michael@0: } michael@0: michael@0: AuthCache authCache = (AuthCache) context.getAttribute(ClientContext.AUTH_CACHE); michael@0: if (authCache == null) { michael@0: this.log.debug("Auth cache not set in the context"); michael@0: return; michael@0: } michael@0: michael@0: CredentialsProvider credsProvider = (CredentialsProvider) context.getAttribute( michael@0: ClientContext.CREDS_PROVIDER); michael@0: if (credsProvider == null) { michael@0: this.log.debug("Credentials provider not set in the context"); michael@0: return; michael@0: } michael@0: michael@0: HttpHost target = (HttpHost) context.getAttribute(ExecutionContext.HTTP_TARGET_HOST); michael@0: AuthState targetState = (AuthState) context.getAttribute(ClientContext.TARGET_AUTH_STATE); michael@0: if (target != null && targetState != null && targetState.getAuthScheme() == null) { michael@0: AuthScheme authScheme = authCache.get(target); michael@0: if (authScheme != null) { michael@0: doPreemptiveAuth(target, authScheme, targetState, credsProvider); michael@0: } michael@0: } michael@0: michael@0: HttpHost proxy = (HttpHost) context.getAttribute(ExecutionContext.HTTP_PROXY_HOST); michael@0: AuthState proxyState = (AuthState) context.getAttribute(ClientContext.PROXY_AUTH_STATE); michael@0: if (proxy != null && proxyState != null && proxyState.getAuthScheme() == null) { michael@0: AuthScheme authScheme = authCache.get(proxy); michael@0: if (authScheme != null) { michael@0: doPreemptiveAuth(proxy, authScheme, proxyState, credsProvider); michael@0: } michael@0: } michael@0: } michael@0: michael@0: private void doPreemptiveAuth( michael@0: final HttpHost host, michael@0: final AuthScheme authScheme, michael@0: final AuthState authState, michael@0: final CredentialsProvider credsProvider) { michael@0: String schemeName = authScheme.getSchemeName(); michael@0: if (this.log.isDebugEnabled()) { michael@0: this.log.debug("Re-using cached '" + schemeName + "' auth scheme for " + host); michael@0: } michael@0: michael@0: AuthScope authScope = new AuthScope(host.getHostName(), host.getPort(), michael@0: AuthScope.ANY_REALM, schemeName); michael@0: Credentials creds = credsProvider.getCredentials(authScope); michael@0: michael@0: if (creds != null) { michael@0: authState.setAuthScheme(authScheme); michael@0: authState.setCredentials(creds); michael@0: } else { michael@0: this.log.debug("No credentials for preemptive authentication"); michael@0: } michael@0: } michael@0: michael@0: }