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.
 
 
 
 
 
 

580 lines
11 KiB

/*++
Copyright (c) 1999-2000 Microsoft Corporation
Module Name:
utils.c
Abstract:
Contains utilities for kd ext and parsing.
Author:
Scott Holden (sholden) 24-Apr-1999
Revision History:
--*/
#include "tcpipxp.h"
#include "tcpipkd.h"
PTCPIP_SRCH
ParseSrch(
PCSTR args,
ULONG ulDefaultOp,
ULONG ulAllowedOps
)
{
PTCPIP_SRCH pSrch = NULL;
char *pszArgs = NULL;
char *pszSrch;
ULONG cbArgs;
ULONG Status = STATUS_SUCCESS;
//
// Make a copy of the argument string to parse with.
//
cbArgs = strlen(args);
pszArgs = LocalAlloc(LPTR, cbArgs + 1);
if (pszArgs == NULL)
{
Status = STATUS_NO_MEMORY;
goto done;
}
strcpy(pszArgs, args);
//
// Allocate the return srch. Set default::assume default has no params.
//
pSrch = LocalAlloc(LPTR, sizeof(TCPIP_SRCH));
if (pSrch == NULL)
{
Status = STATUS_NO_MEMORY;
goto done;
}
pSrch->ulOp = ulDefaultOp;
// If we need to parse out an addr, do it first...
// Get first token in arg list.
pszSrch = mystrtok(pszArgs, " \t\n");
if (pszSrch == NULL)
{
// If default is invalid (i.e. NO default), return error.
if (pSrch->ulOp == 0 ||
(ulAllowedOps & TCPIP_SRCH_PTR_LIST))
{
Status = STATUS_INVALID_PARAMETER;
dprintf("xxx1\n");
}
goto done;
}
if (ulAllowedOps & TCPIP_SRCH_PTR_LIST)
{
pSrch->ListAddr = GetExpression(pszSrch);
pszSrch = mystrtok(NULL, " \t\n");
if (pszSrch == NULL)
{
if (pSrch->ulOp == 0)
{
Status = STATUS_INVALID_PARAMETER;
}
goto done;
}
}
if (strcmp(pszSrch, "all") == 0)
{
pSrch->ulOp = TCPIP_SRCH_ALL;
}
else if (strcmp(pszSrch, "ipaddr") == 0)
{
CHAR szIP[20];
CHAR *p;
ULONG i;
pSrch->ulOp = TCPIP_SRCH_IPADDR;
pSrch->ipaddr = 0;
pszSrch = mystrtok(NULL, " \t\n");
if (pszSrch == NULL || strlen(pszSrch) >= 15)
{
Status = STATUS_INVALID_PARAMETER;
goto done;
}
strcpy(szIP, pszSrch);
p = mystrtok(szIP, ".");
for (i = 0; i < 4; i++)
{
pSrch->ipaddr |= (atoi(p) << (i*8));
p = mystrtok(NULL, ".");
if (p == NULL)
{
break;
}
}
if (i != 3)
{
Status = STATUS_INVALID_PARAMETER;
goto done;
}
}
else if (strcmp(pszSrch, "port") == 0)
{
pSrch->ulOp = TCPIP_SRCH_PORT;
pszSrch = mystrtok(NULL, " \t\n");
if (pszSrch == NULL)
{
Status = STATUS_INVALID_PARAMETER;
goto done;
}
pSrch->port = (USHORT)atoi(pszSrch);
}
else if (strcmp(pszSrch, "prot") == 0)
{
pSrch->ulOp = TCPIP_SRCH_PROT;
pszSrch = mystrtok(NULL, " \t\n");
if (pszSrch == NULL)
{
Status = STATUS_INVALID_PARAMETER;
goto done;
}
if (strcmp(pszSrch, "raw") == 0)
{
pSrch->prot = PROTOCOL_RAW;
}
else if (strcmp(pszSrch, "udp") == 0)
{
pSrch->prot = PROTOCOL_UDP;
}
else if (strcmp(pszSrch, "tcp") == 0)
{
pSrch->prot = PROTOCOL_TCP;
}
else
{
Status = STATUS_INVALID_PARAMETER;
goto done;
}
}
else if (strcmp(pszSrch, "context") == 0)
{
pSrch->ulOp = TCPIP_SRCH_CONTEXT;
pszSrch = mystrtok(NULL, " \t\n");
if (pszSrch == NULL)
{
Status = STATUS_INVALID_PARAMETER;
goto done;
}
pSrch->context = atoi(pszSrch);
}
else if (strcmp(pszSrch, "stats") == 0)
{
pSrch->ulOp = TCPIP_SRCH_STATS;
}
else
{
// Invalid srch request. Fail.
Status = STATUS_INVALID_PARAMETER;
goto done;
}
// Now see if this is an expected type!!!
if ((pSrch->ulOp & ulAllowedOps) == 0)
{
dprintf("invalid operation for current srch" ENDL);
Status = STATUS_INVALID_PARAMETER;
goto done;
}
done:
if (pszArgs)
{
LocalFree(pszArgs);
}
if (Status != STATUS_SUCCESS)
{
if (pSrch)
{
LocalFree(pSrch);
pSrch = NULL;
}
}
return (pSrch);
}
char *mystrtok ( char *string, char * control )
{
static unsigned char *str;
char *p, *s;
if( string )
str = string;
if( str == NULL || *str == '\0' )
return NULL;
//
// Skip leading delimiters...
//
for( ; *str; str++ ) {
for( s=control; *s; s++ ) {
if( *str == *s )
break;
}
if( *s == '\0' )
break;
}
//
// Was it was all delimiters?
//
if( *str == '\0' ) {
str = NULL;
return NULL;
}
//
// We've got a string, terminate it at first delimeter
//
for( p = str+1; *p; p++ ) {
for( s = control; *s; s++ ) {
if( *p == *s ) {
s = str;
*p = '\0';
str = p+1;
return s;
}
}
}
//
// We've got a string that ends with the NULL
//
s = str;
str = NULL;
return s;
}
ULONG
GetUlongValue (
PCHAR String
)
{
ULONG Location;
ULONG Value;
ULONG result;
Location = GetExpression( String );
if (!Location) {
dprintf("unable to get %s\n",String);
return 0;
}
if ((!ReadMemory((DWORD)Location,&Value,sizeof(ULONG),&result)) ||
(result < sizeof(ULONG))) {
dprintf("unable to get %s\n",String);
return 0;
}
return Value;
}
BOOL
GetData(
PVOID pvData,
ULONG cbData,
ULONG_PTR Address,
PCSTR pszDataType
)
{
BOOL b;
ULONG cbRead;
ULONG count = cbData;
while(cbData > 0)
{
if (count >= 3000)
count = 3000;
b = ReadMemory(Address, pvData, count, &cbRead );
if (!b || cbRead != count )
{
dprintf( "Unable to read %u bytes at %X, for %s\n", cbData, Address, pszDataType );
return (FALSE);
}
Address += count;
cbData -= count;
pvData = (PVOID)((ULONG_PTR)pvData + count);
// if (CheckControlC())
// {
// dprintf("ctrl-c\n");
// return (FALSE);
// }
}
return (TRUE);
}
BOOL
KDDump_ULONG(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
ULONG Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "ULONG");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
dprintf("%-10u", Value);
return (TRUE);
}
BOOL
KDDump_LONG(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
LONG Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "LONG");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
dprintf("%-10d", Value);
return (TRUE);
}
BOOL
KDDump_BOOLEAN(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
BOOLEAN Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "BOOLEAN");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
dprintf("%-10s", Value == TRUE ? "TRUE" : "FALSE");
return (TRUE);
}
BOOL
KDDump_uchar(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
uchar Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "uchar");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
dprintf("%-10u", Value);
return (TRUE);
}
BOOL
KDDump_ushort(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
ushort Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "ushort");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
dprintf("%-10hu", Value);
return (TRUE);
}
BOOL
KDDump_PtrSymbol(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
ULONG_PTR Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "ULONG_PTR");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
DumpPtrSymbol((PVOID) Value);
return (TRUE);
}
BOOL
KDDump_Queue(
PCHAR pVar,
PCHAR pName
)
{
ULONG_PTR Addr;
Queue Value;
BOOL fStatus;
Addr = GetExpression(pVar);
if (Addr == 0)
{
dprintf("Failed to get %s" ENDL, pVar);
return (FALSE);
}
fStatus = GetData(&Value, sizeof(Value), Addr, "Queue");
if (fStatus == FALSE)
{
dprintf("Failed to read %s @ %x" ENDL, pVar, Addr);
return (FALSE);
}
PrintFieldName(pName);
dprintf("q_next = %-10lx", Value.q_next);
dprintf("q_prev = %-10lx", Value.q_prev);
dprintf("%s", (Value.q_next == (Queue *)Addr) ? "[Empty]" : "");
return (TRUE);
}