Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

510 lines
12 KiB

/*++
Copyright (c) 1990 Microsoft Corporation
Module Name:
segment.c
Abstract:
This module contains the debugging support needed to track
16-bit VDM segment notifications.
Author:
Neil Sandlin (neilsa) 1-Mar-1997 Rewrote it
Revision History:
--*/
#include <precomp.h>
#pragma hdrstop
#include <stdio.h>
SEGENTRY SegListHead = {0};
//----------------------------------------------------------------------------
// VDMGetSegtablePointer
//
// This is an undocumented entry point that allows VDMEXTS to dump the
// segment list
//
//----------------------------------------------------------------------------
PSEGENTRY
WINAPI
VDMGetSegtablePointer(
VOID
)
{
return SegListHead.Next;
}
//----------------------------------------------------------------------------
// VDMIsModuleLoaded
//
// Given the path parameter, this routine determines if there are any
// segments in the segment list from the specified executable.
//
//----------------------------------------------------------------------------
BOOL
WINAPI
VDMIsModuleLoaded(
LPSTR szPath
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
if (!*szPath) {
return FALSE;
}
while (pSegEntry) {
if ( _stricmp(pSegEntry->szExePath, szPath) == 0 ) {
return TRUE;
}
pSegEntry = pSegEntry->Next;
}
return FALSE;
}
//----------------------------------------------------------------------------
// SegmentLoad
//
// This routine adds an entry to the segment list based on the parameters
// of a client SegmentLoad notification.
//
//----------------------------------------------------------------------------
BOOL
SegmentLoad(
WORD selector,
WORD segment,
LPSTR szExePath
)
{
PSEGENTRY pSegEntry;
if (strlen(szExePath) >= MAX_PATH16) {
return FALSE;
}
pSegEntry = MALLOC(sizeof(SEGENTRY));
if (pSegEntry == NULL) {
return FALSE;
}
pSegEntry->Next = SegListHead.Next;
SegListHead.Next = pSegEntry;
pSegEntry->selector = selector;
pSegEntry->segment = segment;
pSegEntry->type = SEGTYPE_PROT;
strcpy( pSegEntry->szExePath, szExePath );
ParseModuleName(pSegEntry->szModule, szExePath);
pSegEntry->length = 0;
return TRUE;
}
//----------------------------------------------------------------------------
// SegmentFree
//
// This routine removes the entry from the segment list that matches the
// pass selector.
//
//----------------------------------------------------------------------------
BOOL
SegmentFree(
WORD selector
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
PSEGENTRY pSegPrev = &SegListHead;
PSEGENTRY pSegTmp;
BOOL fResult = FALSE;
while (pSegEntry) {
if ((pSegEntry->type == SEGTYPE_PROT) &&
(pSegEntry->selector == selector)) {
pSegPrev->Next = pSegEntry->Next;
pSegTmp = pSegEntry;
pSegEntry = pSegTmp->Next;
FREE(pSegTmp);
fResult = TRUE;
} else {
pSegEntry = pSegEntry->Next;
}
}
return fResult;
}
//----------------------------------------------------------------------------
// ModuleLoad
//
// This routine adds an entry to the segment list based on the parameters
// of a client ModuleLoad notification.
//
//----------------------------------------------------------------------------
BOOL
ModuleLoad(
WORD selector,
WORD segment,
DWORD length,
LPSTR szExePath
)
{
PSEGENTRY pSegEntry;
if (strlen(szExePath) >= MAX_PATH16) {
return FALSE;
}
pSegEntry = MALLOC(sizeof(SEGENTRY));
if (pSegEntry == NULL) {
return FALSE;
}
pSegEntry->Next = SegListHead.Next;
SegListHead.Next = pSegEntry;
pSegEntry->selector = selector;
pSegEntry->segment = segment;
pSegEntry->type = SEGTYPE_V86;
strcpy( pSegEntry->szExePath, szExePath );
ParseModuleName(pSegEntry->szModule, szExePath);
pSegEntry->length = length;
return TRUE;
}
//----------------------------------------------------------------------------
// ModuleFree
//
// This routine removes all entries from the segment list that contain
// the specified path name.
//
//----------------------------------------------------------------------------
BOOL
ModuleFree(
LPSTR szExePath
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
PSEGENTRY pSegPrev = &SegListHead;
PSEGENTRY pSegTmp;
BOOL fResult = FALSE;
while (pSegEntry) {
if ( _stricmp(pSegEntry->szExePath, szExePath) == 0 ) {
pSegPrev->Next = pSegEntry->Next;
pSegTmp = pSegEntry;
pSegEntry = pSegTmp->Next;
FREE(pSegTmp);
fResult = TRUE;
} else {
pSegEntry = pSegEntry->Next;
}
}
return fResult;
}
BOOL
V86SegmentMove(
WORD Selector,
WORD segment,
DWORD length,
LPSTR szExePath
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
PSEGENTRY pSegPrev = &SegListHead;
//
// first see if one exists already
//
pSegEntry = SegListHead.Next;
while (pSegEntry) {
if ((pSegEntry->type == SEGTYPE_V86) &&
(pSegEntry->segment == segment)) {
// Normal segmove, just update selector
pSegEntry->selector = Selector;
return TRUE;
}
pSegEntry = pSegEntry->Next;
}
//
// An entry for this segment doesn't exist, so create one
//
ModuleLoad(Selector, segment, length, szExePath);
//
// Now delete segment zero for this module. This prevents
// confusion in the symbol routines
//
pSegEntry = SegListHead.Next;
pSegPrev = &SegListHead;
while (pSegEntry) {
if ((pSegEntry->type == SEGTYPE_V86) &&
( _stricmp(pSegEntry->szExePath, szExePath) == 0 ) &&
(pSegEntry->segment == 0)) {
// Unlink and free it
pSegPrev->Next = pSegEntry->Next;
FREE(pSegEntry);
break;
}
pSegEntry = pSegEntry->Next;
}
return TRUE;
}
BOOL
PMSegmentMove(
WORD Selector1,
WORD Selector2
)
{
PSEGENTRY pSegEntry;
if (!Selector2) {
return (SegmentFree(Selector1));
}
// Look for the segment entry
pSegEntry = SegListHead.Next;
while (pSegEntry) {
if ((pSegEntry->type == SEGTYPE_PROT) &&
(pSegEntry->selector == Selector1)) {
// Normal segmove, just update selector
pSegEntry->selector = Selector2;
return TRUE;
}
pSegEntry = pSegEntry->Next;
}
return FALSE;
}
//----------------------------------------------------------------------------
// ProcessSegmentNotification
//
// This routine is the main entry point for the following debugger
// notifications:
// DBG_SEGLOAD
// DBG_SEGFREE
// DBG_SEGMOVE
// DBG_MODLOAD
// DBG_MODFREE
//
// It is called from VDMProcessException.
//
//----------------------------------------------------------------------------
VOID
ProcessSegmentNotification(
LPDEBUG_EVENT lpDebugEvent
)
{
BOOL b;
DWORD lpNumberOfBytesRead;
LPDWORD lpdw;
SEGMENT_NOTE se;
HANDLE hProcess;
PSEGENTRY pSegEntry, pSegPrev;
lpdw = &(lpDebugEvent->u.Exception.ExceptionRecord.ExceptionInformation[0]);
hProcess = OpenProcess( PROCESS_VM_READ, FALSE, lpDebugEvent->dwProcessId );
if ( hProcess == HANDLE_NULL ) {
return;
}
b = ReadProcessMemory(hProcess,
(LPVOID)lpdw[2],
&se,
sizeof(se),
&lpNumberOfBytesRead );
if ( !b || lpNumberOfBytesRead != sizeof(se) ) {
return;
}
switch(LOWORD(lpdw[0])) {
case DBG_SEGLOAD:
SegmentLoad(se.Selector1, se.Segment, se.FileName);
break;
case DBG_SEGMOVE:
if (se.Type == SN_V86) {
V86SegmentMove(se.Selector2, se.Segment, se.Length, se.FileName);
} else {
PMSegmentMove(se.Selector1, se.Selector2);
}
break;
case DBG_SEGFREE:
// Here, se.Type is a boolean to tell whether to restore
// any breakpoints in the segment. That was done in the api
// because wdeb386 didn't know how to move the breakpoint
// definitions during a SEGMOVE. Currently, we ignore it, but
// it would be nice to either support the flag, or better to
// have ntsd update the breakpoints based on it.
SegmentFree(se.Selector1);
break;
case DBG_MODFREE:
ModuleFree(se.FileName);
break;
case DBG_MODLOAD:
ModuleLoad(se.Selector1, 0, se.Length, se.FileName);
break;
}
CloseHandle( hProcess );
}
void
CopySegmentInfo(
VDM_SEGINFO *si,
PSEGENTRY pSegEntry
)
{
si->Selector = pSegEntry->selector;
si->SegNumber = pSegEntry->segment;
si->Length = pSegEntry->length;
si->Type = (pSegEntry->type == SEGTYPE_V86) ? 0 : 1;
strcpy(si->ModuleName, pSegEntry->szModule);
strcpy(si->FileName, pSegEntry->szExePath);
}
//----------------------------------------------------------------------------
// VDMGetSegmentInfo
//
// This routine fills in a VDM_SEGINFO structure for the segment that matches
// the specified parameters.
// notifications:
// DBG_SEGLOAD
// DBG_SEGFREE
// DBG_SEGMOVE
// DBG_MODLOAD
// DBG_MODFREE
//
// It is called from VDMProcessException.
//
//----------------------------------------------------------------------------
BOOL
WINAPI
VDMGetSegmentInfo(
WORD Selector,
ULONG Offset,
BOOL bProtectMode,
VDM_SEGINFO *si
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
PSEGENTRY pSegPrev = &SegListHead;
int mode = bProtectMode ? SEGTYPE_PROT : SEGTYPE_V86;
ULONG Base, BaseEnd, Target;
while (pSegEntry) {
if (pSegEntry->type == mode) {
switch(mode) {
case SEGTYPE_PROT:
if (pSegEntry->selector == Selector) {
CopySegmentInfo(si, pSegEntry);
return TRUE;
}
break;
case SEGTYPE_V86:
Base = pSegEntry->selector << 4;
BaseEnd = Base + pSegEntry->length;
Target = (Selector << 4) + Offset;
if ((Target >= Base) && (Target < BaseEnd)) {
CopySegmentInfo(si, pSegEntry);
return TRUE;
}
break;
}
}
pSegEntry = pSegEntry->Next;
}
return FALSE;
}
BOOL
GetInfoBySegmentNumber(
LPSTR szModule,
WORD SegNumber,
VDM_SEGINFO *si
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
PSEGENTRY pSegPrev = &SegListHead;
ULONG Base, BaseEnd, Target;
while (pSegEntry) {
if (_stricmp(szModule, pSegEntry->szModule) == 0) {
if (pSegEntry->segment == 0 || pSegEntry->segment == SegNumber) {
CopySegmentInfo(si, pSegEntry);
return TRUE;
}
}
pSegEntry = pSegEntry->Next;
}
return FALSE;
}
BOOL
EnumerateModulesForValue(
BOOL (WINAPI *pfnEnumModuleProc)(LPSTR,LPSTR,PWORD,PDWORD,PWORD),
LPSTR szSymbol,
PWORD pSelector,
PDWORD pOffset,
PWORD pType
)
{
PSEGENTRY pSegEntry = SegListHead.Next;
PSEGENTRY pSegPrev = &SegListHead;
ULONG Base, BaseEnd, Target;
while (pSegEntry) {
if (pSegEntry->szModule) {
//
// BUGBUG should optimize this so that it only calls
// the enum proc once per module, instead of once per
// segment
//
if ((*pfnEnumModuleProc)(pSegEntry->szModule,
szSymbol,
pSelector,
pOffset,
pType)) {
return TRUE;
}
}
pSegEntry = pSegEntry->Next;
}
return FALSE;
}