forked from mirrors/gecko-dev
3f2298c2f6
MozReview-Commit-ID: 666LNaHyiuQ --HG-- extra : rebase_source : f3a24e0342d8dcdf9020dd473bc6b1045cbc533b
155 lines
3.6 KiB
C++
155 lines
3.6 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/* 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
|