forked from mirrors/gecko-dev
722f6a1679
Backed out changeset 3ceaf46f8f55 (bug 1851992) Backed out changeset c9d322362e22 (bug 1851992) Backed out changeset 673df3f83249 (bug 1851992) Backed out changeset 46e18c56dd39 (bug 1851992) Backed out changeset f9f9143ac713 (bug 1851992) Backed out changeset 38c40d735ab7 (bug 1851992)
102 lines
3.1 KiB
C++
102 lines
3.1 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=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 "MutableBlobStreamListener.h"
|
|
#include "MutableBlobStorage.h"
|
|
#include "nsIInputStream.h"
|
|
#include "nsThreadUtils.h"
|
|
|
|
namespace mozilla::dom {
|
|
|
|
MutableBlobStreamListener::MutableBlobStreamListener(
|
|
MutableBlobStorage::MutableBlobStorageType aStorageType,
|
|
const nsACString& aContentType, MutableBlobStorageCallback* aCallback,
|
|
nsIEventTarget* aEventTarget)
|
|
: mCallback(aCallback),
|
|
mStorageType(aStorageType),
|
|
mContentType(aContentType),
|
|
mEventTarget(aEventTarget) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(aCallback);
|
|
|
|
if (!mEventTarget) {
|
|
mEventTarget = GetMainThreadSerialEventTarget();
|
|
}
|
|
|
|
MOZ_ASSERT(mEventTarget);
|
|
}
|
|
|
|
MutableBlobStreamListener::~MutableBlobStreamListener() {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
}
|
|
|
|
NS_IMPL_ISUPPORTS(MutableBlobStreamListener, nsIStreamListener,
|
|
nsIThreadRetargetableStreamListener, nsIRequestObserver)
|
|
|
|
NS_IMETHODIMP
|
|
MutableBlobStreamListener::OnStartRequest(nsIRequest* aRequest) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(!mStorage);
|
|
MOZ_ASSERT(mEventTarget);
|
|
|
|
mStorage = new MutableBlobStorage(mStorageType, mEventTarget);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
MutableBlobStreamListener::OnStopRequest(nsIRequest* aRequest,
|
|
nsresult aStatus) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(mStorage);
|
|
|
|
// Resetting mStorage to nullptr.
|
|
RefPtr<MutableBlobStorage> storage;
|
|
storage.swap(mStorage);
|
|
|
|
// Let's propagate the error simulating a failure of the storage.
|
|
if (NS_FAILED(aStatus)) {
|
|
mCallback->BlobStoreCompleted(storage, nullptr, aStatus);
|
|
return NS_OK;
|
|
}
|
|
|
|
storage->GetBlobImplWhenReady(mContentType, mCallback);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
MutableBlobStreamListener::OnDataAvailable(nsIRequest* aRequest,
|
|
nsIInputStream* aStream,
|
|
uint64_t aSourceOffset,
|
|
uint32_t aCount) {
|
|
// This method could be called on any thread.
|
|
MOZ_ASSERT(mStorage);
|
|
|
|
uint32_t countRead;
|
|
return aStream->ReadSegments(WriteSegmentFun, this, aCount, &countRead);
|
|
}
|
|
|
|
nsresult MutableBlobStreamListener::WriteSegmentFun(
|
|
nsIInputStream* aWriterStream, void* aClosure, const char* aFromSegment,
|
|
uint32_t aToOffset, uint32_t aCount, uint32_t* aWriteCount) {
|
|
// This method could be called on any thread.
|
|
|
|
MutableBlobStreamListener* self =
|
|
static_cast<MutableBlobStreamListener*>(aClosure);
|
|
MOZ_ASSERT(self->mStorage);
|
|
|
|
nsresult rv = self->mStorage->Append(aFromSegment, aCount);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
*aWriteCount = aCount;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
MutableBlobStreamListener::CheckListenerChain() { return NS_OK; }
|
|
|
|
} // namespace mozilla::dom
|