Monitor Lock the TargetList.txt file instead of using semaphore_inject

Author: michaelchen1996

APIHOOK/DllInject/DllInject.cpp

@@ -92,9 +92,10 @@ DWORD ReadTargetList(LPCWCHAR szCurrentDirectory, LPDWORD dwPIdList) {
 	StringCbCat(szListDirectory, MAX_PATH, L"TargetList.txt");
 	OutputDebugString(szListDirectory);
 	OutputDebugString(L"\n");
-	hFile = CreateFile(szListDirectory, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+	hFile = CreateFile(szListDirectory, GENERIC_READ, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
 	if (INVALID_HANDLE_VALUE == hFile)
 	{
+		printf_s("error: %d\n", GetLastError()); system("PAUSE");
 		OutputDebugString(L"TargetList.txt NOT FOUND\n");
 		return 0;
 	}

APIHOOK/DllUnInject/DllUnInject.cpp

@@ -59,9 +59,10 @@ void ReadTargetListAndDo(LPCWCHAR szCurrentDirectory) {
 	StringCbCat(szListDirectory, MAX_PATH, L"TargetList.txt");
 	OutputDebugString(szListDirectory);
 	OutputDebugString(L"\n");
-	hFile = CreateFile(szListDirectory, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+	hFile = CreateFile(szListDirectory, GENERIC_READ, FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
 	if (INVALID_HANDLE_VALUE == hFile)
 	{
+		printf_s("error: %d\n", GetLastError()); system("PAUSE");
 		OutputDebugString(L"TargetList.txt NOT FOUND\n");
 		return;
 	}

APIHOOK/Monitor/Monitor.cpp

@@ -59,6 +59,7 @@ void StartMonitor()
 {
 	OutputDebugString(TEXT("Do start \n"));
 
+	/*
 	HANDLE hMutexSingleton = NULL;
 
 	hMutexSingleton = CreateMutex(NULL, TRUE, TEXT("APIHOOK_Monitor_Mutex_Singleton"));
@@ -74,21 +75,51 @@ void StartMonitor()
 		OutputDebugString(TEXT("CreateMutex ERROR\n"));
 		return;
 	}
+	*/
+
+	HANDLE hTargetListFile = NULL;
+	TCHAR szTargetListFilePath[MAX_PATH];
+
+	GetCurrentDirectory(MAX_PATH, szTargetListFilePath);
+	StringCbCat(szTargetListFilePath, MAX_PATH, TEXT("\\TargetList.txt"));
+	hTargetListFile = CreateFile(szTargetListFilePath, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
+	if (INVALID_HANDLE_VALUE == hTargetListFile)
+	{
+		OutputDebugString(TEXT("CreateFile ERROR\n"));
+		switch (GetLastError())
+		{
+		case ERROR_FILE_NOT_FOUND:
+			OutputDebugString(TEXT("TargetList.txt is not found\n"));
+			printf_s("TargetList.txt is not found\n");
+			break;
+		case ERROR_SHARING_VIOLATION:
+			OutputDebugString(TEXT("Monitor.exe is already running\n"));
+			printf_s("Monitor.exe is already running\n");
+			break;
+		default:
+			printf_s("create file targetlist error: %d\n", GetLastError());
+			break;
+		}
+		return;
+	}
 
 	Log();
 
+	CloseHandle(hTargetListFile);
+	hTargetListFile = NULL;
+
+	/*
 	ReleaseMutex(hMutexSingleton);
 	CloseHandle(hMutexSingleton);
 	hMutexSingleton = NULL;
-
+	*/
 }
 
 
 void StopMonitor()
 {
 	OutputDebugString(TEXT("Do stop \n"));
-
-	MyCreateProcess(TEXT("DllUnInject.dll"));
+	MyCreateProcess(TEXT("DllUnInject.exe"));
 
 }
 
@@ -103,15 +134,13 @@ void RestartMonitor()
 
 void MyCreateProcess(LPCTSTR szProcessName)
 {
-	TCHAR szCurrentDirectory[MAX_PATH];
 	TCHAR szProcessPath[MAX_PATH];
 	STARTUPINFO si;
 	PROCESS_INFORMATION pi;
 
 	ZeroMemory(&si, sizeof(si));
 	ZeroMemory(&pi, sizeof(pi));
-	GetCurrentDirectory(MAX_PATH, szCurrentDirectory);
-	StringCbCopy(szProcessPath, MAX_PATH, szCurrentDirectory);
+	GetCurrentDirectory(MAX_PATH, szProcessPath);
 	StringCbCat(szProcessPath, MAX_PATH, TEXT("\\"));
 	StringCbCat(szProcessPath, MAX_PATH, szProcessName);
 	OutputDebugString(TEXT("CreateProcess "));
@@ -127,4 +156,4 @@ void MyCreateProcess(LPCTSTR szProcessName)
 	WaitForSingleObject(pi.hProcess, INFINITE);
 	OutputDebugString(szProcessName);
 	OutputDebugString(TEXT(" Finished\n"));
-}
+}