1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/netwerk/base/src/nsPreloadedStream.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,153 @@
1.4 +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.7 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include "nsPreloadedStream.h"
1.10 +#include "nsIRunnable.h"
1.11 +
1.12 +#include "nsThreadUtils.h"
1.13 +#include <algorithm>
1.14 +
1.15 +namespace mozilla {
1.16 +namespace net {
1.17 +
1.18 +NS_IMPL_ISUPPORTS(nsPreloadedStream,
1.19 + nsIInputStream,
1.20 + nsIAsyncInputStream)
1.21 +
1.22 +nsPreloadedStream::nsPreloadedStream(nsIAsyncInputStream *aStream,
1.23 + const char *data, uint32_t datalen)
1.24 + : mStream(aStream),
1.25 + mOffset(0),
1.26 + mLen(datalen)
1.27 +{
1.28 + mBuf = (char *) moz_xmalloc(datalen);
1.29 + memcpy(mBuf, data, datalen);
1.30 +}
1.31 +
1.32 +nsPreloadedStream::~nsPreloadedStream()
1.33 +{
1.34 + moz_free(mBuf);
1.35 +}
1.36 +
1.37 +NS_IMETHODIMP
1.38 +nsPreloadedStream::Close()
1.39 +{
1.40 + mLen = 0;
1.41 + return mStream->Close();
1.42 +}
1.43 +
1.44 +
1.45 +NS_IMETHODIMP
1.46 +nsPreloadedStream::Available(uint64_t *_retval)
1.47 +{
1.48 + uint64_t avail = 0;
1.49 +
1.50 + nsresult rv = mStream->Available(&avail);
1.51 + if (NS_FAILED(rv))
1.52 + return rv;
1.53 + *_retval = avail + mLen;
1.54 + return NS_OK;
1.55 +}
1.56 +
1.57 +NS_IMETHODIMP
1.58 +nsPreloadedStream::Read(char *aBuf, uint32_t aCount,
1.59 + uint32_t *_retval)
1.60 +{
1.61 + if (!mLen)
1.62 + return mStream->Read(aBuf, aCount, _retval);
1.63 +
1.64 + uint32_t toRead = std::min(mLen, aCount);
1.65 + memcpy(aBuf, mBuf + mOffset, toRead);
1.66 + mOffset += toRead;
1.67 + mLen -= toRead;
1.68 + *_retval = toRead;
1.69 + return NS_OK;
1.70 +}
1.71 +
1.72 +NS_IMETHODIMP
1.73 +nsPreloadedStream::ReadSegments(nsWriteSegmentFun aWriter,
1.74 + void *aClosure, uint32_t aCount,
1.75 + uint32_t *result)
1.76 +{
1.77 + if (!mLen)
1.78 + return mStream->ReadSegments(aWriter, aClosure, aCount, result);
1.79 +
1.80 + *result = 0;
1.81 + while (mLen > 0 && aCount > 0) {
1.82 + uint32_t toRead = std::min(mLen, aCount);
1.83 + uint32_t didRead = 0;
1.84 + nsresult rv;
1.85 +
1.86 + rv = aWriter(this, aClosure, mBuf + mOffset, *result, toRead, &didRead);
1.87 +
1.88 + if (NS_FAILED(rv))
1.89 + return NS_OK;
1.90 +
1.91 + *result += didRead;
1.92 + mOffset += didRead;
1.93 + mLen -= didRead;
1.94 + aCount -= didRead;
1.95 + }
1.96 +
1.97 + return NS_OK;
1.98 +}
1.99 +
1.100 +NS_IMETHODIMP
1.101 +nsPreloadedStream::IsNonBlocking(bool *_retval)
1.102 +{
1.103 + return mStream->IsNonBlocking(_retval);
1.104 +}
1.105 +
1.106 +NS_IMETHODIMP
1.107 +nsPreloadedStream::CloseWithStatus(nsresult aStatus)
1.108 +{
1.109 + mLen = 0;
1.110 + return mStream->CloseWithStatus(aStatus);
1.111 +}
1.112 +
1.113 +class RunOnThread : public nsRunnable
1.114 +{
1.115 +public:
1.116 + RunOnThread(nsIAsyncInputStream *aStream,
1.117 + nsIInputStreamCallback *aCallback)
1.118 + : mStream(aStream),
1.119 + mCallback(aCallback) {}
1.120 +
1.121 + virtual ~RunOnThread() {}
1.122 +
1.123 + NS_IMETHOD Run()
1.124 + {
1.125 + mCallback->OnInputStreamReady(mStream);
1.126 + return NS_OK;
1.127 + }
1.128 +
1.129 +private:
1.130 + nsCOMPtr<nsIAsyncInputStream> mStream;
1.131 + nsCOMPtr<nsIInputStreamCallback> mCallback;
1.132 +};
1.133 +
1.134 +NS_IMETHODIMP
1.135 +nsPreloadedStream::AsyncWait(nsIInputStreamCallback *aCallback,
1.136 + uint32_t aFlags,
1.137 + uint32_t aRequestedCount,
1.138 + nsIEventTarget *aEventTarget)
1.139 +{
1.140 + if (!mLen)
1.141 + return mStream->AsyncWait(aCallback, aFlags, aRequestedCount,
1.142 + aEventTarget);
1.143 +
1.144 + if (!aCallback)
1.145 + return NS_OK;
1.146 +
1.147 + if (!aEventTarget)
1.148 + return aCallback->OnInputStreamReady(this);
1.149 +
1.150 + nsCOMPtr<nsIRunnable> event =
1.151 + new RunOnThread(this, aCallback);
1.152 + return aEventTarget->Dispatch(event, nsIEventTarget::DISPATCH_NORMAL);
1.153 +}
1.154 +
1.155 +} // namespace net
1.156 +} // namespace mozilla
