forked from mirrors/gecko-dev
852d02ec16
This only changes the behaviour when called with a TaskQueue or other type using SerialEventTargetGuard on the stack. They are being switched over as the existing GetCurrentEventTarget method is being removed, as it is somewhat confusing, and poorly documented. Callers which need to get the current thread even when on a threadpool or behind a TaskQueue were switched to GetCurrentEventTarget in the previous part. Differential Revision: https://phabricator.services.mozilla.com/D166607
117 lines
3.9 KiB
C++
117 lines
3.9 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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/. */
|
|
|
|
// HttpLog.h should generally be included first
|
|
#include "HttpLog.h"
|
|
|
|
#include "HTTPSRecordResolver.h"
|
|
#include "nsIDNSByTypeRecord.h"
|
|
#include "nsIDNSAdditionalInfo.h"
|
|
#include "nsIDNSService.h"
|
|
#include "nsHttpConnectionInfo.h"
|
|
#include "nsNetCID.h"
|
|
#include "nsAHttpTransaction.h"
|
|
#include "nsServiceManagerUtils.h"
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
NS_IMPL_ISUPPORTS(HTTPSRecordResolver, nsIDNSListener)
|
|
|
|
HTTPSRecordResolver::HTTPSRecordResolver(nsAHttpTransaction* aTransaction)
|
|
: mTransaction(aTransaction),
|
|
mConnInfo(aTransaction->ConnectionInfo()),
|
|
mCaps(aTransaction->Caps()) {}
|
|
|
|
HTTPSRecordResolver::~HTTPSRecordResolver() = default;
|
|
|
|
nsresult HTTPSRecordResolver::FetchHTTPSRRInternal(
|
|
nsIEventTarget* aTarget, nsICancelable** aDNSRequest) {
|
|
NS_ENSURE_ARG_POINTER(aTarget);
|
|
|
|
// Only fetch HTTPS RR for https.
|
|
if (!mConnInfo->FirstHopSSL()) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsCOMPtr<nsIDNSService> dns = do_GetService(NS_DNSSERVICE_CONTRACTID);
|
|
if (!dns) {
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
}
|
|
|
|
nsIDNSService::DNSFlags flags =
|
|
nsIDNSService::GetFlagsFromTRRMode(mConnInfo->GetTRRMode());
|
|
if (mCaps & NS_HTTP_REFRESH_DNS) {
|
|
flags |= nsIDNSService::RESOLVE_BYPASS_CACHE;
|
|
}
|
|
|
|
nsCOMPtr<nsIDNSAdditionalInfo> info;
|
|
if (mConnInfo->OriginPort() != NS_HTTPS_DEFAULT_PORT) {
|
|
dns->NewAdditionalInfo(""_ns, mConnInfo->OriginPort(),
|
|
getter_AddRefs(info));
|
|
}
|
|
return dns->AsyncResolveNative(
|
|
mConnInfo->GetOrigin(), nsIDNSService::RESOLVE_TYPE_HTTPSSVC, flags, info,
|
|
this, aTarget, mConnInfo->GetOriginAttributes(), aDNSRequest);
|
|
}
|
|
|
|
NS_IMETHODIMP HTTPSRecordResolver::OnLookupComplete(nsICancelable* aRequest,
|
|
nsIDNSRecord* aRecord,
|
|
nsresult aStatus) {
|
|
nsCOMPtr<nsIDNSAddrRecord> addrRecord = do_QueryInterface(aRecord);
|
|
// This will be called again when receving the result of speculatively loading
|
|
// the addr records of the target name. In this case, just return NS_OK.
|
|
if (addrRecord) {
|
|
return NS_OK;
|
|
}
|
|
|
|
if (!mTransaction) {
|
|
// The connecttion is not interesed in a response anymore.
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCOMPtr<nsIDNSHTTPSSVCRecord> record = do_QueryInterface(aRecord);
|
|
if (!record || NS_FAILED(aStatus)) {
|
|
return mTransaction->OnHTTPSRRAvailable(nullptr, nullptr);
|
|
}
|
|
|
|
nsCOMPtr<nsISVCBRecord> svcbRecord;
|
|
if (NS_FAILED(record->GetServiceModeRecord(mCaps & NS_HTTP_DISALLOW_SPDY,
|
|
mCaps & NS_HTTP_DISALLOW_HTTP3,
|
|
getter_AddRefs(svcbRecord)))) {
|
|
return mTransaction->OnHTTPSRRAvailable(record, nullptr);
|
|
}
|
|
|
|
return mTransaction->OnHTTPSRRAvailable(record, svcbRecord);
|
|
}
|
|
|
|
void HTTPSRecordResolver::PrefetchAddrRecord(const nsACString& aTargetName,
|
|
bool aRefreshDNS) {
|
|
MOZ_ASSERT(mTransaction);
|
|
nsCOMPtr<nsIDNSService> dns = do_GetService(NS_DNSSERVICE_CONTRACTID);
|
|
if (!dns) {
|
|
return;
|
|
}
|
|
|
|
nsIDNSService::DNSFlags flags = nsIDNSService::GetFlagsFromTRRMode(
|
|
mTransaction->ConnectionInfo()->GetTRRMode());
|
|
if (aRefreshDNS) {
|
|
flags |= nsIDNSService::RESOLVE_BYPASS_CACHE;
|
|
}
|
|
|
|
nsCOMPtr<nsICancelable> tmpOutstanding;
|
|
|
|
Unused << dns->AsyncResolveNative(
|
|
aTargetName, nsIDNSService::RESOLVE_TYPE_DEFAULT,
|
|
flags | nsIDNSService::RESOLVE_SPECULATE, nullptr, this,
|
|
GetCurrentSerialEventTarget(),
|
|
mTransaction->ConnectionInfo()->GetOriginAttributes(),
|
|
getter_AddRefs(tmpOutstanding));
|
|
}
|
|
|
|
void HTTPSRecordResolver::Close() { mTransaction = nullptr; }
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|