WebCore: Adding a way to get the set of all open database handles pointing
[WebKit-https.git] / WebKit / chromium / src / WebDatabase.cpp
1 /*
2  * Copyright (C) 2009 Google 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 are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32 #include "WebDatabase.h"
33
34 #include "Database.h"
35 #include "DatabaseTask.h"
36 #include "DatabaseThread.h"
37 #include "DatabaseTracker.h"
38 #include "Document.h"
39 #include "KURL.h"
40 #include "QuotaTracker.h"
41 #include "SecurityOrigin.h"
42 #include "WebDatabaseObserver.h"
43 #include "WebString.h"
44 #include <wtf/PassRefPtr.h>
45 #include <wtf/RefPtr.h>
46
47 using namespace WebCore;
48
49 namespace WebKit {
50
51 static WebDatabaseObserver* databaseObserver = 0;
52
53 class WebDatabasePrivate : public Database {
54 };
55
56 void WebDatabase::reset()
57 {
58     assign(0);
59 }
60
61 void WebDatabase::assign(const WebDatabase& other)
62 {
63     WebDatabasePrivate* d = const_cast<WebDatabasePrivate*>(other.m_private);
64     if (d)
65         d->ref();
66     assign(d);
67 }
68
69 WebString WebDatabase::name() const
70 {
71     ASSERT(m_private);
72     return m_private->stringIdentifier();
73 }
74
75 WebString WebDatabase::displayName() const
76 {
77     ASSERT(m_private);
78     return m_private->displayName();
79 }
80
81 unsigned long WebDatabase::estimatedSize() const
82 {
83     ASSERT(m_private);
84     return m_private->estimatedSize();
85 }
86
87 WebSecurityOrigin WebDatabase::securityOrigin() const
88 {
89     ASSERT(m_private);
90     return WebSecurityOrigin(m_private->securityOrigin());
91 }
92
93 void WebDatabase::setObserver(WebDatabaseObserver* observer)
94 {
95     databaseObserver = observer;
96 }
97
98 WebDatabaseObserver* WebDatabase::observer()
99 {
100     return databaseObserver;
101 }
102
103 void WebDatabase::updateDatabaseSize(
104     const WebString& originIdentifier, const WebString& databaseName,
105     unsigned long long databaseSize, unsigned long long spaceAvailable)
106 {
107     WebCore::QuotaTracker::instance().updateDatabaseSizeAndSpaceAvailableToOrigin(
108         originIdentifier, databaseName, databaseSize, spaceAvailable);
109 }
110
111 void WebDatabase::closeDatabaseImmediately(const WebString& originIdentifier, const WebString& databaseName)
112 {
113     HashSet<RefPtr<Database> > databaseHandles;
114     PassRefPtr<SecurityOrigin> originPrp(*WebSecurityOrigin::createFromDatabaseIdentifier(originIdentifier));
115     RefPtr<SecurityOrigin> origin = originPrp;
116     DatabaseTracker::tracker().getOpenDatabases(origin.get(), databaseName, &databaseHandles);
117     for (HashSet<RefPtr<Database> >::iterator it = databaseHandles.begin(); it != databaseHandles.end(); ++it) {
118         Database* database = it->get();
119         DatabaseThread* databaseThread = database->scriptExecutionContext()->databaseThread();
120         if (databaseThread && !databaseThread->terminationRequested()) {
121             database->stop();
122             databaseThread->scheduleTask(DatabaseCloseTask::create(database, 0));
123         }
124     }
125 }
126
127 WebDatabase::WebDatabase(const WTF::PassRefPtr<Database>& database)
128     : m_private(static_cast<WebDatabasePrivate*>(database.releaseRef()))
129 {
130 }
131
132 WebDatabase& WebDatabase::operator=(const WTF::PassRefPtr<Database>& database)
133 {
134     assign(static_cast<WebDatabasePrivate*>(database.releaseRef()));
135     return *this;
136 }
137
138 WebDatabase::operator WTF::PassRefPtr<Database>() const
139 {
140     return PassRefPtr<Database>(const_cast<WebDatabasePrivate*>(m_private));
141 }
142
143 void WebDatabase::assign(WebDatabasePrivate* d)
144 {
145     // d is already ref'd for us by the caller
146     if (m_private)
147         m_private->deref();
148     m_private = d;
149 }
150
151 } // namespace WebKit