The Tor Browser / file diff

     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/netwerk/base/src/ArrayBufferInputStream.cpp	Wed Dec 31 06:09:35 2014 +0100
     1.3 @@ -0,0 +1,119 @@
     1.4 +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
     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 <algorithm>
    1.10 +#include "ArrayBufferInputStream.h"
    1.11 +#include "nsStreamUtils.h"
    1.12 +#include "jsapi.h"
    1.13 +#include "jsfriendapi.h"
    1.14 +
    1.15 +NS_IMPL_ISUPPORTS(ArrayBufferInputStream, nsIArrayBufferInputStream, nsIInputStream);
    1.16 +
    1.17 +ArrayBufferInputStream::ArrayBufferInputStream()
    1.18 +: mBuffer(nullptr)
    1.19 +, mBufferLength(0)
    1.20 +, mOffset(0)
    1.21 +, mPos(0)
    1.22 +, mClosed(false)
    1.23 +{
    1.24 +}
    1.25 +
    1.26 +NS_IMETHODIMP
    1.27 +ArrayBufferInputStream::SetData(JS::Handle<JS::Value> aBuffer,
    1.28 +                                uint32_t aByteOffset,
    1.29 +                                uint32_t aLength,
    1.30 +                                JSContext* aCx)
    1.31 +{
    1.32 +  if (!aBuffer.isObject()) {
    1.33 +    return NS_ERROR_FAILURE;
    1.34 +  }
    1.35 +  JS::RootedObject arrayBuffer(aCx, &aBuffer.toObject());
    1.36 +  if (!JS_IsArrayBufferObject(arrayBuffer)) {
    1.37 +    return NS_ERROR_FAILURE;
    1.38 +  }
    1.39 +
    1.40 +  mArrayBuffer.construct(aCx, aBuffer);
    1.41 +
    1.42 +  uint32_t buflen = JS_GetArrayBufferByteLength(arrayBuffer);
    1.43 +  mOffset = std::min(buflen, aByteOffset);
    1.44 +  mBufferLength = std::min(buflen - mOffset, aLength);
    1.45 +  mBuffer = JS_GetStableArrayBufferData(aCx, arrayBuffer);
    1.46 +  if (!mBuffer) {
    1.47 +      return NS_ERROR_FAILURE;
    1.48 +  }
    1.49 +  return NS_OK;
    1.50 +}
    1.51 +
    1.52 +NS_IMETHODIMP
    1.53 +ArrayBufferInputStream::Close()
    1.54 +{
    1.55 +  mClosed = true;
    1.56 +  return NS_OK;
    1.57 +}
    1.58 +
    1.59 +NS_IMETHODIMP
    1.60 +ArrayBufferInputStream::Available(uint64_t* aCount)
    1.61 +{
    1.62 +  if (mClosed) {
    1.63 +    return NS_BASE_STREAM_CLOSED;
    1.64 +  }
    1.65 +  *aCount = mBufferLength - mPos;
    1.66 +  return NS_OK;
    1.67 +}
    1.68 +
    1.69 +NS_IMETHODIMP
    1.70 +ArrayBufferInputStream::Read(char* aBuf, uint32_t aCount, uint32_t *aReadCount)
    1.71 +{
    1.72 +  return ReadSegments(NS_CopySegmentToBuffer, aBuf, aCount, aReadCount);
    1.73 +}
    1.74 +
    1.75 +NS_IMETHODIMP
    1.76 +ArrayBufferInputStream::ReadSegments(nsWriteSegmentFun writer, void *closure,
    1.77 +                                     uint32_t aCount, uint32_t *result)
    1.78 +{
    1.79 +  NS_ASSERTION(result, "null ptr");
    1.80 +  NS_ASSERTION(mBufferLength >= mPos, "bad stream state");
    1.81 +
    1.82 +  if (mClosed) {
    1.83 +    return NS_BASE_STREAM_CLOSED;
    1.84 +  }
    1.85 +
    1.86 +  uint32_t remaining = mBufferLength - mPos;
    1.87 +  if (!mArrayBuffer.empty()) {
    1.88 +    JSObject* buf = &mArrayBuffer.ref().get().toObject();
    1.89 +    uint32_t byteLength = JS_GetArrayBufferByteLength(buf);
    1.90 +    if (byteLength == 0 && remaining != 0) {
    1.91 +      mClosed = true;
    1.92 +      return NS_BASE_STREAM_CLOSED;
    1.93 +    }
    1.94 +  } else {
    1.95 +    MOZ_ASSERT(remaining == 0, "stream inited incorrectly");
    1.96 +  }
    1.97 +
    1.98 +  if (!remaining) {
    1.99 +    *result = 0;
   1.100 +    return NS_OK;
   1.101 +  }
   1.102 +
   1.103 +  if (aCount > remaining) {
   1.104 +    aCount = remaining;
   1.105 +  }
   1.106 +  nsresult rv = writer(this, closure, (char*)(mBuffer + mOffset) + mPos,
   1.107 +                       0, aCount, result);
   1.108 +  if (NS_SUCCEEDED(rv)) {
   1.109 +    NS_ASSERTION(*result <= aCount,
   1.110 +                 "writer should not write more than we asked it to write");
   1.111 +    mPos += *result;
   1.112 +  }
   1.113 +
   1.114 +  return NS_OK;
   1.115 +}
   1.116 +
   1.117 +NS_IMETHODIMP
   1.118 +ArrayBufferInputStream::IsNonBlocking(bool *aNonBlocking)
   1.119 +{
   1.120 +  *aNonBlocking = true;
   1.121 +  return NS_OK;
   1.122 +}