484 lines
10 KiB
C++
484 lines
10 KiB
C++
/*++
|
|
|
|
Copyright (c) 1999-2000 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
w64cpuex.cpp
|
|
|
|
Abstract:
|
|
|
|
Debugger extension DLL for debugging the CPU
|
|
|
|
Author:
|
|
|
|
27-Sept-1999 BarryBo
|
|
|
|
Revision History:
|
|
|
|
|
|
--*/
|
|
|
|
#define _WOW64CPUDBGAPI_
|
|
#define DECLARE_CPU_DEBUGGER_INTERFACE
|
|
#include <nt.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include <windows.h>
|
|
#include <imagehlp.h>
|
|
#include <dbgeng.h>
|
|
#include <ntosp.h>
|
|
|
|
#if defined _X86_
|
|
#define WOW64EXTS_386
|
|
#endif
|
|
|
|
#include "wow64.h"
|
|
#include "wow64cpu.h"
|
|
#include "ia64cpu.h"
|
|
|
|
// Safe release and NULL.
|
|
|
|
#define EXT_RELEASE(Unk) \
|
|
((Unk) != NULL ? ((Unk)->Release(), (Unk) = NULL) : NULL)
|
|
|
|
#define CONTEXT_OFFSET FIELD_OFFSET(TEB64, TlsSlots[WOW64_TLS_CPURESERVED])
|
|
|
|
|
|
CPUCONTEXT LocalCpuContext;
|
|
PWOW64GETCPUDATA CpuGetData;
|
|
PDEBUG_ADVANCED g_ExtAdvanced;
|
|
PDEBUG_CLIENT g_ExtClient;
|
|
PDEBUG_CONTROL g_ExtControl;
|
|
PDEBUG_DATA_SPACES g_ExtData;
|
|
PDEBUG_REGISTERS g_ExtRegisters;
|
|
PDEBUG_SYMBOLS g_ExtSymbols;
|
|
PDEBUG_SYSTEM_OBJECTS g_ExtSystem;
|
|
void
|
|
ExtRelease(void)
|
|
{
|
|
g_ExtClient = NULL;
|
|
EXT_RELEASE(g_ExtAdvanced);
|
|
EXT_RELEASE(g_ExtControl);
|
|
EXT_RELEASE(g_ExtData);
|
|
EXT_RELEASE(g_ExtRegisters);
|
|
EXT_RELEASE(g_ExtSymbols);
|
|
EXT_RELEASE(g_ExtSystem);
|
|
}
|
|
|
|
// Queries for all debugger interfaces.
|
|
HRESULT
|
|
ExtQuery(PDEBUG_CLIENT Client)
|
|
{
|
|
HRESULT Status;
|
|
|
|
if ((Status = Client->QueryInterface(__uuidof(IDebugAdvanced),
|
|
(void **)&g_ExtAdvanced)) != S_OK)
|
|
{
|
|
goto Fail;
|
|
}
|
|
if ((Status = Client->QueryInterface(__uuidof(IDebugControl),
|
|
(void **)&g_ExtControl)) != S_OK)
|
|
{
|
|
goto Fail;
|
|
}
|
|
if ((Status = Client->QueryInterface(__uuidof(IDebugDataSpaces),
|
|
(void **)&g_ExtData)) != S_OK)
|
|
{
|
|
goto Fail;
|
|
}
|
|
if ((Status = Client->QueryInterface(__uuidof(IDebugRegisters),
|
|
(void **)&g_ExtRegisters)) != S_OK)
|
|
{
|
|
goto Fail;
|
|
}
|
|
if ((Status = Client->QueryInterface(__uuidof(IDebugSymbols),
|
|
(void **)&g_ExtSymbols)) != S_OK)
|
|
{
|
|
goto Fail;
|
|
}
|
|
if ((Status = Client->QueryInterface(__uuidof(IDebugSystemObjects),
|
|
(void **)&g_ExtSystem)) != S_OK)
|
|
{
|
|
goto Fail;
|
|
}
|
|
|
|
g_ExtClient = Client;
|
|
|
|
return S_OK;
|
|
|
|
Fail:
|
|
ExtRelease();
|
|
return Status;
|
|
}
|
|
// Normal output.
|
|
void __cdecl
|
|
ExtOut(PCSTR Format, ...)
|
|
{
|
|
va_list Args;
|
|
|
|
va_start(Args, Format);
|
|
g_ExtControl->OutputVaList(DEBUG_OUTPUT_NORMAL, Format, Args);
|
|
va_end(Args);
|
|
}
|
|
|
|
// Error output.
|
|
void __cdecl
|
|
ExtErr(PCSTR Format, ...)
|
|
{
|
|
va_list Args;
|
|
|
|
va_start(Args, Format);
|
|
g_ExtControl->OutputVaList(DEBUG_OUTPUT_ERROR, Format, Args);
|
|
va_end(Args);
|
|
}
|
|
|
|
// Warning output.
|
|
void __cdecl
|
|
ExtWarn(PCSTR Format, ...)
|
|
{
|
|
va_list Args;
|
|
|
|
va_start(Args, Format);
|
|
g_ExtControl->OutputVaList(DEBUG_OUTPUT_WARNING, Format, Args);
|
|
va_end(Args);
|
|
}
|
|
|
|
// Verbose output.
|
|
void __cdecl
|
|
ExtVerb(PCSTR Format, ...)
|
|
{
|
|
va_list Args;
|
|
|
|
va_start(Args, Format);
|
|
g_ExtControl->OutputVaList(DEBUG_OUTPUT_VERBOSE, Format, Args);
|
|
va_end(Args);
|
|
}
|
|
|
|
WOW64CPUDBGAPI VOID
|
|
CpuDbgInitEngapi(
|
|
PWOW64GETCPUDATA lpGetData
|
|
)
|
|
{
|
|
CpuGetData = lpGetData;
|
|
}
|
|
|
|
|
|
HRESULT
|
|
EngGetContextThread(
|
|
IN OUT PCONTEXT32 Context)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine extract the context record of any thread. This is a generic routine.
|
|
When entered, if the target thread isn't the current thread, then it should be
|
|
guaranteed that the target thread is suspended at a proper CPU state.
|
|
|
|
Arguments:
|
|
|
|
Context - Context record to fill
|
|
|
|
Return Value:
|
|
|
|
HRESULT.
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr;
|
|
NTSTATUS NtStatus;
|
|
CONTEXT ContextEM;
|
|
ULONG64 CpuRemoteContext;
|
|
ULONG64 Teb;
|
|
CPUCONTEXT CpuContext;
|
|
|
|
ContextEM.ContextFlags = CONTEXT_FULL;
|
|
|
|
hr = g_ExtSystem->GetCurrentThreadTeb(&Teb);
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
hr = g_ExtData->ReadVirtual(Teb + CONTEXT_OFFSET,
|
|
&CpuRemoteContext,
|
|
sizeof(CpuRemoteContext),
|
|
NULL);
|
|
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
hr = g_ExtData->ReadVirtual(CpuRemoteContext,
|
|
&CpuContext,
|
|
sizeof(CpuContext),
|
|
NULL);
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
NtStatus = GetContextRecord(&CpuContext, Context);
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
return E_FAIL;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
|
|
WOW64CPUDBGAPI BOOL
|
|
CpuDbgGetRemoteContext(
|
|
PDEBUG_CLIENT Client,
|
|
PVOID CpuData
|
|
)
|
|
{
|
|
BOOL bRet = FALSE;
|
|
HRESULT hr;
|
|
CONTEXT Context;
|
|
|
|
hr = ExtQuery(Client);
|
|
if (FAILED(hr)) {
|
|
return FALSE;
|
|
}
|
|
LocalCpuContext.Context.ContextFlags = CONTEXT32_FULLFLOAT;
|
|
hr = EngGetContextThread(&LocalCpuContext.Context);
|
|
if (FAILED(hr)) {
|
|
goto Done;
|
|
}
|
|
|
|
bRet = TRUE;
|
|
Done:
|
|
ExtRelease();
|
|
return bRet;
|
|
}
|
|
|
|
HRESULT
|
|
EngSetContextThread(
|
|
IN OUT PCONTEXT32 Context)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine sets the context record of any thread. This is a generic routine.
|
|
When entered, if the target thread isn't the currently executing thread, then it should be
|
|
guaranteed that the target thread is suspended at a proper CPU state.
|
|
|
|
Arguments:
|
|
|
|
Context - Context record to set
|
|
|
|
Return Value:
|
|
|
|
HRESULT.
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr;
|
|
NTSTATUS NtStatus;
|
|
CONTEXT ContextEM;
|
|
ULONG64 CpuRemoteContext;
|
|
ULONG64 Teb;
|
|
CPUCONTEXT CpuContext;
|
|
|
|
ContextEM.ContextFlags = CONTEXT_FULL;
|
|
|
|
hr = g_ExtAdvanced->SetThreadContext(&ContextEM, sizeof(ContextEM));
|
|
|
|
hr = g_ExtSystem->GetCurrentThreadTeb(&Teb);
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
hr = g_ExtData->ReadVirtual(Teb + CONTEXT_OFFSET,
|
|
&CpuRemoteContext,
|
|
sizeof(CpuRemoteContext),
|
|
NULL);
|
|
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
hr = g_ExtData->ReadVirtual(CpuRemoteContext,
|
|
&CpuContext,
|
|
sizeof(CpuContext),
|
|
NULL);
|
|
if (FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
NtStatus = SetContextRecord(&CpuContext, Context);
|
|
if (!NT_SUCCESS(NtStatus)) {
|
|
return E_FAIL;
|
|
}
|
|
hr = g_ExtData->WriteVirtual(CpuRemoteContext,
|
|
&CpuContext,
|
|
sizeof(CpuContext),
|
|
NULL);
|
|
return hr;
|
|
}
|
|
|
|
|
|
WOW64CPUDBGAPI BOOL
|
|
CpuDbgSetRemoteContext(
|
|
PDEBUG_CLIENT Client
|
|
)
|
|
{
|
|
BOOL bRet = FALSE;
|
|
HRESULT hr;
|
|
NTSTATUS Status;
|
|
CONTEXT Context;
|
|
|
|
hr = ExtQuery(Client);
|
|
if (FAILED(hr)) {
|
|
return FALSE;
|
|
}
|
|
|
|
LocalCpuContext.Context.ContextFlags = CONTEXT32_FULLFLOAT;
|
|
hr = EngSetContextThread(&LocalCpuContext.Context);
|
|
if (FAILED(hr)) {
|
|
ExtOut("CpuDbgSetRemoteContext: Error %x writing CPU context\n", hr);
|
|
goto Done;
|
|
}
|
|
|
|
bRet = TRUE;
|
|
Done:
|
|
ExtRelease();
|
|
return bRet;
|
|
}
|
|
|
|
WOW64CPUDBGAPI BOOL
|
|
CpuDbgGetLocalContext(
|
|
PDEBUG_CLIENT Client,
|
|
PCONTEXT32 Context
|
|
)
|
|
{
|
|
return NT_SUCCESS(GetContextRecord(&LocalCpuContext,
|
|
Context));
|
|
}
|
|
|
|
WOW64CPUDBGAPI BOOL
|
|
CpuDbgSetLocalContext(
|
|
PDEBUG_CLIENT Client,
|
|
PCONTEXT32 Context
|
|
)
|
|
{
|
|
return NT_SUCCESS(SetContextRecord(&LocalCpuContext,
|
|
Context));
|
|
}
|
|
|
|
|
|
WOW64CPUDBGAPI VOID
|
|
CpuDbgFlushInstructionCacheWithHandle(
|
|
HANDLE Process,
|
|
PVOID Addr,
|
|
DWORD Length
|
|
)
|
|
{
|
|
|
|
NtFlushInstructionCache((HANDLE)Process, Addr, Length);
|
|
}
|
|
|
|
|
|
WOW64CPUDBGAPI VOID
|
|
CpuDbgFlushInstructionCache(
|
|
PDEBUG_CLIENT Client,
|
|
PVOID Addr,
|
|
DWORD Length
|
|
)
|
|
{
|
|
HRESULT hr;
|
|
ULONG64 Process;
|
|
|
|
hr = ExtQuery(Client);
|
|
if (FAILED(hr)) {
|
|
return;
|
|
}
|
|
hr = g_ExtSystem->GetCurrentProcessHandle(&Process);
|
|
if (FAILED(hr)) {
|
|
ExtOut("CpuDbgFlushInstructionCache: failed to get Process Handle!\n");
|
|
return;
|
|
}
|
|
CpuDbgFlushInstructionCacheWithHandle((HANDLE)Process, Addr, Length);
|
|
ExtRelease();
|
|
}
|
|
|
|
|
|
VOID SetEax(ULONG ul) {
|
|
LocalCpuContext.Context.Eax = ul;
|
|
}
|
|
VOID SetEbx(ULONG ul) {
|
|
LocalCpuContext.Context.Ebx = ul;
|
|
}
|
|
VOID SetEcx(ULONG ul) {
|
|
LocalCpuContext.Context.Ecx = ul;
|
|
}
|
|
VOID SetEdx(ULONG ul) {
|
|
LocalCpuContext.Context.Edx = ul;
|
|
}
|
|
VOID SetEsi(ULONG ul) {
|
|
LocalCpuContext.Context.Esi = ul;
|
|
}
|
|
VOID SetEdi(ULONG ul) {
|
|
LocalCpuContext.Context.Edi = ul;
|
|
}
|
|
VOID SetEbp(ULONG ul) {
|
|
LocalCpuContext.Context.Ebp = ul;
|
|
}
|
|
VOID SetEsp(ULONG ul) {
|
|
LocalCpuContext.Context.Esp = ul;
|
|
}
|
|
VOID SetEip(ULONG ul) {
|
|
LocalCpuContext.Context.Eip = ul;
|
|
}
|
|
VOID SetEfl(ULONG ul) {
|
|
LocalCpuContext.Context.EFlags = ul;
|
|
}
|
|
|
|
ULONG GetEax(VOID) {
|
|
return LocalCpuContext.Context.Eax;
|
|
}
|
|
ULONG GetEbx(VOID) {
|
|
return LocalCpuContext.Context.Ebx;
|
|
}
|
|
ULONG GetEcx(VOID) {
|
|
return LocalCpuContext.Context.Ecx;
|
|
}
|
|
ULONG GetEdx(VOID) {
|
|
return LocalCpuContext.Context.Edx;
|
|
}
|
|
ULONG GetEsi(VOID) {
|
|
return LocalCpuContext.Context.Esi;
|
|
}
|
|
ULONG GetEdi(VOID) {
|
|
return LocalCpuContext.Context.Edi;
|
|
}
|
|
ULONG GetEbp(VOID) {
|
|
return LocalCpuContext.Context.Ebp;
|
|
}
|
|
ULONG GetEsp(VOID) {
|
|
return LocalCpuContext.Context.Esp;
|
|
}
|
|
ULONG GetEip(VOID) {
|
|
return LocalCpuContext.Context.Eip;
|
|
}
|
|
ULONG GetEfl(VOID) {
|
|
return LocalCpuContext.Context.EFlags;
|
|
}
|
|
|
|
CPUREGFUNCS CpuRegFuncs[] = {
|
|
{ "Eax", SetEax, GetEax },
|
|
{ "Ebx", SetEbx, GetEbx },
|
|
{ "Ecx", SetEcx, GetEcx },
|
|
{ "Edx", SetEdx, GetEdx },
|
|
{ "Esi", SetEsi, GetEsi },
|
|
{ "Edi", SetEdi, GetEdi },
|
|
{ "Ebp", SetEbp, GetEbp },
|
|
{ "Esp", SetEsp, GetEsp },
|
|
{ "Eip", SetEip, GetEip },
|
|
{ "Efl", SetEfl, GetEfl },
|
|
{ NULL, NULL, NULL}
|
|
};
|
|
|
|
WOW64CPUDBGAPI PCPUREGFUNCS
|
|
CpuDbgGetRegisterFuncs(
|
|
void
|
|
)
|
|
{
|
|
return CpuRegFuncs;
|
|
}
|