DllUnInject add sem release, Monitor wait for prrocess

Author: michaelchen1996

APIHOOK/DllUnInject/DllUnInject.cpp

@@ -6,6 +6,19 @@
 
 int main()
 {
+	HANDLE hSemaphoreStatus;
+
+	//hSemaphoreStatus = OpenSemaphore(SYNCHRONIZE, FALSE, L"APIHOOK_Monitor_Semaphore_Status");
+	hSemaphoreStatus = CreateSemaphore(NULL, 0, 1, L"APIHOOK_Monitor_Semaphore_Status");
+	if (!hSemaphoreStatus)
+	{
+		printf("CreateSemaphore ERROR\n");
+		return 0;
+	}
+	ReleaseSemaphore(hSemaphoreStatus, 1, NULL);
+	CloseHandle(hSemaphoreStatus);
+	hSemaphoreStatus = NULL;
+
     return 0;
 }
 

APIHOOK/DllUnInject/DllUnInject.h

@@ -0,0 +1 @@
+#pragma once

APIHOOK/DllUnInject/stdafx.h

@@ -9,6 +9,7 @@
 
 #include <stdio.h>
 #include <tchar.h>
+#include <windows.h>
 
 
 

APIHOOK/Monitor/Monitor.cpp

@@ -89,7 +89,8 @@ void StartMonitor()
 		return;
 	}
 	OutputDebugString(L"CreateProcess DllInject.exe\n");
-
+	WaitForSingleObject(pi.hProcess, INFINITE);
+	OutputDebugString(L"DllInject.exe Finished\n");
 
 	WaitForSingleObject(hThreadLog, INFINITE);
 	ReleaseMutex(hMutexSingleton);
@@ -117,10 +118,14 @@ void StopMonitor()
 		return;
 	}
 	OutputDebugString(L"CreateProcess DllUnInject.exe\n");
+	WaitForSingleObject(pi.hProcess, INFINITE);
+	OutputDebugString(L"DllUnInject.exe Finished\n");
 
 }
 
 void RestartMonitor()
 {
 	OutputDebugString(L"Do restart \n");
+	StopMonitor();
+	StartMonitor();
 }