minor changes for LuaBridge compatibility

gmh5225 · Aug 7, 2018 · b295339 · b295339
1 parent 2e7eaa9
commit b295339
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 7 deletions.
diff --git a/AttackServices/HiddenKernelDLLInjectionAttack/AttackTest.cpp b/AttackServices/HiddenKernelDLLInjectionAttack/AttackTest.cpp
@@ -117,7 +117,7 @@ void HiddenKernelDLLInjectionAttackTest() {
         hProcess = (HANDLE)GetProcessByName("TestApp.exe");
     }
 
-    if (!StealthInject("TestApp.exe", "InjectedDLL.dll")) {
+    if (!StealthInject("DarkSoulsIII.exe", "InjectedDLL.dll")) {
         cout << "HiddenKernelDLLInjectionAttack() failed" << endl;
     } else {
         cout << "HiddenKernelDLLInjectionAttack() success" << endl;

diff --git a/AttackServices/HiddenKernelDLLInjectionAttack/AttackTest.exe b/AttackServices/HiddenKernelDLLInjectionAttack/AttackTest.exe
diff --git a/PenetrationRoutines/AttackProvider/AttackProvider.h b/PenetrationRoutines/AttackProvider/AttackProvider.h
@@ -10,6 +10,7 @@ class AttackProvider {
     // virtual BOOL WriteProcessMemory(HANDLE hProcess, void* address, void* writeBuf, SIZE_T writeSize, SIZE_T* bytesWritten) {return TRUE;};
     // virtual BOOL SetTargetProcessByName(wstring){return FALSE;};
     // virtual BOOL ExecuteAttack(){return FALSE;};
+    AttackProvider(){}
     string GetAttackResults() {
         return results;
     };

diff --git a/PenetrationRoutines/DLLInjectionProvider/DLLInjectionProvider.cpp b/PenetrationRoutines/DLLInjectionProvider/DLLInjectionProvider.cpp
@@ -2,7 +2,8 @@
 #include <Shlwapi.h>
 #include "DLLInjectionProvider.h"
 
-BOOL DLLInjectionProvider::SetTargetProcessByName(wstring name) {
+bool DLLInjectionProvider::SetTargetProcessByName(const string _name) {
+    wstring name(_name.begin(), _name.end());
     hProcess = GetProcessHandleByName(name);
     if (hProcess == NULL) {
         results += "SetTargetProcessByName() failed. Could not get HANDLE to target process. System Error Code: ";
@@ -13,7 +14,8 @@ BOOL DLLInjectionProvider::SetTargetProcessByName(wstring name) {
     return TRUE;
 }
 
-BOOL DLLInjectionProvider::SetTargetDLL(wstring dllPath) {
+bool DLLInjectionProvider::SetTargetDLL(const string _dllPath) {
+    wstring dllPath(_dllPath.begin(), _dllPath.end());
     if (!PathFileExistsW(dllPath.c_str())) {
         results += "SetTargetDLL() failed. File not found. System Error Code: ";
         results += to_string(GetLastError());
@@ -24,7 +26,7 @@ BOOL DLLInjectionProvider::SetTargetDLL(wstring dllPath) {
     return TRUE;
 }
 
-BOOL DLLInjectionProvider::ExecuteAttack() {
+bool DLLInjectionProvider::ExecuteAttack() {
     if (dllPath == L"" || hProcess == NULL) {
         return FALSE;
     }

diff --git a/PenetrationRoutines/DLLInjectionProvider/DLLInjectionProvider.h b/PenetrationRoutines/DLLInjectionProvider/DLLInjectionProvider.h
@@ -7,9 +7,10 @@ using namespace std;
 
 class DLLInjectionProvider : public AttackProvider {
 public:
-    BOOL SetTargetProcessByName(wstring) ;
-    BOOL ExecuteAttack() ;
-    BOOL SetTargetDLL(wstring);
+    DLLInjectionProvider(){}
+    bool SetTargetProcessByName(const string);
+    bool ExecuteAttack();
+    bool SetTargetDLL(const string);
 
 protected:
     wstring dllPath = L"";