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
reset()56 void WebDatabase::reset()
57 {
58 assign(0);
59 }
60
assign(const WebDatabase & other)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
name() const69 WebString WebDatabase::name() const
70 {
71 ASSERT(m_private);
72 return m_private->stringIdentifier();
73 }
74
displayName() const75 WebString WebDatabase::displayName() const
76 {
77 ASSERT(m_private);
78 return m_private->displayName();
79 }
80
estimatedSize() const81 unsigned long WebDatabase::estimatedSize() const
82 {
83 ASSERT(m_private);
84 return m_private->estimatedSize();
85 }
86
securityOrigin() const87 WebSecurityOrigin WebDatabase::securityOrigin() const
88 {
89 ASSERT(m_private);
90 return WebSecurityOrigin(m_private->securityOrigin());
91 }
92
setObserver(WebDatabaseObserver * observer)93 void WebDatabase::setObserver(WebDatabaseObserver* observer)
94 {
95 databaseObserver = observer;
96 }
97
observer()98 WebDatabaseObserver* WebDatabase::observer()
99 {
100 return databaseObserver;
101 }
102
updateDatabaseSize(const WebString & originIdentifier,const WebString & databaseName,unsigned long long databaseSize,unsigned long long spaceAvailable)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
closeDatabaseImmediately(const WebString & originIdentifier,const WebString & databaseName)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
WebDatabase(const WTF::PassRefPtr<Database> & database)127 WebDatabase::WebDatabase(const WTF::PassRefPtr<Database>& database)
128 : m_private(static_cast<WebDatabasePrivate*>(database.releaseRef()))
129 {
130 }
131
operator =(const WTF::PassRefPtr<Database> & database)132 WebDatabase& WebDatabase::operator=(const WTF::PassRefPtr<Database>& database)
133 {
134 assign(static_cast<WebDatabasePrivate*>(database.releaseRef()));
135 return *this;
136 }
137
operator WTF::PassRefPtr<Database>() const138 WebDatabase::operator WTF::PassRefPtr<Database>() const
139 {
140 return PassRefPtr<Database>(const_cast<WebDatabasePrivate*>(m_private));
141 }
142
assign(WebDatabasePrivate * d)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
152