WebDatabaseManager.cpp [plain text]
#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&);
virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject);
virtual ULONG STDMETHODCALLTYPE AddRef();
virtual ULONG STDMETHODCALLTYPE Release();
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::createInstance(const DatabaseDetails& details)
{
DatabaseDetailsPropertyBag* instance = new DatabaseDetailsPropertyBag(details);
instance->AddRef();
return instance;
}
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;
}
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::createInstance()
{
WebDatabaseManager* manager = new WebDatabaseManager();
manager->AddRef();
return manager;
}
WebDatabaseManager::WebDatabaseManager()
: m_refCount(0)
{
gClassCount++;
gClassNameCount.add("WebDatabaseManager");
}
WebDatabaseManager::~WebDatabaseManager()
{
gClassCount--;
gClassNameCount.remove("WebDatabaseManager");
}
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> > {};
HRESULT STDMETHODCALLTYPE WebDatabaseManager::sharedWebDatabaseManager(
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(
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(
IWebSecurityOrigin* origin,
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(
BSTR databaseName,
IWebSecurityOrigin* origin,
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(
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(
BSTR databaseName,
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(
BSTR origin,
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