forked from mirrors/gecko-dev
98304d1200
This patch replaces the previous ContractID-based lookup system for protocol handlers, and replaces it with a new custom system in nsIOService. It will be pre-populated with non-overridable static protocol handlers using the StaticComponents infrastructure added in the previous part, and callers can also dynamically register new protocol handlers at runtime. This new system is intended to provide access to the default port and non-dynamic protocol flags off-main-thread, by requiring these values to be provided up-front as constants, rather than getting them from the xpcom interface. The data is then guarded by an RWLock. Callers which look up specific handlers by their contractID are not changed, as the contract IDs for existing handlers have not been changed, so the lookup will still succeed. This change as-implemented breaks the nsGIOProtocolHandler on Linux, as it removes the special code which would try to use that handler for some protocols. This will be fixed in a later part by making the nsGIOProtocolHandler use the dynamic registration APIs to register and un-register protocol handlers at runtime in response to the GIO pref. Differential Revision: https://phabricator.services.mozilla.com/D162804
86 lines
2.9 KiB
C++
86 lines
2.9 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 "ProtocolHandlerInfo.h"
|
|
#include "StaticComponents.h"
|
|
#include "nsIProtocolHandler.h"
|
|
|
|
namespace mozilla::net {
|
|
|
|
uint32_t ProtocolHandlerInfo::StaticProtocolFlags() const {
|
|
uint32_t flags = mInner.match(
|
|
[&](const xpcom::StaticProtocolHandler* handler) {
|
|
return handler->mProtocolFlags;
|
|
},
|
|
[&](const RuntimeProtocolHandler& handler) {
|
|
return handler.mProtocolFlags;
|
|
});
|
|
#if !IS_ORIGIN_IS_FULL_SPEC_DEFINED
|
|
MOZ_RELEASE_ASSERT(!(flags & nsIProtocolHandler::ORIGIN_IS_FULL_SPEC),
|
|
"ORIGIN_IS_FULL_SPEC is unsupported but used");
|
|
#endif
|
|
return flags;
|
|
}
|
|
|
|
int32_t ProtocolHandlerInfo::DefaultPort() const {
|
|
return mInner.match(
|
|
[&](const xpcom::StaticProtocolHandler* handler) {
|
|
return handler->mDefaultPort;
|
|
},
|
|
[&](const RuntimeProtocolHandler& handler) {
|
|
return handler.mDefaultPort;
|
|
});
|
|
}
|
|
|
|
nsresult ProtocolHandlerInfo::DynamicProtocolFlags(nsIURI* aURI,
|
|
uint32_t* aFlags) const {
|
|
MOZ_DIAGNOSTIC_ASSERT(NS_IsMainThread());
|
|
|
|
// If we're querying dynamic flags, we'll need to fetch the actual xpcom
|
|
// component in order to check them.
|
|
if (HasDynamicFlags()) {
|
|
nsCOMPtr<nsIProtocolHandler> handler = Handler();
|
|
if (nsCOMPtr<nsIProtocolHandlerWithDynamicFlags> dynamic =
|
|
do_QueryInterface(handler)) {
|
|
nsresult rv = dynamic->GetFlagsForURI(aURI, aFlags);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
MOZ_DIAGNOSTIC_ASSERT(
|
|
(StaticProtocolFlags() & ~nsIProtocolHandler::DYNAMIC_URI_FLAGS) ==
|
|
(*aFlags & ~nsIProtocolHandler::DYNAMIC_URI_FLAGS),
|
|
"only DYNAMIC_URI_FLAGS may be changed by a "
|
|
"nsIProtocolHandlerWithDynamicFlags implementation");
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
// Otherwise, just check against static flags.
|
|
*aFlags = StaticProtocolFlags();
|
|
return NS_OK;
|
|
}
|
|
|
|
bool ProtocolHandlerInfo::HasDynamicFlags() const {
|
|
return mInner.match(
|
|
[&](const xpcom::StaticProtocolHandler* handler) {
|
|
return handler->mHasDynamicFlags;
|
|
},
|
|
[&](const RuntimeProtocolHandler&) { return false; });
|
|
}
|
|
|
|
already_AddRefed<nsIProtocolHandler> ProtocolHandlerInfo::Handler() const {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
nsCOMPtr<nsIProtocolHandler> retval;
|
|
mInner.match(
|
|
[&](const xpcom::StaticProtocolHandler* handler) {
|
|
retval = handler->Module().GetService();
|
|
},
|
|
[&](const RuntimeProtocolHandler& handler) {
|
|
retval = handler.mHandler.get();
|
|
});
|
|
return retval.forget();
|
|
}
|
|
|
|
} // namespace mozilla::net
|