57 lines
1.6 KiB
C
57 lines
1.6 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
// inject.h
|
||
|
|
// 1/27/2013 jichi
|
||
|
|
|
||
|
|
#include "windbg/windbg.h"
|
||
|
|
|
||
|
|
#include <windows.h>
|
||
|
|
|
||
|
|
WINDBG_BEGIN_NAMESPACE
|
||
|
|
|
||
|
|
enum { PROCESS_INJECT_ACCESS = PROCESS_CREATE_THREAD|PROCESS_QUERY_INFORMATION|PROCESS_VM_OPERATION|PROCESS_VM_WRITE|PROCESS_VM_READ };
|
||
|
|
enum { INJECT_TIMEOUT = 3000 }; // wait at most 3 seconds for creating remote threads
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Inject function with 1 argument
|
||
|
|
* Either pid or the process handle should be specified
|
||
|
|
* @param addr LONG function memory address
|
||
|
|
* @param arg LPVOID arg1 data
|
||
|
|
* @param argSize int arg1 data size
|
||
|
|
* @param pid process id
|
||
|
|
* @param hProcess process handle
|
||
|
|
* @param timeout msec
|
||
|
|
* @return BOOL
|
||
|
|
*/
|
||
|
|
BOOL injectFunction1(_In_ LPCVOID addr, _In_ LPCVOID arg, _In_ SIZE_T argSize,
|
||
|
|
_In_ DWORD pid = 0, _In_ HANDLE hProcess = INVALID_HANDLE_VALUE,
|
||
|
|
_In_ INT timeout = INJECT_TIMEOUT);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Either pid or the process handle should be specified
|
||
|
|
* @param dllpath ABSOLUTE path to dll
|
||
|
|
* @param pid process id
|
||
|
|
* @param hProcess process handle
|
||
|
|
* @param timeout msec
|
||
|
|
* @return BOOL
|
||
|
|
*/
|
||
|
|
BOOL injectDllW(_In_ LPCWSTR dllPath,
|
||
|
|
_In_ DWORD pid = 0, _In_ HANDLE hProcess = INVALID_HANDLE_VALUE,
|
||
|
|
_In_ INT timeout = INJECT_TIMEOUT);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Either pid or the process handle should be specified
|
||
|
|
* @param hDll dll module handle
|
||
|
|
* @param pid process id
|
||
|
|
* @param hProcess process handle
|
||
|
|
* @param timeout msec
|
||
|
|
* @return BOOL
|
||
|
|
*/
|
||
|
|
BOOL ejectDll(_In_ HANDLE hDll,
|
||
|
|
_In_ DWORD pid = 0, _In_ HANDLE hProcess = INVALID_HANDLE_VALUE,
|
||
|
|
_In_ INT timeout = INJECT_TIMEOUT);
|
||
|
|
|
||
|
|
WINDBG_END_NAMESPACE
|
||
|
|
|
||
|
|
// EOF
|