Emilio Cobos Álvarez
abbeff4fd9
Autogenerated with:
cp dom/.clang-format netwerk/ && mach format netwerk/**.{cpp,h,mm}
Manual fixes:
* mozurl/cbindgen.toml (including headers earlier)
* NetlinkService.h (forward-declaring a necessary struct)
* nsIWebSocketChannel.idl (forward-declaring OriginAttributes).
* nsAsyncRedirectVerifyHelper.h (forward-declaring nsIEventTarget)
* CapsuleEncoder.h gets a missing Capsule forward-decl.
* WebSocketConnectionBase.h gets some missing includes.
* HappyEyeballsConnectionAttempt gets some missing includes.
* nsUDPSocket missed nsSocketTransportService2.h for OnSocketThread().
* HttpInfo missed a forward declaration.
* MockHttpAuth missed an nsCOMPtr include.
* nsHttpAuthManager missed a forward declaration for nsIHttpAuthCache.
* WebTransportSessionProxy was using mozilla::Mutex in a member without
including it.
* win32 fixes to keep include order because windows headers suck.
* nsInputStreamPump needs an Atomics.h include.
Differential Revision: https://phabricator.services.mozilla.com/D311696
140 lines
3.7 KiB
C++
140 lines
3.7 KiB
C++
/* 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 "ArrayBufferInputStream.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "js/ArrayBuffer.h" // JS::{GetArrayBuffer{ByteLength,Data},IsArrayBufferObject}
|
|
#include "js/RootingAPI.h" // JS::{Handle,Rooted}
|
|
#include "js/Value.h" // JS::Value
|
|
#include "mozilla/UniquePtrExtensions.h"
|
|
#include "mozilla/dom/ScriptSettings.h"
|
|
#include "nsStreamUtils.h"
|
|
|
|
using mozilla::dom::RootingCx;
|
|
|
|
NS_IMPL_ISUPPORTS(ArrayBufferInputStream, nsIArrayBufferInputStream,
|
|
nsIInputStream);
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::SetDataFromJS(JS::Handle<JS::Value> aBuffer,
|
|
uint64_t aByteOffset, uint64_t aLength) {
|
|
NS_ASSERT_OWNINGTHREAD(ArrayBufferInputStream);
|
|
|
|
if (!aBuffer.isObject()) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
JS::Rooted<JSObject*> arrayBuffer(RootingCx(), &aBuffer.toObject());
|
|
if (!JS::IsArrayBufferObject(arrayBuffer)) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
uint64_t buflen = JS::GetArrayBufferByteLength(arrayBuffer);
|
|
uint64_t offset = std::min(buflen, aByteOffset);
|
|
uint64_t bufferLength = std::min(buflen - offset, aLength);
|
|
|
|
// Prevent truncation.
|
|
if (bufferLength > UINT32_MAX) {
|
|
return NS_ERROR_INVALID_ARG;
|
|
}
|
|
|
|
mArrayBuffer = mozilla::MakeUniqueFallible<uint8_t[]>(bufferLength);
|
|
if (!mArrayBuffer) {
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
mBufferLength = bufferLength;
|
|
|
|
JS::AutoCheckCannotGC nogc;
|
|
bool isShared;
|
|
uint8_t* src = JS::GetArrayBufferData(arrayBuffer, &isShared, nogc) + offset;
|
|
memcpy(&mArrayBuffer[0], src, mBufferLength);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult ArrayBufferInputStream::SetData(mozilla::UniquePtr<uint8_t[]> aBytes,
|
|
uint64_t aByteLen) {
|
|
mArrayBuffer = std::move(aBytes);
|
|
mBufferLength = aByteLen;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::Close() {
|
|
mClosed = true;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::Available(uint64_t* aCount) {
|
|
if (mClosed) {
|
|
return NS_BASE_STREAM_CLOSED;
|
|
}
|
|
if (mArrayBuffer) {
|
|
*aCount = mBufferLength ? mBufferLength - mPos : 0;
|
|
} else {
|
|
*aCount = 0;
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::StreamStatus() {
|
|
return mClosed ? NS_BASE_STREAM_CLOSED : NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::Read(char* aBuf, uint32_t aCount,
|
|
uint32_t* aReadCount) {
|
|
return ReadSegments(NS_CopySegmentToBuffer, aBuf, aCount, aReadCount);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::ReadSegments(nsWriteSegmentFun writer, void* closure,
|
|
uint32_t aCount, uint32_t* result) {
|
|
NS_ASSERTION(result, "null ptr");
|
|
NS_ASSERTION(mBufferLength >= mPos, "bad stream state");
|
|
|
|
if (mClosed) {
|
|
return NS_BASE_STREAM_CLOSED;
|
|
}
|
|
|
|
MOZ_ASSERT(mArrayBuffer || (mPos == mBufferLength),
|
|
"stream inited incorrectly");
|
|
|
|
*result = 0;
|
|
while (mPos < mBufferLength) {
|
|
uint32_t remaining = mBufferLength - mPos;
|
|
MOZ_ASSERT(mArrayBuffer);
|
|
|
|
uint32_t count = std::min(aCount, remaining);
|
|
if (count == 0) {
|
|
break;
|
|
}
|
|
|
|
uint32_t written;
|
|
nsresult rv = writer(this, closure, (char*)&mArrayBuffer[0] + mPos, *result,
|
|
count, &written);
|
|
if (NS_FAILED(rv)) {
|
|
// InputStreams do not propagate errors to caller.
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_ASSERTION(written <= count,
|
|
"writer should not write more than we asked it to write");
|
|
mPos += written;
|
|
*result += written;
|
|
aCount -= written;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
ArrayBufferInputStream::IsNonBlocking(bool* aNonBlocking) {
|
|
*aNonBlocking = true;
|
|
return NS_OK;
|
|
}
|