зеркало из https://github.com/mozilla/gecko-dev.git
78 строки
2.6 KiB
C++
78 строки
2.6 KiB
C++
/* -*- Mode: C++; tab-width: 8; 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/. */
|
|
|
|
#include "InjectCrashReporter.h"
|
|
#include "nsDirectoryServiceUtils.h"
|
|
#include "nsDirectoryServiceDefs.h"
|
|
#include "windows/crash_generation/crash_generation_client.h"
|
|
#include "nsExceptionHandler.h"
|
|
#include "LoadLibraryRemote.h"
|
|
#include "nsWindowsHelpers.h"
|
|
|
|
using CrashReporter::GetChildNotificationPipe;
|
|
using google_breakpad::CrashGenerationClient;
|
|
|
|
namespace mozilla {
|
|
|
|
InjectCrashRunnable::InjectCrashRunnable(DWORD pid)
|
|
: Runnable("InjectCrashRunnable"), mPID(pid) {
|
|
nsCOMPtr<nsIFile> dll;
|
|
nsresult rv = NS_GetSpecialDirectory(NS_GRE_DIR, getter_AddRefs(dll));
|
|
if (NS_SUCCEEDED(rv)) {
|
|
dll->Append(u"breakpadinjector.dll"_ns);
|
|
dll->GetPath(mInjectorPath);
|
|
}
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
InjectCrashRunnable::Run() {
|
|
if (mInjectorPath.IsEmpty()) return NS_OK;
|
|
|
|
nsAutoHandle hProcess(OpenProcess(
|
|
PROCESS_CREATE_THREAD | PROCESS_QUERY_INFORMATION | PROCESS_DUP_HANDLE |
|
|
PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ,
|
|
FALSE, mPID));
|
|
if (!hProcess) {
|
|
NS_WARNING(
|
|
"Unable to open remote process handle for crashreporter injection.");
|
|
return NS_OK;
|
|
}
|
|
|
|
void* proc =
|
|
LoadRemoteLibraryAndGetAddress(hProcess, mInjectorPath.get(), "Start");
|
|
if (!proc) {
|
|
NS_WARNING("Unable to inject crashreporter DLL.");
|
|
return NS_OK;
|
|
}
|
|
|
|
HANDLE hRemotePipe = CrashGenerationClient::DuplicatePipeToClientProcess(
|
|
NS_ConvertASCIItoUTF16(GetChildNotificationPipe()).get(), hProcess);
|
|
if (INVALID_HANDLE_VALUE == hRemotePipe) {
|
|
NS_WARNING("Unable to duplicate crash reporter pipe to process.");
|
|
return NS_OK;
|
|
}
|
|
|
|
nsAutoHandle hThread(CreateRemoteThread(hProcess, nullptr, 0,
|
|
(LPTHREAD_START_ROUTINE)proc,
|
|
(void*)hRemotePipe, 0, nullptr));
|
|
if (!hThread) {
|
|
NS_WARNING("Unable to CreateRemoteThread");
|
|
|
|
// We have to close the remote pipe or else our crash generation client
|
|
// will be stuck unable to accept other remote requests.
|
|
HANDLE toClose = INVALID_HANDLE_VALUE;
|
|
if (DuplicateHandle(hProcess, hRemotePipe, ::GetCurrentProcess(), &toClose,
|
|
0, FALSE,
|
|
DUPLICATE_CLOSE_SOURCE | DUPLICATE_SAME_ACCESS)) {
|
|
CloseHandle(toClose);
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
} // namespace mozilla
|