org.eclipse.jetty.client.HttpCookieStore Maven / Gradle / Ivy

Go to download
//
//  ========================================================================
//  Copyright (c) 1995-2012 Mort Bay Consulting Pty. Ltd.
//  ------------------------------------------------------------------------
//  All rights reserved. This program and the accompanying materials
//  are made available under the terms of the Eclipse Public License v1.0
//  and Apache License v2.0 which accompanies this distribution.
//
//      The Eclipse Public License is available at
//      http://www.eclipse.org/legal/epl-v10.html
//
//      The Apache License v2.0 is available at
//      http://www.opensource.org/licenses/apache2.0.php
//
//  You may elect to redistribute this code under either of these licenses.
//  ========================================================================
//

package org.eclipse.jetty.client;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ConcurrentMap;

import org.eclipse.jetty.client.api.CookieStore;
import org.eclipse.jetty.client.api.Destination;
import org.eclipse.jetty.http.HttpCookie;

public class HttpCookieStore implements CookieStore
{
    private final ConcurrentMap> allCookies = new ConcurrentHashMap<>();

    @Override
    public List findCookies(Destination destination, String path)
    {
        List result = new ArrayList<>();

        String host = destination.getHost();
        int port = destination.getPort();
        String key = host + ":" + port;

        // Root path lookup
        Queue cookies = allCookies.get(key + "/");
        if (cookies != null)
            accumulateCookies(destination, cookies, result);

        // Path lookup
        String[] split = path.split("/");
        for (int i = 1; i < split.length; i++)
        {
            String segment = split[i];
            key += "/" + segment;
            cookies = allCookies.get(key);
            if (cookies != null)
                accumulateCookies(destination, cookies, result);
            if (segment.length() > 0)
                key += "/";
        }

        // Domain lookup
        int domains = host.split("\\.").length - 1;
        for (int i = 2; i <= domains; ++i)
        {
            String[] hostParts = host.split("\\.", i);
            host = hostParts[hostParts.length - 1];
            key = host + ":" + port + "/";
            cookies = allCookies.get(key);
            if (cookies != null)
                accumulateCookies(destination, cookies, result);
        }

        return result;
    }

    private void accumulateCookies(Destination destination, Queue cookies, List result)
    {
        for (Iterator iterator = cookies.iterator(); iterator.hasNext(); )
        {
            HttpCookie cookie = iterator.next();
            if (cookie.isExpired(System.nanoTime()))
            {
                iterator.remove();
            }
            else
            {
                if (!"https".equalsIgnoreCase(destination.getScheme()) && cookie.isSecure())
                    continue;
                result.add(cookie);
            }
        }
    }

    @Override
    public boolean addCookie(Destination destination, HttpCookie cookie)
    {
        String destinationDomain = destination.getHost() + ":" + destination.getPort();

        // Check whether it is the same domain
        String domain = cookie.getDomain();
        if (domain == null)
            domain = destinationDomain;

        if (domain.indexOf(':') < 0)
            domain += ":" + ("https".equalsIgnoreCase(destination.getScheme()) ? 443 : 80);

        // Cookie domains may start with a ".", like ".domain.com"
        // This also avoids that a request to sub.domain.com sets a cookie for domain.com
        if (!domain.endsWith(destinationDomain))
            return false;

        // Normalize the path
        String path = cookie.getPath();
        if (path == null || path.length() == 0)
            path = "/";

        String key = destination.getHost() + ":" + destination.getPort() + path;
        Queue cookies = allCookies.get(key);
        if (cookies == null)
        {
            cookies = new ConcurrentLinkedQueue<>();
            Queue existing = allCookies.putIfAbsent(key, cookies);
            if (existing != null)
                cookies = existing;
        }
        cookies.add(cookie);
        return true;
    }

    @Override
    public void clear()
    {
        allCookies.clear();
    }
}