WebKit/win/WebDatabaseManager.cpp
changeset 0 4f2f89ce4247
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebKit/win/WebDatabaseManager.cpp	Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,385 @@
+/*
+ * Copyright (C) 2007, 2008 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1.  Redistributions of source code must retain the above copyright
+ *     notice, this list of conditions and the following disclaimer.
+ * 2.  Redistributions in binary form must reproduce the above copyright
+ *     notice, this list of conditions and the following disclaimer in the
+ *     documentation and/or other materials provided with the distribution.
+ * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
+ *     its contributors may be used to endorse or promote products derived
+ *     from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "config.h"
+#include "WebDatabaseManager.h"
+#include "WebKitDLL.h"
+
+#if ENABLE(DATABASE)
+
+#include "CFDictionaryPropertyBag.h"
+#include "COMEnumVariant.h"
+#include "MarshallingHelpers.h"
+#include "WebNotificationCenter.h"
+#include "WebSecurityOrigin.h"
+
+#include <WebCore/BString.h>
+#include <WebCore/COMPtr.h>
+#include <WebCore/DatabaseTracker.h>
+#include <WebCore/FileSystem.h>
+#include <WebCore/SecurityOrigin.h>
+
+using namespace WebCore;
+
+static inline bool isEqual(LPCWSTR s1, LPCWSTR s2)
+{
+    return !wcscmp(s1, s2);
+}
+
+class DatabaseDetailsPropertyBag : public IPropertyBag, public Noncopyable {
+public:
+    static DatabaseDetailsPropertyBag* createInstance(const DatabaseDetails&);
+
+    // IUnknown
+    virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject);
+    virtual ULONG STDMETHODCALLTYPE AddRef();
+    virtual ULONG STDMETHODCALLTYPE Release();
+
+    // IPropertyBag
+    virtual HRESULT STDMETHODCALLTYPE Read(LPCOLESTR pszPropName, VARIANT* pVar, IErrorLog* pErrorLog);
+    virtual HRESULT STDMETHODCALLTYPE Write(LPCOLESTR pszPropName, VARIANT* pVar);
+private:
+    DatabaseDetailsPropertyBag(const DatabaseDetails& details) 
+        : m_refCount(0)
+        , m_details(details) { }
+    ~DatabaseDetailsPropertyBag() { }
+
+    ULONG m_refCount;
+    DatabaseDetails m_details;
+};
+
+// DatabaseDetailsPropertyBag ------------------------------------------------------
+DatabaseDetailsPropertyBag* DatabaseDetailsPropertyBag::createInstance(const DatabaseDetails& details)
+{
+    DatabaseDetailsPropertyBag* instance = new DatabaseDetailsPropertyBag(details);
+    instance->AddRef();
+    return instance;
+}
+
+// IUnknown ------------------------------------------------------------------------
+ULONG STDMETHODCALLTYPE DatabaseDetailsPropertyBag::AddRef()
+{
+    return ++m_refCount;
+}
+
+ULONG STDMETHODCALLTYPE DatabaseDetailsPropertyBag::Release()
+{
+    ULONG newRef = --m_refCount;
+    if (!newRef)
+        delete this;
+
+    return newRef;
+}
+
+HRESULT STDMETHODCALLTYPE DatabaseDetailsPropertyBag::QueryInterface(REFIID riid, void** ppvObject)
+{
+    *ppvObject = 0;
+    if (IsEqualGUID(riid, IID_IUnknown))
+        *ppvObject = static_cast<DatabaseDetailsPropertyBag*>(this);
+    else if (IsEqualGUID(riid, IID_IPropertyBag))
+        *ppvObject = static_cast<DatabaseDetailsPropertyBag*>(this);
+    else
+        return E_NOINTERFACE;
+
+    AddRef();
+    return S_OK;
+}
+
+// IPropertyBag --------------------------------------------------------------------
+HRESULT STDMETHODCALLTYPE DatabaseDetailsPropertyBag::Read(LPCOLESTR pszPropName, VARIANT* pVar, IErrorLog*)
+{
+    if (!pszPropName || !pVar)
+        return E_POINTER;
+
+    VariantInit(pVar);
+
+    if (isEqual(pszPropName, WebDatabaseDisplayNameKey)) {
+        COMVariantSetter<String>::setVariant(pVar, m_details.displayName());
+        return S_OK;
+    } else if (isEqual(pszPropName, WebDatabaseExpectedSizeKey)) {
+        COMVariantSetter<unsigned long long>::setVariant(pVar, m_details.expectedUsage());
+        return S_OK;
+    } else if (isEqual(pszPropName, WebDatabaseUsageKey)) {
+        COMVariantSetter<unsigned long long>::setVariant(pVar, m_details.currentUsage());
+        return S_OK;
+    }
+
+    return E_INVALIDARG;
+}
+
+HRESULT STDMETHODCALLTYPE DatabaseDetailsPropertyBag::Write(LPCOLESTR pszPropName, VARIANT* pVar)
+{
+    if (!pszPropName || !pVar)
+        return E_POINTER;
+
+    return E_FAIL;
+}
+
+static COMPtr<WebDatabaseManager> s_sharedWebDatabaseManager;
+
+// WebDatabaseManager --------------------------------------------------------------
+WebDatabaseManager* WebDatabaseManager::createInstance()
+{
+    WebDatabaseManager* manager = new WebDatabaseManager();
+    manager->AddRef();
+    return manager;    
+}
+
+WebDatabaseManager::WebDatabaseManager()
+    : m_refCount(0)
+{
+    gClassCount++;
+    gClassNameCount.add("WebDatabaseManager");
+}
+
+WebDatabaseManager::~WebDatabaseManager()
+{
+    gClassCount--;
+    gClassNameCount.remove("WebDatabaseManager");
+}
+
+// IUnknown ------------------------------------------------------------------------
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::QueryInterface(REFIID riid, void** ppvObject)
+{
+    *ppvObject = 0;
+    if (IsEqualGUID(riid, IID_IUnknown))
+        *ppvObject = static_cast<WebDatabaseManager*>(this);
+    else if (IsEqualGUID(riid, IID_IWebDatabaseManager))
+        *ppvObject = static_cast<WebDatabaseManager*>(this);
+    else
+        return E_NOINTERFACE;
+
+    AddRef();
+    return S_OK;
+}
+
+ULONG STDMETHODCALLTYPE WebDatabaseManager::AddRef()
+{
+    return ++m_refCount;
+}
+
+ULONG STDMETHODCALLTYPE WebDatabaseManager::Release()
+{
+    ULONG newRef = --m_refCount;
+    if (!newRef)
+        delete this;
+
+    return newRef;
+}
+
+template<> struct COMVariantSetter<RefPtr<SecurityOrigin> > : COMIUnknownVariantSetter<WebSecurityOrigin, RefPtr<SecurityOrigin> > {};
+
+// IWebDatabaseManager -------------------------------------------------------------
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::sharedWebDatabaseManager( 
+    /* [retval][out] */ IWebDatabaseManager** result)
+{
+    if (!s_sharedWebDatabaseManager) {
+        s_sharedWebDatabaseManager.adoptRef(WebDatabaseManager::createInstance());
+        DatabaseTracker::tracker().setClient(s_sharedWebDatabaseManager.get());
+    }
+
+    return s_sharedWebDatabaseManager.copyRefTo(result);
+}
+
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::origins( 
+    /* [retval][out] */ IEnumVARIANT** result)
+{
+    if (!result)
+        return E_POINTER;
+
+    *result = 0;
+
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    Vector<RefPtr<SecurityOrigin> > origins;
+    DatabaseTracker::tracker().origins(origins);
+        COMPtr<COMEnumVariant<Vector<RefPtr<SecurityOrigin> > > > enumVariant(AdoptCOM, COMEnumVariant<Vector<RefPtr<SecurityOrigin> > >::adopt(origins));
+
+    *result = enumVariant.releaseRef();
+    return S_OK;
+}
+    
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::databasesWithOrigin( 
+    /* [in] */ IWebSecurityOrigin* origin,
+    /* [retval][out] */ IEnumVARIANT** result)
+{
+    if (!origin || !result)
+        return E_POINTER;
+
+    *result = 0;
+
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    COMPtr<WebSecurityOrigin> webSecurityOrigin(Query, origin);
+    if (!webSecurityOrigin)
+        return E_FAIL;
+
+    Vector<String> databaseNames;
+    DatabaseTracker::tracker().databaseNamesForOrigin(webSecurityOrigin->securityOrigin(), databaseNames);
+
+    COMPtr<COMEnumVariant<Vector<String> > > enumVariant(AdoptCOM, COMEnumVariant<Vector<String> >::adopt(databaseNames));
+
+    *result = enumVariant.releaseRef();
+    return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::detailsForDatabase( 
+    /* [in] */ BSTR databaseName,
+    /* [in] */ IWebSecurityOrigin* origin,
+    /* [retval][out] */ IPropertyBag** result)
+{
+    if (!origin || !result)
+        return E_POINTER;
+
+    *result = 0;
+
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    COMPtr<WebSecurityOrigin> webSecurityOrigin(Query, origin);
+    if (!webSecurityOrigin)
+        return E_FAIL;
+
+    DatabaseDetails details = DatabaseTracker::tracker().detailsForNameAndOrigin(String(databaseName, SysStringLen(databaseName)),
+        webSecurityOrigin->securityOrigin());
+
+    if (details.name().isNull())
+        return E_INVALIDARG;
+
+    *result = DatabaseDetailsPropertyBag::createInstance(details);
+    return S_OK;
+}
+    
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::deleteAllDatabases()
+{
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    DatabaseTracker::tracker().deleteAllDatabases();
+
+    return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::deleteOrigin( 
+    /* [in] */ IWebSecurityOrigin* origin)
+{
+    if (!origin)
+        return E_POINTER;
+
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    COMPtr<WebSecurityOrigin> webSecurityOrigin(Query, origin);
+    if (!webSecurityOrigin)
+        return E_FAIL;
+
+    DatabaseTracker::tracker().deleteOrigin(webSecurityOrigin->securityOrigin());
+
+    return S_OK;
+}
+    
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::deleteDatabase( 
+    /* [in] */ BSTR databaseName,
+    /* [in] */ IWebSecurityOrigin* origin)
+{
+    if (!origin)
+        return E_POINTER;
+
+    if (!databaseName)
+        return E_INVALIDARG;
+
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    COMPtr<WebSecurityOrigin> webSecurityOrigin(Query, origin);
+    if (!webSecurityOrigin)
+        return E_FAIL;
+
+    DatabaseTracker::tracker().deleteDatabase(webSecurityOrigin->securityOrigin(), String(databaseName, SysStringLen(databaseName)));
+
+    return S_OK;
+}
+
+void WebDatabaseManager::dispatchDidModifyOrigin(SecurityOrigin* origin)
+{
+    static BSTR databaseDidModifyOriginName = SysAllocString(WebDatabaseDidModifyOriginNotification);
+    IWebNotificationCenter* notifyCenter = WebNotificationCenter::defaultCenterInternal();
+
+    COMPtr<WebSecurityOrigin> securityOrigin(AdoptCOM, WebSecurityOrigin::createInstance(origin));
+    notifyCenter->postNotificationName(databaseDidModifyOriginName, securityOrigin.get(), 0);
+}
+
+HRESULT STDMETHODCALLTYPE WebDatabaseManager::setQuota(
+    /* [in] */ BSTR origin,
+    /* [in] */ unsigned long long quota)
+{
+    if (!origin)
+        return E_POINTER;
+
+    if (this != s_sharedWebDatabaseManager)
+        return E_FAIL;
+
+    DatabaseTracker::tracker().setQuota(SecurityOrigin::createFromString(origin).get(), quota);
+
+    return S_OK;
+}
+
+void WebDatabaseManager::dispatchDidModifyDatabase(SecurityOrigin* origin, const String& databaseName)
+{
+    static BSTR databaseDidModifyOriginName = SysAllocString(WebDatabaseDidModifyDatabaseNotification);
+    IWebNotificationCenter* notifyCenter = WebNotificationCenter::defaultCenterInternal();
+
+    COMPtr<WebSecurityOrigin> securityOrigin(AdoptCOM, WebSecurityOrigin::createInstance(origin));
+
+    RetainPtr<CFMutableDictionaryRef> userInfo(AdoptCF, CFDictionaryCreateMutable(0, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks));
+
+    static CFStringRef databaseNameKey = MarshallingHelpers::LPCOLESTRToCFStringRef(WebDatabaseNameKey);
+    RetainPtr<CFStringRef> str(AdoptCF, databaseName.createCFString());
+    CFDictionarySetValue(userInfo.get(), databaseNameKey, str.get());
+
+    COMPtr<CFDictionaryPropertyBag> userInfoBag = CFDictionaryPropertyBag::createInstance();
+    userInfoBag->setDictionary(userInfo.get());
+
+    notifyCenter->postNotificationName(databaseDidModifyOriginName, securityOrigin.get(), userInfoBag.get());
+}
+
+void WebKitInitializeWebDatabasesIfNecessary()
+{
+    static bool initialized = false;
+    if (initialized)
+        return;
+
+    WebCore::String databasesDirectory = WebCore::pathByAppendingComponent(WebCore::localUserSpecificStorageDirectory(), "Databases");
+    WebCore::DatabaseTracker::initializeTracker(databasesDirectory);
+
+    initialized = true;
+}
+
+#endif