summaryrefslogtreecommitdiffstats
path: root/dom/base/MultipartBlobImpl.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'dom/base/MultipartBlobImpl.cpp')
-rw-r--r--dom/base/MultipartBlobImpl.cpp455
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;
+}