docshell/base/nsDocShellLoadInfo.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 "nsDocShellLoadInfo.h"
   8 #include "nsISHEntry.h"
   9 #include "nsIInputStream.h"
  10 #include "nsIURI.h"
  11 #include "nsIDocShell.h"
  12 #include "mozilla/net/ReferrerPolicy.h"
  13 #include "mozilla/Unused.h"
  14
  15 namespace mozilla {
  16
  17 void
  18 GetMaybeResultPrincipalURI(nsIDocShellLoadInfo* aLoadInfo, Maybe<nsCOMPtr<nsIURI>>& aRPURI)
  19 {
  20   if (!aLoadInfo) {
  21     return;
  22   }
  23
  24   nsresult rv;
  25
  26   bool isSome;
  27   rv = aLoadInfo->GetResultPrincipalURIIsSome(&isSome);
  28   if (NS_WARN_IF(NS_FAILED(rv))) {
  29     return;
  30   }
  31
  32   aRPURI.reset();
  33
  34   if (!isSome) {
  35     return;
  36   }
  37
  38   nsCOMPtr<nsIURI> uri;
  39   rv = aLoadInfo->GetResultPrincipalURI(getter_AddRefs(uri));
  40   if (NS_WARN_IF(NS_FAILED(rv))) {
  41     return;
  42   }
  43
  44   aRPURI.emplace(std::move(uri));
  45 }
  46
  47 void
  48 SetMaybeResultPrincipalURI(nsIDocShellLoadInfo* aLoadInfo, Maybe<nsCOMPtr<nsIURI>> const& aRPURI)
  49 {
  50   if (!aLoadInfo) {
  51     return;
  52   }
  53
  54   nsresult rv;
  55
  56   rv = aLoadInfo->SetResultPrincipalURI(aRPURI.refOr(nullptr));
  57   Unused << NS_WARN_IF(NS_FAILED(rv));
  58
  59   rv = aLoadInfo->SetResultPrincipalURIIsSome(aRPURI.isSome());
  60   Unused << NS_WARN_IF(NS_FAILED(rv));
  61 }
  62
  63 } // mozilla
  64
  65 nsDocShellLoadInfo::nsDocShellLoadInfo()
  66   : mResultPrincipalURIIsSome(false)
  67   , mLoadReplace(false)
  68   , mInheritPrincipal(false)
  69   , mPrincipalIsExplicit(false)
  70   , mForceAllowDataURI(false)
  71   , mOriginalFrameSrc(false)
  72   , mSendReferrer(true)
  73   , mReferrerPolicy(mozilla::net::RP_Unset)
  74   , mLoadType(nsIDocShellLoadInfo::loadNormal)
  75   , mIsSrcdocLoad(false)
  76 {
  77 }
  78
  79 nsDocShellLoadInfo::~nsDocShellLoadInfo()
  80 {
  81 }
  82
  83 NS_IMPL_ADDREF(nsDocShellLoadInfo)
  84 NS_IMPL_RELEASE(nsDocShellLoadInfo)
  85
  86 NS_INTERFACE_MAP_BEGIN(nsDocShellLoadInfo)
  87   NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIDocShellLoadInfo)
  88   NS_INTERFACE_MAP_ENTRY(nsIDocShellLoadInfo)
  89 NS_INTERFACE_MAP_END
  90
  91 NS_IMETHODIMP
  92 nsDocShellLoadInfo::GetReferrer(nsIURI** aReferrer)
  93 {
  94   NS_ENSURE_ARG_POINTER(aReferrer);
  95
  96   *aReferrer = mReferrer;
  97   NS_IF_ADDREF(*aReferrer);
  98   return NS_OK;
  99 }
 100
 101 NS_IMETHODIMP
 102 nsDocShellLoadInfo::SetReferrer(nsIURI* aReferrer)
 103 {
 104   mReferrer = aReferrer;
 105   return NS_OK;
 106 }
 107
 108 NS_IMETHODIMP
 109 nsDocShellLoadInfo::GetOriginalURI(nsIURI** aOriginalURI)
 110 {
 111   NS_ENSURE_ARG_POINTER(aOriginalURI);
 112
 113   *aOriginalURI = mOriginalURI;
 114   NS_IF_ADDREF(*aOriginalURI);
 115   return NS_OK;
 116 }
 117
 118 NS_IMETHODIMP
 119 nsDocShellLoadInfo::SetOriginalURI(nsIURI* aOriginalURI)
 120 {
 121   mOriginalURI = aOriginalURI;
 122   return NS_OK;
 123 }
 124
 125 NS_IMETHODIMP
 126 nsDocShellLoadInfo::GetResultPrincipalURI(nsIURI** aResultPrincipalURI)
 127 {
 128   NS_ENSURE_ARG_POINTER(aResultPrincipalURI);
 129
 130   *aResultPrincipalURI = mResultPrincipalURI;
 131   NS_IF_ADDREF(*aResultPrincipalURI);
 132   return NS_OK;
 133 }
 134
 135 NS_IMETHODIMP
 136 nsDocShellLoadInfo::SetResultPrincipalURI(nsIURI* aResultPrincipalURI)
 137 {
 138   mResultPrincipalURI = aResultPrincipalURI;
 139   return NS_OK;
 140 }
 141
 142 NS_IMETHODIMP
 143 nsDocShellLoadInfo::GetResultPrincipalURIIsSome(bool* aIsSome)
 144 {
 145   *aIsSome = mResultPrincipalURIIsSome;
 146   return NS_OK;
 147 }
 148
 149 NS_IMETHODIMP
 150 nsDocShellLoadInfo::SetResultPrincipalURIIsSome(bool aIsSome)
 151 {
 152   mResultPrincipalURIIsSome = aIsSome;
 153   return NS_OK;
 154 }
 155
 156 NS_IMETHODIMP
 157 nsDocShellLoadInfo::GetLoadReplace(bool* aLoadReplace)
 158 {
 159   *aLoadReplace = mLoadReplace;
 160   return NS_OK;
 161 }
 162
 163 NS_IMETHODIMP
 164 nsDocShellLoadInfo::SetLoadReplace(bool aLoadReplace)
 165 {
 166   mLoadReplace = aLoadReplace;
 167   return NS_OK;
 168 }
 169
 170 NS_IMETHODIMP
 171 nsDocShellLoadInfo::GetTriggeringPrincipal(nsIPrincipal** aTriggeringPrincipal)
 172 {
 173   NS_ENSURE_ARG_POINTER(aTriggeringPrincipal);
 174   NS_IF_ADDREF(*aTriggeringPrincipal = mTriggeringPrincipal);
 175   return NS_OK;
 176 }
 177
 178 NS_IMETHODIMP
 179 nsDocShellLoadInfo::SetTriggeringPrincipal(nsIPrincipal* aTriggeringPrincipal)
 180 {
 181   mTriggeringPrincipal = aTriggeringPrincipal;
 182   return NS_OK;
 183 }
 184
 185 NS_IMETHODIMP
 186 nsDocShellLoadInfo::GetInheritPrincipal(bool* aInheritPrincipal)
 187 {
 188   NS_ENSURE_ARG_POINTER(aInheritPrincipal);
 189   *aInheritPrincipal = mInheritPrincipal;
 190   return NS_OK;
 191 }
 192
 193 NS_IMETHODIMP
 194 nsDocShellLoadInfo::SetInheritPrincipal(bool aInheritPrincipal)
 195 {
 196   mInheritPrincipal = aInheritPrincipal;
 197   return NS_OK;
 198 }
 199
 200 NS_IMETHODIMP
 201 nsDocShellLoadInfo::GetPrincipalIsExplicit(bool* aPrincipalIsExplicit)
 202 {
 203   *aPrincipalIsExplicit = mPrincipalIsExplicit;
 204   return NS_OK;
 205 }
 206
 207 NS_IMETHODIMP
 208 nsDocShellLoadInfo::SetPrincipalIsExplicit(bool aPrincipalIsExplicit)
 209 {
 210   mPrincipalIsExplicit = aPrincipalIsExplicit;
 211   return NS_OK;
 212 }
 213
 214 NS_IMETHODIMP
 215 nsDocShellLoadInfo::GetForceAllowDataURI(bool* aForceAllowDataURI)
 216 {
 217   *aForceAllowDataURI = mForceAllowDataURI;
 218   return NS_OK;
 219 }
 220
 221 NS_IMETHODIMP
 222 nsDocShellLoadInfo::SetForceAllowDataURI(bool aForceAllowDataURI)
 223 {
 224   mForceAllowDataURI = aForceAllowDataURI;
 225   return NS_OK;
 226 }
 227
 228 NS_IMETHODIMP
 229 nsDocShellLoadInfo::GetOriginalFrameSrc(bool* aOriginalFrameSrc)
 230 {
 231   *aOriginalFrameSrc = mOriginalFrameSrc;
 232   return NS_OK;
 233 }
 234
 235 NS_IMETHODIMP
 236 nsDocShellLoadInfo::SetOriginalFrameSrc(bool aOriginalFrameSrc)
 237 {
 238   mOriginalFrameSrc = aOriginalFrameSrc;
 239   return NS_OK;
 240 }
 241
 242 NS_IMETHODIMP
 243 nsDocShellLoadInfo::GetLoadType(nsDocShellInfoLoadType* aLoadType)
 244 {
 245   NS_ENSURE_ARG_POINTER(aLoadType);
 246
 247   *aLoadType = mLoadType;
 248   return NS_OK;
 249 }
 250
 251 NS_IMETHODIMP
 252 nsDocShellLoadInfo::SetLoadType(nsDocShellInfoLoadType aLoadType)
 253 {
 254   mLoadType = aLoadType;
 255   return NS_OK;
 256 }
 257
 258 NS_IMETHODIMP
 259 nsDocShellLoadInfo::GetSHEntry(nsISHEntry** aSHEntry)
 260 {
 261   NS_ENSURE_ARG_POINTER(aSHEntry);
 262
 263   *aSHEntry = mSHEntry;
 264   NS_IF_ADDREF(*aSHEntry);
 265   return NS_OK;
 266 }
 267
 268 NS_IMETHODIMP
 269 nsDocShellLoadInfo::SetSHEntry(nsISHEntry* aSHEntry)
 270 {
 271   mSHEntry = aSHEntry;
 272   return NS_OK;
 273 }
 274
 275 NS_IMETHODIMP
 276 nsDocShellLoadInfo::GetTarget(char16_t** aTarget)
 277 {
 278   NS_ENSURE_ARG_POINTER(aTarget);
 279
 280   *aTarget = ToNewUnicode(mTarget);
 281
 282   return NS_OK;
 283 }
 284
 285 NS_IMETHODIMP
 286 nsDocShellLoadInfo::SetTarget(const char16_t* aTarget)
 287 {
 288   mTarget.Assign(aTarget);
 289   return NS_OK;
 290 }
 291
 292 NS_IMETHODIMP
 293 nsDocShellLoadInfo::GetPostDataStream(nsIInputStream** aResult)
 294 {
 295   NS_ENSURE_ARG_POINTER(aResult);
 296
 297   *aResult = mPostDataStream;
 298
 299   NS_IF_ADDREF(*aResult);
 300   return NS_OK;
 301 }
 302
 303 NS_IMETHODIMP
 304 nsDocShellLoadInfo::SetPostDataStream(nsIInputStream* aStream)
 305 {
 306   mPostDataStream = aStream;
 307   return NS_OK;
 308 }
 309
 310 NS_IMETHODIMP
 311 nsDocShellLoadInfo::GetHeadersStream(nsIInputStream** aHeadersStream)
 312 {
 313   NS_ENSURE_ARG_POINTER(aHeadersStream);
 314   *aHeadersStream = mHeadersStream;
 315   NS_IF_ADDREF(*aHeadersStream);
 316   return NS_OK;
 317 }
 318 NS_IMETHODIMP
 319 nsDocShellLoadInfo::SetHeadersStream(nsIInputStream* aHeadersStream)
 320 {
 321   mHeadersStream = aHeadersStream;
 322   return NS_OK;
 323 }
 324
 325 NS_IMETHODIMP
 326 nsDocShellLoadInfo::GetSendReferrer(bool* aSendReferrer)
 327 {
 328   NS_ENSURE_ARG_POINTER(aSendReferrer);
 329
 330   *aSendReferrer = mSendReferrer;
 331   return NS_OK;
 332 }
 333
 334 NS_IMETHODIMP
 335 nsDocShellLoadInfo::SetSendReferrer(bool aSendReferrer)
 336 {
 337   mSendReferrer = aSendReferrer;
 338   return NS_OK;
 339 }
 340
 341 NS_IMETHODIMP
 342 nsDocShellLoadInfo::GetReferrerPolicy(
 343     nsDocShellInfoReferrerPolicy* aReferrerPolicy)
 344 {
 345   *aReferrerPolicy = mReferrerPolicy;
 346   return NS_OK;
 347 }
 348
 349 NS_IMETHODIMP
 350 nsDocShellLoadInfo::SetReferrerPolicy(
 351     nsDocShellInfoReferrerPolicy aReferrerPolicy)
 352 {
 353   mReferrerPolicy = aReferrerPolicy;
 354   return NS_OK;
 355 }
 356
 357 NS_IMETHODIMP
 358 nsDocShellLoadInfo::GetIsSrcdocLoad(bool* aIsSrcdocLoad)
 359 {
 360   *aIsSrcdocLoad = mIsSrcdocLoad;
 361   return NS_OK;
 362 }
 363
 364 NS_IMETHODIMP
 365 nsDocShellLoadInfo::GetSrcdocData(nsAString& aSrcdocData)
 366 {
 367   aSrcdocData = mSrcdocData;
 368   return NS_OK;
 369 }
 370
 371 NS_IMETHODIMP
 372 nsDocShellLoadInfo::SetSrcdocData(const nsAString& aSrcdocData)
 373 {
 374   mSrcdocData = aSrcdocData;
 375   mIsSrcdocLoad = true;
 376   return NS_OK;
 377 }
 378
 379 NS_IMETHODIMP
 380 nsDocShellLoadInfo::GetSourceDocShell(nsIDocShell** aSourceDocShell)
 381 {
 382   MOZ_ASSERT(aSourceDocShell);
 383   nsCOMPtr<nsIDocShell> result = mSourceDocShell;
 384   result.forget(aSourceDocShell);
 385   return NS_OK;
 386 }
 387
 388 NS_IMETHODIMP
 389 nsDocShellLoadInfo::SetSourceDocShell(nsIDocShell* aSourceDocShell)
 390 {
 391   mSourceDocShell = aSourceDocShell;
 392   return NS_OK;
 393 }
 394
 395 NS_IMETHODIMP
 396 nsDocShellLoadInfo::GetBaseURI(nsIURI** aBaseURI)
 397 {
 398   NS_ENSURE_ARG_POINTER(aBaseURI);
 399
 400   *aBaseURI = mBaseURI;
 401   NS_IF_ADDREF(*aBaseURI);
 402   return NS_OK;
 403 }
 404
 405 NS_IMETHODIMP
 406 nsDocShellLoadInfo::SetBaseURI(nsIURI* aBaseURI)
 407 {
 408   mBaseURI = aBaseURI;
 409   return NS_OK;
 410 }