зеркало из https://github.com/mozilla/gecko-dev.git
146 строки
4.2 KiB
C++
146 строки
4.2 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* 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 "StreamBlobImpl.h"
|
|
#include "nsStringStream.h"
|
|
|
|
namespace mozilla {
|
|
namespace dom {
|
|
|
|
NS_IMPL_ISUPPORTS_INHERITED(StreamBlobImpl, BlobImpl, nsIMemoryReporter)
|
|
|
|
/* static */ already_AddRefed<StreamBlobImpl>
|
|
StreamBlobImpl::Create(nsIInputStream* aInputStream,
|
|
const nsAString& aContentType,
|
|
uint64_t aLength)
|
|
{
|
|
RefPtr<StreamBlobImpl> blobImplStream =
|
|
new StreamBlobImpl(aInputStream, aContentType, aLength);
|
|
blobImplStream->MaybeRegisterMemoryReporter();
|
|
return blobImplStream.forget();
|
|
}
|
|
|
|
/* static */ already_AddRefed<StreamBlobImpl>
|
|
StreamBlobImpl::Create(nsIInputStream* aInputStream,
|
|
const nsAString& aName,
|
|
const nsAString& aContentType,
|
|
int64_t aLastModifiedDate,
|
|
uint64_t aLength)
|
|
{
|
|
RefPtr<StreamBlobImpl> blobImplStream =
|
|
new StreamBlobImpl(aInputStream, aName, aContentType, aLastModifiedDate,
|
|
aLength);
|
|
blobImplStream->MaybeRegisterMemoryReporter();
|
|
return blobImplStream.forget();
|
|
}
|
|
|
|
StreamBlobImpl::StreamBlobImpl(nsIInputStream* aInputStream,
|
|
const nsAString& aContentType,
|
|
uint64_t aLength)
|
|
: BaseBlobImpl(aContentType, aLength)
|
|
, mInputStream(aInputStream)
|
|
, mIsDirectory(false)
|
|
, mFileId(-1)
|
|
{
|
|
mImmutable = true;
|
|
}
|
|
|
|
StreamBlobImpl::StreamBlobImpl(StreamBlobImpl* aOther,
|
|
const nsAString& aContentType,
|
|
uint64_t aStart, uint64_t aLength)
|
|
: BaseBlobImpl(aContentType, aOther->mStart + aStart, aLength)
|
|
, mInputStream(new SlicedInputStream(aOther->mInputStream, aStart, aLength))
|
|
, mIsDirectory(false)
|
|
, mFileId(-1)
|
|
{
|
|
mImmutable = true;
|
|
}
|
|
|
|
StreamBlobImpl::StreamBlobImpl(nsIInputStream* aInputStream,
|
|
const nsAString& aName,
|
|
const nsAString& aContentType,
|
|
int64_t aLastModifiedDate,
|
|
uint64_t aLength)
|
|
: BaseBlobImpl(aName, aContentType, aLength, aLastModifiedDate)
|
|
, mInputStream(aInputStream)
|
|
, mIsDirectory(false)
|
|
, mFileId(-1)
|
|
{
|
|
mImmutable = true;
|
|
}
|
|
|
|
StreamBlobImpl::~StreamBlobImpl()
|
|
{
|
|
UnregisterWeakMemoryReporter(this);
|
|
}
|
|
|
|
void
|
|
StreamBlobImpl::GetInternalStream(nsIInputStream** aStream, ErrorResult& aRv)
|
|
{
|
|
nsCOMPtr<nsIInputStream> clonedStream;
|
|
nsCOMPtr<nsIInputStream> replacementStream;
|
|
|
|
aRv = NS_CloneInputStream(mInputStream, getter_AddRefs(clonedStream),
|
|
getter_AddRefs(replacementStream));
|
|
if (NS_WARN_IF(aRv.Failed())) {
|
|
return;
|
|
}
|
|
|
|
if (replacementStream) {
|
|
mInputStream = replacementStream.forget();
|
|
}
|
|
|
|
clonedStream.forget(aStream);
|
|
}
|
|
|
|
already_AddRefed<BlobImpl>
|
|
StreamBlobImpl::CreateSlice(uint64_t aStart, uint64_t aLength,
|
|
const nsAString& aContentType, ErrorResult& aRv)
|
|
{
|
|
if (!aLength) {
|
|
RefPtr<BlobImpl> impl = new EmptyBlobImpl(aContentType);
|
|
return impl.forget();
|
|
}
|
|
|
|
RefPtr<BlobImpl> impl =
|
|
new StreamBlobImpl(this, aContentType, aStart, aLength);
|
|
return impl.forget();
|
|
}
|
|
|
|
void
|
|
StreamBlobImpl::MaybeRegisterMemoryReporter()
|
|
{
|
|
// We report only stringInputStream.
|
|
nsCOMPtr<nsIStringInputStream> stringInputStream =
|
|
do_QueryInterface(mInputStream);
|
|
if (!stringInputStream) {
|
|
return;
|
|
}
|
|
|
|
RegisterWeakMemoryReporter(this);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
StreamBlobImpl::CollectReports(nsIHandleReportCallback* aHandleReport,
|
|
nsISupports* aData, bool aAnonymize)
|
|
{
|
|
nsCOMPtr<nsIStringInputStream> stringInputStream =
|
|
do_QueryInterface(mInputStream);
|
|
if (!stringInputStream) {
|
|
return NS_OK;
|
|
}
|
|
|
|
MOZ_COLLECT_REPORT(
|
|
"explicit/dom/memory-file-data/stream", KIND_HEAP, UNITS_BYTES,
|
|
stringInputStream->SizeOfIncludingThis(MallocSizeOf),
|
|
"Memory used to back a File/Blob based on an input stream.");
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
} // namespace dom
|
|
} // namespace mozilla
|