hardsploit_gui 2.2 → 2.3

data/lib/class/{I2C → i2c}/I2c_command.rb

@@ -9,38 +9,41 @@ require_relative '../../gui/gui_i2c_command'
 class I2c_command < Qt::Widget
   slots 'open_generic_cmd()'
 
-  def initialize(api, chip, bus_id, parent)
+  def initialize(chip, bus_id, parent)
     super()
-    @i2c_command_gui = Ui_I2c_command.new
+    @view = Ui_I2c_command.new
     centerWindow(self)
-    @i2c_command_gui.setupUi(self)
-    @i2c_command_gui.rbn_read.setChecked(true)
-    inputRestrict(@i2c_command_gui.lie_size, 0)
+    @view.setupUi(self)
+    @view.rbn_read.setChecked(true)
+    inputRestrict(@view.lie_size, 0)
     @chip = chip
-    @chip_settings = I2C.find_by(i2c_chip: chip.chip_id)
-    @api = api
+    @chip_settings = chip.i2c_setting
     @bus_id = bus_id
     @parent = parent
   end
 
   def open_generic_cmd
     return 0 if check_form_param.zero?
-    if @i2c_command_gui.rbn_read.isChecked
+    if @view.rbn_read.isChecked
       mode = "r"
-      addr = @chip_settings.i2c_address_r
+      addr = @chip_settings.address_r
     else
       mode = "w"
-      addr = @chip_settings.i2c_address_w
+      addr = @chip_settings.address_w
     end
-    i2c_cmd_form = Command_editor.new(0, nil, @chip, @bus_id, @parent, @api, :mode => mode, :size => @i2c_command_gui.lie_size.text, :addr => addr)
+    i2c_cmd_form = Command_editor.new(0, nil, @chip, @bus_id, @parent, :mode => mode, :size => @view.lie_size.text, :addr => addr)
     i2c_cmd_form.setWindowModality(Qt::ApplicationModal)
     i2c_cmd_form.show
     close
   end
 
   def check_form_param
-    if @i2c_command_gui.lie_size.text.empty? || @i2c_command_gui.lie_size.text.to_i.zero?
-      Qt::MessageBox.new(Qt::MessageBox::Information, "Form invalid", "Payload size must be filled and superior to 0").exec
+    if @view.lie_size.text.empty? || @view.lie_size.text.to_i.zero?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        "Form invalid",
+        "Payload size must be filled and superior to 0"
+      ).exec
       return 0
     end
     return 1

data/lib/class/i2c/I2c_export.rb

@@ -0,0 +1,95 @@
+#===================================================
+#  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_generic_export'
+require_relative '../../HardsploitAPI/Modules/I2C/HardsploitAPI_I2C'
+class I2c_export < Qt::Widget
+  slots 'export()'
+  slots 'select_export_file()'
+
+  def initialize(chip)
+    super()
+    @view = Ui_Generic_export.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    inputRestrict(@view.lie_start, 0)
+    inputRestrict(@view.lie_stop, 0)
+    @chip = chip
+  end
+
+  def select_export_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.lbl_selected_file.setText("#{@filepath.split("/").last}")
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def export
+    $file = File.open("#{@filepath}", 'w') unless @filepath.nil?
+    if sender.objectName == 'btn_full_export'
+      return false unless control_export_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')
+      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.show
+
+    if [40, 100, 400, 1000].include?(@chip.i2c_setting.frequency)
+      speed = 0 if @chip.i2c_setting.frequency == 100
+      speed = 1 if @chip.i2c_setting.frequency == 400
+      speed = 2 if @chip.i2c_setting.frequency == 1000
+      speed = 3 if @chip.i2c_setting.frequency == 40
+      i2c = HardsploitAPI_I2C.new(speed: speed)
+      i2c.i2c_Generic_Dump(
+        i2cBaseAddress: @chip.i2c_setting.address_w.to_i(16),
+        startAddress:   start,
+        stopAddress:    stop,
+        sizeMax:        @chip.i2c_setting.total_size
+      )
+    end
+    $file.close unless $file.nil?
+    ErrorMsg.new.filesize_error unless control == File.size(@filepath)
+  rescue HardsploitAPI::ERROR::HARDSPLOIT_NOT_FOUND
+    ErrorMsg.new.hardsploit_not_found
+  rescue HardsploitAPI::ERROR::USB_ERROR
+    ErrorMsg.new.usb_error
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def control_export_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?
+    return ErrorMsg.new.full_size_error    if @chip.i2c_setting.total_size.nil?
+    return ErrorMsg.new.full_size_error    if @chip.i2c_setting.total_size.zero?
+    if type == 'partial'
+      return ErrorMsg.new.start_stop_missing if @view.lie_start.text.empty?
+      return ErrorMsg.new.start_stop_missing if @view.lie_stop.text.empty?
+      start = @view.lie_start.text.to_i
+      stop = @view.lie_stop.text.to_i
+      total_size = @chip.i2c_setting.total_size
+      return ErrorMsg.new.start_neq_stop    if start == stop
+      return ErrorMsg.new.start_inf_to_stop if start > stop
+      return ErrorMsg.new.inf_to_total_size if start > (total_size - 1)
+      return ErrorMsg.new.inf_to_total_size if stop > (total_size - 1)
+    end
+    return true
+  end
+end

data/lib/class/i2c/I2c_import.rb

@@ -0,0 +1,117 @@
+#===================================================
+#  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_generic_import'
+require_relative '../../HardsploitAPI/Modules/I2C/HardsploitAPI_I2C'
+class I2c_import < Qt::Widget
+  slots 'import()'
+  slots 'select_import_file()'
+
+  def initialize(chip)
+    super()
+    @view = Ui_Generic_import.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    inputRestrict(@view.lie_start, 0)
+    @chip = chip
+  end
+
+  def select_import_file
+    @filepath = Qt::FileDialog.getOpenFileName(self, tr('Select a file'), '/', tr('*.*'))
+    unless @filepath.nil?
+      @view.btn_import.setEnabled(true)
+      @view.lbl_selected_file.setText("#{@filepath.split("/").last}")
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def import
+    return 0 if control_import_settings.zero?
+    Firmware.new('I2C')
+    $pgb = Progress_bar.new("I²C: Exporting...")
+    $pgb.show
+    if [40, 100, 400, 1000].include?(@chip.i2c_setting.frequency)
+      speed = 0 if @chip.i2c_setting.frequency == 100
+      speed = 1 if @chip.i2c_setting.frequency == 400
+      speed = 2 if @chip.i2c_setting.frequency == 1000
+      speed = 3 if @chip.i2c_setting.frequency == 40
+      i2c = HardsploitAPI_I2C.new(speed: speed)
+      i2c.i2c_Generic_Import(
+        i2cBaseAddress:   @chip.i2c_setting.address_w.to_i(16),
+        startAddress:     @view.lie_start.text.to_i,
+        pageSize:         @chip.i2c_setting.page_size,
+        memorySize:       @chip.i2c_setting.total_size,
+        dataFile:         @filepath,
+        writePageLatency: @chip.i2c_setting.write_page_latency / 1000.0
+      )
+    end
+  rescue HardsploitAPI::ERROR::HARDSPLOIT_NOT_FOUND
+    ErrorMsg.new.hardsploit_not_found
+  rescue HardsploitAPI::ERROR::USB_ERROR
+    ErrorMsg.new.usb_error
+  rescue HardsploitAPI::ERROR::I2CWrongSpeed
+    ErrorMsg.new.i2c_error
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def control_import_settings
+    if @chip.i2c_setting.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing I2C setting',
+        'No settings saved for this chip'
+      ).exec
+      return 0
+    end
+    if @chip.i2c_setting.total_size.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing I2C setting',
+        'Total size setting missing'
+      ).exec
+      return 0
+    end
+    if @chip.i2c_setting.page_size.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing I2C setting',
+        'Page size setting missing'
+      ).exec
+      return 0
+    end
+    if @chip.i2c_setting.write_page_latency.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing I2C setting',
+        'Write page latency setting missing'
+      ).exec
+      return 0
+    end
+    if @chip.i2c_setting.frequency.nil? || @chip.i2c_setting.address_w.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing I2C setting',
+        'Write base address or frequency settings missing'
+      ).exec
+      return 0
+    end
+    if @view.lie_start.text.empty?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing start address',
+        'Please fill the Start address field'
+      ).exec
+      return 0
+    end
+    return 1
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+end

data/lib/class/i2c/I2c_scanner.rb

@@ -0,0 +1,114 @@
+#===================================================
+#  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_signal_scanner'
+require_relative '../../HardsploitAPI/Modules/I2C/HardsploitAPI_I2C'
+
+class I2c_scanner < Qt::Widget
+  slots 'scan()'
+  slots 'autowiring()'
+  slots 'update_tbl(QString)'
+  slots 'update_cbx(QString)'
+
+  def initialize
+    super()
+    @view = Ui_Signal_scanner.new
+    centerWindow(self)
+    @view.setupUi(self)
+    update_cbx('B0')
+  end
+
+  def scan
+    @view.cbx_start.setEnabled(false)
+    @view.cbx_stop.setEnabled(false)
+    @view.btn_scan.setEnabled(false)
+    update_tbl(@view.cbx_stop.currentText)
+    Firmware.new('I2C')
+    i2c = HardsploitAPI_I2C.new(speed: 3)
+    @result = i2c.find(
+      start_from: @view.cbx_start.currentText[1].to_i + 9,
+      stop_to:    @view.cbx_stop.currentText[1].to_i + 9
+    )
+    unless @result[1].is_a? Array
+      @view.tbl_result.setItem(@result.index(0), 1, Qt::TableWidgetItem.new("CLK"))
+      @view.tbl_result.setItem(@result.index(1), 1, Qt::TableWidgetItem.new("SDA"))
+    end
+    @view.cbx_start.setEnabled(true)
+    @view.cbx_stop.setEnabled(true)
+    @view.btn_scan.setEnabled(true)
+  rescue HardsploitAPI::ERROR::HARDSPLOIT_NOT_FOUND
+    ErrorMsg.new.hardsploit_not_found
+    return false
+  rescue HardsploitAPI::ERROR::USB_ERROR
+    ErrorMsg.new.usb_error
+    return false
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+    return false
+  end
+
+  def autowiring
+    signal_found = false
+    @view.tbl_result.rowCount.times do |i|
+      if @view.tbl_result.item(i, 1).text == 'SDA'
+        signal_found = true
+        current_signal = Signall.find_by(name: 'SDA')
+        current_signal.update(pin: @view.tbl_result.item(i, 0).text)
+      elsif @view.tbl_result.item(i, 1).text == 'CLK'
+        signal_found = true
+        current_signal = Signall.find_by(name: 'I2C_CLK')
+        current_signal.update(pin: @view.tbl_result.item(i, 0).text)
+      else
+        # Next row
+      end
+    end
+    if signal_found
+      Qt::MessageBox.new(
+  			Qt::MessageBox::Information,
+  			'Hardsploit Autowiring',
+  			'Wiring saved. To change it, go to Menu > Signal mapper'
+  		).exec
+    else
+      Qt::MessageBox.new(
+  			Qt::MessageBox::Information,
+  			'Hardsploit Autowiring',
+  			'No signals found in the array'
+  		).exec
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+    return false
+  end
+
+  def update_tbl(pin)
+    unless @view.cbx_stop.count.zero?
+      @view.tbl_result.setRowCount(0)
+      nbr = pin[1].to_i - (@view.cbx_start.currentText[1].to_i) + 1
+      nbr.times do |i|
+        @view.tbl_result.insertRow(i)
+        @view.tbl_result.setItem(i, 0, Qt::TableWidgetItem.new("B#{(i + @view.cbx_start.currentText[1].to_i)}"))
+        @view.tbl_result.setItem(i, 1, Qt::TableWidgetItem.new('-'))
+      end
+      resize_to_content
+    end
+  end
+
+  def resize_to_content
+    @view.tbl_result.resizeColumnsToContents
+    @view.tbl_result.resizeRowsToContents
+    @view.tbl_result.horizontalHeader.stretchLastSection = true
+  end
+
+  def update_cbx(pin)
+    @view.cbx_stop.clear
+    start_at = pin[1].to_i
+    end_at = 7
+    (end_at - start_at).times do |i|
+      @view.cbx_stop.addItem("B#{(i + start_at) + 1}")
+    end
+  end
+end

data/lib/class/i2c/I2c_settings.rb

@@ -0,0 +1,148 @@
+#===================================================
+#  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_i2c_settings'
+class I2c_settings < Qt::Widget
+  slots 'save_settings()'
+  slots 'bus_scan()'
+
+  def initialize(chip)
+    super()
+    @view = Ui_I2c_settings.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    inputRestrict(@view.lie_total_size,         0)
+    inputRestrict(@view.lie_page_size,          0)
+    inputRestrict(@view.lie_write_page_latency, 0)
+    inputRestrict(@view.lie_address_r,          3)
+    inputRestrict(@view.lie_address_w,          3)
+    @view.tbl_bus_scan.resizeColumnsToContents
+    @view.tbl_bus_scan.resizeRowsToContents
+    @view.tbl_bus_scan.horizontalHeader.stretchLastSection = true
+    @chip = chip
+    feed_settings_form unless @chip.i2c_setting.nil?
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def save_settings
+    @chip.i2c_setting.nil? ? create : update
+    @chip.reload
+  end
+
+  def feed_settings_form
+    @view.lie_address_w.setText(@chip.i2c_setting.address_w)
+    @view.lie_address_r.setText(@chip.i2c_setting.address_r)
+    @view.cbx_frequency.setCurrentIndex(@view.cbx_frequency.findText(@chip.i2c_setting.frequency.to_s))
+    @view.lie_write_page_latency.setText(@chip.i2c_setting.write_page_latency.to_s)
+    @view.lie_page_size.setText(@chip.i2c_setting.page_size.to_s)
+    @view.lie_total_size.setText(@chip.i2c_setting.total_size.to_s)
+  end
+
+  def create
+    chip_setting = I2cSetting.create(
+      address_w:          @view.lie_address_w.text,
+      address_r:          @view.lie_address_r.text,
+      frequency:          @view.cbx_frequency.currentText.to_i,
+      write_page_latency: @view.lie_write_page_latency.text,
+      page_size:          @view.lie_page_size.text,
+      total_size:         @view.lie_total_size.text,
+      chip_id:            @chip.id
+    )
+    unless check_for_errors(chip_setting)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'I2C settings saved'
+      ).exec
+      close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def update
+    @chip.i2c_setting.update(
+      address_w:          @view.lie_address_w.text,
+      address_r:          @view.lie_address_r.text,
+      frequency:          @view.cbx_frequency.currentText.to_i,
+      write_page_latency: @view.lie_write_page_latency.text,
+      page_size:          @view.lie_page_size.text,
+      total_size:         @view.lie_total_size.text
+    )
+    unless check_for_errors(@chip.i2c_setting)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'I2C settings updated'
+      ).exec
+      close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def bus_scan
+    @view.tbl_bus_scan.setRowCount(0)
+    Firmware.new('I2C')
+    speed = 0
+    if [40, 100, 400, 1000].include?(@chip.i2c_setting.frequency)
+      speed = 0 if @chip.i2c_setting.frequency == 100
+      speed = 1 if @chip.i2c_setting.frequency == 400
+      speed = 2 if @chip.i2c_setting.frequency == 1000
+      speed = 3 if @chip.i2c_setting.frequency == 40
+    end
+    i2c = HardsploitAPI_I2C.new(speed: speed)
+    scan_result = i2c.i2c_Scan
+    if scan_result.include?(1)
+      scan_result.each_with_index do |v, i|
+        if v == 1
+          @view.tbl_bus_scan.insertRow(@view.tbl_bus_scan.rowCount)
+          @view.tbl_bus_scan.setItem(
+            @view.tbl_bus_scan.rowCount - 1,
+            0,
+            Qt::TableWidgetItem.new("0x#{i.to_s(16).upcase}")
+          )
+          if i % 2 == 0
+            @view.tbl_bus_scan.setItem(
+              @view.tbl_bus_scan.rowCount - 1,
+              1,
+              Qt::TableWidgetItem.new('Write')
+            )
+          else
+            @view.tbl_bus_scan.setItem(
+              @view.tbl_bus_scan.rowCount - 1,
+              1,
+              Qt::TableWidgetItem.new('Read')
+            )
+          end
+        end
+      end
+      @view.tbl_bus_scan.resizeColumnsToContents
+      @view.tbl_bus_scan.resizeRowsToContents
+      @view.tbl_bus_scan.horizontalHeader.stretchLastSection = true
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        "Bus Scan",
+        "Bus scan ended correctly: #{@view.tbl_bus_scan.rowCount} address(es) found"
+      ).exec
+    else
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        "Bus addresses",
+        "No valid addresses have been returned by the scan"
+      ).exec
+    end
+  rescue HardsploitAPI::ERROR::HARDSPLOIT_NOT_FOUND
+    ErrorMsg.new.hardsploit_not_found
+  rescue HardsploitAPI::ERROR::USB_ERROR
+    ErrorMsg.new.usb_error
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+end