Allow detecting injection success

At least this will hint to issues that can be discovered by looking at the log.
devlooped · Sep 23, 2022 · b578ed3 · b578ed3
1 parent 6ca7e08
commit b578ed3
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/src/Bootstrap/Injector.cpp b/src/Bootstrap/Injector.cpp
@@ -16,7 +16,7 @@ static HHOOK _messageHookHandle;
 //-----------------------------------------------------------------------------
 //Spying Process functions follow
 //-----------------------------------------------------------------------------
-void Injector::Launch(System::IntPtr windowHandle, System::String^ assemblyFile, System::String^ typeFullName, System::String^ methodName)
+bool Injector::Launch(System::IntPtr windowHandle, System::String^ assemblyFile, System::String^ typeFullName, System::String^ methodName)
 {
 	System::String^ assemblyClassAndMethod = assemblyFile + "$" + typeFullName + "$" + methodName;
 	pin_ptr<const wchar_t> acmLocal = PtrToStringChars(assemblyClassAndMethod);
@@ -52,6 +52,7 @@ void Injector::Launch(System::IntPtr windowHandle, System::String^ assemblyFile,
 						LogMessage("SetWindowsHookEx successful", true);
 						::SendMessage((HWND)windowHandle.ToPointer(), WM_GOBABYGO, (WPARAM)acmRemote, 0);
 						::UnhookWindowsHookEx(_messageHookHandle);
+                        return true;
 					}
 
 					::VirtualFreeEx(hProcess, acmRemote, 0, MEM_RELEASE);
@@ -69,6 +70,8 @@ void Injector::Launch(System::IntPtr windowHandle, System::String^ assemblyFile,
         }
 		::FreeLibrary(hinstDLL);
 	}
+
+    return false;
 }
 
 void Injector::LogMessage(String^ message, bool append)

diff --git a/src/Bootstrap/Injector.h b/src/Bootstrap/Injector.h
@@ -16,7 +16,7 @@ namespace Devlooped
 	{
 		public:
 
-		static void Launch(System::IntPtr windowHandle, System::String^ assemblyFile, System::String^ typeFullName, System::String^ methodName);
+		static bool Launch(System::IntPtr windowHandle, System::String^ assemblyFile, System::String^ typeFullName, System::String^ methodName);
 
 		static void LogMessage(System::String^ message, bool append);
 	};

diff --git a/src/Injector/Program.cs b/src/Injector/Program.cs
@@ -31,8 +31,10 @@ static int Main(string[] args)
             var typeName = args[2].Trim('"');
             var methodName = args[3].Trim('"');
 
-            Injector.Launch(mainWindow, assemblyFile, typeName, methodName);
-            return 0;
+            if (Injector.Launch(mainWindow, assemblyFile, typeName, methodName))
+                return 0;
+            else
+                return -1;
         }
     }
 }