diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /dom/base/MultipartBlobImpl.cpp | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip |
Add m-esr52 at 52.6.0
Diffstat (limited to 'dom/base/MultipartBlobImpl.cpp')
-rw-r--r-- | dom/base/MultipartBlobImpl.cpp | 455 |
1 files changed, 455 insertions, 0 deletions
diff --git a/dom/base/MultipartBlobImpl.cpp b/dom/base/MultipartBlobImpl.cpp new file mode 100644 index 000000000..ba26d07f9 --- /dev/null +++ b/dom/base/MultipartBlobImpl.cpp @@ -0,0 +1,455 @@ +/* -*- 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 "MultipartBlobImpl.h" +#include "jsfriendapi.h" +#include "mozilla/dom/BlobSet.h" +#include "mozilla/dom/FileBinding.h" +#include "mozilla/dom/UnionTypes.h" +#include "nsDOMClassInfoID.h" +#include "nsIMultiplexInputStream.h" +#include "nsStringStream.h" +#include "nsTArray.h" +#include "nsJSUtils.h" +#include "nsContentUtils.h" +#include "nsIScriptError.h" +#include "nsIXPConnect.h" +#include <algorithm> + +using namespace mozilla; +using namespace mozilla::dom; + +NS_IMPL_ISUPPORTS_INHERITED0(MultipartBlobImpl, BlobImpl) + +/* static */ already_AddRefed<MultipartBlobImpl> +MultipartBlobImpl::Create(nsTArray<RefPtr<BlobImpl>>&& aBlobImpls, + const nsAString& aName, + const nsAString& aContentType, + ErrorResult& aRv) +{ + RefPtr<MultipartBlobImpl> blobImpl = + new MultipartBlobImpl(Move(aBlobImpls), aName, aContentType); + blobImpl->SetLengthAndModifiedDate(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + return blobImpl.forget(); +} + +/* static */ already_AddRefed<MultipartBlobImpl> +MultipartBlobImpl::Create(nsTArray<RefPtr<BlobImpl>>&& aBlobImpls, + const nsAString& aContentType, + ErrorResult& aRv) +{ + RefPtr<MultipartBlobImpl> blobImpl = + new MultipartBlobImpl(Move(aBlobImpls), aContentType); + blobImpl->SetLengthAndModifiedDate(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + return blobImpl.forget(); +} + +void +MultipartBlobImpl::GetInternalStream(nsIInputStream** aStream, + ErrorResult& aRv) +{ + *aStream = nullptr; + + nsCOMPtr<nsIMultiplexInputStream> stream = + do_CreateInstance("@mozilla.org/io/multiplex-input-stream;1"); + if (NS_WARN_IF(!stream)) { + aRv.Throw(NS_ERROR_FAILURE); + return; + } + + uint32_t i; + for (i = 0; i < mBlobImpls.Length(); i++) { + nsCOMPtr<nsIInputStream> scratchStream; + BlobImpl* blobImpl = mBlobImpls.ElementAt(i).get(); + + blobImpl->GetInternalStream(getter_AddRefs(scratchStream), aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + aRv = stream->AppendStream(scratchStream); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + } + + stream.forget(aStream); +} + +already_AddRefed<BlobImpl> +MultipartBlobImpl::CreateSlice(uint64_t aStart, uint64_t aLength, + const nsAString& aContentType, + ErrorResult& aRv) +{ + // If we clamped to nothing we create an empty blob + nsTArray<RefPtr<BlobImpl>> blobImpls; + + uint64_t length = aLength; + uint64_t skipStart = aStart; + + // Prune the list of blobs if we can + uint32_t i; + for (i = 0; length && skipStart && i < mBlobImpls.Length(); i++) { + BlobImpl* blobImpl = mBlobImpls[i].get(); + + uint64_t l = blobImpl->GetSize(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + if (skipStart < l) { + uint64_t upperBound = std::min<uint64_t>(l - skipStart, length); + + RefPtr<BlobImpl> firstBlobImpl = + blobImpl->CreateSlice(skipStart, upperBound, + aContentType, aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + // Avoid wrapping a single blob inside an MultipartBlobImpl + if (length == upperBound) { + return firstBlobImpl.forget(); + } + + blobImpls.AppendElement(firstBlobImpl); + length -= upperBound; + i++; + break; + } + skipStart -= l; + } + + // Now append enough blobs until we're done + for (; length && i < mBlobImpls.Length(); i++) { + BlobImpl* blobImpl = mBlobImpls[i].get(); + + uint64_t l = blobImpl->GetSize(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + if (length < l) { + RefPtr<BlobImpl> lastBlobImpl = + blobImpl->CreateSlice(0, length, aContentType, aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + blobImpls.AppendElement(lastBlobImpl); + } else { + blobImpls.AppendElement(blobImpl); + } + length -= std::min<uint64_t>(l, length); + } + + // we can create our blob now + RefPtr<BlobImpl> impl = Create(Move(blobImpls), aContentType, aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + return impl.forget(); +} + +void +MultipartBlobImpl::InitializeBlob(ErrorResult& aRv) +{ + SetLengthAndModifiedDate(aRv); + NS_WARNING_ASSERTION(!aRv.Failed(), "SetLengthAndModifiedDate failed"); +} + +void +MultipartBlobImpl::InitializeBlob(JSContext* aCx, + const Sequence<Blob::BlobPart>& aData, + const nsAString& aContentType, + bool aNativeEOL, + ErrorResult& aRv) +{ + mContentType = aContentType; + BlobSet blobSet; + + for (uint32_t i = 0, len = aData.Length(); i < len; ++i) { + const Blob::BlobPart& data = aData[i]; + + if (data.IsBlob()) { + RefPtr<Blob> blob = data.GetAsBlob().get(); + blobSet.AppendBlobImpl(blob->Impl()); + } + + else if (data.IsUSVString()) { + aRv = blobSet.AppendString(data.GetAsUSVString(), aNativeEOL, aCx); + if (aRv.Failed()) { + return; + } + } + + else if (data.IsArrayBuffer()) { + const ArrayBuffer& buffer = data.GetAsArrayBuffer(); + buffer.ComputeLengthAndData(); + aRv = blobSet.AppendVoidPtr(buffer.Data(), buffer.Length()); + if (aRv.Failed()) { + return; + } + } + + else if (data.IsArrayBufferView()) { + const ArrayBufferView& buffer = data.GetAsArrayBufferView(); + buffer.ComputeLengthAndData(); + aRv = blobSet.AppendVoidPtr(buffer.Data(), buffer.Length()); + if (aRv.Failed()) { + return; + } + } + + else { + MOZ_CRASH("Impossible blob data type."); + } + } + + + mBlobImpls = blobSet.GetBlobImpls(); + SetLengthAndModifiedDate(aRv); + NS_WARNING_ASSERTION(!aRv.Failed(), "SetLengthAndModifiedDate failed"); +} + +void +MultipartBlobImpl::SetLengthAndModifiedDate(ErrorResult& aRv) +{ + MOZ_ASSERT(mLength == UINT64_MAX); + MOZ_ASSERT(mLastModificationDate == INT64_MAX); + + uint64_t totalLength = 0; + int64_t lastModified = 0; + bool lastModifiedSet = false; + + for (uint32_t index = 0, count = mBlobImpls.Length(); index < count; index++) { + RefPtr<BlobImpl>& blob = mBlobImpls[index]; + +#ifdef DEBUG + MOZ_ASSERT(!blob->IsSizeUnknown()); + MOZ_ASSERT(!blob->IsDateUnknown()); +#endif + + uint64_t subBlobLength = blob->GetSize(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + MOZ_ASSERT(UINT64_MAX - subBlobLength >= totalLength); + totalLength += subBlobLength; + + if (blob->IsFile()) { + int64_t partLastModified = blob->GetLastModified(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + if (lastModified < partLastModified) { + lastModified = partLastModified; + lastModifiedSet = true; + } + } + } + + mLength = totalLength; + + if (mIsFile) { + // We cannot use PR_Now() because bug 493756 and, for this reason: + // var x = new Date(); var f = new File(...); + // x.getTime() < f.dateModified.getTime() + // could fail. + mLastModificationDate = + lastModifiedSet ? lastModified * PR_USEC_PER_MSEC : JS_Now(); + } +} + +void +MultipartBlobImpl::GetMozFullPathInternal(nsAString& aFilename, + ErrorResult& aRv) const +{ + if (!mIsFromNsIFile || mBlobImpls.Length() == 0) { + BlobImplBase::GetMozFullPathInternal(aFilename, aRv); + return; + } + + BlobImpl* blobImpl = mBlobImpls.ElementAt(0).get(); + if (!blobImpl) { + BlobImplBase::GetMozFullPathInternal(aFilename, aRv); + return; + } + + blobImpl->GetMozFullPathInternal(aFilename, aRv); +} + +nsresult +MultipartBlobImpl::SetMutable(bool aMutable) +{ + nsresult rv; + + // This looks a little sketchy since BlobImpl objects are supposed to be + // threadsafe. However, we try to enforce that all BlobImpl objects must be + // set to immutable *before* being passed to another thread, so this should + // be safe. + if (!aMutable && !mImmutable && !mBlobImpls.IsEmpty()) { + for (uint32_t index = 0, count = mBlobImpls.Length(); + index < count; + index++) { + rv = mBlobImpls[index]->SetMutable(aMutable); + if (NS_WARN_IF(NS_FAILED(rv))) { + return rv; + } + } + } + + rv = BlobImplBase::SetMutable(aMutable); + if (NS_WARN_IF(NS_FAILED(rv))) { + return rv; + } + + MOZ_ASSERT_IF(!aMutable, mImmutable); + + return NS_OK; +} + +void +MultipartBlobImpl::InitializeChromeFile(Blob& aBlob, + const ChromeFilePropertyBag& aBag, + ErrorResult& aRv) +{ + NS_ASSERTION(!mImmutable, "Something went wrong ..."); + + if (mImmutable) { + aRv.Throw(NS_ERROR_UNEXPECTED); + return; + } + + MOZ_ASSERT(nsContentUtils::ThreadsafeIsCallerChrome()); + + mName = aBag.mName; + mContentType = aBag.mType; + mIsFromNsIFile = true; + + // XXXkhuey this is terrible + if (mContentType.IsEmpty()) { + aBlob.GetType(mContentType); + } + + + BlobSet blobSet; + blobSet.AppendBlobImpl(aBlob.Impl()); + mBlobImpls = blobSet.GetBlobImpls(); + + SetLengthAndModifiedDate(aRv); + NS_WARNING_ASSERTION(!aRv.Failed(), "SetLengthAndModifiedDate failed"); +} + +void +MultipartBlobImpl::InitializeChromeFile(nsPIDOMWindowInner* aWindow, + nsIFile* aFile, + const ChromeFilePropertyBag& aBag, + bool aIsFromNsIFile, + ErrorResult& aRv) +{ + NS_ASSERTION(!mImmutable, "Something went wrong ..."); + if (mImmutable) { + aRv.Throw(NS_ERROR_UNEXPECTED); + return; + } + + MOZ_ASSERT(nsContentUtils::IsCallerChrome()); + + mName = aBag.mName; + mContentType = aBag.mType; + mIsFromNsIFile = aIsFromNsIFile; + + bool exists; + aRv = aFile->Exists(&exists); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + if (!exists) { + aRv.Throw(NS_ERROR_FILE_NOT_FOUND); + return; + } + + bool isDir; + aRv = aFile->IsDirectory(&isDir); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + if (isDir) { + aRv.Throw(NS_ERROR_FILE_IS_DIRECTORY); + return; + } + + if (mName.IsEmpty()) { + aFile->GetLeafName(mName); + } + + RefPtr<File> blob = File::CreateFromFile(aWindow, aFile, aBag.mTemporary); + + // Pre-cache size. + blob->GetSize(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + // Pre-cache modified date. + blob->GetLastModified(aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + // XXXkhuey this is terrible + if (mContentType.IsEmpty()) { + blob->GetType(mContentType); + } + + BlobSet blobSet; + blobSet.AppendBlobImpl(static_cast<File*>(blob.get())->Impl()); + mBlobImpls = blobSet.GetBlobImpls(); + + SetLengthAndModifiedDate(aRv); + NS_WARNING_ASSERTION(!aRv.Failed(), "SetLengthAndModifiedDate failed"); +} + +void +MultipartBlobImpl::InitializeChromeFile(nsPIDOMWindowInner* aWindow, + const nsAString& aData, + const ChromeFilePropertyBag& aBag, + ErrorResult& aRv) +{ + nsCOMPtr<nsIFile> file; + aRv = NS_NewLocalFile(aData, false, getter_AddRefs(file)); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + InitializeChromeFile(aWindow, file, aBag, false, aRv); +} + +bool +MultipartBlobImpl::MayBeClonedToOtherThreads() const +{ + for (uint32_t i = 0; i < mBlobImpls.Length(); ++i) { + if (!mBlobImpls[i]->MayBeClonedToOtherThreads()) { + return false; + } + } + + return true; +} |