diff options
Diffstat (limited to 'dom/base/FormData.cpp')
| -rw-r--r-- | dom/base/FormData.cpp | 430 |
1 files changed, 430 insertions, 0 deletions
diff --git a/dom/base/FormData.cpp b/dom/base/FormData.cpp new file mode 100644 index 000000000..6095286be --- /dev/null +++ b/dom/base/FormData.cpp @@ -0,0 +1,430 @@ +/* -*- 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 "FormData.h" +#include "nsIVariant.h" +#include "nsIInputStream.h" +#include "mozilla/dom/File.h" +#include "mozilla/dom/Directory.h" +#include "mozilla/dom/HTMLFormElement.h" + +#include "MultipartBlobImpl.h" + +using namespace mozilla; +using namespace mozilla::dom; + +FormData::FormData(nsISupports* aOwner) + : HTMLFormSubmission(NS_LITERAL_CSTRING("UTF-8"), nullptr) + , mOwner(aOwner) +{ +} + +namespace { + +already_AddRefed<File> +GetOrCreateFileCalledBlob(Blob& aBlob, ErrorResult& aRv) +{ + // If this is file, we can just use it + RefPtr<File> file = aBlob.ToFile(); + if (file) { + return file.forget(); + } + + // Forcing 'blob' as filename + file = aBlob.ToFile(NS_LITERAL_STRING("blob"), aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + return file.forget(); +} + +already_AddRefed<File> +GetBlobForFormDataStorage(Blob& aBlob, const Optional<nsAString>& aFilename, + ErrorResult& aRv) +{ + // Forcing a filename + if (aFilename.WasPassed()) { + RefPtr<File> file = aBlob.ToFile(aFilename.Value(), aRv); + if (NS_WARN_IF(aRv.Failed())) { + return nullptr; + } + + return file.forget(); + } + + return GetOrCreateFileCalledBlob(aBlob, aRv); +} + +} // namespace + +// ------------------------------------------------------------------------- +// nsISupports + +NS_IMPL_CYCLE_COLLECTION_CLASS(FormData) + +NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(FormData) + NS_IMPL_CYCLE_COLLECTION_UNLINK(mOwner) + + for (uint32_t i = 0, len = tmp->mFormData.Length(); i < len; ++i) { + ImplCycleCollectionUnlink(tmp->mFormData[i].value); + } + + NS_IMPL_CYCLE_COLLECTION_UNLINK_PRESERVED_WRAPPER +NS_IMPL_CYCLE_COLLECTION_UNLINK_END + +NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(FormData) + NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mOwner) + + for (uint32_t i = 0, len = tmp->mFormData.Length(); i < len; ++i) { + ImplCycleCollectionTraverse(cb, tmp->mFormData[i].value, + "mFormData[i].GetAsBlob()", 0); + } + + NS_IMPL_CYCLE_COLLECTION_TRAVERSE_SCRIPT_OBJECTS +NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END + +NS_IMPL_CYCLE_COLLECTION_TRACE_WRAPPERCACHE(FormData) + +NS_IMPL_CYCLE_COLLECTING_ADDREF(FormData) +NS_IMPL_CYCLE_COLLECTING_RELEASE(FormData) + +NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(FormData) + NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY + NS_INTERFACE_MAP_ENTRY(nsIDOMFormData) + NS_INTERFACE_MAP_ENTRY(nsIXHRSendable) + NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIDOMFormData) +NS_INTERFACE_MAP_END + +// ------------------------------------------------------------------------- +// HTMLFormSubmission +nsresult +FormData::GetEncodedSubmission(nsIURI* aURI, + nsIInputStream** aPostDataStream) +{ + NS_NOTREACHED("Shouldn't call FormData::GetEncodedSubmission"); + return NS_OK; +} + +void +FormData::Append(const nsAString& aName, const nsAString& aValue, + ErrorResult& aRv) +{ + AddNameValuePair(aName, aValue); +} + +void +FormData::Append(const nsAString& aName, Blob& aBlob, + const Optional<nsAString>& aFilename, + ErrorResult& aRv) +{ + RefPtr<File> file = GetBlobForFormDataStorage(aBlob, aFilename, aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + AddNameBlobOrNullPair(aName, file); +} + +void +FormData::Append(const nsAString& aName, Directory* aDirectory) +{ + AddNameDirectoryPair(aName, aDirectory); +} + +void +FormData::Delete(const nsAString& aName) +{ + // We have to use this slightly awkward for loop since uint32_t >= 0 is an + // error for being always true. + for (uint32_t i = mFormData.Length(); i-- > 0; ) { + if (aName.Equals(mFormData[i].name)) { + mFormData.RemoveElementAt(i); + } + } +} + +void +FormData::Get(const nsAString& aName, + Nullable<OwningBlobOrDirectoryOrUSVString>& aOutValue) +{ + for (uint32_t i = 0; i < mFormData.Length(); ++i) { + if (aName.Equals(mFormData[i].name)) { + aOutValue.SetValue() = mFormData[i].value; + return; + } + } + + aOutValue.SetNull(); +} + +void +FormData::GetAll(const nsAString& aName, + nsTArray<OwningBlobOrDirectoryOrUSVString>& aValues) +{ + for (uint32_t i = 0; i < mFormData.Length(); ++i) { + if (aName.Equals(mFormData[i].name)) { + OwningBlobOrDirectoryOrUSVString* element = aValues.AppendElement(); + *element = mFormData[i].value; + } + } +} + +bool +FormData::Has(const nsAString& aName) +{ + for (uint32_t i = 0; i < mFormData.Length(); ++i) { + if (aName.Equals(mFormData[i].name)) { + return true; + } + } + + return false; +} + +nsresult +FormData::AddNameBlobOrNullPair(const nsAString& aName, Blob* aBlob) +{ + RefPtr<File> file; + + if (!aBlob) { + FormDataTuple* data = mFormData.AppendElement(); + SetNameValuePair(data, aName, EmptyString(), true /* aWasNullBlob */); + return NS_OK; + } + + ErrorResult rv; + file = GetOrCreateFileCalledBlob(*aBlob, rv); + if (NS_WARN_IF(rv.Failed())) { + return rv.StealNSResult(); + } + + FormDataTuple* data = mFormData.AppendElement(); + SetNameFilePair(data, aName, file); + return NS_OK; +} + +nsresult +FormData::AddNameDirectoryPair(const nsAString& aName, Directory* aDirectory) +{ + MOZ_ASSERT(aDirectory); + + FormDataTuple* data = mFormData.AppendElement(); + SetNameDirectoryPair(data, aName, aDirectory); + return NS_OK; +} + +FormData::FormDataTuple* +FormData::RemoveAllOthersAndGetFirstFormDataTuple(const nsAString& aName) +{ + FormDataTuple* lastFoundTuple = nullptr; + uint32_t lastFoundIndex = mFormData.Length(); + // We have to use this slightly awkward for loop since uint32_t >= 0 is an + // error for being always true. + for (uint32_t i = mFormData.Length(); i-- > 0; ) { + if (aName.Equals(mFormData[i].name)) { + if (lastFoundTuple) { + // The one we found earlier was not the first one, we can remove it. + mFormData.RemoveElementAt(lastFoundIndex); + } + + lastFoundTuple = &mFormData[i]; + lastFoundIndex = i; + } + } + + return lastFoundTuple; +} + +void +FormData::Set(const nsAString& aName, Blob& aBlob, + const Optional<nsAString>& aFilename, + ErrorResult& aRv) +{ + FormDataTuple* tuple = RemoveAllOthersAndGetFirstFormDataTuple(aName); + if (tuple) { + RefPtr<File> file = GetBlobForFormDataStorage(aBlob, aFilename, aRv); + if (NS_WARN_IF(aRv.Failed())) { + return; + } + + SetNameFilePair(tuple, aName, file); + } else { + Append(aName, aBlob, aFilename, aRv); + } +} + +void +FormData::Set(const nsAString& aName, const nsAString& aValue, + ErrorResult& aRv) +{ + FormDataTuple* tuple = RemoveAllOthersAndGetFirstFormDataTuple(aName); + if (tuple) { + SetNameValuePair(tuple, aName, aValue); + } else { + Append(aName, aValue, aRv); + } +} + +uint32_t +FormData::GetIterableLength() const +{ + return mFormData.Length(); +} + +const nsAString& +FormData::GetKeyAtIndex(uint32_t aIndex) const +{ + MOZ_ASSERT(aIndex < mFormData.Length()); + return mFormData[aIndex].name; +} + +const OwningBlobOrDirectoryOrUSVString& +FormData::GetValueAtIndex(uint32_t aIndex) const +{ + MOZ_ASSERT(aIndex < mFormData.Length()); + return mFormData[aIndex].value; +} + +void +FormData::SetNameValuePair(FormDataTuple* aData, + const nsAString& aName, + const nsAString& aValue, + bool aWasNullBlob) +{ + MOZ_ASSERT(aData); + aData->name = aName; + aData->wasNullBlob = aWasNullBlob; + aData->value.SetAsUSVString() = aValue; +} + +void +FormData::SetNameFilePair(FormDataTuple* aData, + const nsAString& aName, + File* aFile) +{ + MOZ_ASSERT(aData); + MOZ_ASSERT(aFile); + + aData->name = aName; + aData->wasNullBlob = false; + aData->value.SetAsBlob() = aFile; +} + +void +FormData::SetNameDirectoryPair(FormDataTuple* aData, + const nsAString& aName, + Directory* aDirectory) +{ + MOZ_ASSERT(aData); + MOZ_ASSERT(aDirectory); + + aData->name = aName; + aData->wasNullBlob = false; + aData->value.SetAsDirectory() = aDirectory; +} + +// ------------------------------------------------------------------------- +// nsIDOMFormData + +NS_IMETHODIMP +FormData::Append(const nsAString& aName, nsIVariant* aValue) +{ + uint16_t dataType; + nsresult rv = aValue->GetDataType(&dataType); + NS_ENSURE_SUCCESS(rv, rv); + + if (dataType == nsIDataType::VTYPE_INTERFACE || + dataType == nsIDataType::VTYPE_INTERFACE_IS) { + nsCOMPtr<nsISupports> supports; + nsID *iid; + rv = aValue->GetAsInterface(&iid, getter_AddRefs(supports)); + NS_ENSURE_SUCCESS(rv, rv); + + free(iid); + + nsCOMPtr<nsIDOMBlob> domBlob = do_QueryInterface(supports); + RefPtr<Blob> blob = static_cast<Blob*>(domBlob.get()); + if (domBlob) { + Optional<nsAString> temp; + ErrorResult rv; + Append(aName, *blob, temp, rv); + if (NS_WARN_IF(rv.Failed())) { + return rv.StealNSResult(); + } + + return NS_OK; + } + } + + char16_t* stringData = nullptr; + uint32_t stringLen = 0; + rv = aValue->GetAsWStringWithSize(&stringLen, &stringData); + NS_ENSURE_SUCCESS(rv, rv); + + nsString valAsString; + valAsString.Adopt(stringData, stringLen); + + ErrorResult error; + Append(aName, valAsString, error); + if (NS_WARN_IF(error.Failed())) { + return error.StealNSResult(); + } + + return NS_OK; +} + +/* virtual */ JSObject* +FormData::WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto) +{ + return FormDataBinding::Wrap(aCx, this, aGivenProto); +} + +/* static */ already_AddRefed<FormData> +FormData::Constructor(const GlobalObject& aGlobal, + const Optional<NonNull<HTMLFormElement> >& aFormElement, + ErrorResult& aRv) +{ + RefPtr<FormData> formData = new FormData(aGlobal.GetAsSupports()); + if (aFormElement.WasPassed()) { + aRv = aFormElement.Value().WalkFormElements(formData); + } + return formData.forget(); +} + +// ------------------------------------------------------------------------- +// nsIXHRSendable + +NS_IMETHODIMP +FormData::GetSendInfo(nsIInputStream** aBody, uint64_t* aContentLength, + nsACString& aContentType, nsACString& aCharset) +{ + FSMultipartFormData fs(NS_LITERAL_CSTRING("UTF-8"), nullptr); + + for (uint32_t i = 0; i < mFormData.Length(); ++i) { + if (mFormData[i].wasNullBlob) { + MOZ_ASSERT(mFormData[i].value.IsUSVString()); + fs.AddNameBlobOrNullPair(mFormData[i].name, nullptr); + } else if (mFormData[i].value.IsUSVString()) { + fs.AddNameValuePair(mFormData[i].name, + mFormData[i].value.GetAsUSVString()); + } else if (mFormData[i].value.IsBlob()) { + fs.AddNameBlobOrNullPair(mFormData[i].name, + mFormData[i].value.GetAsBlob()); + } else { + MOZ_ASSERT(mFormData[i].value.IsDirectory()); + fs.AddNameDirectoryPair(mFormData[i].name, + mFormData[i].value.GetAsDirectory()); + } + } + + fs.GetContentType(aContentType); + aCharset.Truncate(); + *aContentLength = 0; + NS_ADDREF(*aBody = fs.GetSubmissionBody(aContentLength)); + + return NS_OK; +} |