summaryrefslogtreecommitdiffstats
path: root/dom/url/URLSearchParams.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'dom/url/URLSearchParams.cpp')
-rw-r--r--dom/url/URLSearchParams.cpp549
1 files changed, 549 insertions, 0 deletions
diff --git a/dom/url/URLSearchParams.cpp b/dom/url/URLSearchParams.cpp
new file mode 100644
index 000000000..d9492f81c
--- /dev/null
+++ b/dom/url/URLSearchParams.cpp
@@ -0,0 +1,549 @@
+/* -*- 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 "URLSearchParams.h"
+#include "mozilla/dom/URLSearchParamsBinding.h"
+#include "mozilla/dom/EncodingUtils.h"
+#include "nsDOMString.h"
+#include "nsIInputStream.h"
+#include "nsStringStream.h"
+
+namespace mozilla {
+namespace dom {
+
+bool
+URLParams::Has(const nsAString& aName)
+{
+ for (uint32_t i = 0, len = mParams.Length(); i < len; ++i) {
+ if (mParams[i].mKey.Equals(aName)) {
+ return true;
+ }
+ }
+
+ return false;
+}
+
+void
+URLParams::Get(const nsAString& aName, nsString& aRetval)
+{
+ SetDOMStringToNull(aRetval);
+
+ for (uint32_t i = 0, len = mParams.Length(); i < len; ++i) {
+ if (mParams[i].mKey.Equals(aName)) {
+ aRetval.Assign(mParams[i].mValue);
+ break;
+ }
+ }
+}
+
+void
+URLParams::GetAll(const nsAString& aName, nsTArray<nsString>& aRetval)
+{
+ aRetval.Clear();
+
+ for (uint32_t i = 0, len = mParams.Length(); i < len; ++i) {
+ if (mParams[i].mKey.Equals(aName)) {
+ aRetval.AppendElement(mParams[i].mValue);
+ }
+ }
+}
+
+void
+URLParams::Append(const nsAString& aName, const nsAString& aValue)
+{
+ Param* param = mParams.AppendElement();
+ param->mKey = aName;
+ param->mValue = aValue;
+}
+
+void
+URLParams::Set(const nsAString& aName, const nsAString& aValue)
+{
+ Param* param = nullptr;
+ for (uint32_t i = 0, len = mParams.Length(); i < len;) {
+ if (!mParams[i].mKey.Equals(aName)) {
+ ++i;
+ continue;
+ }
+ if (!param) {
+ param = &mParams[i];
+ ++i;
+ continue;
+ }
+ // Remove duplicates.
+ mParams.RemoveElementAt(i);
+ --len;
+ }
+
+ if (!param) {
+ param = mParams.AppendElement();
+ param->mKey = aName;
+ }
+
+ param->mValue = aValue;
+}
+
+bool
+URLParams::Delete(const nsAString& aName)
+{
+ bool found = false;
+ for (uint32_t i = 0; i < mParams.Length();) {
+ if (mParams[i].mKey.Equals(aName)) {
+ mParams.RemoveElementAt(i);
+ found = true;
+ } else {
+ ++i;
+ }
+ }
+
+ return found;
+}
+
+void
+URLParams::ConvertString(const nsACString& aInput, nsAString& aOutput)
+{
+ aOutput.Truncate();
+
+ if (!mDecoder) {
+ mDecoder = EncodingUtils::DecoderForEncoding("UTF-8");
+ if (!mDecoder) {
+ MOZ_ASSERT(mDecoder, "Failed to create a decoder.");
+ return;
+ }
+ }
+
+ int32_t inputLength = aInput.Length();
+ int32_t outputLength = 0;
+
+ nsresult rv = mDecoder->GetMaxLength(aInput.BeginReading(), inputLength,
+ &outputLength);
+ if (NS_WARN_IF(NS_FAILED(rv))) {
+ return;
+ }
+
+ if (!aOutput.SetLength(outputLength, fallible)) {
+ return;
+ }
+
+ int32_t newOutputLength = outputLength;
+ rv = mDecoder->Convert(aInput.BeginReading(), &inputLength,
+ aOutput.BeginWriting(), &newOutputLength);
+ if (NS_FAILED(rv)) {
+ aOutput.Truncate();
+ return;
+ }
+ if (newOutputLength < outputLength) {
+ aOutput.Truncate(newOutputLength);
+ }
+}
+
+void
+URLParams::DecodeString(const nsACString& aInput, nsAString& aOutput)
+{
+ nsACString::const_iterator start, end;
+ aInput.BeginReading(start);
+ aInput.EndReading(end);
+
+ nsCString unescaped;
+
+ while (start != end) {
+ // replace '+' with U+0020
+ if (*start == '+') {
+ unescaped.Append(' ');
+ ++start;
+ continue;
+ }
+
+ // Percent decode algorithm
+ if (*start == '%') {
+ nsACString::const_iterator first(start);
+ ++first;
+
+ nsACString::const_iterator second(first);
+ ++second;
+
+#define ASCII_HEX_DIGIT( x ) \
+ ((x >= 0x41 && x <= 0x46) || \
+ (x >= 0x61 && x <= 0x66) || \
+ (x >= 0x30 && x <= 0x39))
+
+#define HEX_DIGIT( x ) \
+ (*x >= 0x30 && *x <= 0x39 \
+ ? *x - 0x30 \
+ : (*x >= 0x41 && *x <= 0x46 \
+ ? *x - 0x37 \
+ : *x - 0x57))
+
+ if (first != end && second != end &&
+ ASCII_HEX_DIGIT(*first) && ASCII_HEX_DIGIT(*second)) {
+ unescaped.Append(HEX_DIGIT(first) * 16 + HEX_DIGIT(second));
+ start = ++second;
+ continue;
+
+ } else {
+ unescaped.Append('%');
+ ++start;
+ continue;
+ }
+ }
+
+ unescaped.Append(*start);
+ ++start;
+ }
+
+ ConvertString(unescaped, aOutput);
+}
+
+void
+URLParams::ParseInput(const nsACString& aInput)
+{
+ // Remove all the existing data before parsing a new input.
+ DeleteAll();
+
+ nsACString::const_iterator start, end;
+ aInput.BeginReading(start);
+ aInput.EndReading(end);
+ nsACString::const_iterator iter(start);
+
+ while (start != end) {
+ nsAutoCString string;
+
+ if (FindCharInReadable('&', iter, end)) {
+ string.Assign(Substring(start, iter));
+ start = ++iter;
+ } else {
+ string.Assign(Substring(start, end));
+ start = end;
+ }
+
+ if (string.IsEmpty()) {
+ continue;
+ }
+
+ nsACString::const_iterator eqStart, eqEnd;
+ string.BeginReading(eqStart);
+ string.EndReading(eqEnd);
+ nsACString::const_iterator eqIter(eqStart);
+
+ nsAutoCString name;
+ nsAutoCString value;
+
+ if (FindCharInReadable('=', eqIter, eqEnd)) {
+ name.Assign(Substring(eqStart, eqIter));
+
+ ++eqIter;
+ value.Assign(Substring(eqIter, eqEnd));
+ } else {
+ name.Assign(string);
+ }
+
+ nsAutoString decodedName;
+ DecodeString(name, decodedName);
+
+ nsAutoString decodedValue;
+ DecodeString(value, decodedValue);
+
+ Append(decodedName, decodedValue);
+ }
+}
+
+namespace {
+
+void SerializeString(const nsCString& aInput, nsAString& aValue)
+{
+ const unsigned char* p = (const unsigned char*) aInput.get();
+ const unsigned char* end = p + aInput.Length();
+
+ while (p != end) {
+ // ' ' to '+'
+ if (*p == 0x20) {
+ aValue.Append(0x2B);
+ // Percent Encode algorithm
+ } else if (*p == 0x2A || *p == 0x2D || *p == 0x2E ||
+ (*p >= 0x30 && *p <= 0x39) ||
+ (*p >= 0x41 && *p <= 0x5A) || *p == 0x5F ||
+ (*p >= 0x61 && *p <= 0x7A)) {
+ aValue.Append(*p);
+ } else {
+ aValue.AppendPrintf("%%%.2X", *p);
+ }
+
+ ++p;
+ }
+}
+
+} // namespace
+
+void
+URLParams::Serialize(nsAString& aValue) const
+{
+ aValue.Truncate();
+ bool first = true;
+
+ for (uint32_t i = 0, len = mParams.Length(); i < len; ++i) {
+ if (first) {
+ first = false;
+ } else {
+ aValue.Append('&');
+ }
+
+ SerializeString(NS_ConvertUTF16toUTF8(mParams[i].mKey), aValue);
+ aValue.Append('=');
+ SerializeString(NS_ConvertUTF16toUTF8(mParams[i].mValue), aValue);
+ }
+}
+
+NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(URLSearchParams, mParent, mObserver)
+NS_IMPL_CYCLE_COLLECTING_ADDREF(URLSearchParams)
+NS_IMPL_CYCLE_COLLECTING_RELEASE(URLSearchParams)
+
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(URLSearchParams)
+ NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
+ NS_INTERFACE_MAP_ENTRY(nsIXHRSendable)
+ NS_INTERFACE_MAP_ENTRY(nsISupports)
+NS_INTERFACE_MAP_END
+
+URLSearchParams::URLSearchParams(nsISupports* aParent,
+ URLSearchParamsObserver* aObserver)
+ : mParams(new URLParams())
+ , mParent(aParent)
+ , mObserver(aObserver)
+{
+}
+
+URLSearchParams::URLSearchParams(nsISupports* aParent,
+ const URLSearchParams& aOther)
+ : mParams(new URLParams(*aOther.mParams.get()))
+ , mParent(aParent)
+ , mObserver(nullptr)
+{
+}
+
+URLSearchParams::~URLSearchParams()
+{
+ DeleteAll();
+}
+
+JSObject*
+URLSearchParams::WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto)
+{
+ return URLSearchParamsBinding::Wrap(aCx, this, aGivenProto);
+}
+
+/* static */ already_AddRefed<URLSearchParams>
+URLSearchParams::Constructor(const GlobalObject& aGlobal,
+ const nsAString& aInit,
+ ErrorResult& aRv)
+{
+ RefPtr<URLSearchParams> sp =
+ new URLSearchParams(aGlobal.GetAsSupports(), nullptr);
+
+ NS_ConvertUTF16toUTF8 input(aInit);
+
+ if (StringBeginsWith(input, NS_LITERAL_CSTRING("?"))) {
+ sp->ParseInput(Substring(input, 1, input.Length() - 1));
+ } else {
+ sp->ParseInput(input);
+ }
+
+ return sp.forget();
+}
+
+/* static */ already_AddRefed<URLSearchParams>
+URLSearchParams::Constructor(const GlobalObject& aGlobal,
+ URLSearchParams& aInit,
+ ErrorResult& aRv)
+{
+ RefPtr<URLSearchParams> sp =
+ new URLSearchParams(aGlobal.GetAsSupports(), aInit);
+
+ return sp.forget();
+}
+
+void
+URLSearchParams::ParseInput(const nsACString& aInput)
+{
+ mParams->ParseInput(aInput);
+}
+
+void
+URLSearchParams::Get(const nsAString& aName, nsString& aRetval)
+{
+ return mParams->Get(aName, aRetval);
+}
+
+void
+URLSearchParams::GetAll(const nsAString& aName, nsTArray<nsString>& aRetval)
+{
+ return mParams->GetAll(aName, aRetval);
+}
+
+void
+URLSearchParams::Set(const nsAString& aName, const nsAString& aValue)
+{
+ mParams->Set(aName, aValue);
+ NotifyObserver();
+}
+
+void
+URLSearchParams::Append(const nsAString& aName, const nsAString& aValue)
+{
+ mParams->Append(aName, aValue);
+ NotifyObserver();
+}
+
+bool
+URLSearchParams::Has(const nsAString& aName)
+{
+ return mParams->Has(aName);
+}
+
+void
+URLSearchParams::Delete(const nsAString& aName)
+{
+ if (mParams->Delete(aName)) {
+ NotifyObserver();
+ }
+}
+
+void
+URLSearchParams::DeleteAll()
+{
+ mParams->DeleteAll();
+}
+
+void
+URLSearchParams::Serialize(nsAString& aValue) const
+{
+ mParams->Serialize(aValue);
+}
+
+void
+URLSearchParams::NotifyObserver()
+{
+ if (mObserver) {
+ mObserver->URLSearchParamsUpdated(this);
+ }
+}
+
+uint32_t
+URLSearchParams::GetIterableLength() const
+{
+ return mParams->Length();
+}
+
+const nsAString&
+URLSearchParams::GetKeyAtIndex(uint32_t aIndex) const
+{
+ return mParams->GetKeyAtIndex(aIndex);
+}
+
+const nsAString&
+URLSearchParams::GetValueAtIndex(uint32_t aIndex) const
+{
+ return mParams->GetValueAtIndex(aIndex);
+}
+
+// Helper functions for structured cloning
+inline bool
+ReadString(JSStructuredCloneReader* aReader, nsString& aString)
+{
+ MOZ_ASSERT(aReader);
+
+ bool read;
+ uint32_t nameLength, zero;
+ read = JS_ReadUint32Pair(aReader, &nameLength, &zero);
+ if (!read) {
+ return false;
+ }
+ MOZ_ASSERT(zero == 0);
+ aString.SetLength(nameLength);
+ size_t charSize = sizeof(nsString::char_type);
+ read = JS_ReadBytes(aReader, (void*) aString.BeginWriting(),
+ nameLength * charSize);
+ if (!read) {
+ return false;
+ }
+
+ return true;
+}
+
+inline bool
+WriteString(JSStructuredCloneWriter* aWriter, const nsString& aString)
+{
+ MOZ_ASSERT(aWriter);
+
+ size_t charSize = sizeof(nsString::char_type);
+ return JS_WriteUint32Pair(aWriter, aString.Length(), 0) &&
+ JS_WriteBytes(aWriter, aString.get(), aString.Length() * charSize);
+}
+
+bool
+URLParams::WriteStructuredClone(JSStructuredCloneWriter* aWriter) const
+{
+ const uint32_t& nParams = mParams.Length();
+ if (!JS_WriteUint32Pair(aWriter, nParams, 0)) {
+ return false;
+ }
+ for (uint32_t i = 0; i < nParams; ++i) {
+ if (!WriteString(aWriter, mParams[i].mKey) ||
+ !WriteString(aWriter, mParams[i].mValue)) {
+ return false;
+ }
+ }
+ return true;
+}
+
+bool
+URLParams::ReadStructuredClone(JSStructuredCloneReader* aReader)
+{
+ MOZ_ASSERT(aReader);
+
+ DeleteAll();
+
+ uint32_t nParams, zero;
+ nsAutoString key, value;
+ if (!JS_ReadUint32Pair(aReader, &nParams, &zero)) {
+ return false;
+ }
+ MOZ_ASSERT(zero == 0);
+ for (uint32_t i = 0; i < nParams; ++i) {
+ if (!ReadString(aReader, key) || !ReadString(aReader, value)) {
+ return false;
+ }
+ Append(key, value);
+ }
+ return true;
+}
+
+bool
+URLSearchParams::WriteStructuredClone(JSStructuredCloneWriter* aWriter) const
+{
+ return mParams->WriteStructuredClone(aWriter);
+}
+
+bool
+URLSearchParams::ReadStructuredClone(JSStructuredCloneReader* aReader)
+{
+ return mParams->ReadStructuredClone(aReader);
+}
+
+NS_IMETHODIMP
+URLSearchParams::GetSendInfo(nsIInputStream** aBody, uint64_t* aContentLength,
+ nsACString& aContentType, nsACString& aCharset)
+{
+ aContentType.AssignLiteral("application/x-www-form-urlencoded");
+ aCharset.AssignLiteral("UTF-8");
+
+ nsAutoString serialized;
+ Serialize(serialized);
+ NS_ConvertUTF16toUTF8 converted(serialized);
+ *aContentLength = converted.Length();
+ return NS_NewCStringInputStream(aBody, converted);
+}
+
+} // namespace dom
+} // namespace mozilla