forked from mirrors/gecko-dev
fccd6eb9ba
* At this point our DLL blocking infra is complicated enough that I decided to bite the bullet and move all of this code out of `mozglue/build` and into its own subdirectory, `mozglue/dllservices`. * We delete the original `UntrustedDllsHandler` code which is now obsolete. * We implement mozglue's `LoaderObserver`: ** When this observer registers itself with the launcher process API, it receives a vector containing all saved records of loaded DLLs that happened until that moment. ** This code handles profiler labels and stackwalking suppression. ** Once a load has completed, we either pass the load on to XUL for further processing, or save it for later if XUL is not initialized yet. * mozglue has its own `ModuleLoadFrame` implementation for the legacy blocklist. * `DllServicesBase` is updated to support the new interfaces. * We implement `FallbackLoaderAPI` for `plugin-container`, `xpcshell`, and any other non-`firefox` processes that do not have a launcher process providing a loader API. * We add some wide to UTF8 conversion functions. Depends on D43157 Differential Revision: https://phabricator.services.mozilla.com/D43158 --HG-- rename : mozglue/build/Authenticode.cpp => mozglue/dllservices/Authenticode.cpp rename : mozglue/build/Authenticode.h => mozglue/dllservices/Authenticode.h rename : browser/app/winlauncher/freestanding/LoaderAPIInterfaces.h => mozglue/dllservices/LoaderAPIInterfaces.h rename : browser/app/winlauncher/freestanding/ModuleLoadInfo.h => mozglue/dllservices/ModuleLoadInfo.h rename : browser/app/winlauncher/NtLoaderAPI.h => mozglue/dllservices/NtLoaderAPI.h rename : mozglue/build/WindowsDllBlocklist.cpp => mozglue/dllservices/WindowsDllBlocklist.cpp rename : mozglue/build/WindowsDllBlocklist.h => mozglue/dllservices/WindowsDllBlocklist.h rename : mozglue/build/WindowsDllBlocklistCommon.h => mozglue/dllservices/WindowsDllBlocklistCommon.h rename : mozglue/build/WindowsDllBlocklistDefs.in => mozglue/dllservices/WindowsDllBlocklistDefs.in rename : mozglue/build/WindowsDllServices.h => mozglue/dllservices/WindowsDllServices.h rename : mozglue/build/gen_dll_blocklist_defs.py => mozglue/dllservices/gen_dll_blocklist_defs.py rename : mozglue/build/moz.build => mozglue/dllservices/moz.build rename : mozglue/build/MozglueUtils.h => mozglue/misc/WinUtils.h extra : moz-landing-system : lando
88 lines
2.8 KiB
C++
88 lines
2.8 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 "ModuleLoadFrame.h"
|
|
#include "mozilla/NativeNt.h"
|
|
#include "mozilla/UniquePtr.h"
|
|
#include "NtLoaderAPI.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "WindowsFallbackLoaderAPI.h"
|
|
|
|
static bool IsNullTerminated(PCUNICODE_STRING aStr) {
|
|
return aStr && (aStr->MaximumLength >= (aStr->Length + sizeof(WCHAR))) &&
|
|
aStr->Buffer && aStr->Buffer[aStr->Length / sizeof(WCHAR)] == 0;
|
|
}
|
|
|
|
static mozilla::FallbackLoaderAPI gFallbackLoaderAPI;
|
|
|
|
namespace mozilla {
|
|
namespace glue {
|
|
|
|
nt::LoaderAPI* ModuleLoadFrame::sLoaderAPI;
|
|
|
|
using GetNtLoaderAPIFn = decltype(&mozilla::GetNtLoaderAPI);
|
|
|
|
/* static */
|
|
void ModuleLoadFrame::StaticInit(nt::LoaderObserver* aNewObserver) {
|
|
const auto pGetNtLoaderAPI = reinterpret_cast<GetNtLoaderAPIFn>(
|
|
::GetProcAddress(::GetModuleHandleW(nullptr), "GetNtLoaderAPI"));
|
|
if (!pGetNtLoaderAPI) {
|
|
// This case occurs in processes other than firefox.exe that do not contain
|
|
// the launcher process blocklist.
|
|
gFallbackLoaderAPI.SetObserver(aNewObserver);
|
|
sLoaderAPI = &gFallbackLoaderAPI;
|
|
return;
|
|
}
|
|
|
|
sLoaderAPI = pGetNtLoaderAPI(aNewObserver);
|
|
}
|
|
|
|
ModuleLoadFrame::ModuleLoadFrame(PCUNICODE_STRING aRequestedDllName)
|
|
: mAlreadyLoaded(false),
|
|
mContext(nullptr),
|
|
mDllLoadStatus(STATUS_UNSUCCESSFUL),
|
|
mLoadInfo(sLoaderAPI->ConstructAndNotifyBeginDllLoad(&mContext,
|
|
aRequestedDllName)) {
|
|
if (!aRequestedDllName) {
|
|
return;
|
|
}
|
|
|
|
UniquePtr<WCHAR[]> nameBuf;
|
|
const WCHAR* name = nullptr;
|
|
|
|
if (IsNullTerminated(aRequestedDllName)) {
|
|
name = aRequestedDllName->Buffer;
|
|
} else {
|
|
USHORT charLenExclNul = aRequestedDllName->Length / sizeof(WCHAR);
|
|
USHORT charLenInclNul = charLenExclNul + 1;
|
|
nameBuf = MakeUnique<WCHAR[]>(charLenInclNul);
|
|
if (!wcsncpy_s(nameBuf.get(), charLenInclNul, aRequestedDllName->Buffer,
|
|
charLenExclNul)) {
|
|
name = nameBuf.get();
|
|
}
|
|
}
|
|
|
|
mAlreadyLoaded = name && !!::GetModuleHandleW(name);
|
|
}
|
|
|
|
ModuleLoadFrame::~ModuleLoadFrame() {
|
|
sLoaderAPI->NotifyEndDllLoad(mContext, mDllLoadStatus, std::move(mLoadInfo));
|
|
}
|
|
|
|
void ModuleLoadFrame::SetLoadStatus(NTSTATUS aNtStatus, HANDLE aHandle) {
|
|
mDllLoadStatus = aNtStatus;
|
|
void* baseAddr = mozilla::nt::PEHeaders::HModuleToBaseAddr(
|
|
reinterpret_cast<HMODULE>(aHandle));
|
|
mLoadInfo.mBaseAddr = baseAddr;
|
|
if (!mAlreadyLoaded) {
|
|
mLoadInfo.mSectionName = sLoaderAPI->GetSectionName(baseAddr);
|
|
}
|
|
}
|
|
|
|
} // namespace glue
|
|
} // namespace mozilla
|