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.HttpResponse; michael@0: import ch.boye.httpclientandroidlib.HttpResponseInterceptor; michael@0: import ch.boye.httpclientandroidlib.annotation.Immutable; michael@0: import ch.boye.httpclientandroidlib.auth.AuthScheme; michael@0: import ch.boye.httpclientandroidlib.auth.AuthState; michael@0: import ch.boye.httpclientandroidlib.client.AuthCache; michael@0: import ch.boye.httpclientandroidlib.client.params.AuthPolicy; michael@0: import ch.boye.httpclientandroidlib.impl.client.BasicAuthCache; michael@0: import ch.boye.httpclientandroidlib.protocol.ExecutionContext; michael@0: import ch.boye.httpclientandroidlib.protocol.HttpContext; michael@0: michael@0: /** michael@0: * Response interceptor that adds successfully completed {@link AuthScheme}s michael@0: * to the local {@link AuthCache} instance. Cached {@link AuthScheme}s can be michael@0: * re-used when executing requests against known hosts, thus avoiding michael@0: * additional authentication round-trips. michael@0: * michael@0: * @since 4.1 michael@0: */ michael@0: @Immutable michael@0: public class ResponseAuthCache implements HttpResponseInterceptor { michael@0: michael@0: public HttpClientAndroidLog log = new HttpClientAndroidLog(getClass()); michael@0: michael@0: public ResponseAuthCache() { michael@0: super(); michael@0: } michael@0: michael@0: public void process(final HttpResponse response, final HttpContext context) michael@0: throws HttpException, IOException { michael@0: if (response == 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: AuthCache authCache = (AuthCache) context.getAttribute(ClientContext.AUTH_CACHE); 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) { michael@0: if (isCachable(targetState)) { michael@0: if (authCache == null) { michael@0: authCache = new BasicAuthCache(); michael@0: context.setAttribute(ClientContext.AUTH_CACHE, authCache); michael@0: } michael@0: cache(authCache, target, targetState); 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) { michael@0: if (isCachable(proxyState)) { michael@0: if (authCache == null) { michael@0: authCache = new BasicAuthCache(); michael@0: context.setAttribute(ClientContext.AUTH_CACHE, authCache); michael@0: } michael@0: cache(authCache, proxy, proxyState); michael@0: } michael@0: } michael@0: } michael@0: michael@0: private boolean isCachable(final AuthState authState) { michael@0: AuthScheme authScheme = authState.getAuthScheme(); michael@0: if (authScheme == null || !authScheme.isComplete()) { michael@0: return false; michael@0: } michael@0: String schemeName = authScheme.getSchemeName(); michael@0: return schemeName.equalsIgnoreCase(AuthPolicy.BASIC) || michael@0: schemeName.equalsIgnoreCase(AuthPolicy.DIGEST); michael@0: } michael@0: michael@0: private void cache(final AuthCache authCache, final HttpHost host, final AuthState authState) { michael@0: AuthScheme authScheme = authState.getAuthScheme(); michael@0: if (authState.getAuthScope() != null) { michael@0: if (authState.getCredentials() != null) { michael@0: if (this.log.isDebugEnabled()) { michael@0: this.log.debug("Caching '" + authScheme.getSchemeName() + michael@0: "' auth scheme for " + host); michael@0: } michael@0: authCache.put(host, authScheme); michael@0: } else { michael@0: authCache.remove(host); michael@0: } michael@0: } michael@0: } michael@0: michael@0: }