mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-09-10 11:54:40 +00:00
267 lines
7.4 KiB
C++
267 lines
7.4 KiB
C++
|
/* -*- 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 "Response.h"
|
||
|
|
||
|
#include "nsISupportsImpl.h"
|
||
|
#include "nsIURI.h"
|
||
|
#include "nsPIDOMWindow.h"
|
||
|
|
||
|
#include "mozilla/ErrorResult.h"
|
||
|
#include "mozilla/dom/FetchBinding.h"
|
||
|
#include "mozilla/dom/Headers.h"
|
||
|
#include "mozilla/dom/Promise.h"
|
||
|
#include "mozilla/dom/URL.h"
|
||
|
#include "mozilla/dom/workers/bindings/URL.h"
|
||
|
|
||
|
#include "nsDOMString.h"
|
||
|
|
||
|
#include "InternalResponse.h"
|
||
|
#include "WorkerPrivate.h"
|
||
|
|
||
|
namespace mozilla {
|
||
|
namespace dom {
|
||
|
|
||
|
NS_IMPL_CYCLE_COLLECTING_ADDREF(Response)
|
||
|
NS_IMPL_CYCLE_COLLECTING_RELEASE(Response)
|
||
|
NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(Response, mOwner, mHeaders)
|
||
|
|
||
|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(Response)
|
||
|
NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
|
||
|
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
||
|
NS_INTERFACE_MAP_END
|
||
|
|
||
|
Response::Response(nsIGlobalObject* aGlobal, InternalResponse* aInternalResponse)
|
||
|
: FetchBody<Response>()
|
||
|
, mOwner(aGlobal)
|
||
|
, mInternalResponse(aInternalResponse)
|
||
|
{
|
||
|
SetMimeType();
|
||
|
}
|
||
|
|
||
|
Response::~Response()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
/* static */ already_AddRefed<Response>
|
||
|
Response::Error(const GlobalObject& aGlobal)
|
||
|
{
|
||
|
nsCOMPtr<nsIGlobalObject> global = do_QueryInterface(aGlobal.GetAsSupports());
|
||
|
RefPtr<InternalResponse> error = InternalResponse::NetworkError();
|
||
|
RefPtr<Response> r = new Response(global, error);
|
||
|
return r.forget();
|
||
|
}
|
||
|
|
||
|
/* static */ already_AddRefed<Response>
|
||
|
Response::Redirect(const GlobalObject& aGlobal, const nsAString& aUrl,
|
||
|
uint16_t aStatus, ErrorResult& aRv)
|
||
|
{
|
||
|
nsAutoString parsedURL;
|
||
|
|
||
|
if (NS_IsMainThread()) {
|
||
|
nsCOMPtr<nsIURI> baseURI;
|
||
|
nsIDocument* doc = GetEntryDocument();
|
||
|
if (doc) {
|
||
|
baseURI = doc->GetBaseURI();
|
||
|
}
|
||
|
nsCOMPtr<nsIURI> resolvedURI;
|
||
|
aRv = NS_NewURI(getter_AddRefs(resolvedURI), aUrl, nullptr, baseURI);
|
||
|
if (NS_WARN_IF(aRv.Failed())) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
nsAutoCString spec;
|
||
|
aRv = resolvedURI->GetSpec(spec);
|
||
|
if (NS_WARN_IF(aRv.Failed())) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
CopyUTF8toUTF16(spec, parsedURL);
|
||
|
} else {
|
||
|
workers::WorkerPrivate* worker = workers::GetCurrentThreadWorkerPrivate();
|
||
|
MOZ_ASSERT(worker);
|
||
|
worker->AssertIsOnWorkerThread();
|
||
|
|
||
|
NS_ConvertUTF8toUTF16 baseURL(worker->GetLocationInfo().mHref);
|
||
|
RefPtr<workers::URL> url =
|
||
|
workers::URL::Constructor(aGlobal, aUrl, baseURL, aRv);
|
||
|
if (aRv.Failed()) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
url->Stringify(parsedURL, aRv);
|
||
|
}
|
||
|
|
||
|
if (aRv.Failed()) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
if (aStatus != 301 && aStatus != 302 && aStatus != 303 && aStatus != 307 && aStatus != 308) {
|
||
|
aRv.ThrowRangeError<MSG_INVALID_REDIRECT_STATUSCODE_ERROR>();
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
Optional<ArrayBufferOrArrayBufferViewOrBlobOrFormDataOrUSVStringOrURLSearchParams> body;
|
||
|
ResponseInit init;
|
||
|
init.mStatus = aStatus;
|
||
|
RefPtr<Response> r = Response::Constructor(aGlobal, body, init, aRv);
|
||
|
if (NS_WARN_IF(aRv.Failed())) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
r->GetInternalHeaders()->Set(NS_LITERAL_CSTRING("Location"),
|
||
|
NS_ConvertUTF16toUTF8(parsedURL), aRv);
|
||
|
if (NS_WARN_IF(aRv.Failed())) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
r->GetInternalHeaders()->SetGuard(HeadersGuardEnum::Immutable, aRv);
|
||
|
MOZ_ASSERT(!aRv.Failed());
|
||
|
|
||
|
return r.forget();
|
||
|
}
|
||
|
|
||
|
/*static*/ already_AddRefed<Response>
|
||
|
Response::Constructor(const GlobalObject& aGlobal,
|
||
|
const Optional<ArrayBufferOrArrayBufferViewOrBlobOrFormDataOrUSVStringOrURLSearchParams>& aBody,
|
||
|
const ResponseInit& aInit, ErrorResult& aRv)
|
||
|
{
|
||
|
nsCOMPtr<nsIGlobalObject> global = do_QueryInterface(aGlobal.GetAsSupports());
|
||
|
|
||
|
if (aInit.mStatus < 200 || aInit.mStatus > 599) {
|
||
|
aRv.ThrowRangeError<MSG_INVALID_RESPONSE_STATUSCODE_ERROR>();
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
nsCString statusText;
|
||
|
if (aInit.mStatusText.WasPassed()) {
|
||
|
statusText = aInit.mStatusText.Value();
|
||
|
nsACString::const_iterator start, end;
|
||
|
statusText.BeginReading(start);
|
||
|
statusText.EndReading(end);
|
||
|
if (FindCharInReadable('\r', start, end)) {
|
||
|
aRv.ThrowTypeError<MSG_RESPONSE_INVALID_STATUSTEXT_ERROR>();
|
||
|
return nullptr;
|
||
|
}
|
||
|
// Reset iterator since FindCharInReadable advances it.
|
||
|
statusText.BeginReading(start);
|
||
|
if (FindCharInReadable('\n', start, end)) {
|
||
|
aRv.ThrowTypeError<MSG_RESPONSE_INVALID_STATUSTEXT_ERROR>();
|
||
|
return nullptr;
|
||
|
}
|
||
|
} else {
|
||
|
// Since we don't support default values for ByteString.
|
||
|
statusText = NS_LITERAL_CSTRING("OK");
|
||
|
}
|
||
|
|
||
|
RefPtr<InternalResponse> internalResponse =
|
||
|
new InternalResponse(aInit.mStatus, statusText);
|
||
|
|
||
|
// Grab a valid channel info from the global so this response is 'valid' for
|
||
|
// interception.
|
||
|
if (NS_IsMainThread()) {
|
||
|
ChannelInfo info;
|
||
|
nsCOMPtr<nsPIDOMWindow> window = do_QueryInterface(global);
|
||
|
if (window) {
|
||
|
nsIDocument* doc = window->GetExtantDoc();
|
||
|
MOZ_ASSERT(doc);
|
||
|
info.InitFromDocument(doc);
|
||
|
} else {
|
||
|
info.InitFromChromeGlobal(global);
|
||
|
}
|
||
|
internalResponse->InitChannelInfo(info);
|
||
|
} else {
|
||
|
workers::WorkerPrivate* worker = workers::GetCurrentThreadWorkerPrivate();
|
||
|
MOZ_ASSERT(worker);
|
||
|
internalResponse->InitChannelInfo(worker->GetChannelInfo());
|
||
|
}
|
||
|
|
||
|
RefPtr<Response> r = new Response(global, internalResponse);
|
||
|
|
||
|
if (aInit.mHeaders.WasPassed()) {
|
||
|
internalResponse->Headers()->Clear();
|
||
|
|
||
|
// Instead of using Fill, create an object to allow the constructor to
|
||
|
// unwrap the HeadersInit.
|
||
|
RefPtr<Headers> headers =
|
||
|
Headers::Create(global, aInit.mHeaders.Value(), aRv);
|
||
|
if (aRv.Failed()) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
internalResponse->Headers()->Fill(*headers->GetInternalHeaders(), aRv);
|
||
|
if (NS_WARN_IF(aRv.Failed())) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (aBody.WasPassed()) {
|
||
|
if (aInit.mStatus == 204 || aInit.mStatus == 205 || aInit.mStatus == 304) {
|
||
|
aRv.ThrowTypeError<MSG_RESPONSE_NULL_STATUS_WITH_BODY>();
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
nsCOMPtr<nsIInputStream> bodyStream;
|
||
|
nsCString contentType;
|
||
|
aRv = ExtractByteStreamFromBody(aBody.Value(), getter_AddRefs(bodyStream), contentType);
|
||
|
internalResponse->SetBody(bodyStream);
|
||
|
|
||
|
if (!contentType.IsVoid() &&
|
||
|
!internalResponse->Headers()->Has(NS_LITERAL_CSTRING("Content-Type"), aRv)) {
|
||
|
// Ignore Append() failing here.
|
||
|
ErrorResult error;
|
||
|
internalResponse->Headers()->Append(NS_LITERAL_CSTRING("Content-Type"), contentType, error);
|
||
|
error.SuppressException();
|
||
|
}
|
||
|
|
||
|
if (aRv.Failed()) {
|
||
|
return nullptr;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
r->SetMimeType();
|
||
|
return r.forget();
|
||
|
}
|
||
|
|
||
|
already_AddRefed<Response>
|
||
|
Response::Clone(ErrorResult& aRv) const
|
||
|
{
|
||
|
if (BodyUsed()) {
|
||
|
aRv.ThrowTypeError<MSG_FETCH_BODY_CONSUMED_ERROR>();
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
RefPtr<InternalResponse> ir = mInternalResponse->Clone();
|
||
|
RefPtr<Response> response = new Response(mOwner, ir);
|
||
|
return response.forget();
|
||
|
}
|
||
|
|
||
|
void
|
||
|
Response::SetBody(nsIInputStream* aBody)
|
||
|
{
|
||
|
MOZ_ASSERT(!BodyUsed());
|
||
|
mInternalResponse->SetBody(aBody);
|
||
|
}
|
||
|
|
||
|
already_AddRefed<InternalResponse>
|
||
|
Response::GetInternalResponse() const
|
||
|
{
|
||
|
RefPtr<InternalResponse> ref = mInternalResponse;
|
||
|
return ref.forget();
|
||
|
}
|
||
|
|
||
|
Headers*
|
||
|
Response::Headers_()
|
||
|
{
|
||
|
if (!mHeaders) {
|
||
|
mHeaders = new Headers(mOwner, mInternalResponse->Headers());
|
||
|
}
|
||
|
|
||
|
return mHeaders;
|
||
|
}
|
||
|
|
||
|
} // namespace dom
|
||
|
} // namespace mozilla
|