Merge pull request #36 from bblenard/issue-35-ps-crashes-powershell-b…

…eacon Issue 35 ps crashes powershell beacon
bats3c · Sep 16, 2020 · 510a734 · 510a734
2 parents 254829a + d4487c6
commit 510a734
Show file tree

Hide file tree

Showing 7 changed files with 20 additions and 23 deletions.
diff --git a/beacon/injectable/core.c b/beacon/injectable/core.c
@@ -26,6 +26,9 @@
 
 #include "settings.h"
 
+// include stdlib function prototypes
+#include "../lib/stdlib/stdlib.h"
+
 // core functions
 BOOL GetBasicUserInfo(struct BasicUserInfo *UserInfo)
 {

diff --git a/beacon/injectable/core.h b/beacon/injectable/core.h
@@ -16,14 +16,6 @@ LPCWSTR* BeaconCallbackC2(LPCSTR CallbackAddress, INT CallbackPort, LPCSTR UserA
 LPVOID DieCleanly();
 LPVOID CheckIfDie(LPCWSTR *ReadBuffer);
 
-char* listdirs(LPCSTR szDir);
-char* readfile(char* szFile);
-char* makedirectory(char* szDirName);
-char* removefile(char* szFileName);
-char* getdir();
-char* changedir(char* Dir);
-char* getpid();
-
 LPVOID ReportExecutionFail();
 
 char* decrypt_string(char* string, int key);

diff --git a/beacon/lib/stdlib/stdlib.h b/beacon/lib/stdlib/stdlib.h
@@ -0,0 +1,11 @@
+#include <windows.h>
+
+char* changedir(char* Dir);
+char* getdir();
+char* getpid();
+char* getps();
+char* listdirs(char* Dir);
+char* makedirectory(char* szDirName);
+char* readfile(char* szFile);
+char* removefile(char* szFileName);
+char* whoami(BOOL DisplayAll, char* oBuffer);
diff --git a/beacon/reflection/core.c b/beacon/reflection/core.c
@@ -26,6 +26,9 @@
 
 #include "settings.h"
 
+// Include stdlib function prototypes
+#include "../lib/stdlib/stdlib.h"
+
 // core functions
 BOOL GetBasicUserInfo(struct BasicUserInfo *UserInfo)
 {

diff --git a/beacon/reflection/core.h b/beacon/reflection/core.h
@@ -16,14 +16,6 @@ LPCWSTR* BeaconCallbackC2(LPCSTR CallbackAddress, INT CallbackPort, LPCSTR UserA
 LPVOID DieCleanly();
 LPVOID CheckIfDie(LPCWSTR *ReadBuffer);
 
-char* listdirs(LPCSTR szDir);
-char* readfile(char* szFile);
-char* makedirectory(char* szDirName);
-char* removefile(char* szFileName);
-char* getdir();
-char* changedir(char* Dir);
-char* getpid();
-
 LPVOID ReportExecutionFail();
 
 char* decrypt_string(char* string, int key);

diff --git a/beacon/src/core.c b/beacon/src/core.c
@@ -28,6 +28,9 @@
 
 #include "settings.h"
 
+// Include stdlib function prototypes
+#include "../lib/stdlib/stdlib.h"
+
 // core functions
 BOOL GetBasicUserInfo(struct BasicUserInfo *UserInfo)
 {

diff --git a/beacon/src/core.h b/beacon/src/core.h
@@ -16,13 +16,6 @@ LPCWSTR* BeaconCallbackC2(LPCSTR CallbackAddress, INT CallbackPort, LPCSTR UserA
 LPVOID DieCleanly();
 LPVOID CheckIfDie(LPCWSTR *ReadBuffer);
 
-char* listdirs(LPCSTR szDir);
-char* readfile(char* szFile);
-char* makedirectory(char* szDirName);
-char* removefile(char* szFileName);
-char* getdir();
-char* changedir(char* Dir);
-
 LPVOID ReportExecutionFail();
 
 char* decrypt_string(char* string, int key);