-processname flag to specify a custom process name.

YaLTeR · Sep 21, 2014 · c25490e · c25490e
1 parent 8934f53
commit c25490e
Showing 1 changed file with 34 additions and 11 deletions.
diff --git a/BunnymodXT-Injector/main.c b/BunnymodXT-Injector/main.c
@@ -5,7 +5,7 @@
 #include <TlHelp32.h>
 #include <tchar.h>
 
-DWORD GetHLProcessID()
+DWORD GetProcessID(const TCHAR* processName)
 {
  HANDLE snapshot;
  PROCESSENTRY32 entry;
@@ -22,7 +22,7 @@ DWORD GetHLProcessID()
  {
  do
  {
- if (_tcsicmp(entry.szExeFile, TEXT("hl.exe")) == 0)
+ if (_tcsicmp(entry.szExeFile, processName) == 0)
  {
  CloseHandle(snapshot);
  return entry.th32ProcessID;
@@ -123,12 +123,31 @@ void DoInjection(HANDLE targetProcess)
  VirtualFreeEx(targetProcess, dllPathAddr, 0, MEM_RELEASE);
 }
 
-BOOL CheckArgs(int argc, char** argv)
+void CheckArgs(int argc, const TCHAR** argv, BOOL* needToWait, size_t processNameSize, TCHAR* processName)
 {
- if ((argc == 2) && !strcmp(argv[1], "-noconfirm"))
- return FALSE;
+ if (needToWait)
+ *needToWait = TRUE;
 
- return TRUE;
+ if (processName)
+ {
+ _tcsncpy(processName, TEXT("hl.exe"), processNameSize);
+ if (processNameSize <= 6)
+ processName[processNameSize] = '\0';
+ }
+
+ for (int i = 1; i < argc; ++i)
+ {
+ if (needToWait && (_tcscmp(argv[i], TEXT("-noconfirm")) == 0))
+ *needToWait = FALSE;
+
+ if (processName && (_tcscmp(argv[i], TEXT("-processname")) == 0) && ((i + 1) < argc))
+ {
+ i++;
+ _tcsncpy(processName, argv[i], processNameSize);
+ if (processNameSize <= _tcslen(argv[i]))
+ processName[processNameSize] = '\0';
+ }
+ }
 }
 
 void waitForKey(BOOL needToWait)
@@ -137,23 +156,27 @@ void waitForKey(BOOL needToWait)
  getc(stdin);
 }
 
-int main(int argc, char** argv)
+int _tmain(int argc, TCHAR** argv)
 {
  HANDLE hlProcess;
- DWORD hlPID = GetHLProcessID();
- BOOL needToWait = CheckArgs(argc, argv);
+ DWORD hlPID;
+ BOOL needToWait;
+ TCHAR processName[16];
+
+ CheckArgs(argc, argv, &needToWait, (sizeof(processName) / sizeof(TCHAR)), processName);
+ hlPID = GetProcessID(processName);
 
  if (hlPID == 0)
  {
- fputs("hl.exe is not running!\n", stderr);
+ _ftprintf(stderr, TEXT("%s is not running!\n"), processName);
  waitForKey(needToWait);
  return 0;
  }
 
  hlProcess = OpenProcess((PROCESS_CREATE_THREAD | PROCESS_QUERY_INFORMATION | PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ), FALSE, hlPID);
  if (hlProcess == NULL)
  {
- fprintf(stderr, "Failed opening the Half-Life process: %d\n", GetLastError());
+ fprintf(stderr, "Failed opening the process: %d\n", GetLastError());
  waitForKey(needToWait);
  return 0;
  }