mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-28 19:38:13 +00:00
4893263a15
--HG-- extra : transplant_source : %95%B5y%11%C5%AB%D9%CF%FF%7C%5E%BB%E5%DA%F5j%83o%29%EB
1066 lines
29 KiB
C++
1066 lines
29 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "base/basictypes.h"
|
|
|
|
#include "IDBCursor.h"
|
|
|
|
#include "mozilla/storage.h"
|
|
#include "nsComponentManagerUtils.h"
|
|
#include "nsContentUtils.h"
|
|
#include "nsDOMClassInfoID.h"
|
|
#include "nsEventDispatcher.h"
|
|
#include "nsJSUtils.h"
|
|
#include "nsThreadUtils.h"
|
|
|
|
#include "AsyncConnectionHelper.h"
|
|
#include "IDBEvents.h"
|
|
#include "IDBIndex.h"
|
|
#include "IDBObjectStore.h"
|
|
#include "IDBTransaction.h"
|
|
#include "TransactionThreadPool.h"
|
|
|
|
#include "ipc/IndexedDBChild.h"
|
|
#include "ipc/IndexedDBParent.h"
|
|
|
|
#include "IndexedDatabaseInlines.h"
|
|
|
|
USING_INDEXEDDB_NAMESPACE
|
|
using namespace mozilla::dom::indexedDB::ipc;
|
|
|
|
MOZ_STATIC_ASSERT(sizeof(size_t) >= sizeof(IDBCursor::Direction),
|
|
"Relying on conversion between size_t and "
|
|
"IDBCursor::Direction");
|
|
|
|
namespace {
|
|
|
|
class CursorHelper : public AsyncConnectionHelper
|
|
{
|
|
public:
|
|
CursorHelper(IDBCursor* aCursor)
|
|
: AsyncConnectionHelper(aCursor->Transaction(), aCursor->Request()),
|
|
mCursor(aCursor), mActor(nullptr)
|
|
{
|
|
NS_ASSERTION(aCursor, "Null cursor!");
|
|
}
|
|
|
|
virtual void ReleaseMainThreadObjects() MOZ_OVERRIDE;
|
|
|
|
virtual nsresult Dispatch(nsIEventTarget* aDatabaseThread) MOZ_OVERRIDE;
|
|
|
|
virtual nsresult
|
|
PackArgumentsForParentProcess(CursorRequestParams& aParams) = 0;
|
|
|
|
virtual nsresult
|
|
UnpackResponseFromParentProcess(const ResponseValue& aResponseValue) = 0;
|
|
|
|
protected:
|
|
nsRefPtr<IDBCursor> mCursor;
|
|
|
|
private:
|
|
IndexedDBCursorRequestChild* mActor;
|
|
};
|
|
|
|
} // anonymous namespace
|
|
|
|
BEGIN_INDEXEDDB_NAMESPACE
|
|
|
|
class ContinueHelper : public CursorHelper
|
|
{
|
|
public:
|
|
ContinueHelper(IDBCursor* aCursor,
|
|
PRInt32 aCount)
|
|
: CursorHelper(aCursor), mCount(aCount)
|
|
{
|
|
NS_ASSERTION(aCount > 0, "Must have a count!");
|
|
}
|
|
|
|
~ContinueHelper()
|
|
{
|
|
IDBObjectStore::ClearStructuredCloneBuffer(mCloneReadInfo.mCloneBuffer);
|
|
}
|
|
|
|
virtual nsresult DoDatabaseWork(mozIStorageConnection* aConnection)
|
|
MOZ_OVERRIDE;
|
|
|
|
virtual nsresult GetSuccessResult(JSContext* aCx,
|
|
jsval* aVal) MOZ_OVERRIDE;
|
|
|
|
virtual nsresult
|
|
PackArgumentsForParentProcess(CursorRequestParams& aParams) MOZ_OVERRIDE;
|
|
|
|
virtual ChildProcessSendResult
|
|
MaybeSendResponseToChildProcess(nsresult aResultCode) MOZ_OVERRIDE;
|
|
|
|
virtual nsresult
|
|
UnpackResponseFromParentProcess(const ResponseValue& aResponseValue)
|
|
MOZ_OVERRIDE;
|
|
|
|
protected:
|
|
virtual nsresult
|
|
BindArgumentsToStatement(mozIStorageStatement* aStatement) = 0;
|
|
|
|
virtual nsresult
|
|
GatherResultsFromStatement(mozIStorageStatement* aStatement) = 0;
|
|
|
|
void UpdateCursorState()
|
|
{
|
|
mCursor->mCachedKey = JSVAL_VOID;
|
|
mCursor->mCachedPrimaryKey = JSVAL_VOID;
|
|
mCursor->mCachedValue = JSVAL_VOID;
|
|
mCursor->mHaveCachedKey = false;
|
|
mCursor->mHaveCachedPrimaryKey = false;
|
|
mCursor->mHaveCachedValue = false;
|
|
mCursor->mContinueCalled = false;
|
|
|
|
if (mKey.IsUnset()) {
|
|
mCursor->mHaveValue = false;
|
|
}
|
|
else {
|
|
NS_ASSERTION(mCursor->mType == IDBCursor::OBJECTSTORE ||
|
|
!mObjectKey.IsUnset(), "Bad key!");
|
|
|
|
// Set new values.
|
|
mCursor->mKey = mKey;
|
|
mCursor->mObjectKey = mObjectKey;
|
|
mCursor->mContinueToKey.Unset();
|
|
|
|
mCursor->mCloneReadInfo.Swap(mCloneReadInfo);
|
|
mCloneReadInfo.mCloneBuffer.clear();
|
|
}
|
|
}
|
|
|
|
PRInt32 mCount;
|
|
Key mKey;
|
|
Key mObjectKey;
|
|
StructuredCloneReadInfo mCloneReadInfo;
|
|
};
|
|
|
|
class ContinueObjectStoreHelper : public ContinueHelper
|
|
{
|
|
public:
|
|
ContinueObjectStoreHelper(IDBCursor* aCursor,
|
|
PRUint32 aCount)
|
|
: ContinueHelper(aCursor, aCount)
|
|
{ }
|
|
|
|
private:
|
|
nsresult BindArgumentsToStatement(mozIStorageStatement* aStatement);
|
|
nsresult GatherResultsFromStatement(mozIStorageStatement* aStatement);
|
|
};
|
|
|
|
class ContinueIndexHelper : public ContinueHelper
|
|
{
|
|
public:
|
|
ContinueIndexHelper(IDBCursor* aCursor,
|
|
PRUint32 aCount)
|
|
: ContinueHelper(aCursor, aCount)
|
|
{ }
|
|
|
|
private:
|
|
nsresult BindArgumentsToStatement(mozIStorageStatement* aStatement);
|
|
nsresult GatherResultsFromStatement(mozIStorageStatement* aStatement);
|
|
};
|
|
|
|
class ContinueIndexObjectHelper : public ContinueIndexHelper
|
|
{
|
|
public:
|
|
ContinueIndexObjectHelper(IDBCursor* aCursor,
|
|
PRUint32 aCount)
|
|
: ContinueIndexHelper(aCursor, aCount)
|
|
{ }
|
|
|
|
private:
|
|
nsresult GatherResultsFromStatement(mozIStorageStatement* aStatement);
|
|
};
|
|
|
|
END_INDEXEDDB_NAMESPACE
|
|
|
|
// static
|
|
already_AddRefed<IDBCursor>
|
|
IDBCursor::Create(IDBRequest* aRequest,
|
|
IDBTransaction* aTransaction,
|
|
IDBObjectStore* aObjectStore,
|
|
Direction aDirection,
|
|
const Key& aRangeKey,
|
|
const nsACString& aContinueQuery,
|
|
const nsACString& aContinueToQuery,
|
|
const Key& aKey,
|
|
StructuredCloneReadInfo& aCloneReadInfo)
|
|
{
|
|
NS_ASSERTION(aObjectStore, "Null pointer!");
|
|
NS_ASSERTION(!aKey.IsUnset(), "Bad key!");
|
|
|
|
nsRefPtr<IDBCursor> cursor =
|
|
IDBCursor::CreateCommon(aRequest, aTransaction, aObjectStore, aDirection,
|
|
aRangeKey, aContinueQuery, aContinueToQuery);
|
|
NS_ASSERTION(cursor, "This shouldn't fail!");
|
|
|
|
cursor->mObjectStore = aObjectStore;
|
|
cursor->mType = OBJECTSTORE;
|
|
cursor->mKey = aKey;
|
|
cursor->mCloneReadInfo.Swap(aCloneReadInfo);
|
|
|
|
return cursor.forget();
|
|
}
|
|
|
|
// static
|
|
already_AddRefed<IDBCursor>
|
|
IDBCursor::Create(IDBRequest* aRequest,
|
|
IDBTransaction* aTransaction,
|
|
IDBIndex* aIndex,
|
|
Direction aDirection,
|
|
const Key& aRangeKey,
|
|
const nsACString& aContinueQuery,
|
|
const nsACString& aContinueToQuery,
|
|
const Key& aKey,
|
|
const Key& aObjectKey)
|
|
{
|
|
NS_ASSERTION(aIndex, "Null pointer!");
|
|
NS_ASSERTION(!aKey.IsUnset(), "Bad key!");
|
|
NS_ASSERTION(!aObjectKey.IsUnset(), "Bad key!");
|
|
|
|
nsRefPtr<IDBCursor> cursor =
|
|
IDBCursor::CreateCommon(aRequest, aTransaction, aIndex->ObjectStore(),
|
|
aDirection, aRangeKey, aContinueQuery,
|
|
aContinueToQuery);
|
|
NS_ASSERTION(cursor, "This shouldn't fail!");
|
|
|
|
cursor->mIndex = aIndex;
|
|
cursor->mType = INDEXKEY;
|
|
cursor->mKey = aKey,
|
|
cursor->mObjectKey = aObjectKey;
|
|
|
|
return cursor.forget();
|
|
}
|
|
|
|
// static
|
|
already_AddRefed<IDBCursor>
|
|
IDBCursor::Create(IDBRequest* aRequest,
|
|
IDBTransaction* aTransaction,
|
|
IDBIndex* aIndex,
|
|
Direction aDirection,
|
|
const Key& aRangeKey,
|
|
const nsACString& aContinueQuery,
|
|
const nsACString& aContinueToQuery,
|
|
const Key& aKey,
|
|
const Key& aObjectKey,
|
|
StructuredCloneReadInfo& aCloneReadInfo)
|
|
{
|
|
NS_ASSERTION(aIndex, "Null pointer!");
|
|
NS_ASSERTION(!aKey.IsUnset(), "Bad key!");
|
|
|
|
nsRefPtr<IDBCursor> cursor =
|
|
IDBCursor::CreateCommon(aRequest, aTransaction, aIndex->ObjectStore(),
|
|
aDirection, aRangeKey, aContinueQuery,
|
|
aContinueToQuery);
|
|
NS_ASSERTION(cursor, "This shouldn't fail!");
|
|
|
|
cursor->mObjectStore = aIndex->ObjectStore();
|
|
cursor->mIndex = aIndex;
|
|
cursor->mType = INDEXOBJECT;
|
|
cursor->mKey = aKey;
|
|
cursor->mObjectKey = aObjectKey;
|
|
cursor->mCloneReadInfo.Swap(aCloneReadInfo);
|
|
|
|
return cursor.forget();
|
|
}
|
|
|
|
// static
|
|
nsresult
|
|
IDBCursor::ParseDirection(const nsAString& aDirection, Direction* aResult)
|
|
{
|
|
if (aDirection.EqualsLiteral("next")) {
|
|
*aResult = NEXT;
|
|
}
|
|
else if (aDirection.EqualsLiteral("nextunique")) {
|
|
*aResult = NEXT_UNIQUE;
|
|
}
|
|
else if (aDirection.EqualsLiteral("prev")) {
|
|
*aResult = PREV;
|
|
}
|
|
else if (aDirection.EqualsLiteral("prevunique")) {
|
|
*aResult = PREV_UNIQUE;
|
|
}
|
|
else {
|
|
return NS_ERROR_TYPE_ERR;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
// static
|
|
already_AddRefed<IDBCursor>
|
|
IDBCursor::CreateCommon(IDBRequest* aRequest,
|
|
IDBTransaction* aTransaction,
|
|
IDBObjectStore* aObjectStore,
|
|
Direction aDirection,
|
|
const Key& aRangeKey,
|
|
const nsACString& aContinueQuery,
|
|
const nsACString& aContinueToQuery)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
NS_ASSERTION(aRequest, "Null pointer!");
|
|
NS_ASSERTION(aTransaction, "Null pointer!");
|
|
NS_ASSERTION(aObjectStore, "Null pointer!");
|
|
NS_ASSERTION(!aContinueQuery.IsEmpty() ||
|
|
!IndexedDatabaseManager::IsMainProcess(),
|
|
"Empty query!");
|
|
NS_ASSERTION(!aContinueToQuery.IsEmpty() ||
|
|
!IndexedDatabaseManager::IsMainProcess(),
|
|
"Empty query!");
|
|
|
|
nsRefPtr<IDBCursor> cursor = new IDBCursor();
|
|
|
|
IDBDatabase* database = aTransaction->Database();
|
|
cursor->mScriptOwner = database->GetScriptOwner();
|
|
|
|
if (cursor->mScriptOwner) {
|
|
if (NS_FAILED(NS_HOLD_JS_OBJECTS(cursor, IDBCursor))) {
|
|
return nullptr;
|
|
}
|
|
|
|
cursor->mRooted = true;
|
|
}
|
|
|
|
cursor->mRequest = aRequest;
|
|
cursor->mTransaction = aTransaction;
|
|
cursor->mObjectStore = aObjectStore;
|
|
cursor->mDirection = aDirection;
|
|
cursor->mContinueQuery = aContinueQuery;
|
|
cursor->mContinueToQuery = aContinueToQuery;
|
|
cursor->mRangeKey = aRangeKey;
|
|
|
|
return cursor.forget();
|
|
}
|
|
|
|
IDBCursor::IDBCursor()
|
|
: mScriptOwner(nullptr),
|
|
mType(OBJECTSTORE),
|
|
mDirection(IDBCursor::NEXT),
|
|
mCachedKey(JSVAL_VOID),
|
|
mCachedPrimaryKey(JSVAL_VOID),
|
|
mCachedValue(JSVAL_VOID),
|
|
mActorChild(nullptr),
|
|
mActorParent(nullptr),
|
|
mHaveCachedKey(false),
|
|
mHaveCachedPrimaryKey(false),
|
|
mHaveCachedValue(false),
|
|
mRooted(false),
|
|
mContinueCalled(false),
|
|
mHaveValue(true)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
}
|
|
|
|
IDBCursor::~IDBCursor()
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
NS_ASSERTION(!mActorParent, "Actor parent owns us, how can we be dying?!");
|
|
if (mActorChild) {
|
|
NS_ASSERTION(!IndexedDatabaseManager::IsMainProcess(), "Wrong process!");
|
|
mActorChild->Send__delete__(mActorChild);
|
|
NS_ASSERTION(!mActorChild, "Should have cleared in Send__delete__!");
|
|
}
|
|
|
|
if (mRooted) {
|
|
NS_DROP_JS_OBJECTS(this, IDBCursor);
|
|
}
|
|
IDBObjectStore::ClearStructuredCloneBuffer(mCloneReadInfo.mCloneBuffer);
|
|
}
|
|
|
|
nsresult
|
|
IDBCursor::ContinueInternal(const Key& aKey,
|
|
PRInt32 aCount)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
NS_ASSERTION(aCount > 0, "Must have a count!");
|
|
|
|
if (!mTransaction->IsOpen()) {
|
|
return NS_ERROR_DOM_INDEXEDDB_TRANSACTION_INACTIVE_ERR;
|
|
}
|
|
|
|
if (!mHaveValue || mContinueCalled) {
|
|
return NS_ERROR_DOM_INDEXEDDB_NOT_ALLOWED_ERR;
|
|
}
|
|
|
|
mContinueToKey = aKey;
|
|
|
|
#ifdef DEBUG
|
|
{
|
|
nsAutoString readyState;
|
|
if (NS_FAILED(mRequest->GetReadyState(readyState))) {
|
|
NS_ERROR("This should never fail!");
|
|
}
|
|
NS_ASSERTION(readyState.EqualsLiteral("done"), "Should be DONE!");
|
|
}
|
|
#endif
|
|
|
|
mRequest->Reset();
|
|
|
|
nsRefPtr<ContinueHelper> helper;
|
|
switch (mType) {
|
|
case OBJECTSTORE:
|
|
helper = new ContinueObjectStoreHelper(this, aCount);
|
|
break;
|
|
|
|
case INDEXKEY:
|
|
helper = new ContinueIndexHelper(this, aCount);
|
|
break;
|
|
|
|
case INDEXOBJECT:
|
|
helper = new ContinueIndexObjectHelper(this, aCount);
|
|
break;
|
|
|
|
default:
|
|
NS_NOTREACHED("Unknown cursor type!");
|
|
}
|
|
|
|
nsresult rv = helper->DispatchToTransactionPool();
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
mContinueCalled = true;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMPL_CYCLE_COLLECTION_CLASS(IDBCursor)
|
|
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(IDBCursor)
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_SCRIPT_OBJECTS
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_NSCOMPTR_AMBIGUOUS(mRequest,
|
|
nsIDOMEventTarget)
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_NSCOMPTR_AMBIGUOUS(mTransaction,
|
|
nsIDOMEventTarget)
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_NSCOMPTR(mObjectStore)
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_NSCOMPTR(mIndex)
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
|
|
|
|
NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN(IDBCursor)
|
|
NS_ASSERTION(tmp->mHaveCachedKey || JSVAL_IS_VOID(tmp->mCachedKey),
|
|
"Should have a cached key");
|
|
NS_ASSERTION(tmp->mHaveCachedPrimaryKey ||
|
|
JSVAL_IS_VOID(tmp->mCachedPrimaryKey),
|
|
"Should have a cached primary key");
|
|
NS_ASSERTION(tmp->mHaveCachedValue || JSVAL_IS_VOID(tmp->mCachedValue),
|
|
"Should have a cached value");
|
|
NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mScriptOwner)
|
|
NS_IMPL_CYCLE_COLLECTION_TRACE_JSVAL_MEMBER_CALLBACK(mCachedKey)
|
|
NS_IMPL_CYCLE_COLLECTION_TRACE_JSVAL_MEMBER_CALLBACK(mCachedPrimaryKey)
|
|
NS_IMPL_CYCLE_COLLECTION_TRACE_JSVAL_MEMBER_CALLBACK(mCachedValue)
|
|
NS_IMPL_CYCLE_COLLECTION_TRACE_END
|
|
|
|
NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(IDBCursor)
|
|
// Don't unlink mObjectStore, mIndex, or mTransaction!
|
|
if (tmp->mRooted) {
|
|
NS_DROP_JS_OBJECTS(tmp, IDBCursor);
|
|
tmp->mScriptOwner = nullptr;
|
|
tmp->mCachedKey = JSVAL_VOID;
|
|
tmp->mCachedPrimaryKey = JSVAL_VOID;
|
|
tmp->mCachedValue = JSVAL_VOID;
|
|
tmp->mHaveCachedKey = false;
|
|
tmp->mHaveCachedPrimaryKey = false;
|
|
tmp->mHaveCachedValue = false;
|
|
tmp->mRooted = false;
|
|
tmp->mHaveValue = false;
|
|
}
|
|
NS_IMPL_CYCLE_COLLECTION_UNLINK_NSCOMPTR(mRequest)
|
|
NS_IMPL_CYCLE_COLLECTION_UNLINK_END
|
|
|
|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(IDBCursor)
|
|
NS_INTERFACE_MAP_ENTRY(nsIIDBCursor)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIIDBCursorWithValue, mType != INDEXKEY)
|
|
NS_DOM_INTERFACE_MAP_ENTRY_CLASSINFO_CONDITIONAL(IDBCursorWithValue,
|
|
mType != INDEXKEY)
|
|
NS_DOM_INTERFACE_MAP_ENTRY_CLASSINFO_CONDITIONAL(IDBCursor,
|
|
mType == INDEXKEY)
|
|
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
NS_IMPL_CYCLE_COLLECTING_ADDREF(IDBCursor)
|
|
NS_IMPL_CYCLE_COLLECTING_RELEASE(IDBCursor)
|
|
|
|
DOMCI_DATA(IDBCursor, IDBCursor)
|
|
DOMCI_DATA(IDBCursorWithValue, IDBCursor)
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::GetDirection(nsAString& aDirection)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
switch (mDirection) {
|
|
case NEXT:
|
|
aDirection.AssignLiteral("next");
|
|
break;
|
|
case NEXT_UNIQUE:
|
|
aDirection.AssignLiteral("nextunique");
|
|
break;
|
|
case PREV:
|
|
aDirection.AssignLiteral("prev");
|
|
break;
|
|
case PREV_UNIQUE:
|
|
aDirection.AssignLiteral("prevunique");
|
|
break;
|
|
|
|
case DIRECTION_INVALID:
|
|
default:
|
|
NS_NOTREACHED("Bad direction value!");
|
|
return NS_ERROR_UNEXPECTED;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::GetSource(nsISupports** aSource)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
return mType == OBJECTSTORE ?
|
|
CallQueryInterface(mObjectStore, aSource) :
|
|
CallQueryInterface(mIndex, aSource);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::GetKey(JSContext* aCx,
|
|
jsval* aKey)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
NS_ASSERTION(!mKey.IsUnset() || !mHaveValue, "Bad key!");
|
|
|
|
if (!mHaveValue) {
|
|
*aKey = JSVAL_VOID;
|
|
return NS_OK;
|
|
}
|
|
|
|
if (!mHaveCachedKey) {
|
|
if (!mRooted) {
|
|
NS_HOLD_JS_OBJECTS(this, IDBCursor);
|
|
mRooted = true;
|
|
}
|
|
|
|
JSAutoRequest ar(aCx);
|
|
|
|
nsresult rv = mKey.ToJSVal(aCx, &mCachedKey);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
mHaveCachedKey = true;
|
|
}
|
|
|
|
*aKey = mCachedKey;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::GetPrimaryKey(JSContext* aCx,
|
|
jsval* aValue)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
if (!mHaveValue) {
|
|
*aValue = JSVAL_VOID;
|
|
return NS_OK;
|
|
}
|
|
|
|
if (!mHaveCachedPrimaryKey) {
|
|
if (!mRooted) {
|
|
NS_HOLD_JS_OBJECTS(this, IDBCursor);
|
|
mRooted = true;
|
|
}
|
|
|
|
JSAutoRequest ar(aCx);
|
|
|
|
NS_ASSERTION(mType == OBJECTSTORE ? !mKey.IsUnset() :
|
|
!mObjectKey.IsUnset(), "Bad key!");
|
|
|
|
const Key& key = mType == OBJECTSTORE ? mKey : mObjectKey;
|
|
|
|
nsresult rv = key.ToJSVal(aCx, &mCachedPrimaryKey);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
mHaveCachedPrimaryKey = true;
|
|
}
|
|
|
|
*aValue = mCachedPrimaryKey;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::GetValue(JSContext* aCx,
|
|
jsval* aValue)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
NS_ASSERTION(mType != INDEXKEY, "GetValue shouldn't exist on index keys");
|
|
|
|
if (!mHaveValue) {
|
|
*aValue = JSVAL_VOID;
|
|
return NS_OK;
|
|
}
|
|
|
|
if (!mHaveCachedValue) {
|
|
if (!mRooted) {
|
|
NS_HOLD_JS_OBJECTS(this, IDBCursor);
|
|
mRooted = true;
|
|
}
|
|
|
|
jsval val;
|
|
if (!IDBObjectStore::DeserializeValue(aCx, mCloneReadInfo, &val)) {
|
|
return NS_ERROR_DOM_DATA_CLONE_ERR;
|
|
}
|
|
|
|
mCloneReadInfo.mCloneBuffer.clear();
|
|
|
|
mCachedValue = val;
|
|
mHaveCachedValue = true;
|
|
}
|
|
|
|
*aValue = mCachedValue;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::Continue(const jsval &aKey,
|
|
JSContext* aCx)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
Key key;
|
|
nsresult rv = key.SetFromJSVal(aCx, aKey);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
if (!key.IsUnset()) {
|
|
switch (mDirection) {
|
|
case IDBCursor::NEXT:
|
|
case IDBCursor::NEXT_UNIQUE:
|
|
if (key <= mKey) {
|
|
return NS_ERROR_DOM_INDEXEDDB_DATA_ERR;
|
|
}
|
|
break;
|
|
|
|
case IDBCursor::PREV:
|
|
case IDBCursor::PREV_UNIQUE:
|
|
if (key >= mKey) {
|
|
return NS_ERROR_DOM_INDEXEDDB_DATA_ERR;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
NS_NOTREACHED("Unknown direction type!");
|
|
}
|
|
}
|
|
|
|
return ContinueInternal(key, 1);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::Update(const jsval& aValue,
|
|
JSContext* aCx,
|
|
nsIIDBRequest** _retval)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
if (!mTransaction->IsOpen()) {
|
|
return NS_ERROR_DOM_INDEXEDDB_TRANSACTION_INACTIVE_ERR;
|
|
}
|
|
|
|
if (!mTransaction->IsWriteAllowed()) {
|
|
return NS_ERROR_DOM_INDEXEDDB_READ_ONLY_ERR;
|
|
}
|
|
|
|
if (!mHaveValue || mType == INDEXKEY) {
|
|
return NS_ERROR_DOM_INDEXEDDB_NOT_ALLOWED_ERR;
|
|
}
|
|
|
|
NS_ASSERTION(mObjectStore, "This cannot be null!");
|
|
NS_ASSERTION(!mKey.IsUnset() , "Bad key!");
|
|
NS_ASSERTION(mType != INDEXOBJECT || !mObjectKey.IsUnset(), "Bad key!");
|
|
|
|
nsresult rv;
|
|
|
|
JSAutoRequest ar(aCx);
|
|
|
|
Key& objectKey = (mType == OBJECTSTORE) ? mKey : mObjectKey;
|
|
|
|
if (mObjectStore->HasValidKeyPath()) {
|
|
// Make sure the object given has the correct keyPath value set on it.
|
|
const KeyPath& keyPath = mObjectStore->GetKeyPath();
|
|
Key key;
|
|
|
|
rv = keyPath.ExtractKey(aCx, aValue, key);
|
|
if (NS_FAILED(rv)) {
|
|
return rv;
|
|
}
|
|
|
|
if (key != objectKey) {
|
|
return NS_ERROR_DOM_INDEXEDDB_DATA_ERR;
|
|
}
|
|
|
|
return mObjectStore->Put(aValue, JSVAL_VOID, aCx, 0, _retval);
|
|
}
|
|
|
|
jsval keyVal;
|
|
rv = objectKey.ToJSVal(aCx, &keyVal);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
return mObjectStore->Put(aValue, keyVal, aCx, 1, _retval);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::Delete(JSContext* aCx,
|
|
nsIIDBRequest** _retval)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
if (!mTransaction->IsOpen()) {
|
|
return NS_ERROR_DOM_INDEXEDDB_TRANSACTION_INACTIVE_ERR;
|
|
}
|
|
|
|
if (!mTransaction->IsWriteAllowed()) {
|
|
return NS_ERROR_DOM_INDEXEDDB_READ_ONLY_ERR;
|
|
}
|
|
|
|
if (!mHaveValue || mType == INDEXKEY) {
|
|
return NS_ERROR_DOM_INDEXEDDB_NOT_ALLOWED_ERR;
|
|
}
|
|
|
|
NS_ASSERTION(mObjectStore, "This cannot be null!");
|
|
NS_ASSERTION(!mKey.IsUnset() , "Bad key!");
|
|
|
|
Key& objectKey = (mType == OBJECTSTORE) ? mKey : mObjectKey;
|
|
|
|
jsval key;
|
|
nsresult rv = objectKey.ToJSVal(aCx, &key);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
return mObjectStore->Delete(key, aCx, _retval);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
IDBCursor::Advance(PRInt64 aCount)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
|
|
if (aCount < 1 || aCount > PR_UINT32_MAX) {
|
|
return NS_ERROR_TYPE_ERR;
|
|
}
|
|
|
|
Key key;
|
|
return ContinueInternal(key, PRInt32(aCount));
|
|
}
|
|
|
|
void
|
|
CursorHelper::ReleaseMainThreadObjects()
|
|
{
|
|
mCursor = nullptr;
|
|
AsyncConnectionHelper::ReleaseMainThreadObjects();
|
|
}
|
|
|
|
nsresult
|
|
CursorHelper::Dispatch(nsIEventTarget* aDatabaseThread)
|
|
{
|
|
if (IndexedDatabaseManager::IsMainProcess()) {
|
|
return AsyncConnectionHelper::Dispatch(aDatabaseThread);
|
|
}
|
|
|
|
IndexedDBCursorChild* cursorActor = mCursor->GetActorChild();
|
|
NS_ASSERTION(cursorActor, "Must have an actor here!");
|
|
|
|
CursorRequestParams params;
|
|
nsresult rv = PackArgumentsForParentProcess(params);
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
NoDispatchEventTarget target;
|
|
rv = AsyncConnectionHelper::Dispatch(&target);
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
mActor = new IndexedDBCursorRequestChild(this, mCursor, params.type());
|
|
cursorActor->SendPIndexedDBRequestConstructor(mActor, params);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueHelper::DoDatabaseWork(mozIStorageConnection* aConnection)
|
|
{
|
|
// We need to pick a query based on whether or not the cursor's mContinueToKey
|
|
// is set. If it is unset then othing was passed to continue so we'll grab the
|
|
// next item in the database that is greater than (less than, if we're running
|
|
// a PREV cursor) the current key. If it is set then a key was passed to
|
|
// continue so we'll grab the next item in the database that is greater than
|
|
// (less than, if we're running a PREV cursor) or equal to the key that was
|
|
// specified.
|
|
|
|
nsCAutoString query;
|
|
if (mCursor->mContinueToKey.IsUnset()) {
|
|
query.Assign(mCursor->mContinueQuery);
|
|
}
|
|
else {
|
|
query.Assign(mCursor->mContinueToQuery);
|
|
}
|
|
NS_ASSERTION(!query.IsEmpty(), "Bad query!");
|
|
|
|
query.AppendInt(mCount);
|
|
|
|
nsCOMPtr<mozIStorageStatement> stmt = mTransaction->GetCachedStatement(query);
|
|
NS_ENSURE_TRUE(stmt, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
mozStorageStatementScoper scoper(stmt);
|
|
|
|
nsresult rv = BindArgumentsToStatement(stmt);
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
NS_ASSERTION(mCount > 0, "Not ok!");
|
|
|
|
bool hasResult;
|
|
for (PRInt32 index = 0; index < mCount; index++) {
|
|
rv = stmt->ExecuteStep(&hasResult);
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
if (!hasResult) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (hasResult) {
|
|
rv = GatherResultsFromStatement(stmt);
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
}
|
|
else {
|
|
mKey.Unset();
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueHelper::GetSuccessResult(JSContext* aCx,
|
|
jsval* aVal)
|
|
{
|
|
UpdateCursorState();
|
|
|
|
if (mKey.IsUnset()) {
|
|
*aVal = JSVAL_NULL;
|
|
}
|
|
else {
|
|
nsresult rv = WrapNative(aCx, mCursor, aVal);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueHelper::PackArgumentsForParentProcess(CursorRequestParams& aParams)
|
|
{
|
|
ContinueParams params;
|
|
|
|
params.key() = mCursor->mContinueToKey;
|
|
params.count() = uint32_t(mCount);
|
|
|
|
aParams = params;
|
|
return NS_OK;
|
|
}
|
|
|
|
HelperBase::ChildProcessSendResult
|
|
ContinueHelper::MaybeSendResponseToChildProcess(nsresult aResultCode)
|
|
{
|
|
NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
|
|
NS_ASSERTION(IndexedDatabaseManager::IsMainProcess(), "Wrong process!");
|
|
|
|
IndexedDBRequestParentBase* actor = mRequest->GetActorParent();
|
|
if (!actor) {
|
|
return Success_NotSent;
|
|
}
|
|
|
|
InfallibleTArray<PBlobParent*> blobsParent;
|
|
|
|
if (NS_SUCCEEDED(aResultCode)) {
|
|
IDBDatabase* database = mTransaction->Database();
|
|
NS_ASSERTION(database, "This should never be null!");
|
|
|
|
ContentParent* contentParent = database->GetContentParent();
|
|
NS_ASSERTION(contentParent, "This should never be null!");
|
|
|
|
FileManager* fileManager = database->Manager();
|
|
NS_ASSERTION(fileManager, "This should never be null!");
|
|
|
|
const nsTArray<StructuredCloneFile>& files = mCloneReadInfo.mFiles;
|
|
|
|
aResultCode =
|
|
IDBObjectStore::ConvertBlobsToActors(contentParent, fileManager, files,
|
|
blobsParent);
|
|
if (NS_FAILED(aResultCode)) {
|
|
NS_WARNING("ConvertBlobsToActors failed!");
|
|
}
|
|
}
|
|
|
|
ResponseValue response;
|
|
if (NS_FAILED(aResultCode)) {
|
|
response = aResultCode;
|
|
}
|
|
else {
|
|
ContinueResponse continueResponse;
|
|
continueResponse.key() = mKey;
|
|
continueResponse.objectKey() = mObjectKey;
|
|
continueResponse.cloneInfo() = mCloneReadInfo;
|
|
continueResponse.blobsParent().SwapElements(blobsParent);
|
|
response = continueResponse;
|
|
}
|
|
|
|
if (!actor->Send__delete__(actor, response)) {
|
|
return Error;
|
|
}
|
|
|
|
UpdateCursorState();
|
|
|
|
return Success_Sent;
|
|
}
|
|
|
|
nsresult
|
|
ContinueHelper::UnpackResponseFromParentProcess(
|
|
const ResponseValue& aResponseValue)
|
|
{
|
|
NS_ASSERTION(aResponseValue.type() == ResponseValue::TContinueResponse,
|
|
"Bad response type!");
|
|
|
|
const ContinueResponse& response = aResponseValue.get_ContinueResponse();
|
|
|
|
mKey = response.key();
|
|
mObjectKey = response.objectKey();
|
|
|
|
const SerializedStructuredCloneReadInfo& cloneInfo = response.cloneInfo();
|
|
|
|
NS_ASSERTION((!cloneInfo.dataLength && !cloneInfo.data) ||
|
|
(cloneInfo.dataLength && cloneInfo.data),
|
|
"Inconsistent clone info!");
|
|
|
|
if (!mCloneReadInfo.SetFromSerialized(cloneInfo)) {
|
|
NS_WARNING("Failed to copy clone buffer!");
|
|
return NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR;
|
|
}
|
|
|
|
IDBObjectStore::ConvertActorsToBlobs(response.blobsChild(),
|
|
mCloneReadInfo.mFiles);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueObjectStoreHelper::BindArgumentsToStatement(
|
|
mozIStorageStatement* aStatement)
|
|
{
|
|
// Bind object store id.
|
|
nsresult rv = aStatement->BindInt64ByName(NS_LITERAL_CSTRING("id"),
|
|
mCursor->mObjectStore->Id());
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
NS_NAMED_LITERAL_CSTRING(currentKeyName, "current_key");
|
|
NS_NAMED_LITERAL_CSTRING(rangeKeyName, "range_key");
|
|
|
|
// Bind current key.
|
|
const Key& currentKey = mCursor->mContinueToKey.IsUnset() ?
|
|
mCursor->mKey :
|
|
mCursor->mContinueToKey;
|
|
|
|
rv = currentKey.BindToStatement(aStatement, currentKeyName);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Bind range key if it is specified.
|
|
const Key& rangeKey = mCursor->mRangeKey;
|
|
|
|
if (!rangeKey.IsUnset()) {
|
|
rv = rangeKey.BindToStatement(aStatement, rangeKeyName);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueObjectStoreHelper::GatherResultsFromStatement(
|
|
mozIStorageStatement* aStatement)
|
|
{
|
|
// Figure out what kind of key we have next.
|
|
nsresult rv = mKey.SetFromStatement(aStatement, 0);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = IDBObjectStore::GetStructuredCloneReadInfoFromStatement(aStatement, 1, 2,
|
|
mDatabase, mCloneReadInfo);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueIndexHelper::BindArgumentsToStatement(mozIStorageStatement* aStatement)
|
|
{
|
|
// Bind index id.
|
|
nsresult rv = aStatement->BindInt64ByName(NS_LITERAL_CSTRING("id"),
|
|
mCursor->mIndex->Id());
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_DOM_INDEXEDDB_UNKNOWN_ERR);
|
|
|
|
NS_NAMED_LITERAL_CSTRING(currentKeyName, "current_key");
|
|
|
|
// Bind current key.
|
|
const Key& currentKey = mCursor->mContinueToKey.IsUnset() ?
|
|
mCursor->mKey :
|
|
mCursor->mContinueToKey;
|
|
|
|
rv = currentKey.BindToStatement(aStatement, currentKeyName);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Bind range key if it is specified.
|
|
if (!mCursor->mRangeKey.IsUnset()) {
|
|
NS_NAMED_LITERAL_CSTRING(rangeKeyName, "range_key");
|
|
rv = mCursor->mRangeKey.BindToStatement(aStatement, rangeKeyName);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
}
|
|
|
|
// Bind object key if duplicates are allowed and we're not continuing to a
|
|
// specific key.
|
|
if ((mCursor->mDirection == IDBCursor::NEXT ||
|
|
mCursor->mDirection == IDBCursor::PREV) &&
|
|
mCursor->mContinueToKey.IsUnset()) {
|
|
NS_ASSERTION(!mCursor->mObjectKey.IsUnset(), "Bad key!");
|
|
|
|
NS_NAMED_LITERAL_CSTRING(objectKeyName, "object_key");
|
|
rv = mCursor->mObjectKey.BindToStatement(aStatement, objectKeyName);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueIndexHelper::GatherResultsFromStatement(
|
|
mozIStorageStatement* aStatement)
|
|
{
|
|
nsresult rv = mKey.SetFromStatement(aStatement, 0);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = mObjectKey.SetFromStatement(aStatement, 1);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
ContinueIndexObjectHelper::GatherResultsFromStatement(
|
|
mozIStorageStatement* aStatement)
|
|
{
|
|
nsresult rv = mKey.SetFromStatement(aStatement, 0);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = mObjectKey.SetFromStatement(aStatement, 1);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = IDBObjectStore::GetStructuredCloneReadInfoFromStatement(aStatement, 2, 3,
|
|
mDatabase, mCloneReadInfo);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
return NS_OK;
|
|
}
|