forked from mirrors/gecko-dev
123 lines
3.2 KiB
C++
123 lines
3.2 KiB
C++
/* -*- Mode: C++; tab-width: 4; 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 "nsPreloadedStream.h"
|
|
#include "nsIRunnable.h"
|
|
|
|
#include "nsThreadUtils.h"
|
|
#include <algorithm>
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
NS_IMPL_ISUPPORTS(nsPreloadedStream, nsIInputStream, nsIAsyncInputStream)
|
|
|
|
nsPreloadedStream::nsPreloadedStream(nsIAsyncInputStream* aStream,
|
|
const char* data, uint32_t datalen)
|
|
: mStream(aStream), mOffset(0), mLen(datalen) {
|
|
mBuf = (char*)moz_xmalloc(datalen);
|
|
memcpy(mBuf, data, datalen);
|
|
}
|
|
|
|
nsPreloadedStream::~nsPreloadedStream() { free(mBuf); }
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::Close() {
|
|
mLen = 0;
|
|
return mStream->Close();
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::Available(uint64_t* _retval) {
|
|
uint64_t avail = 0;
|
|
|
|
nsresult rv = mStream->Available(&avail);
|
|
if (NS_FAILED(rv)) return rv;
|
|
*_retval = avail + mLen;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::Read(char* aBuf, uint32_t aCount, uint32_t* _retval) {
|
|
if (!mLen) return mStream->Read(aBuf, aCount, _retval);
|
|
|
|
uint32_t toRead = std::min(mLen, aCount);
|
|
memcpy(aBuf, mBuf + mOffset, toRead);
|
|
mOffset += toRead;
|
|
mLen -= toRead;
|
|
*_retval = toRead;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::ReadSegments(nsWriteSegmentFun aWriter, void* aClosure,
|
|
uint32_t aCount, uint32_t* result) {
|
|
if (!mLen) return mStream->ReadSegments(aWriter, aClosure, aCount, result);
|
|
|
|
*result = 0;
|
|
while (mLen > 0 && aCount > 0) {
|
|
uint32_t toRead = std::min(mLen, aCount);
|
|
uint32_t didRead = 0;
|
|
nsresult rv;
|
|
|
|
rv = aWriter(this, aClosure, mBuf + mOffset, *result, toRead, &didRead);
|
|
|
|
if (NS_FAILED(rv)) return NS_OK;
|
|
|
|
*result += didRead;
|
|
mOffset += didRead;
|
|
mLen -= didRead;
|
|
aCount -= didRead;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::IsNonBlocking(bool* _retval) {
|
|
return mStream->IsNonBlocking(_retval);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::CloseWithStatus(nsresult aStatus) {
|
|
mLen = 0;
|
|
return mStream->CloseWithStatus(aStatus);
|
|
}
|
|
|
|
class RunOnThread : public Runnable {
|
|
public:
|
|
RunOnThread(nsIAsyncInputStream* aStream, nsIInputStreamCallback* aCallback)
|
|
: Runnable("net::RunOnThread"), mStream(aStream), mCallback(aCallback) {}
|
|
|
|
virtual ~RunOnThread() = default;
|
|
|
|
NS_IMETHOD Run() override {
|
|
mCallback->OnInputStreamReady(mStream);
|
|
return NS_OK;
|
|
}
|
|
|
|
private:
|
|
nsCOMPtr<nsIAsyncInputStream> mStream;
|
|
nsCOMPtr<nsIInputStreamCallback> mCallback;
|
|
};
|
|
|
|
NS_IMETHODIMP
|
|
nsPreloadedStream::AsyncWait(nsIInputStreamCallback* aCallback, uint32_t aFlags,
|
|
uint32_t aRequestedCount,
|
|
nsIEventTarget* aEventTarget) {
|
|
if (!mLen) {
|
|
return mStream->AsyncWait(aCallback, aFlags, aRequestedCount, aEventTarget);
|
|
}
|
|
|
|
if (!aCallback) return NS_OK;
|
|
|
|
if (!aEventTarget) return aCallback->OnInputStreamReady(this);
|
|
|
|
nsCOMPtr<nsIRunnable> event = new RunOnThread(this, aCallback);
|
|
return aEventTarget->Dispatch(event, nsIEventTarget::DISPATCH_NORMAL);
|
|
}
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|