Merge pull request #11 from odedshimon/add-file-extracting

Add file extracting
odedshimon · May 22, 2020 · 8554c12 · 8554c12
2 parents 5023a2c + 0720c6d
commit 8554c12
Show file tree

Hide file tree

Showing 32 changed files with 1,004 additions and 126 deletions.
diff --git a/BruteShark/BruteSharkDesktop/BruteSharkDesktop.csproj b/BruteShark/BruteSharkDesktop/BruteSharkDesktop.csproj
@@ -57,6 +57,18 @@
     <Compile Include="Casting.cs" />
     <Compile Include="Objects\TcpPacket.cs" />
     <Compile Include="Objects\TcpSession.cs" />
+    <Compile Include="UserControls\FilesControl\FilePreviewUserControl.cs">
+      <SubType>UserControl</SubType>
+    </Compile>
+    <Compile Include="UserControls\FilesControl\FilePreviewUserControl.Designer.cs">
+      <DependentUpon>FilePreviewUserControl.cs</DependentUpon>
+    </Compile>
+    <Compile Include="UserControls\FilesControl\FilesUserControl.cs">
+      <SubType>UserControl</SubType>
+    </Compile>
+    <Compile Include="UserControls\FilesControl\FilesUserControl.Designer.cs">
+      <DependentUpon>FilesUserControl.cs</DependentUpon>
+    </Compile>
     <Compile Include="UserControls\NetworkMapControl\NetworkMapEdge.cs" />
     <Compile Include="UserControls\SessionsControl\SessionsExplorerUserControl.cs">
       <SubType>UserControl</SubType>
@@ -96,6 +108,12 @@
     </Compile>
     <Compile Include="Program.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <EmbeddedResource Include="UserControls\FilesControl\FilePreviewUserControl.resx">
+      <DependentUpon>FilePreviewUserControl.cs</DependentUpon>
+    </EmbeddedResource>
+    <EmbeddedResource Include="UserControls\FilesControl\FilesUserControl.resx">
+      <DependentUpon>FilesUserControl.cs</DependentUpon>
+    </EmbeddedResource>
     <EmbeddedResource Include="UserControls\SessionsControl\SessionsExplorerUserControl.resx">
       <DependentUpon>SessionsExplorerUserControl.cs</DependentUpon>
     </EmbeddedResource>

diff --git a/BruteShark/BruteSharkDesktop/MainForm.Designer.cs b/BruteShark/BruteSharkDesktop/MainForm.Designer.cs
diff --git a/BruteShark/BruteSharkDesktop/MainForm.cs b/BruteShark/BruteSharkDesktop/MainForm.cs
@@ -1,4 +1,5 @@
-using System;
+using BruteSharkDesktop;
+using System;
 using System.Collections.Generic;
 using System.ComponentModel;
 using System.Data;
@@ -22,6 +23,7 @@ public partial class MainForm : Form
         private HashesUserControl _hashesUserControl;
         private NetworkMapUserControl _networkMapUserControl;
         private SessionsExplorerUserControl _sessionsExplorerUserControl;
+        private FilesUserControl _filesUserControl;
 
 
         public MainForm()
@@ -44,6 +46,8 @@ public MainForm()
             _hashesUserControl.Dock = DockStyle.Fill;
             _passwordsUserControl = new GenericTableUserControl();
             _passwordsUserControl.Dock = DockStyle.Fill;
+            _filesUserControl = new FilesUserControl();
+            _filesUserControl.Dock = DockStyle.Fill;
 
             // Contract the events.
             _processor.UdpPacketArived += (s, e) => _analyzer.Analyze(Casting.CastProcessorUdpPacketToAnalyzerUdpPacket(e.Packet));
@@ -57,9 +61,18 @@ public MainForm()
             _processor.ProcessingFinished += (s, e) => SwitchToMainThreadContext(() => OnProcessingFinished(s, e));
 
             InitilizeFilesIconsList();
+            InitilizeModulesCheckedListBox();
             this.modulesTreeView.ExpandAll();
         }
 
+        private void InitilizeModulesCheckedListBox()
+        {
+            foreach (var module_name in _analyzer.AvailableModulesNames)
+            {
+                this.modulesCheckedListBox.Items.Add(module_name, isChecked: false);
+            }
+        }
+
         private void OnProcessingFinished(object sender, EventArgs e)
         {
             this.progressBar.Value = this.progressBar.Maximum;
@@ -164,10 +177,14 @@ private void OnParsedItemDetected(object sender, PcapAnalyzer.ParsedItemDetected
                 _networkMapUserControl.AddEdge(connection.Source, connection.Destination);
                 this.modulesTreeView.Nodes["NetworkNode"].Nodes["NetworkMapNode"].Text = $"Network Map ({_networkMapUserControl.NodesCount})";
             }
+            else if (e.ParsedItem is PcapAnalyzer.NetworkFile)
+            {
+                var fileObject = e.ParsedItem as PcapAnalyzer.NetworkFile;
+                _filesUserControl.AddFile(fileObject);
+                this.modulesTreeView.Nodes["DataNode"].Nodes["FilesNode"].Text = $"Files ({_filesUserControl.FilesCount})";
+            }
         }
 
-
-
         private void addFilesButton_Click(object sender, EventArgs e)
         {
             var openFileDialog = new OpenFileDialog();
@@ -223,6 +240,9 @@ private void modulesTreeView_AfterSelect(object sender, TreeViewEventArgs e)
                 case "SessionsNode":
                     this.modulesSplitContainer.Panel2.Controls.Add(_sessionsExplorerUserControl);
                     break;
+                case "FilesNode":
+                    this.modulesSplitContainer.Panel2.Controls.Add(_filesUserControl);
+                    break;
                 default:
                     break;
             }
@@ -236,6 +256,21 @@ private void removeFilesButton_Click(object sender, EventArgs e)
                 item.Remove();
             } 
         }
+
+        private void modulesCheckedListBox_ItemCheck(object sender, ItemCheckEventArgs e)
+        {
+            var module_name = ((CheckedListBox)sender).Text;
+
+            if (e.NewValue == CheckState.Checked)
+            {
+                _analyzer.AddModule(module_name);
+            }
+            else
+            {
+                _analyzer.RemoveModule(module_name);
+            }
+        }
+
     }
 }