From 9eeee826995901295b5666b69d74710776c0764a Mon Sep 17 00:00:00 2001
From: XaFF-XaFF <ksawery-czapczynski@wp.pl>
Date: Mon, 20 Dec 2021 09:34:46 +0100
Subject: [PATCH] Show registers value in real time

---
 Shellcodevv/Core/AssemblyHandler.cs | 12 ++++----
 Shellcodevv/MainWindow.xaml         | 15 +++++-----
 Shellcodevv/MainWindow.xaml.cs      | 43 ++++++++++++++++++++++++++---
 3 files changed, 53 insertions(+), 17 deletions(-)

diff --git a/Shellcodevv/Core/AssemblyHandler.cs b/Shellcodevv/Core/AssemblyHandler.cs
index fa8c665..5e4c0dc 100644
--- a/Shellcodevv/Core/AssemblyHandler.cs
+++ b/Shellcodevv/Core/AssemblyHandler.cs
@@ -145,7 +145,7 @@ private void AppendRegisters(API.Registers registers)
             }
 
             string str = string.Join(" ", list);
-            main.registersBox.AppendText(str);
+            main.registersBox.Text = str;
         }
 
         private void AppendIndexes(API.Registers registers)
@@ -164,7 +164,7 @@ private void AppendIndexes(API.Registers registers)
             }
 
             string str = string.Join(" ", list);
-            main.indexesBox.AppendText(str);
+            main.indexesBox.Text = str;
         }
 
         private void AppendPointers(API.Registers registers)
@@ -183,7 +183,7 @@ private void AppendPointers(API.Registers registers)
             }
 
             string str = string.Join(" ", list);
-            main.pointersBox.AppendText(str);
+            main.pointersBox.Text = str;
         }
 
         private string Clear(string instruction)
@@ -251,9 +251,9 @@ public unsafe void SetRegisters(string instruction, API.PROCESS_INFORMATION pi)
 
             MainWindow.registers = Configure(registers, MainWindow.registers, instruction);
 
-            //AppendRegisters(Main.registers);
-            //AppendIndexes(Main.registers);
-            //AppendPointers(Main.registers);
+            AppendRegisters(MainWindow.registers);
+            AppendIndexes(MainWindow.registers);
+            AppendPointers(MainWindow.registers);
         }
         #endregion
     }
diff --git a/Shellcodevv/MainWindow.xaml b/Shellcodevv/MainWindow.xaml
index a549a06..c9ec5cf 100644
--- a/Shellcodevv/MainWindow.xaml
+++ b/Shellcodevv/MainWindow.xaml
@@ -84,10 +84,10 @@
                 HorizontalAlignment="Left"
                 VerticalAlignment="Top" Margin="5,39,0,0" Height="29" Width="59"/>
 
-            <RichTextBox
+            <TextBox
                Name="registersBox"
                Margin="10,65,10,131"
-               Style="{StaticResource RichTxtboxTheme}" />
+               />
 
             <Label
                 Content="Indexes"
@@ -96,10 +96,10 @@
                 HorizontalAlignment="Left"
                 VerticalAlignment="Top" Margin="5,89,0,0" Height="29" Width="57"/>
 
-            <RichTextBox
+            <TextBox
                Name="indexesBox"  
                Margin="10,115,10,81"
-               Style="{StaticResource RichTxtboxTheme}" />
+               />
 
             <Label
                 Content="Pointers"
@@ -108,10 +108,10 @@
                 HorizontalAlignment="Left"
                 VerticalAlignment="Top" Margin="5,139,0,0" Height="29" Width="60"/>
 
-            <RichTextBox
+            <TextBox
                Name="pointersBox"
                Margin="10,165,10,31"
-               Style="{StaticResource RichTxtboxTheme}" />
+               />
 
             <!--Row 0 Column 1-->
             <Button
@@ -254,7 +254,8 @@
                 Background="Transparent"
                 Foreground="White"
                 Grid.Row="1"
-                Margin="10,449,204,7"/>
+                Margin="10,449,204,7"
+                KeyDown="instructionTxt_KeyDown"/>
 
             <!--Row 1 Column 1-->
             <Label
diff --git a/Shellcodevv/MainWindow.xaml.cs b/Shellcodevv/MainWindow.xaml.cs
index 92b92fe..841fb5a 100644
--- a/Shellcodevv/MainWindow.xaml.cs
+++ b/Shellcodevv/MainWindow.xaml.cs
@@ -23,6 +23,36 @@ public MainWindow()
         {
             InitializeComponent();
             instance = this;
+
+            AssemblyHandler handler = new AssemblyHandler();
+            InitProcess();
+
+            handler.SetRegisters("xor eax,eax", pi);
+            handler.SetRegisters("xor ebx,ebx", pi);
+            handler.SetRegisters("xor ecx,ecx", pi);
+        }
+
+        private void InitProcess()
+        {
+            API.STARTUPINFO si = new API.STARTUPINFO();
+            pi = new API.PROCESS_INFORMATION();
+
+            bool createproc = API.CreateProcess(
+                AppDomain.CurrentDomain.FriendlyName,
+                null,
+                IntPtr.Zero,
+                IntPtr.Zero,
+                false,
+                0x00000002,
+                IntPtr.Zero,
+                null,
+                ref si, out pi);
+
+            if(!createproc)
+            {
+                MessageBox.Show("ERROR! CreateProcess Failed");
+                this.Close();
+            }
         }
 
         #region Buttons
@@ -50,16 +80,21 @@ private void addBtn_Click(object sender, RoutedEventArgs e)
             if (string.IsNullOrEmpty(instruction))
                 return;
 
-            Console.WriteLine(instruction);
-
             Instructions instructions = new Instructions();
             AssemblyHandler handler = new AssemblyHandler();
 
             instructions.instruction = instruction;
-
             instructionGrid.Items.Add(instructions);
 
-            Console.WriteLine(handler.Assembler(instruction));
+            handler.SetRegisters(instruction, pi);
+
+            instructionTxt.SelectAll();
+        }
+
+        private void instructionTxt_KeyDown(object sender, KeyEventArgs e)
+        {
+            if (e.Key == Key.Enter)
+                addBtn_Click(sender, e);
         }
     }