2009-03-31 14:26:16 +00:00
|
|
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
2009-04-13 16:29:41 +00:00
|
|
|
* vim: sw=2 ts=2 et lcs=trail\:.,tab\:>~ :
|
2006-02-16 02:59:42 +00:00
|
|
|
* ***** BEGIN LICENSE BLOCK *****
|
2004-10-09 00:04:10 +00:00
|
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
|
|
*
|
|
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
|
|
* the License. You may obtain a copy of the License at
|
|
|
|
* http://www.mozilla.org/MPL/
|
|
|
|
*
|
|
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
|
|
* for the specific language governing rights and limitations under the
|
|
|
|
* License.
|
|
|
|
*
|
|
|
|
* The Original Code is Oracle Corporation code.
|
|
|
|
*
|
|
|
|
* The Initial Developer of the Original Code is
|
|
|
|
* Oracle Corporation
|
|
|
|
* Portions created by the Initial Developer are Copyright (C) 2004
|
|
|
|
* the Initial Developer. All Rights Reserved.
|
|
|
|
*
|
|
|
|
* Contributor(s):
|
|
|
|
* Vladimir Vukicevic <vladimir.vukicevic@oracle.com>
|
2006-02-16 02:59:42 +00:00
|
|
|
* Brett Wilson <brettw@gmail.com>
|
2007-06-12 23:31:59 +00:00
|
|
|
* Shawn Wilsher <me@shawnwilsher.com>
|
2009-07-15 17:49:05 +00:00
|
|
|
* Drew Willcoxon <adw@mozilla.com>
|
2004-10-09 00:04:10 +00:00
|
|
|
*
|
|
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
|
|
* the provisions above, a recipient may use your version of this file under
|
|
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
|
|
*
|
|
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
|
|
|
|
#include "mozStorageService.h"
|
|
|
|
#include "mozStorageConnection.h"
|
2008-03-20 01:37:04 +00:00
|
|
|
#include "prinit.h"
|
|
|
|
#include "nsAutoPtr.h"
|
2009-07-15 17:49:05 +00:00
|
|
|
#include "nsCollationCID.h"
|
2008-11-01 00:38:58 +00:00
|
|
|
#include "nsEmbedCID.h"
|
2010-07-24 18:21:17 +00:00
|
|
|
#include "nsThreadUtils.h"
|
2008-10-30 22:50:00 +00:00
|
|
|
#include "mozStoragePrivateHelpers.h"
|
2009-07-15 17:49:05 +00:00
|
|
|
#include "nsILocale.h"
|
|
|
|
#include "nsILocaleService.h"
|
2008-11-19 06:11:30 +00:00
|
|
|
#include "nsIXPConnect.h"
|
2009-01-21 21:52:16 +00:00
|
|
|
#include "nsIObserverService.h"
|
Bug 560095 - Use mozilla::services::GetObserverService(). r=biesi,dveditz,gavin,josh,jst,mrbkap,roc,sdwilsh,shaver,sicking,smontagu,surkov
2010-04-29 16:59:13 +00:00
|
|
|
#include "mozilla/Services.h"
|
2004-10-09 00:04:10 +00:00
|
|
|
|
|
|
|
#include "sqlite3.h"
|
|
|
|
|
2008-11-01 00:38:58 +00:00
|
|
|
#include "nsIPromptService.h"
|
2010-05-19 20:46:08 +00:00
|
|
|
#include "nsIMemoryReporter.h"
|
2008-11-01 00:38:58 +00:00
|
|
|
|
2010-05-19 23:22:19 +00:00
|
|
|
#include "mozilla/FunctionTimer.h"
|
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
namespace mozilla {
|
|
|
|
namespace storage {
|
2006-02-16 02:59:42 +00:00
|
|
|
|
2010-07-22 17:08:59 +00:00
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
//// Memory Reporting
|
|
|
|
|
|
|
|
static PRInt64
|
|
|
|
GetStorageSQLitePageCacheMemoryUsed(void *)
|
|
|
|
{
|
|
|
|
int current, high;
|
|
|
|
int rc = ::sqlite3_status(SQLITE_STATUS_PAGECACHE_OVERFLOW, ¤t, &high,
|
|
|
|
0);
|
|
|
|
return rc == SQLITE_OK ? current : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PRInt64
|
|
|
|
GetStorageSQLiteOtherMemoryUsed(void *)
|
|
|
|
{
|
|
|
|
int pageCacheCurrent, pageCacheHigh;
|
|
|
|
int rc = ::sqlite3_status(SQLITE_STATUS_PAGECACHE_OVERFLOW, &pageCacheCurrent,
|
|
|
|
&pageCacheHigh, 0);
|
|
|
|
return rc == SQLITE_OK ? ::sqlite3_memory_used() - pageCacheCurrent : 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_MEMORY_REPORTER_IMPLEMENT(StorageSQLitePageCacheMemoryUsed,
|
|
|
|
"storage/sqlite/pagecache",
|
|
|
|
"Memory in use by SQLite for the page cache",
|
|
|
|
GetStorageSQLitePageCacheMemoryUsed,
|
|
|
|
nsnull)
|
|
|
|
|
|
|
|
NS_MEMORY_REPORTER_IMPLEMENT(StorageSQLiteOtherMemoryUsed,
|
|
|
|
"storage/sqlite/other",
|
|
|
|
"Memory in use by SQLite for other various reasons",
|
|
|
|
GetStorageSQLiteOtherMemoryUsed,
|
|
|
|
nsnull)
|
|
|
|
|
2010-07-24 18:21:17 +00:00
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
//// Helpers
|
|
|
|
|
|
|
|
class ServiceMainThreadInitializer : public nsRunnable
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
ServiceMainThreadInitializer(nsIObserver *aObserver,
|
|
|
|
nsIXPConnect **aXPConnectPtr)
|
|
|
|
: mObserver(aObserver)
|
|
|
|
, mXPConnectPtr(aXPConnectPtr)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHOD Run()
|
|
|
|
{
|
|
|
|
NS_PRECONDITION(NS_IsMainThread(), "Must be running on the main thread!");
|
|
|
|
|
|
|
|
// NOTE: All code that can only run on the main thread and needs to be run
|
|
|
|
// during initialization should be placed here. During the off-
|
|
|
|
// chance that storage is initialized on a background thread, this
|
|
|
|
// will ensure everything that isn't threadsafe is initialized in
|
|
|
|
// the right place.
|
|
|
|
|
|
|
|
// Register for xpcom-shutdown so we can cleanup after ourselves. The
|
|
|
|
// observer service can only be used on the main thread.
|
|
|
|
nsCOMPtr<nsIObserverService> os =
|
|
|
|
mozilla::services::GetObserverService();
|
|
|
|
NS_ENSURE_TRUE(os, NS_ERROR_FAILURE);
|
|
|
|
nsresult rv = os->AddObserver(mObserver, "xpcom-shutdown", PR_FALSE);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// We cache XPConnect for our language helpers. XPConnect can only be
|
|
|
|
// used on the main thread.
|
|
|
|
(void)CallGetService(nsIXPConnect::GetCID(), mXPConnectPtr);
|
|
|
|
|
|
|
|
// Register our SQLite memory reporters. Registration can only happen on
|
|
|
|
// the main thread (otherwise you'll get cryptic crashes).
|
|
|
|
NS_RegisterMemoryReporter(new NS_MEMORY_REPORTER_NAME(StorageSQLitePageCacheMemoryUsed));
|
|
|
|
NS_RegisterMemoryReporter(new NS_MEMORY_REPORTER_NAME(StorageSQLiteOtherMemoryUsed));
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
nsIObserver *mObserver;
|
|
|
|
nsIXPConnect **mXPConnectPtr;
|
|
|
|
};
|
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
//// Service
|
2007-06-19 02:22:01 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_IMPL_THREADSAFE_ISUPPORTS2(
|
|
|
|
Service,
|
|
|
|
mozIStorageService,
|
|
|
|
nsIObserver
|
|
|
|
)
|
2008-09-25 18:28:29 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
Service *Service::gService = nsnull;
|
2008-11-01 00:38:58 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
Service *
|
|
|
|
Service::getSingleton()
|
|
|
|
{
|
|
|
|
if (gService) {
|
|
|
|
NS_ADDREF(gService);
|
|
|
|
return gService;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ensure that we are using the same version of SQLite that we compiled with
|
|
|
|
// or newer. Our configure check ensures we are using a new enough version
|
|
|
|
// at compile time.
|
|
|
|
if (SQLITE_VERSION_NUMBER > ::sqlite3_libversion_number()) {
|
|
|
|
nsCOMPtr<nsIPromptService> ps(do_GetService(NS_PROMPTSERVICE_CONTRACTID));
|
|
|
|
if (ps) {
|
|
|
|
nsAutoString title, message;
|
|
|
|
title.AppendASCII("SQLite Version Error");
|
|
|
|
message.AppendASCII("The application has been updated, but your version "
|
|
|
|
"of SQLite is too old and the application cannot "
|
|
|
|
"run.");
|
|
|
|
(void)ps->Alert(nsnull, title.get(), message.get());
|
2007-06-19 02:22:01 +00:00
|
|
|
}
|
2009-03-31 14:26:16 +00:00
|
|
|
::PR_Abort();
|
|
|
|
}
|
2008-09-25 18:28:29 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
gService = new Service();
|
|
|
|
if (gService) {
|
|
|
|
NS_ADDREF(gService);
|
|
|
|
if (NS_FAILED(gService->initialize()))
|
|
|
|
NS_RELEASE(gService);
|
|
|
|
}
|
|
|
|
|
|
|
|
return gService;
|
2004-10-09 00:04:10 +00:00
|
|
|
}
|
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
nsIXPConnect *Service::sXPConnect = nsnull;
|
|
|
|
|
2009-01-21 21:52:16 +00:00
|
|
|
already_AddRefed<nsIXPConnect>
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::getXPConnect()
|
2008-11-19 06:11:30 +00:00
|
|
|
{
|
2010-07-24 18:21:17 +00:00
|
|
|
NS_PRECONDITION(NS_IsMainThread(),
|
|
|
|
"Must only get XPConnect on the main thread!");
|
|
|
|
NS_PRECONDITION(gService,
|
|
|
|
"Can not get XPConnect without an instance of our service!");
|
2009-03-31 14:26:16 +00:00
|
|
|
|
|
|
|
// If we've been shutdown, sXPConnect will be null. To prevent leaks, we do
|
|
|
|
// not cache the service after this point.
|
|
|
|
nsCOMPtr<nsIXPConnect> xpc(sXPConnect);
|
|
|
|
if (!xpc)
|
|
|
|
xpc = do_GetService(nsIXPConnect::GetCID());
|
|
|
|
NS_ASSERTION(xpc, "Could not get XPConnect!");
|
|
|
|
return xpc.forget();
|
2008-11-19 06:11:30 +00:00
|
|
|
}
|
|
|
|
|
2009-07-15 17:49:05 +00:00
|
|
|
Service::Service()
|
|
|
|
: mMutex("Service::mMutex")
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::~Service()
|
2004-10-09 00:04:10 +00:00
|
|
|
{
|
2009-03-31 14:26:16 +00:00
|
|
|
// Shutdown the sqlite3 API. Warn if shutdown did not turn out okay, but
|
|
|
|
// there is nothing actionable we can do in that case.
|
|
|
|
int rc = ::sqlite3_shutdown();
|
|
|
|
if (rc != SQLITE_OK)
|
|
|
|
NS_WARNING("sqlite3 did not shutdown cleanly.");
|
|
|
|
|
2010-07-24 18:21:17 +00:00
|
|
|
bool shutdownObserved = !sXPConnect;
|
|
|
|
NS_ASSERTION(shutdownObserved, "Shutdown was not observed!");
|
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
gService = nsnull;
|
2009-01-21 21:52:16 +00:00
|
|
|
}
|
2008-11-19 06:11:30 +00:00
|
|
|
|
2009-01-21 21:52:16 +00:00
|
|
|
void
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::shutdown()
|
2009-01-21 21:52:16 +00:00
|
|
|
{
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_IF_RELEASE(sXPConnect);
|
2006-02-16 02:59:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
nsresult
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::initialize()
|
2006-02-16 02:59:42 +00:00
|
|
|
{
|
2010-05-19 23:22:19 +00:00
|
|
|
NS_TIME_FUNCTION;
|
|
|
|
|
2010-05-19 20:46:08 +00:00
|
|
|
int rc;
|
2009-03-31 14:26:16 +00:00
|
|
|
|
|
|
|
// Explicitly initialize sqlite3. Although this is implicitly called by
|
|
|
|
// various sqlite3 functions (and the sqlite3_open calls in our case),
|
|
|
|
// the documentation suggests calling this directly. So we do.
|
|
|
|
rc = ::sqlite3_initialize();
|
|
|
|
if (rc != SQLITE_OK)
|
2009-05-09 00:29:56 +00:00
|
|
|
return convertResultCode(rc);
|
2009-03-31 14:26:16 +00:00
|
|
|
|
|
|
|
// This makes multiple connections to the same database share the same pager
|
2009-07-15 17:49:05 +00:00
|
|
|
// cache. We do not need to lock here with mMutex because this function is
|
2009-03-31 14:26:16 +00:00
|
|
|
// only ever called from Service::GetSingleton, which will only
|
|
|
|
// call this function once, and will not return until this function returns.
|
|
|
|
// (It does not matter where this is called relative to sqlite3_initialize.)
|
|
|
|
rc = ::sqlite3_enable_shared_cache(1);
|
|
|
|
if (rc != SQLITE_OK)
|
2009-05-09 00:29:56 +00:00
|
|
|
return convertResultCode(rc);
|
2009-03-31 14:26:16 +00:00
|
|
|
|
2010-07-24 18:21:17 +00:00
|
|
|
// Run the things that need to run on the main thread there.
|
|
|
|
nsCOMPtr<nsIRunnable> event =
|
|
|
|
new ServiceMainThreadInitializer(this, &sXPConnect);
|
|
|
|
if (event && ::NS_IsMainThread()) {
|
|
|
|
(void)event->Run();
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
(void)::NS_DispatchToMainThread(event);
|
|
|
|
}
|
2009-03-31 14:26:16 +00:00
|
|
|
|
|
|
|
return NS_OK;
|
2004-10-09 00:04:10 +00:00
|
|
|
}
|
|
|
|
|
2009-07-15 17:49:05 +00:00
|
|
|
int
|
|
|
|
Service::localeCompareStrings(const nsAString &aStr1,
|
|
|
|
const nsAString &aStr2,
|
|
|
|
PRInt32 aComparisonStrength)
|
|
|
|
{
|
|
|
|
// The implementation of nsICollation.CompareString() is platform-dependent.
|
|
|
|
// On Linux it's not thread-safe. It may not be on Windows and OS X either,
|
|
|
|
// but it's more difficult to tell. We therefore synchronize this method.
|
|
|
|
MutexAutoLock mutex(mMutex);
|
|
|
|
|
|
|
|
nsICollation *coll = getLocaleCollation();
|
|
|
|
if (!coll) {
|
|
|
|
NS_ERROR("Storage service has no collation");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
PRInt32 res;
|
|
|
|
nsresult rv = coll->CompareString(aComparisonStrength, aStr1, aStr2, &res);
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
NS_ERROR("Collation compare string failed");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
nsICollation *
|
|
|
|
Service::getLocaleCollation()
|
|
|
|
{
|
|
|
|
mMutex.AssertCurrentThreadOwns();
|
|
|
|
|
|
|
|
if (mLocaleCollation)
|
|
|
|
return mLocaleCollation;
|
|
|
|
|
|
|
|
nsCOMPtr<nsILocaleService> svc(do_GetService(NS_LOCALESERVICE_CONTRACTID));
|
|
|
|
if (!svc) {
|
|
|
|
NS_WARNING("Could not get locale service");
|
|
|
|
return nsnull;
|
|
|
|
}
|
|
|
|
|
|
|
|
nsCOMPtr<nsILocale> appLocale;
|
|
|
|
nsresult rv = svc->GetApplicationLocale(getter_AddRefs(appLocale));
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
NS_WARNING("Could not get application locale");
|
|
|
|
return nsnull;
|
|
|
|
}
|
|
|
|
|
|
|
|
nsCOMPtr<nsICollationFactory> collFact =
|
|
|
|
do_CreateInstance(NS_COLLATIONFACTORY_CONTRACTID);
|
|
|
|
if (!collFact) {
|
|
|
|
NS_WARNING("Could not create collation factory");
|
|
|
|
return nsnull;
|
|
|
|
}
|
|
|
|
|
|
|
|
rv = collFact->CreateCollation(appLocale, getter_AddRefs(mLocaleCollation));
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
NS_WARNING("Could not create collation");
|
|
|
|
return nsnull;
|
|
|
|
}
|
|
|
|
|
|
|
|
return mLocaleCollation;
|
|
|
|
}
|
|
|
|
|
2009-01-21 21:52:16 +00:00
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
//// mozIStorageService
|
|
|
|
|
2004-10-09 00:04:10 +00:00
|
|
|
#ifndef NS_APP_STORAGE_50_FILE
|
|
|
|
#define NS_APP_STORAGE_50_FILE "UStor"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::OpenSpecialDatabase(const char *aStorageKey,
|
|
|
|
mozIStorageConnection **_connection)
|
2004-10-09 00:04:10 +00:00
|
|
|
{
|
2009-03-31 14:26:16 +00:00
|
|
|
nsresult rv;
|
2004-10-09 00:04:10 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
nsCOMPtr<nsIFile> storageFile;
|
|
|
|
if (::strcmp(aStorageKey, "memory") == 0) {
|
|
|
|
// just fall through with NULL storageFile, this will cause the storage
|
|
|
|
// connection to use a memory DB.
|
|
|
|
}
|
|
|
|
else if (::strcmp(aStorageKey, "profile") == 0) {
|
2007-06-12 23:31:59 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
rv = NS_GetSpecialDirectory(NS_APP_STORAGE_50_FILE,
|
|
|
|
getter_AddRefs(storageFile));
|
2007-06-12 23:31:59 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2004-10-09 00:04:10 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
nsString filename;
|
|
|
|
storageFile->GetPath(filename);
|
|
|
|
nsCString filename8 = NS_ConvertUTF16toUTF8(filename.get());
|
|
|
|
// fall through to DB initialization
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
}
|
|
|
|
|
2009-04-17 21:19:31 +00:00
|
|
|
Connection *msc = new Connection(this);
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ENSURE_TRUE(msc, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
2009-04-17 21:19:31 +00:00
|
|
|
rv = msc->initialize(storageFile);
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
NS_ADDREF(*_connection = msc);
|
|
|
|
return NS_OK;
|
2004-10-09 00:04:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::OpenDatabase(nsIFile *aDatabaseFile,
|
|
|
|
mozIStorageConnection **_connection)
|
2004-10-09 00:04:10 +00:00
|
|
|
{
|
2010-05-19 23:22:19 +00:00
|
|
|
#ifdef NS_FUNCTION_TIMER
|
|
|
|
nsCString leafname;
|
|
|
|
(void)aDatabaseFile->GetNativeLeafName(leafname);
|
|
|
|
NS_TIME_FUNCTION_FMT("mozIStorageService::OpenDatabase(%s)", leafname.get());
|
|
|
|
#endif
|
|
|
|
|
2009-04-17 21:19:31 +00:00
|
|
|
nsRefPtr<Connection> msc = new Connection(this);
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ENSURE_TRUE(msc, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
|
|
|
{
|
2009-07-15 17:49:05 +00:00
|
|
|
MutexAutoLock mutex(mMutex);
|
2009-04-17 21:19:31 +00:00
|
|
|
nsresult rv = msc->initialize(aDatabaseFile);
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
}
|
2008-02-09 19:05:49 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ADDREF(*_connection = msc);
|
|
|
|
return NS_OK;
|
2004-10-09 00:04:10 +00:00
|
|
|
}
|
2008-01-29 23:34:19 +00:00
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::OpenUnsharedDatabase(nsIFile *aDatabaseFile,
|
|
|
|
mozIStorageConnection **_connection)
|
2008-01-29 23:34:19 +00:00
|
|
|
{
|
2010-05-19 23:22:19 +00:00
|
|
|
#ifdef NS_FUNCTION_TIMER
|
|
|
|
nsCString leafname;
|
|
|
|
(void)aDatabaseFile->GetNativeLeafName(leafname);
|
|
|
|
NS_TIME_FUNCTION_FMT("mozIStorageService::OpenUnsharedDatabase(%s)",
|
|
|
|
leafname.get());
|
|
|
|
#endif
|
|
|
|
|
2009-04-17 21:19:31 +00:00
|
|
|
nsRefPtr<Connection> msc = new Connection(this);
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ENSURE_TRUE(msc, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
|
|
|
// Initialize the connection, temporarily turning off shared caches so the
|
|
|
|
// new connection gets its own cache. Database connections are assigned
|
|
|
|
// caches when they are opened, and they retain those caches for their
|
|
|
|
// lifetimes, unaffected by changes to the shared caches setting, so we can
|
|
|
|
// disable shared caches temporarily while we initialize the new connection
|
|
|
|
// without affecting the caches currently in use by other connections.
|
|
|
|
nsresult rv;
|
|
|
|
{
|
2009-07-15 17:49:05 +00:00
|
|
|
MutexAutoLock mutex(mMutex);
|
2009-03-31 14:26:16 +00:00
|
|
|
int rc = ::sqlite3_enable_shared_cache(0);
|
|
|
|
if (rc != SQLITE_OK)
|
2009-05-09 00:29:56 +00:00
|
|
|
return convertResultCode(rc);
|
2008-02-09 19:05:49 +00:00
|
|
|
|
2009-04-17 21:19:31 +00:00
|
|
|
rv = msc->initialize(aDatabaseFile);
|
2009-03-31 14:26:16 +00:00
|
|
|
|
|
|
|
rc = ::sqlite3_enable_shared_cache(1);
|
|
|
|
if (rc != SQLITE_OK)
|
2009-05-09 00:29:56 +00:00
|
|
|
return convertResultCode(rc);
|
2009-03-31 14:26:16 +00:00
|
|
|
}
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
NS_ADDREF(*_connection = msc);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
2008-03-24 22:14:38 +00:00
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::BackupDatabaseFile(nsIFile *aDBFile,
|
|
|
|
const nsAString &aBackupFileName,
|
|
|
|
nsIFile *aBackupParentDirectory,
|
|
|
|
nsIFile **backup)
|
2008-03-24 22:14:38 +00:00
|
|
|
{
|
2009-03-31 14:26:16 +00:00
|
|
|
nsresult rv;
|
|
|
|
nsCOMPtr<nsIFile> parentDir = aBackupParentDirectory;
|
|
|
|
if (!parentDir) {
|
|
|
|
// This argument is optional, and defaults to the same parent directory
|
|
|
|
// as the current file.
|
|
|
|
rv = aDBFile->GetParent(getter_AddRefs(parentDir));
|
2008-03-24 22:14:38 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2009-03-31 14:26:16 +00:00
|
|
|
}
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
nsCOMPtr<nsIFile> backupDB;
|
|
|
|
rv = parentDir->Clone(getter_AddRefs(backupDB));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
rv = backupDB->Append(aBackupFileName);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
rv = backupDB->CreateUnique(nsIFile::NORMAL_FILE_TYPE, 0600);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
nsAutoString fileName;
|
|
|
|
rv = backupDB->GetLeafName(fileName);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
rv = backupDB->Remove(PR_FALSE);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2008-03-24 22:14:38 +00:00
|
|
|
|
2009-03-31 14:26:16 +00:00
|
|
|
backupDB.forget(backup);
|
|
|
|
|
|
|
|
return aDBFile->CopyTo(parentDir, fileName);
|
2008-03-24 22:14:38 +00:00
|
|
|
}
|
|
|
|
|
2009-01-21 21:52:16 +00:00
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
//// nsIObserver
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2009-03-31 14:26:16 +00:00
|
|
|
Service::Observe(nsISupports *, const char *aTopic, const PRUnichar *)
|
2009-01-21 21:52:16 +00:00
|
|
|
{
|
2009-03-31 14:26:16 +00:00
|
|
|
if (strcmp(aTopic, "xpcom-shutdown") == 0)
|
|
|
|
shutdown();
|
|
|
|
return NS_OK;
|
2009-01-21 21:52:16 +00:00
|
|
|
}
|
2009-03-31 14:26:16 +00:00
|
|
|
|
|
|
|
} // namespace storage
|
|
|
|
} // namespace mozilla
|