hardsploit_gui 2.4.0 → 2.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 33cac9a06b6835cd33abb3a2b9937152f2d9d0f1
-  data.tar.gz: 0cd2a410c28964b3cbd40be6a1ccfe533e4d7fa7
+  metadata.gz: 801459f50d14a2e5b5d94fb574cd6163e630b1ae
+  data.tar.gz: 5fa96b5a0281ce5557a14701b2e121e92a4fa861
 SHA512:
-  metadata.gz: 2356ccef654cc626ba4ab9b23e50708727f73233d0b46d92e61a0ac11f468f8b2d4a4faf03995f6a383289ef4eb6a110e47940bfc8a979d19209dc0557936e30
-  data.tar.gz: 1ec86ba60d5fffe6c512c6bf0709a595ad24f702507fbc521e9dfc5c7dfe341c1c335c33f1d78946272be9a7302b5b66c7a3a8053d5c9bfab080cc08525d996b
+  metadata.gz: f57c188bc439fe59bb84b655be76781915b3873c8ca3d020a7cb7a17b9ad3436fa8f8f382fc602d897639a7cbc25706fa5fa2394f9d246e1a8429d182c3df109
+  data.tar.gz: e716df40ec6768c808ab85e357facac2be65a54a3174af8b3de3b9a24e7f64ff9661d618764d040621d16b3b94bb136bb2e538d1ce8cfbc91a26870abe59792f

data/lib/Hardsploit_gui.rb

@@ -11,14 +11,14 @@ require 'active_record'
 require_relative 'db/database.rb'
 require_relative 'gui/gui_chip_management'
 require_relative 'class/Chip_management'
-require_relative 'HardsploitAPI/Core/HardsploitAPI'
-require_relative 'Firmwares/UC/VersionUC'
-require_relative 'Firmwares/FPGA/VersionFPGA'
+require_relative 'hardsploit-api/HardsploitAPI/Core/HardsploitAPI'
+require_relative 'hardsploit-api/Firmwares/UC/VersionUC'
+require_relative 'hardsploit-api/Firmwares/FPGA/VersionFPGA'
 include VersionFPGA::VERSION_FPGA
 include VersionUC::VERSION_UC
 
 class Hardsploit_gui
-  VERSION = "2.4.0"
+  VERSION = "2.5.0"
   def initialize
     HardsploitAPI.callbackInfo = method(:callbackInfo)
     HardsploitAPI.callbackData = method(:callbackData)
@@ -28,7 +28,6 @@ class Hardsploit_gui
     $currentFirmware = nil
     $logFilePath = File.expand_path(File.dirname(__FILE__)) + "/logs/error.log"
     $dbFilePath = File.expand_path(File.dirname(__FILE__)) + "/db/hs.db"
-    # Launch GUI
     Qt::Application.new(ARGV) do
       $app = self
       w = Chip_management.new(VERSION)

data/lib/class/Chip_clone.rb

@@ -0,0 +1,70 @@
+#===================================================
+#  Hardsploit GUI - By Opale Security
+#  www.opale-security.com || www.hardsploit.io
+#  License: GNU General Public License v3
+#  License URI: http://www.gnu.org/licenses/gpl.txt
+#===================================================
+
+require_relative '../gui/gui_chip_clone'
+class Chip_clone < Qt::Widget
+  slots 'clone()'
+
+  def initialize(parent, chip)
+    super()
+    @parent = parent
+    @chip = chip
+    @view = Ui_Chip_clone.new
+    centerWindow(self)
+    @view.setupUi(self)
+    inputRestrict(@view.lie_reference,   2)
+  end
+
+  def clone
+    #Chip
+    clone = @chip.dup
+    clone.reference = @view.lie_reference.text
+    clone.save
+    #Pins
+    @chip.pins.each do |pin|
+      npin = pin.dup
+      npin.chip_id = clone.id
+      npin.save
+    end
+    #Setting(s)
+    unless @chip.parallel_setting.nil?
+    clone.parralel_setting = @chip.parallel_setting.dup
+    clone.parralel_setting.save
+    end
+    unless @chip.spi_setting.nil?
+    clone.spi_setting = @chip.spi_setting.dup
+    clone.spi_setting.save
+    end
+    unless @chip.i2c_setting.nil?
+      clone.i2c_setting = @chip.i2c_setting.dup
+      clone.i2c_setting.save
+    end
+    unless @chip.swd_setting.nil?
+      clone.swd_setting = @chip.swd_setting.dup
+      clone.swd_setting.save
+    end
+    unless @chip.uart_setting.nil?
+      clone.uart_setting = @chip.uart_setting.dup
+      clone.uart_setting.save
+    end
+    #Command(s) and cmd bytes
+    unless @chip.commands.nil?
+      @chip.commands.each do |cmd|
+        clone_cmd = cmd.dup
+        clone_cmd.chip_id = clone.id
+        clone_cmd.save
+        cmd.bytes.each do |byte|
+          clone_byte = byte.dup
+          clone_byte.command_id = clone_cmd.id
+          clone_byte.save
+        end
+      end
+    end
+    @parent.feed_chip_array
+    self.close
+  end
+end

data/lib/class/Chip_editor.rb

@@ -23,6 +23,7 @@ class Chip_editor < Qt::Widget
     @view.setupUi(self)
     @parent = parent
     @chip = chip
+    @action = action
 
     inputRestrict(@view.lie_pack_name,   2)
     inputRestrict(@view.lie_pack_pin,    0)

data/lib/class/Chip_management.rb

@@ -46,12 +46,25 @@ class Chip_management < Qt::MainWindow
 		feed_type_cbx
     @console = Console.new(@view.tbl_console)
     @versionGUI = versionGUI
+    @hardsploit_board_status = Qt::Label.new
+    statusBar.addWidget(@hardsploit_board_status);
     check_hardsploit_connection
   rescue Exception => msg
     ErrorMsg.new.unknown(msg)
     return false
   end
 
+  def set_hs_board_status(connected)
+  	if connected
+  		@hardsploit_board_status.setText("Hardsploit board: connected " +
+        		"- version #{HardsploitAPI.instance.getVersionNumber} " +
+        		"- api V#{HardsploitAPI::VERSION::API}"
+  		)
+  	else
+  		@hardsploit_board_status.setText("Hardsploit board: disconnected");
+  	end
+  end
+
 	def feed_type_cbx
       Manufacturer.all.each do |m|
         @view.cbx_manufacturer.addItem(m.name)
@@ -109,9 +122,9 @@ class Chip_management < Qt::MainWindow
     edit_lvl = Qt::TreeWidgetItem.new
     edit_lvl.setText(0, "Edit")
     parent_node.addChild(edit_lvl)
-    template_lvl = Qt::TreeWidgetItem.new
-    template_lvl.setText(0, "Template")
-    parent_node.addChild(template_lvl)
+    clone_lvl = Qt::TreeWidgetItem.new
+    clone_lvl.setText(0, "Clone")
+    parent_node.addChild(clone_lvl)
     delete_lvl = Qt::TreeWidgetItem.new
     delete_lvl.setText(0, "Delete")
     parent_node.addChild(delete_lvl)
@@ -137,12 +150,12 @@ class Chip_management < Qt::MainWindow
       detect_lvl = Qt::TreeWidgetItem.new
       detect_lvl.setText(0, "Detect")
       parent_node.addChild(detect_lvl)
-      import_lvl = Qt::TreeWidgetItem.new
-      import_lvl.setText(0, "Import")
-      parent_node.addChild(import_lvl)
-      export_lvl = Qt::TreeWidgetItem.new
-      export_lvl.setText(0, "Dump")
-      parent_node.addChild(export_lvl)
+      write_lvl = Qt::TreeWidgetItem.new
+      write_lvl.setText(0, "Write")
+      parent_node.addChild(write_lvl)
+      read_lvl = Qt::TreeWidgetItem.new
+      read_lvl.setText(0, "Read")
+      parent_node.addChild(read_lvl)
       erase_lvl = Qt::TreeWidgetItem.new
       erase_lvl.setText(0, "Erase")
       parent_node.addChild(erase_lvl)
@@ -153,12 +166,12 @@ class Chip_management < Qt::MainWindow
       cmd_lvl = Qt::TreeWidgetItem.new
       cmd_lvl.setText(0, "Commands")
       parent_node.addChild(cmd_lvl)
-      import_lvl = Qt::TreeWidgetItem.new
-      import_lvl.setText(0, "Import")
-      parent_node.addChild(import_lvl)
-      export_lvl = Qt::TreeWidgetItem.new
-      export_lvl.setText(0, "Export")
-      parent_node.addChild(export_lvl)
+      write_lvl = Qt::TreeWidgetItem.new
+      write_lvl.setText(0, "Write")
+      parent_node.addChild(write_lvl)
+      read_lvl = Qt::TreeWidgetItem.new
+      read_lvl.setText(0, "Read")
+      parent_node.addChild(read_lvl)
       sniffer_lvl = Qt::TreeWidgetItem.new
       sniffer_lvl.setText(0, "Sniffer")
       parent_node.addChild(sniffer_lvl)
@@ -172,23 +185,23 @@ class Chip_management < Qt::MainWindow
       cmd_lvl = Qt::TreeWidgetItem.new
       cmd_lvl.setText(0, "Commands")
       parent_node.addChild(cmd_lvl)
-      import_lvl = Qt::TreeWidgetItem.new
-      import_lvl.setText(0, "Import")
-      parent_node.addChild(import_lvl)
-      export_lvl = Qt::TreeWidgetItem.new
-      export_lvl.setText(0, "Export")
-      parent_node.addChild(export_lvl)
+      write_lvl = Qt::TreeWidgetItem.new
+      write_lvl.setText(0, "Write")
+      parent_node.addChild(write_lvl)
+      read_lvl = Qt::TreeWidgetItem.new
+      read_lvl.setText(0, "Read")
+      parent_node.addChild(read_lvl)
     when 'PARALLEL'
       settings_lvl = Qt::TreeWidgetItem.new
       settings_lvl.setText(0, "Settings")
       parent_node.addChild(settings_lvl)
       # Not implemented yet
-      #import_lvl = Qt::TreeWidgetItem.new
-      #import_lvl.setText(0, "Import")
-      #parent_node.addChild(import_lvl)
-      export_lvl = Qt::TreeWidgetItem.new
-      export_lvl.setText(0, "Export")
-      parent_node.addChild(export_lvl)
+      #write_lvl = Qt::TreeWidgetItem.new
+      #write_lvl.setText(0, "write")
+      #parent_node.addChild(write_lvl)
+      read_lvl = Qt::TreeWidgetItem.new
+      read_lvl.setText(0, "Read")
+      parent_node.addChild(read_lvl)
     end
     return parent_node
   end
@@ -250,7 +263,7 @@ class Chip_management < Qt::MainWindow
       case item.text(0)
       when 'Wiring'   ; wire_chip
       when 'Edit'     ; edit_chip
-      when 'Template' ; add_chip
+      when 'Clone'    ; clone_chip
       when 'Delete'   ; delete_chip
       end
     else
@@ -268,8 +281,8 @@ class Chip_management < Qt::MainWindow
     case spi_module
     when 'Settings' ; Spi_settings.new(@chip_clicked).show
     when 'Commands' ; Generic_commands.new(@chip_clicked, 'SPI').show
-    when 'Import'   ; Spi_import.new(@chip_clicked).show
-    when 'Export'   ; Spi_export.new(@chip_clicked).show
+    when 'Write'    ; Spi_write.new(@chip_clicked).show
+    when 'Read'     ; Spi_read.new(@chip_clicked).show
     when 'Sniffer'  ; Spi_sniffer.new(@chip_clicked).show
     end
   end
@@ -279,31 +292,39 @@ class Chip_management < Qt::MainWindow
     when 'Settings'    ; I2c_settings.new(@chip_clicked).show
     when 'PIN Scanner' ; I2c_scanner.new.show
     when 'Commands'    ; Generic_commands.new(@chip_clicked, 'I2C').show
-    when 'Import'      ; I2c_import.new(@chip_clicked).show
-    when 'Export'      ; I2c_export.new(@chip_clicked).show
+    when 'Write'       ; I2c_write.new(@chip_clicked).show
+    when 'Read'        ; I2c_read.new(@chip_clicked).show
     end
   end
 
   def load_parallel_module(parallel_module)
     case parallel_module
     when 'Settings' ; Parallel_settings.new(@chip_clicked).show
-    when 'Import'   ; Parallel_import.new(@chip_clicked).show
-    when 'Export'   ; Parallel_export.new(@chip_clicked).show
+    when 'Write'    ; Parallel_write.new(@chip_clicked).show
+    when 'Read'     ; Parallel_read.new(@chip_clicked).show
     end
   end
 
   def load_swd_module(swd_module)
     case swd_module
-    when 'Settings'     ; Swd_settings.new(@chip_clicked).show
-    when 'PIN Scanner'  ; Swd_scanner.new.show
-    when 'Detect'       ; Swd.new(@chip_clicked, @console).detect
-    when 'Import'
+    when 'Settings'; Swd_settings.new(@chip_clicked).show
+    when 'PIN Scanner'
+      return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
+      Swd_scanner.new.show
+    when 'Detect'
+      return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
+      Swd.new(@chip_clicked, @console).detect
+    when 'Write'
+      return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
       filepath = Qt::FileDialog.getOpenFileName(self, tr('Select a file'), '/')
-      Swd.new(@chip_clicked, @console).import(filepath) unless filepath.nil?
-    when 'Dump'
+      Swd.new(@chip_clicked, @console).write(filepath) unless filepath.nil?
+    when 'Read'
+      return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
       filepath = Qt::FileDialog.getSaveFileName(self, tr('Select a file'), '/')
-      Swd.new(@chip_clicked, @console).export(filepath) unless filepath.nil?
-    when 'Erase'    ; Swd.new(@chip_clicked, @console).erase
+      Swd.new(@chip_clicked, @console).read(filepath) unless filepath.nil?
+    when 'Erase'
+      return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
+      Swd.new(@chip_clicked, @console).erase
     end
   end
 
@@ -314,6 +335,7 @@ class Chip_management < Qt::MainWindow
       settingsUart.setWindowModality(Qt::ApplicationModal)
       settingsUart.show
     when 'Console'
+      return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
       consoleUart = Uart_console.new(@chip_clicked)
       consoleUart.setWindowModality(Qt::ApplicationModal)
       consoleUart.show
@@ -321,16 +343,7 @@ class Chip_management < Qt::MainWindow
   end
 
   def add_chip
-
-    if sender.objectName == 'tw_chip' or sender.objectName == 'actionTemplate'
-      unless @chip_clicked.nil?
-        add_chip = Chip_editor.new(self, @chip_clicked, 'temp')
-      else
-        ErrorMsg.new.no_chip_loaded
-      end
-    else
-      add_chip = Chip_editor.new(self, 'none', 'new')
-    end
+    add_chip = Chip_editor.new(self, 'none', 'new')
     add_chip.show
   end
 
@@ -340,6 +353,12 @@ class Chip_management < Qt::MainWindow
     edit_chip.show
   end
 
+  def clone_chip
+    return ErrorMsg.new.no_chip_loaded if @chip_clicked.nil?
+    clone_chip = Chip_clone.new(self, @chip_clicked)
+    clone_chip.show
+  end
+
   def delete_chip
     return ErrorMsg.new.no_chip_loaded if @chip_clicked.nil?
     msg = Qt::MessageBox.new
@@ -486,11 +505,13 @@ class Chip_management < Qt::MainWindow
         "BOARD : #{HardsploitAPI.instance.getVersionNumber}"
       )
       @console.print('Hardsploit ready to suck chip souls !')
+      set_hs_board_status(true)
     else
       @console.print(
         'Hardsploit board unconnected: '+
         'Wiring and command execution disabled'
       )
+      set_hs_board_status(false)
     end
   end
 end

data/lib/class/ErrorMsg.rb

@@ -302,6 +302,7 @@ class ErrorMsg
 
 # Unknown
 	def unknown(msg)
+		Dir.mkdir('logs') unless File.exists?('logs')
 		Logger.new($logFilePath).error(msg)
 		Qt::MessageBox.new(
 			Qt::MessageBox::Critical,

data/lib/class/Firmware.rb

@@ -12,7 +12,7 @@ class Firmware
         $pgb = Progress_bar.new("Upload firmware :")
         $pgb.show
       end
-      base_path = File.expand_path(File.dirname(__FILE__)) + '/../Firmwares/FPGA/'
+      base_path = File.expand_path(File.dirname(__FILE__)) + '/../hardsploit-api/Firmwares/FPGA/'
       case firmware
       when 'I2C'
         firmware_path = base_path + 'I2C/I2C_INTERACT/HARDSPLOIT_FIRMWARE_FPGA_I2C_INTERACT.rpd'

data/lib/class/Progress_bar.rb

@@ -14,7 +14,7 @@ class Progress_bar < Qt::Widget
     centerWindow(self)
     @pgb_ui.setupUi(self)
     self.update_status(status)
-    @pgb_ui.lbl_close.setEnabled(false)
+    @pgb_ui.lbl_close.setEnabled(true)
     self.display_time("Total duration:")
   end
 

data/lib/class/Wire_helper.rb

@@ -6,7 +6,7 @@
 #===================================================
 
 require_relative '../gui/gui_wire_helper'
-require_relative '../HardsploitAPI/Core/HardsploitAPI'
+require_relative '../hardsploit-api/HardsploitAPI/Core/HardsploitAPI'
 
 class Wire_helper < Qt::Widget
   slots 'rotate_scene()'
@@ -158,7 +158,10 @@ class CustomItem < Qt::GraphicsTextItem
       pin.setColor
       pin.instance_variable_get('@signalTxt').clearFocus
       pin.instance_variable_get('@nbrTxt').clearFocus
-      return false if @api_value == 'NA'
+      if @api_value.name == 'NA'
+        HardsploitAPI.instance.setWiringLeds(value: 0x0000000000000000)
+        return false
+      end
       pin_group = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H']
       hardsploit_pin_number = pin_group.index(@api_value.pin[0]) * 8 + @api_value.pin[1].to_i
       HardsploitAPI.instance.setWiringLeds(value: 2**hardsploit_pin_number)
@@ -196,7 +199,7 @@ class UniqPin
 
     @nbrTxt = CustomItem.new(pinNum.to_s)
     @nbrTxt.cursor = Qt::Cursor.new(Qt::PointingHandCursor)
-    @nbrTxt.setPin(self, pinNum.to_s)
+    @nbrTxt.setPin(self, signal_name)
     @nbrTxt.setTextInteractionFlags(Qt::TextSelectableByMouse)
     @nbrTxt.setX(xNum)
     @nbrTxt.setY(yNum)

data/lib/class/{I2C/I2c_export.rb → i2c/I2c_read.rb}

@@ -5,15 +5,15 @@
 #  License URI: http://www.gnu.org/licenses/gpl.txt
 #===================================================
 
-require_relative '../../gui/gui_generic_export'
-require_relative '../../HardsploitAPI/Modules/I2C/HardsploitAPI_I2C'
-class I2c_export < Qt::Widget
-  slots 'export()'
-  slots 'select_export_file()'
+require_relative '../../gui/gui_generic_read'
+require_relative '../../hardsploit-api/HardsploitAPI/Modules/I2C/HardsploitAPI_I2C'
+class I2c_read < Qt::Widget
+  slots 'read()'
+  slots 'select_read_file()'
 
   def initialize(chip)
     super()
-    @view = Ui_Generic_export.new
+    @view = Ui_Generic_read.new
     centerWindow(self)
     @view.setupUi(self)
     @view.lbl_chip.setText(chip.reference)
@@ -22,32 +22,32 @@ class I2c_export < Qt::Widget
     @chip = chip
   end
 
-  def select_export_file
+  def select_read_file
     @filepath = Qt::FileDialog.getSaveFileName(self, tr('Select a file'), '/', tr('*.*'))
     unless @filepath.nil?
-      @view.btn_export.setEnabled(true)
-      @view.btn_full_export.setEnabled(true)
+      @view.btn_read.setEnabled(true)
       @view.lbl_selected_file.setText("#{@filepath.split("/").last}")
     end
   rescue Exception => msg
     ErrorMsg.new.unknown(msg)
   end
 
-  def export
+  def read
+    return ErrorMsg.new.hardsploit_not_found unless HardsploitAPI.getNumberOfBoardAvailable > 0
     $file = File.open("#{@filepath}", 'w') unless @filepath.nil?
-    if sender.objectName == 'btn_full_export'
-      return false unless control_export_settings('full')
+    if @view.rbn_full.isChecked
+      return false unless control_read_settings('full')
       start   = 0
       stop    = @chip.i2c_setting.total_size - 1
       control = @chip.i2c_setting.total_size
     else
-      return false unless control_export_settings('partial')
+      return false unless control_read_settings('partial')
       start   =  @view.lie_start.text.to_i
       stop    =  @view.lie_stop.text.to_i
       control = (stop - start) + 1
     end
     Firmware.new('I2C')
-    $pgb = Progress_bar.new("I²C: Exporting...")
+    $pgb = Progress_bar.new("I²C: Reading...")
     $pgb.show
 
     if [40, 100, 400, 1000].include?(@chip.i2c_setting.frequency)
@@ -73,7 +73,7 @@ class I2c_export < Qt::Widget
     ErrorMsg.new.unknown(msg)
   end
 
-  def control_export_settings(type)
+  def control_read_settings(type)
     return ErrorMsg.new.settings_missing   if @chip.i2c_setting.nil?
     return ErrorMsg.new.frequency_missing  if @chip.i2c_setting.frequency.nil?
     return ErrorMsg.new.mode_missing       if @chip.i2c_setting.address_w.nil?