Added backup & fixed formatting

main.cpp

@@ -1,28 +1,22 @@
 #include <Windows.h>
 #include <iostream>
+
 #include "memory.hpp"
+#include "ifexist.hpp"
 
 using namespace std;
 
 Injector inj;
 
 DWORD pid;
 
-bool DoesFileExist(const char* name) {
-	if (FILE* file = fopen(name, "r")) {
-		fclose(file);
-		return true;
-	}
-	else {
-		return false;
-	}
-}
+//now it's global
+LPVOID ntOpenFile = GetProcAddress(LoadLibraryW(L"ntdll"), "NtOpenFile");
 
 void bypass()
 {
 	// Restore original NtOpenFile from external process
 	//credits: Daniel Krupiñski(pozdro dla ciebie byczku <3)
-	LPVOID ntOpenFile = GetProcAddress(LoadLibraryW(L"ntdll"), "NtOpenFile");
 	if (ntOpenFile) {
 		char originalBytes[5];
 		memcpy(originalBytes, ntOpenFile, 5);
@@ -31,6 +25,24 @@ void bypass()
 	else
 	{
 		cout << "Unable to bypass :(\n";
+		Sleep(2000);
+		exit(-1);
+	}
+}
+
+void Backup()
+{
+	if (ntOpenFile) {
+		//So, when I patching first 5 bytes I need to backup them to 0? (I think)
+		char originalBytes[5];
+		memcpy(originalBytes, ntOpenFile, 5);
+		WriteProcessMemory(inj.process, ntOpenFile, originalBytes, 0, NULL);
+	}
+	else
+	{
+		cout << "Unable to backup :(\n";
+		Sleep(2000);
+		exit(-1);
 	}
 }
 
@@ -49,26 +61,28 @@ int main()
 	inj.clientDLL = inj.GetModule(pid, "client.dll");
 
 	if (DoesFileExist("cheat.dll")) {
-	bypass();
+		bypass();
 
 		if (inj.inject(pid, "cheat.dll")) {
 			cout << "module injected!\n\n" << endl;
+			Backup();
 			Sleep(2000);
 			exit(0);
 		}
 		else
 		{
 			cout << "Injection failed!\n\n" << endl;
+			Backup();
 			Sleep(2000);
-			exit(0);
+			exit(-1);
 		}
 
 	}
 	else
 	{
 		cout << "cannot find cheat.dll\n\n";
 		Sleep(2000);
-		exit(0);
+		exit(-1);
 	}
 
 	return 0;