Reduce PassRefPtr in WebKit2 - 3
[WebKit-https.git] / Source / WebKit / Storage / StorageNamespaceImpl.cpp
1 /*
2  * Copyright (C) 2008 Apple Inc. All Rights Reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "StorageNamespaceImpl.h"
27
28 #include "StorageAreaImpl.h"
29 #include "StorageSyncManager.h"
30 #include "StorageTracker.h"
31 #include <WebCore/StorageMap.h>
32 #include <wtf/MainThread.h>
33 #include <wtf/NeverDestroyed.h>
34 #include <wtf/text/StringHash.h>
35
36 namespace WebCore {
37
38 static HashMap<String, StorageNamespaceImpl*>& localStorageNamespaceMap()
39 {
40     static NeverDestroyed<HashMap<String, StorageNamespaceImpl*>> localStorageNamespaceMap;
41
42     return localStorageNamespaceMap;
43 }
44
45 RefPtr<StorageNamespaceImpl> StorageNamespaceImpl::createSessionStorageNamespace(unsigned quota)
46 {
47     return adoptRef(new StorageNamespaceImpl(SessionStorage, String(), quota));
48 }
49
50 RefPtr<StorageNamespaceImpl> StorageNamespaceImpl::getOrCreateLocalStorageNamespace(const String& databasePath, unsigned quota)
51 {
52     ASSERT(!databasePath.isNull());
53
54     auto& slot = localStorageNamespaceMap().add(databasePath, nullptr).iterator->value;
55     if (slot)
56         return slot;
57
58     RefPtr<StorageNamespaceImpl> storageNamespace = adoptRef(new StorageNamespaceImpl(LocalStorage, databasePath, quota));
59     slot = storageNamespace.get();
60
61     return storageNamespace;
62 }
63
64 StorageNamespaceImpl::StorageNamespaceImpl(StorageType storageType, const String& path, unsigned quota)
65     : m_storageType(storageType)
66     , m_path(path.isolatedCopy())
67     , m_syncManager(0)
68     , m_quota(quota)
69     , m_isShutdown(false)
70 {
71     if (m_storageType == LocalStorage && !m_path.isEmpty())
72         m_syncManager = StorageSyncManager::create(m_path);
73 }
74
75 StorageNamespaceImpl::~StorageNamespaceImpl()
76 {
77     ASSERT(isMainThread());
78
79     if (m_storageType == LocalStorage) {
80         ASSERT(localStorageNamespaceMap().get(m_path) == this);
81         localStorageNamespaceMap().remove(m_path);
82     }
83
84     if (!m_isShutdown)
85         close();
86 }
87
88 RefPtr<StorageNamespace> StorageNamespaceImpl::copy(Page*)
89 {
90     ASSERT(isMainThread());
91     ASSERT(!m_isShutdown);
92     ASSERT(m_storageType == SessionStorage);
93
94     RefPtr<StorageNamespaceImpl> newNamespace = adoptRef(new StorageNamespaceImpl(m_storageType, m_path, m_quota));
95
96     StorageAreaMap::iterator end = m_storageAreaMap.end();
97     for (StorageAreaMap::iterator i = m_storageAreaMap.begin(); i != end; ++i)
98         newNamespace->m_storageAreaMap.set(i->key, i->value->copy());
99     return newNamespace;
100 }
101
102 RefPtr<StorageArea> StorageNamespaceImpl::storageArea(PassRefPtr<SecurityOrigin> prpOrigin)
103 {
104     ASSERT(isMainThread());
105     ASSERT(!m_isShutdown);
106
107     RefPtr<SecurityOrigin> origin = prpOrigin;
108     RefPtr<StorageAreaImpl> storageArea;
109     if ((storageArea = m_storageAreaMap.get(origin)))
110         return storageArea;
111
112     storageArea = StorageAreaImpl::create(m_storageType, origin, m_syncManager, m_quota);
113     m_storageAreaMap.set(origin.release(), storageArea);
114     return storageArea;
115 }
116
117 void StorageNamespaceImpl::close()
118 {
119     ASSERT(isMainThread());
120
121     if (m_isShutdown)
122         return;
123
124     // If we're session storage, we shouldn't need to do any work here.
125     if (m_storageType == SessionStorage) {
126         ASSERT(!m_syncManager);
127         return;
128     }
129
130     StorageAreaMap::iterator end = m_storageAreaMap.end();
131     for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
132         it->value->close();
133
134     if (m_syncManager)
135         m_syncManager->close();
136
137     m_isShutdown = true;
138 }
139
140 void StorageNamespaceImpl::clearOriginForDeletion(SecurityOrigin* origin)
141 {
142     ASSERT(isMainThread());
143
144     RefPtr<StorageAreaImpl> storageArea = m_storageAreaMap.get(origin);
145     if (storageArea)
146         storageArea->clearForOriginDeletion();
147 }
148
149 void StorageNamespaceImpl::clearAllOriginsForDeletion()
150 {
151     ASSERT(isMainThread());
152
153     StorageAreaMap::iterator end = m_storageAreaMap.end();
154     for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
155         it->value->clearForOriginDeletion();
156 }
157     
158 void StorageNamespaceImpl::sync()
159 {
160     ASSERT(isMainThread());
161     StorageAreaMap::iterator end = m_storageAreaMap.end();
162     for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
163         it->value->sync();
164 }
165
166 void StorageNamespaceImpl::closeIdleLocalStorageDatabases()
167 {
168     ASSERT(isMainThread());
169     StorageAreaMap::iterator end = m_storageAreaMap.end();
170     for (StorageAreaMap::iterator it = m_storageAreaMap.begin(); it != end; ++it)
171         it->value->closeDatabaseIfIdle();
172 }
173
174 } // namespace WebCore