ipc/glue/URIUtils.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "URIUtils.h"
   8
   9 #include "mozilla/ArrayUtils.h"
  10 #include "mozilla/Assertions.h"
  11 #include "mozilla/dom/BlobURL.h"
  12 #include "mozilla/net/DefaultURI.h"
  13 #include "mozilla/net/SubstitutingURL.h"
  14 #include "mozilla/NullPrincipalURI.h"
  15 #include "nsComponentManagerUtils.h"
  16 #include "nsDebug.h"
  17 #include "nsID.h"
  18 #include "nsJARURI.h"
  19 #include "nsIIconURI.h"
  20 #include "nsJSProtocolHandler.h"
  21 #include "nsNetCID.h"
  22 #include "nsSimpleNestedURI.h"
  23 #include "nsThreadUtils.h"
  24 #include "nsIURIMutator.h"
  25
  26 using namespace mozilla::ipc;
  27 using mozilla::ArrayLength;
  28
  29 namespace {
  30
  31 NS_DEFINE_CID(kSimpleURIMutatorCID, NS_SIMPLEURIMUTATOR_CID);
  32 NS_DEFINE_CID(kStandardURLMutatorCID, NS_STANDARDURLMUTATOR_CID);
  33 NS_DEFINE_CID(kJARURIMutatorCID, NS_JARURIMUTATOR_CID);
  34 NS_DEFINE_CID(kIconURIMutatorCID, NS_MOZICONURIMUTATOR_CID);
  35
  36 }  // namespace
  37
  38 namespace mozilla {
  39 namespace ipc {
  40
  41 void SerializeURI(nsIURI* aURI, URIParams& aParams) {
  42   MOZ_ASSERT(NS_IsMainThread());
  43   MOZ_ASSERT(aURI);
  44
  45   aURI->Serialize(aParams);
  46   if (aParams.type() == URIParams::T__None) {
  47     MOZ_CRASH("Serialize failed!");
  48   }
  49 }
  50
  51 void SerializeURI(nsIURI* aURI, Maybe<URIParams>& aParams) {
  52   MOZ_ASSERT(NS_IsMainThread());
  53
  54   if (aURI) {
  55     URIParams params;
  56     SerializeURI(aURI, params);
  57     aParams = Some(std::move(params));
  58   } else {
  59     aParams = Nothing();
  60   }
  61 }
  62
  63 already_AddRefed<nsIURI> DeserializeURI(const URIParams& aParams) {
  64   MOZ_ASSERT(NS_IsMainThread());
  65
  66   nsCOMPtr<nsIURIMutator> mutator;
  67
  68   switch (aParams.type()) {
  69     case URIParams::TSimpleURIParams:
  70       mutator = do_CreateInstance(kSimpleURIMutatorCID);
  71       break;
  72
  73     case URIParams::TStandardURLParams:
  74       if (aParams.get_StandardURLParams().isSubstituting()) {
  75         mutator = new net::SubstitutingURL::Mutator();
  76       } else {
  77         mutator = do_CreateInstance(kStandardURLMutatorCID);
  78       }
  79       break;
  80
  81     case URIParams::TJARURIParams:
  82       mutator = do_CreateInstance(kJARURIMutatorCID);
  83       break;
  84
  85     case URIParams::TJSURIParams:
  86       mutator = new nsJSURI::Mutator();
  87       break;
  88
  89     case URIParams::TIconURIParams:
  90       mutator = do_CreateInstance(kIconURIMutatorCID);
  91       break;
  92
  93     case URIParams::TNullPrincipalURIParams:
  94       mutator = new NullPrincipalURI::Mutator();
  95       break;
  96
  97     case URIParams::TSimpleNestedURIParams:
  98       mutator = new net::nsSimpleNestedURI::Mutator();
  99       break;
 100
 101     case URIParams::THostObjectURIParams:
 102       mutator = new mozilla::dom::BlobURL::Mutator();
 103       break;
 104
 105     case URIParams::TDefaultURIParams:
 106       mutator = new mozilla::net::DefaultURI::Mutator();
 107       break;
 108
 109     default:
 110       MOZ_CRASH("Unknown params!");
 111   }
 112
 113   MOZ_ASSERT(mutator);
 114
 115   nsresult rv = mutator->Deserialize(aParams);
 116   if (NS_FAILED(rv)) {
 117     MOZ_ASSERT(false, "Deserialize failed!");
 118     return nullptr;
 119   }
 120
 121   nsCOMPtr<nsIURI> uri;
 122   DebugOnly<nsresult> rv2 = mutator->Finalize(getter_AddRefs(uri));
 123   MOZ_ASSERT(uri);
 124   MOZ_ASSERT(NS_SUCCEEDED(rv2));
 125
 126   return uri.forget();
 127 }
 128
 129 already_AddRefed<nsIURI> DeserializeURI(const Maybe<URIParams>& aParams) {
 130   MOZ_ASSERT(NS_IsMainThread());
 131
 132   nsCOMPtr<nsIURI> uri;
 133
 134   if (aParams.isSome()) {
 135     uri = DeserializeURI(aParams.ref());
 136   }
 137
 138   return uri.forget();
 139 }
 140
 141 }  // namespace ipc
 142 }  // namespace mozilla