1/*
2 * Copyright (C) 2012 Motorola Mobility Inc.
3 * Copyright (C) 2013 Google Inc. All Rights Reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * 1.  Redistributions of source code must retain the above copyright
10 *     notice, this list of conditions and the following disclaimer.
11 * 2.  Redistributions in binary form must reproduce the above copyright
12 *     notice, this list of conditions and the following disclaimer in the
13 *     documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
16 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
19 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27#include "config.h"
28#include "core/html/PublicURLManager.h"
29
30#include "core/fetch/MemoryCache.h"
31#include "core/html/URLRegistry.h"
32#include "platform/weborigin/KURL.h"
33#include "wtf/Vector.h"
34#include "wtf/text/StringHash.h"
35
36namespace blink {
37
38PassOwnPtr<PublicURLManager> PublicURLManager::create(ExecutionContext* context)
39{
40    OwnPtr<PublicURLManager> publicURLManager(adoptPtr(new PublicURLManager(context)));
41    publicURLManager->suspendIfNeeded();
42    return publicURLManager.release();
43}
44
45PublicURLManager::PublicURLManager(ExecutionContext* context)
46    : ActiveDOMObject(context)
47    , m_isStopped(false)
48{
49}
50
51void PublicURLManager::registerURL(SecurityOrigin* origin, const KURL& url, URLRegistrable* registrable, const String& uuid)
52{
53    if (m_isStopped)
54        return;
55
56    RegistryURLMap::ValueType* found = m_registryToURL.add(&registrable->registry(), URLMap()).storedValue;
57    found->key->registerURL(origin, url, registrable);
58    found->value.add(url.string(), uuid);
59}
60
61void PublicURLManager::revoke(const KURL& url)
62{
63    for (RegistryURLMap::iterator i = m_registryToURL.begin(); i != m_registryToURL.end(); ++i) {
64        if (i->value.contains(url.string())) {
65            i->key->unregisterURL(url);
66            i->value.remove(url.string());
67            break;
68        }
69    }
70}
71
72void PublicURLManager::revoke(const String& uuid)
73{
74    // A linear scan; revoking by UUID is assumed rare.
75    Vector<String> urlsToRemove;
76    for (RegistryURLMap::iterator i = m_registryToURL.begin(); i != m_registryToURL.end(); ++i) {
77        URLRegistry* registry = i->key;
78        URLMap& registeredURLs = i->value;
79        for (URLMap::iterator j = registeredURLs.begin(); j != registeredURLs.end(); ++j) {
80            if (uuid == j->value) {
81                KURL url(ParsedURLString, j->key);
82                MemoryCache::removeURLFromCache(executionContext(), url);
83                registry->unregisterURL(url);
84                urlsToRemove.append(j->key);
85            }
86        }
87        for (unsigned j = 0; j < urlsToRemove.size(); j++)
88            registeredURLs.remove(urlsToRemove[j]);
89        urlsToRemove.clear();
90    }
91}
92
93void PublicURLManager::stop()
94{
95    if (m_isStopped)
96        return;
97
98    m_isStopped = true;
99    for (RegistryURLMap::iterator i = m_registryToURL.begin(); i != m_registryToURL.end(); ++i) {
100        for (URLMap::iterator j = i->value.begin(); j != i->value.end(); ++j)
101            i->key->unregisterURL(KURL(ParsedURLString, j->key));
102    }
103
104    m_registryToURL.clear();
105}
106
107}
108