diff --git a/inject/src/Inject.c b/inject/src/Inject.c index d7918fa..bf0bd5a 100644 --- a/inject/src/Inject.c +++ b/inject/src/Inject.c @@ -31,6 +31,7 @@ #include #include #include +#include #include "LoadLibraryR.h" #pragma comment(lib,"Advapi32.lib") @@ -59,8 +60,26 @@ int main( int argc, char * argv[] ) do { - // Usage: inject.exe [string] [pid] [dll_file] - + // Usage: inject.exe [pid] [dll_file] + + //inject by the name of your process + HANDLE processList = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); + PROCESSENTRY32 pInfo; + BOOL st = TRUE; + pInfo.dwSize = sizeof(PROCESSENTRY32); + Process32First(processList, &pInfo); + int myPid = 0; + + do + { + if (strcmp(pInfo.szExeFile, "explorer.exe") == 0) + { + myPid = pInfo.th32ProcessID; + break; + } + Process32Next(processList, &pInfo); + } while (st != FALSE); + if( argc == 2 ) dwProcessId = GetCurrentProcessId(); else @@ -121,4 +140,4 @@ int main( int argc, char * argv[] ) CloseHandle( hProcess ); return dwExitCode; -} \ No newline at end of file +}