192 lines
3.9 KiB
C++
192 lines
3.9 KiB
C++
/*++
|
||
|
||
Copyright (c) 1995-1999 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
exec.cxx
|
||
|
||
Abstract:
|
||
|
||
This module contains an NTSD debugger extension for executing
|
||
external commands.
|
||
|
||
Author:
|
||
|
||
Keith Moore (keithmo) 12-Nov-1997
|
||
|
||
Revision History:
|
||
|
||
--*/
|
||
|
||
#include "inetdbgp.h"
|
||
|
||
|
||
/************************************************************
|
||
* Execute
|
||
************************************************************/
|
||
|
||
|
||
DECLARE_API( exec )
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This function is called as an NTSD extension to ...
|
||
|
||
Arguments:
|
||
|
||
hCurrentProcess - Supplies a handle to the current process (at the
|
||
time the extension was called).
|
||
|
||
hCurrentThread - Supplies a handle to the current thread (at the
|
||
time the extension was called).
|
||
|
||
CurrentPc - Supplies the current pc at the time the extension is
|
||
called.
|
||
|
||
lpExtensionApis - Supplies the address of the functions callable
|
||
by this extension.
|
||
|
||
lpArgumentString - Supplies the asciiz string that describes the
|
||
ansi string to be dumped.
|
||
|
||
Return Value:
|
||
|
||
None.
|
||
|
||
--*/
|
||
|
||
{
|
||
|
||
BOOL result;
|
||
STARTUPINFO startInfo;
|
||
PROCESS_INFORMATION processInfo;
|
||
|
||
INIT_API();
|
||
|
||
//
|
||
// Skip leading blanks.
|
||
//
|
||
|
||
while( *lpArgumentString == ' ' ||
|
||
*lpArgumentString == '\t' ) {
|
||
lpArgumentString++;
|
||
}
|
||
|
||
//
|
||
// Use "cmd.exe" by default.
|
||
//
|
||
|
||
if( *lpArgumentString == '\0' ) {
|
||
lpArgumentString = "cmd";
|
||
}
|
||
|
||
//
|
||
// Set the prompt environment variable so the user will have clue.
|
||
//
|
||
|
||
SetEnvironmentVariable(
|
||
"PROMPT",
|
||
"!inetdbg.exec - $p$g"
|
||
);
|
||
|
||
//
|
||
// Launch it.
|
||
//
|
||
|
||
ZeroMemory(
|
||
&startInfo,
|
||
sizeof(startInfo)
|
||
);
|
||
|
||
ZeroMemory(
|
||
&processInfo,
|
||
sizeof(processInfo)
|
||
);
|
||
|
||
startInfo.cb = sizeof(startInfo);
|
||
|
||
result = CreateProcess(
|
||
NULL, // lpszImageName
|
||
lpArgumentString, // lpszCommandLine
|
||
NULL, // lpsaProcess
|
||
NULL, // lpsaThread
|
||
TRUE, // fInheritHandles
|
||
0, // fdwCreate
|
||
NULL, // lpvEnvironment
|
||
NULL, // lpszCurDir
|
||
&startInfo, // lpsiStartInfo
|
||
&processInfo // lppiProcessInfo
|
||
);
|
||
|
||
if( result ) {
|
||
|
||
//
|
||
// Wait for the child process to terminate, then cleanup.
|
||
//
|
||
|
||
WaitForSingleObject( processInfo.hProcess, INFINITE );
|
||
CloseHandle( processInfo.hProcess );
|
||
CloseHandle( processInfo.hThread );
|
||
|
||
} else {
|
||
|
||
//
|
||
// Could not launch the process.
|
||
//
|
||
|
||
dprintf(
|
||
"cannot launch \"%s\", error %lu\n",
|
||
lpArgumentString,
|
||
GetLastError()
|
||
);
|
||
|
||
}
|
||
|
||
} // DECLARE_API( exec )
|
||
|
||
|
||
|
||
/*
|
||
CallExtension allows one extension to call another extension in a
|
||
different DLL. From TomCan.
|
||
|
||
DECLARE_API(test)
|
||
{
|
||
NTSTATUS status = ERROR_SUCCESS;
|
||
status = DO_EXTENSION("kdextx86.dll", "pool", args);
|
||
dprintf("\n\nStatus: %08X\n", status);
|
||
}
|
||
*/
|
||
|
||
NTSTATUS
|
||
CallExtension(
|
||
PCSTR szModuleName,
|
||
PCSTR szFunction,
|
||
HANDLE hCurrentProcess,
|
||
HANDLE hCurrentThread,
|
||
ULONG dwCurrentPc,
|
||
ULONG dwProcessor,
|
||
PCSTR args)
|
||
{
|
||
PWINDBG_EXTENSION_FUNCTION pFn = NULL;
|
||
HINSTANCE hlib = NULL;
|
||
NTSTATUS status = ERROR_SUCCESS;
|
||
hlib = LoadLibrary(szModuleName);
|
||
if (hlib != NULL)
|
||
{
|
||
pFn = (PWINDBG_EXTENSION_FUNCTION)GetProcAddress(hlib, szFunction);
|
||
if (pFn != NULL)
|
||
{
|
||
(pFn)(hCurrentProcess, hCurrentThread, dwCurrentPc,
|
||
dwProcessor, args);
|
||
}
|
||
FreeLibrary(hlib);
|
||
}
|
||
status = GetLastError();
|
||
return status;
|
||
}
|
||
|