michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "URLSearchParams.h"
michael@0: #include "mozilla/dom/URLSearchParamsBinding.h"
michael@0: 
michael@0: namespace mozilla {
michael@0: namespace dom {
michael@0: 
michael@0: NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE_1(URLSearchParams, mObservers)
michael@0: NS_IMPL_CYCLE_COLLECTING_ADDREF(URLSearchParams)
michael@0: NS_IMPL_CYCLE_COLLECTING_RELEASE(URLSearchParams)
michael@0: 
michael@0: NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(URLSearchParams)
michael@0:   NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
michael@0:   NS_INTERFACE_MAP_ENTRY(nsISupports)
michael@0: NS_INTERFACE_MAP_END
michael@0: 
michael@0: URLSearchParams::URLSearchParams()
michael@0: {
michael@0:   SetIsDOMBinding();
michael@0: }
michael@0: 
michael@0: URLSearchParams::~URLSearchParams()
michael@0: {
michael@0:   DeleteAll();
michael@0: }
michael@0: 
michael@0: JSObject*
michael@0: URLSearchParams::WrapObject(JSContext* aCx)
michael@0: {
michael@0:   return URLSearchParamsBinding::Wrap(aCx, this);
michael@0: }
michael@0: 
michael@0: /* static */ already_AddRefed<URLSearchParams>
michael@0: URLSearchParams::Constructor(const GlobalObject& aGlobal,
michael@0:                              const nsAString& aInit,
michael@0:                              ErrorResult& aRv)
michael@0: {
michael@0:   nsRefPtr<URLSearchParams> sp = new URLSearchParams();
michael@0:   sp->ParseInput(NS_ConvertUTF16toUTF8(aInit), nullptr);
michael@0:   return sp.forget();
michael@0: }
michael@0: 
michael@0: /* static */ already_AddRefed<URLSearchParams>
michael@0: URLSearchParams::Constructor(const GlobalObject& aGlobal,
michael@0:                              URLSearchParams& aInit,
michael@0:                              ErrorResult& aRv)
michael@0: {
michael@0:   nsRefPtr<URLSearchParams> sp = new URLSearchParams();
michael@0:   aInit.mSearchParams.EnumerateRead(CopyEnumerator, sp);
michael@0:   return sp.forget();
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::ParseInput(const nsACString& aInput,
michael@0:                             URLSearchParamsObserver* aObserver)
michael@0: {
michael@0:   // Remove all the existing data before parsing a new input.
michael@0:   DeleteAll();
michael@0: 
michael@0:   nsACString::const_iterator start, end;
michael@0:   aInput.BeginReading(start);
michael@0:   aInput.EndReading(end);
michael@0:   nsACString::const_iterator iter(start);
michael@0: 
michael@0:   while (start != end) {
michael@0:     nsAutoCString string;
michael@0: 
michael@0:     if (FindCharInReadable('&', iter, end)) {
michael@0:       string.Assign(Substring(start, iter));
michael@0:       start = ++iter;
michael@0:     } else {
michael@0:       string.Assign(Substring(start, end));
michael@0:       start = end;
michael@0:     }
michael@0: 
michael@0:     if (string.IsEmpty()) {
michael@0:       continue;
michael@0:     }
michael@0: 
michael@0:     nsACString::const_iterator eqStart, eqEnd;
michael@0:     string.BeginReading(eqStart);
michael@0:     string.EndReading(eqEnd);
michael@0:     nsACString::const_iterator eqIter(eqStart);
michael@0: 
michael@0:     nsAutoCString name;
michael@0:     nsAutoCString value;
michael@0: 
michael@0:     if (FindCharInReadable('=', eqIter, eqEnd)) {
michael@0:       name.Assign(Substring(eqStart, eqIter));
michael@0: 
michael@0:       ++eqIter;
michael@0:       value.Assign(Substring(eqIter, eqEnd));
michael@0:     } else {
michael@0:       name.Assign(string);
michael@0:     }
michael@0: 
michael@0:     nsAutoCString decodedName;
michael@0:     DecodeString(name, decodedName);
michael@0: 
michael@0:     nsAutoCString decodedValue;
michael@0:     DecodeString(value, decodedValue);
michael@0: 
michael@0:     AppendInternal(NS_ConvertUTF8toUTF16(decodedName),
michael@0:                    NS_ConvertUTF8toUTF16(decodedValue));
michael@0:   }
michael@0: 
michael@0:   NotifyObservers(aObserver);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::DecodeString(const nsACString& aInput, nsACString& aOutput)
michael@0: {
michael@0:   nsACString::const_iterator start, end;
michael@0:   aInput.BeginReading(start);
michael@0:   aInput.EndReading(end);
michael@0: 
michael@0:   while (start != end) {
michael@0:     // replace '+' with U+0020
michael@0:     if (*start == '+') {
michael@0:       aOutput.Append(' ');
michael@0:       ++start;
michael@0:       continue;
michael@0:     }
michael@0: 
michael@0:     // Percent decode algorithm
michael@0:     if (*start == '%') {
michael@0:       nsACString::const_iterator first(start);
michael@0:       ++first;
michael@0: 
michael@0:       nsACString::const_iterator second(first);
michael@0:       ++second;
michael@0: 
michael@0: #define ASCII_HEX_DIGIT( x )    \
michael@0:   ((x >= 0x41 && x <= 0x46) ||  \
michael@0:    (x >= 0x61 && x <= 0x66) ||  \
michael@0:    (x >= 0x30 && x <= 0x39))
michael@0: 
michael@0: #define HEX_DIGIT( x )              \
michael@0:    (*x >= 0x30 && *x <= 0x39        \
michael@0:      ? *x - 0x30                    \
michael@0:      : (*x >= 0x41 && *x <= 0x46    \
michael@0:         ? *x - 0x37                 \
michael@0:         : *x - 0x57))
michael@0: 
michael@0:       if (first != end && second != end &&
michael@0:           ASCII_HEX_DIGIT(*first) && ASCII_HEX_DIGIT(*second)) {
michael@0:         aOutput.Append(HEX_DIGIT(first) * 16 + HEX_DIGIT(second));
michael@0:         start = ++second;
michael@0:         continue;
michael@0: 
michael@0:       } else {
michael@0:         aOutput.Append('%');
michael@0:         ++start;
michael@0:         continue;
michael@0:       }
michael@0:     }
michael@0: 
michael@0:     aOutput.Append(*start);
michael@0:     ++start;
michael@0:   }
michael@0: }
michael@0: 
michael@0: /* static */ PLDHashOperator
michael@0: URLSearchParams::CopyEnumerator(const nsAString& aName,
michael@0:                                 nsTArray<nsString>* aArray,
michael@0:                                 void *userData)
michael@0: {
michael@0:   URLSearchParams* aSearchParams = static_cast<URLSearchParams*>(userData);
michael@0: 
michael@0:   nsTArray<nsString>* newArray = new nsTArray<nsString>();
michael@0:   newArray->AppendElements(*aArray);
michael@0: 
michael@0:   aSearchParams->mSearchParams.Put(aName, newArray);
michael@0:   return PL_DHASH_NEXT;
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::AddObserver(URLSearchParamsObserver* aObserver)
michael@0: {
michael@0:   MOZ_ASSERT(aObserver);
michael@0:   MOZ_ASSERT(!mObservers.Contains(aObserver));
michael@0:   mObservers.AppendElement(aObserver);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::RemoveObserver(URLSearchParamsObserver* aObserver)
michael@0: {
michael@0:   MOZ_ASSERT(aObserver);
michael@0:   mObservers.RemoveElement(aObserver);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::Get(const nsAString& aName, nsString& aRetval)
michael@0: {
michael@0:   nsTArray<nsString>* array;
michael@0:   if (!mSearchParams.Get(aName, &array)) {
michael@0:     aRetval.Truncate();
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   aRetval.Assign(array->ElementAt(0));
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::GetAll(const nsAString& aName, nsTArray<nsString>& aRetval)
michael@0: {
michael@0:   nsTArray<nsString>* array;
michael@0:   if (!mSearchParams.Get(aName, &array)) {
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   aRetval.AppendElements(*array);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::Set(const nsAString& aName, const nsAString& aValue)
michael@0: {
michael@0:   nsTArray<nsString>* array;
michael@0:   if (!mSearchParams.Get(aName, &array)) {
michael@0:     array = new nsTArray<nsString>();
michael@0:     array->AppendElement(aValue);
michael@0:     mSearchParams.Put(aName, array);
michael@0:   } else {
michael@0:     array->ElementAt(0) = aValue;
michael@0:   }
michael@0: 
michael@0:   NotifyObservers(nullptr);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::Append(const nsAString& aName, const nsAString& aValue)
michael@0: {
michael@0:   AppendInternal(aName, aValue);
michael@0:   NotifyObservers(nullptr);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::AppendInternal(const nsAString& aName, const nsAString& aValue)
michael@0: {
michael@0:   nsTArray<nsString>* array;
michael@0:   if (!mSearchParams.Get(aName, &array)) {
michael@0:     array = new nsTArray<nsString>();
michael@0:     mSearchParams.Put(aName, array);
michael@0:   }
michael@0: 
michael@0:   array->AppendElement(aValue);
michael@0: }
michael@0: 
michael@0: bool
michael@0: URLSearchParams::Has(const nsAString& aName)
michael@0: {
michael@0:   return mSearchParams.Get(aName, nullptr);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::Delete(const nsAString& aName)
michael@0: {
michael@0:   nsTArray<nsString>* array;
michael@0:   if (!mSearchParams.Get(aName, &array)) {
michael@0:     return;
michael@0:   }
michael@0: 
michael@0:   mSearchParams.Remove(aName);
michael@0: 
michael@0:   NotifyObservers(nullptr);
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::DeleteAll()
michael@0: {
michael@0:   mSearchParams.Clear();
michael@0: }
michael@0: 
michael@0: class MOZ_STACK_CLASS SerializeData
michael@0: {
michael@0: public:
michael@0:   SerializeData()
michael@0:     : mFirst(true)
michael@0:   {}
michael@0: 
michael@0:   nsAutoString mValue;
michael@0:   bool mFirst;
michael@0: 
michael@0:   void Serialize(const nsCString& aInput)
michael@0:   {
michael@0:     const unsigned char* p = (const unsigned char*) aInput.get();
michael@0: 
michael@0:     while (p && *p) {
michael@0:       // ' ' to '+'
michael@0:       if (*p == 0x20) {
michael@0:         mValue.Append(0x2B);
michael@0:       // Percent Encode algorithm
michael@0:       } else if (*p == 0x2A || *p == 0x2D || *p == 0x2E ||
michael@0:                  (*p >= 0x30 && *p <= 0x39) ||
michael@0:                  (*p >= 0x41 && *p <= 0x5A) || *p == 0x5F ||
michael@0:                  (*p >= 0x61 && *p <= 0x7A)) {
michael@0:         mValue.Append(*p);
michael@0:       } else {
michael@0:         mValue.AppendPrintf("%%%X", *p);
michael@0:       }
michael@0: 
michael@0:       ++p;
michael@0:     }
michael@0:   }
michael@0: };
michael@0: 
michael@0: void
michael@0: URLSearchParams::Serialize(nsAString& aValue) const
michael@0: {
michael@0:   SerializeData data;
michael@0:   mSearchParams.EnumerateRead(SerializeEnumerator, &data);
michael@0:   aValue.Assign(data.mValue);
michael@0: }
michael@0: 
michael@0: /* static */ PLDHashOperator
michael@0: URLSearchParams::SerializeEnumerator(const nsAString& aName,
michael@0:                                      nsTArray<nsString>* aArray,
michael@0:                                      void *userData)
michael@0: {
michael@0:   SerializeData* data = static_cast<SerializeData*>(userData);
michael@0: 
michael@0:   for (uint32_t i = 0, len = aArray->Length(); i < len; ++i) {
michael@0:     if (data->mFirst) {
michael@0:       data->mFirst = false;
michael@0:     } else {
michael@0:       data->mValue.Append(NS_LITERAL_STRING("&"));
michael@0:     }
michael@0: 
michael@0:     data->Serialize(NS_ConvertUTF16toUTF8(aName));
michael@0:     data->mValue.Append(NS_LITERAL_STRING("="));
michael@0:     data->Serialize(NS_ConvertUTF16toUTF8(aArray->ElementAt(i)));
michael@0:   }
michael@0: 
michael@0:   return PL_DHASH_NEXT;
michael@0: }
michael@0: 
michael@0: void
michael@0: URLSearchParams::NotifyObservers(URLSearchParamsObserver* aExceptObserver)
michael@0: {
michael@0:   for (uint32_t i = 0; i < mObservers.Length(); ++i) {
michael@0:     if (mObservers[i] != aExceptObserver) {
michael@0:       mObservers[i]->URLSearchParamsUpdated();
michael@0:     }
michael@0:   }
michael@0: }
michael@0: 
michael@0: } // namespace dom
michael@0: } // namespace mozilla