mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-28 07:13:20 +00:00
438 lines
13 KiB
C++
438 lines
13 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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 "PartiallySeekableInputStream.h"
|
|
#include "mozilla/ipc/InputStreamUtils.h"
|
|
#include "nsISeekableStream.h"
|
|
#include "nsStreamUtils.h"
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
NS_IMPL_ADDREF(PartiallySeekableInputStream);
|
|
NS_IMPL_RELEASE(PartiallySeekableInputStream);
|
|
|
|
NS_INTERFACE_MAP_BEGIN(PartiallySeekableInputStream)
|
|
NS_INTERFACE_MAP_ENTRY(nsIInputStream)
|
|
NS_INTERFACE_MAP_ENTRY(nsISeekableStream)
|
|
NS_INTERFACE_MAP_ENTRY(nsITellableStream)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsICloneableInputStream,
|
|
mWeakCloneableInputStream)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIIPCSerializableInputStream,
|
|
mWeakIPCSerializableInputStream)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIAsyncInputStream, mWeakAsyncInputStream)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIInputStreamCallback,
|
|
mWeakAsyncInputStream)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIInputStreamLength,
|
|
mWeakInputStreamLength)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIAsyncInputStreamLength,
|
|
mWeakAsyncInputStreamLength)
|
|
NS_INTERFACE_MAP_ENTRY_CONDITIONAL(nsIInputStreamLengthCallback,
|
|
mWeakAsyncInputStreamLength)
|
|
NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIInputStream)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
PartiallySeekableInputStream::PartiallySeekableInputStream(
|
|
already_AddRefed<nsIInputStream> aInputStream, uint64_t aBufferSize)
|
|
: mInputStream(std::move(aInputStream)),
|
|
mWeakCloneableInputStream(nullptr),
|
|
mWeakIPCSerializableInputStream(nullptr),
|
|
mWeakAsyncInputStream(nullptr),
|
|
mWeakInputStreamLength(nullptr),
|
|
mWeakAsyncInputStreamLength(nullptr),
|
|
mBufferSize(aBufferSize),
|
|
mPos(0),
|
|
mClosed(false),
|
|
mMutex("PartiallySeekableInputStream::mMutex") {
|
|
Init();
|
|
}
|
|
|
|
PartiallySeekableInputStream::PartiallySeekableInputStream(
|
|
already_AddRefed<nsIInputStream> aClonedBaseStream,
|
|
PartiallySeekableInputStream* aClonedFrom)
|
|
: mInputStream(std::move(aClonedBaseStream)),
|
|
mWeakCloneableInputStream(nullptr),
|
|
mWeakIPCSerializableInputStream(nullptr),
|
|
mWeakAsyncInputStream(nullptr),
|
|
mWeakInputStreamLength(nullptr),
|
|
mWeakAsyncInputStreamLength(nullptr),
|
|
mCachedBuffer(aClonedFrom->mCachedBuffer.Clone()),
|
|
mBufferSize(aClonedFrom->mBufferSize),
|
|
mPos(aClonedFrom->mPos),
|
|
mClosed(aClonedFrom->mClosed),
|
|
mMutex("PartiallySeekableInputStream::mMutex") {
|
|
Init();
|
|
}
|
|
|
|
void PartiallySeekableInputStream::Init() {
|
|
MOZ_ASSERT(mInputStream);
|
|
|
|
#ifdef DEBUG
|
|
nsCOMPtr<nsISeekableStream> seekableStream = do_QueryInterface(mInputStream);
|
|
MOZ_ASSERT(!seekableStream);
|
|
#endif
|
|
|
|
nsCOMPtr<nsICloneableInputStream> cloneableStream =
|
|
do_QueryInterface(mInputStream);
|
|
if (cloneableStream && SameCOMIdentity(mInputStream, cloneableStream)) {
|
|
mWeakCloneableInputStream = cloneableStream;
|
|
}
|
|
|
|
nsCOMPtr<nsIIPCSerializableInputStream> serializableStream =
|
|
do_QueryInterface(mInputStream);
|
|
if (serializableStream && SameCOMIdentity(mInputStream, serializableStream)) {
|
|
mWeakIPCSerializableInputStream = serializableStream;
|
|
}
|
|
|
|
nsCOMPtr<nsIAsyncInputStream> asyncInputStream =
|
|
do_QueryInterface(mInputStream);
|
|
if (asyncInputStream && SameCOMIdentity(mInputStream, asyncInputStream)) {
|
|
mWeakAsyncInputStream = asyncInputStream;
|
|
}
|
|
|
|
nsCOMPtr<nsIInputStreamLength> inputStreamLength =
|
|
do_QueryInterface(mInputStream);
|
|
if (inputStreamLength && SameCOMIdentity(mInputStream, inputStreamLength)) {
|
|
mWeakInputStreamLength = inputStreamLength;
|
|
}
|
|
|
|
nsCOMPtr<nsIAsyncInputStreamLength> asyncInputStreamLength =
|
|
do_QueryInterface(mInputStream);
|
|
if (asyncInputStreamLength &&
|
|
SameCOMIdentity(mInputStream, asyncInputStreamLength)) {
|
|
mWeakAsyncInputStreamLength = asyncInputStreamLength;
|
|
}
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Close() {
|
|
mInputStream->Close();
|
|
mCachedBuffer.Clear();
|
|
mPos = 0;
|
|
mClosed = true;
|
|
return NS_OK;
|
|
}
|
|
|
|
// nsIInputStream interface
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Available(uint64_t* aLength) {
|
|
if (mClosed) {
|
|
return NS_BASE_STREAM_CLOSED;
|
|
}
|
|
|
|
nsresult rv = mInputStream->Available(aLength);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
if (mPos < mCachedBuffer.Length()) {
|
|
*aLength += mCachedBuffer.Length() - mPos;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Read(char* aBuffer, uint32_t aCount,
|
|
uint32_t* aReadCount) {
|
|
*aReadCount = 0;
|
|
|
|
if (mClosed) {
|
|
return NS_OK;
|
|
}
|
|
|
|
uint32_t byteRead = 0;
|
|
|
|
if (mPos < mCachedBuffer.Length()) {
|
|
// We are reading from the cached buffer.
|
|
byteRead = XPCOM_MIN(mCachedBuffer.Length() - mPos, (uint64_t)aCount);
|
|
memcpy(aBuffer, mCachedBuffer.Elements() + mPos, byteRead);
|
|
*aReadCount = byteRead;
|
|
mPos += byteRead;
|
|
}
|
|
|
|
if (byteRead < aCount) {
|
|
MOZ_ASSERT(mPos >= mCachedBuffer.Length());
|
|
MOZ_ASSERT_IF(mPos > mCachedBuffer.Length(),
|
|
mCachedBuffer.Length() == mBufferSize);
|
|
|
|
// We can read from the stream.
|
|
uint32_t byteWritten;
|
|
nsresult rv =
|
|
mInputStream->Read(aBuffer + byteRead, aCount - byteRead, &byteWritten);
|
|
if (NS_WARN_IF(NS_FAILED(rv)) || byteWritten == 0) {
|
|
return rv;
|
|
}
|
|
|
|
*aReadCount += byteWritten;
|
|
|
|
// Maybe we have to cache something.
|
|
if (mPos < mBufferSize) {
|
|
uint32_t size = XPCOM_MIN(mPos + byteWritten, mBufferSize);
|
|
mCachedBuffer.SetLength(size);
|
|
memcpy(mCachedBuffer.Elements() + mPos, aBuffer + byteRead, size - mPos);
|
|
}
|
|
|
|
mPos += byteWritten;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::ReadSegments(nsWriteSegmentFun aWriter,
|
|
void* aClosure, uint32_t aCount,
|
|
uint32_t* aResult) {
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::IsNonBlocking(bool* aNonBlocking) {
|
|
return mInputStream->IsNonBlocking(aNonBlocking);
|
|
}
|
|
|
|
// nsICloneableInputStream interface
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::GetCloneable(bool* aCloneable) {
|
|
NS_ENSURE_STATE(mWeakCloneableInputStream);
|
|
|
|
return mWeakCloneableInputStream->GetCloneable(aCloneable);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Clone(nsIInputStream** aResult) {
|
|
NS_ENSURE_STATE(mWeakCloneableInputStream);
|
|
|
|
nsCOMPtr<nsIInputStream> clonedStream;
|
|
nsresult rv = mWeakCloneableInputStream->Clone(getter_AddRefs(clonedStream));
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
nsCOMPtr<nsIInputStream> stream =
|
|
new PartiallySeekableInputStream(clonedStream.forget(), this);
|
|
|
|
stream.forget(aResult);
|
|
return NS_OK;
|
|
}
|
|
|
|
// nsIAsyncInputStream interface
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::CloseWithStatus(nsresult aStatus) {
|
|
NS_ENSURE_STATE(mWeakAsyncInputStream);
|
|
|
|
return mWeakAsyncInputStream->CloseWithStatus(aStatus);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::AsyncWait(nsIInputStreamCallback* aCallback,
|
|
uint32_t aFlags,
|
|
uint32_t aRequestedCount,
|
|
nsIEventTarget* aEventTarget) {
|
|
if (mClosed) {
|
|
if (aCallback) {
|
|
if (aEventTarget) {
|
|
nsCOMPtr<nsIInputStreamCallback> callable = NS_NewInputStreamReadyEvent(
|
|
"PartiallySeekableInputStream::OnInputStreamReady", aCallback,
|
|
aEventTarget);
|
|
callable->OnInputStreamReady(this);
|
|
} else {
|
|
aCallback->OnInputStreamReady(this);
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
{
|
|
MutexAutoLock lock(mMutex);
|
|
if (mAsyncWaitCallback && aCallback) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
mAsyncWaitCallback = aCallback;
|
|
}
|
|
|
|
NS_ENSURE_STATE(mWeakAsyncInputStream);
|
|
nsCOMPtr<nsIInputStreamCallback> callback = aCallback ? this : nullptr;
|
|
return mWeakAsyncInputStream->AsyncWait(callback, aFlags, aRequestedCount,
|
|
aEventTarget);
|
|
}
|
|
|
|
// nsIInputStreamCallback
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::OnInputStreamReady(nsIAsyncInputStream* aStream) {
|
|
MOZ_ASSERT(mWeakAsyncInputStream);
|
|
MOZ_ASSERT(mWeakAsyncInputStream == aStream);
|
|
|
|
nsCOMPtr<nsIInputStreamCallback> callback;
|
|
|
|
{
|
|
MutexAutoLock lock(mMutex);
|
|
|
|
// We have been canceled in the meanwhile.
|
|
if (!mAsyncWaitCallback) {
|
|
return NS_OK;
|
|
}
|
|
|
|
callback.swap(mAsyncWaitCallback);
|
|
}
|
|
|
|
MOZ_ASSERT(callback);
|
|
return callback->OnInputStreamReady(this);
|
|
}
|
|
|
|
// nsIIPCSerializableInputStream
|
|
|
|
void PartiallySeekableInputStream::Serialize(
|
|
mozilla::ipc::InputStreamParams& aParams,
|
|
FileDescriptorArray& aFileDescriptors, bool aDelayedStart,
|
|
uint32_t aMaxSize, uint32_t* aSizeUsed,
|
|
mozilla::ipc::ParentToChildStreamActorManager* aManager) {
|
|
SerializeInternal(aParams, aFileDescriptors, aDelayedStart, aMaxSize,
|
|
aSizeUsed, aManager);
|
|
}
|
|
|
|
void PartiallySeekableInputStream::Serialize(
|
|
mozilla::ipc::InputStreamParams& aParams,
|
|
FileDescriptorArray& aFileDescriptors, bool aDelayedStart,
|
|
uint32_t aMaxSize, uint32_t* aSizeUsed,
|
|
mozilla::ipc::ChildToParentStreamActorManager* aManager) {
|
|
SerializeInternal(aParams, aFileDescriptors, aDelayedStart, aMaxSize,
|
|
aSizeUsed, aManager);
|
|
}
|
|
|
|
template <typename M>
|
|
void PartiallySeekableInputStream::SerializeInternal(
|
|
mozilla::ipc::InputStreamParams& aParams,
|
|
FileDescriptorArray& aFileDescriptors, bool aDelayedStart,
|
|
uint32_t aMaxSize, uint32_t* aSizeUsed, M* aManager) {
|
|
MOZ_ASSERT(mWeakIPCSerializableInputStream);
|
|
MOZ_DIAGNOSTIC_ASSERT(mCachedBuffer.IsEmpty());
|
|
mozilla::ipc::InputStreamHelper::SerializeInputStream(
|
|
mInputStream, aParams, aFileDescriptors, aDelayedStart, aMaxSize,
|
|
aSizeUsed, aManager);
|
|
}
|
|
|
|
bool PartiallySeekableInputStream::Deserialize(
|
|
const mozilla::ipc::InputStreamParams& aParams,
|
|
const FileDescriptorArray& aFileDescriptors) {
|
|
MOZ_CRASH("This method should never be called!");
|
|
return false;
|
|
}
|
|
|
|
// nsISeekableStream
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Seek(int32_t aWhence, int64_t aOffset) {
|
|
if (mClosed) {
|
|
return NS_BASE_STREAM_CLOSED;
|
|
}
|
|
|
|
int64_t offset;
|
|
|
|
switch (aWhence) {
|
|
case NS_SEEK_SET:
|
|
offset = aOffset;
|
|
break;
|
|
case NS_SEEK_CUR:
|
|
offset = mPos + aOffset;
|
|
break;
|
|
case NS_SEEK_END: {
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
default:
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
}
|
|
|
|
if (offset < 0) {
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
}
|
|
|
|
if ((uint64_t)offset >= mCachedBuffer.Length() || mPos > mBufferSize) {
|
|
return NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
mPos = offset;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Tell(int64_t* aResult) {
|
|
if (mClosed) {
|
|
return NS_BASE_STREAM_CLOSED;
|
|
}
|
|
|
|
*aResult = mPos;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::SetEOF() { return Close(); }
|
|
|
|
// nsIInputStreamLength
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::Length(int64_t* aLength) {
|
|
NS_ENSURE_STATE(mWeakInputStreamLength);
|
|
return mWeakInputStreamLength->Length(aLength);
|
|
}
|
|
|
|
// nsIAsyncInputStreamLength
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::AsyncLengthWait(
|
|
nsIInputStreamLengthCallback* aCallback, nsIEventTarget* aEventTarget) {
|
|
if (mClosed) {
|
|
if (aCallback) {
|
|
const RefPtr<PartiallySeekableInputStream> self = this;
|
|
const nsCOMPtr<nsIInputStreamLengthCallback> callback = aCallback;
|
|
nsCOMPtr<nsIRunnable> runnable = NS_NewRunnableFunction(
|
|
"PartiallySeekableInputStream::OnInputStreamLengthReady",
|
|
[self, callback] { callback->OnInputStreamLengthReady(self, -1); });
|
|
if (aEventTarget) {
|
|
aEventTarget->Dispatch(runnable, NS_DISPATCH_NORMAL);
|
|
} else {
|
|
runnable->Run();
|
|
}
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_ENSURE_STATE(mWeakAsyncInputStreamLength);
|
|
|
|
nsCOMPtr<nsIInputStreamLengthCallback> callback = aCallback ? this : nullptr;
|
|
{
|
|
MutexAutoLock lock(mMutex);
|
|
mAsyncInputStreamLengthCallback = aCallback;
|
|
}
|
|
|
|
return mWeakAsyncInputStreamLength->AsyncLengthWait(callback, aEventTarget);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
PartiallySeekableInputStream::OnInputStreamLengthReady(
|
|
nsIAsyncInputStreamLength* aStream, int64_t aLength) {
|
|
nsCOMPtr<nsIInputStreamLengthCallback> callback;
|
|
{
|
|
MutexAutoLock lock(mMutex);
|
|
// We have been canceled in the meanwhile.
|
|
if (!mAsyncInputStreamLengthCallback) {
|
|
return NS_OK;
|
|
}
|
|
|
|
callback.swap(mAsyncInputStreamLengthCallback);
|
|
}
|
|
|
|
return callback->OnInputStreamLengthReady(this, aLength);
|
|
}
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|