hardsploit_gui 2.3 → 2.4.0

data/lib/class/Import.rb

@@ -1,193 +1,193 @@
-#===================================================
-#  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 'json'
-require_relative '../gui/gui_import'
-class Import < Qt::Widget
-  slots 'import()'
-  slots 'import_file()'
-
-  def initialize(cm, chip = 'none')
-    super()
-    @import_ui = Ui_Import.new
-    centerWindow(self)
-    @import_ui.setupUi(self)
-    @chip = chip
-    @cm = cm
-    @import_ui.lbl_import.setText("Import:")
-  end
-
-  def import_file
-    @filepath = Qt::FileDialog.getOpenFileName(
-      self,
-      tr('Select a file'),
-      '/',
-      tr('JSON file (*.json)')
-    )
-  end
-
-  def import
-    unless @filepath.nil?
-      @file = File.read("#{@filepath}")
-      data_hash = JSON.parse(@file)
-      data_hash.each_with_index do |element, index|
-        @data_chip = element[1]['chip']
-        @data_commands =  element[1]['commands']
-        if @import_ui.rbn_both.isChecked
-            import_chip
-            import_commands
-        elsif @import_ui.rbn_comp.isChecked
-            import_chip
-        else
-            import_commands
-        end
-      end
-      @cm.feed_chip_array
-      Qt::MessageBox.new(
-        Qt::MessageBox::Information,
-        'Import status',
-        'Import complete'
-      ).exec
-      self.close
-    else
-      Qt::MessageBox.new(
-        Qt::MessageBox::Warning,
-        'File missing',
-        'Choose a file before importing'
-      ).exec
-    end
-  end
-
-  def import_package
-    package = Package.new
-    package.from_json(@data_chip['package'].to_json)
-    if Package.exists?(name: package.name)
-      return Package.find_by(name: package.name).id
-    else
-      Package.last.nil? ? package.id = 1 : package.id = Package.last.id.next
-      package.save
-      return package.id
-    end
-  end
-
-  def import_type
-    chip_type = ChipType.new
-    chip_type.from_json(@data_chip['chip_type'].to_json)
-    if ChipType.exists?(name: chip_type.name)
-      return ChipType.find_by(name: chip_type.name).id
-    else
-      ChipType.last.nil? ? chip_type.id = 1 : chip_type.id = ChipType.last.id.next
-      chip_type.save
-      return chip_type.id
-    end
-  end
-
-  def import_manufacturer
-    manufacturer = Manufacturer.new
-    manufacturer.from_json(@data_chip['manufacturer'].to_json)
-    if Manufacturer.exists?(name: manufacturer.name)
-      return Manufacturer.find_by(name: manufacturer.name).id
-    else
-      Manufacturer.last.nil? ? manufacturer.id = 1 : manufacturer.id = Manufacturer.last.id.next
-      manufacturer.save
-      return manufacturer.id
-    end
-  end
-
-  def import_chip
-    @chip = Chip.new
-    @chip.from_json(@data_chip['characteristics'].to_json)
-    if Chip.exists?(reference: @chip.reference)
-      Qt::MessageBox.new(
-        Qt::MessageBox::Warning,
-        'Import aborted',
-        'This component already exists'
-      ).exec
-      return 0
-    else
-      Chip.last.nil? ? @chip.id = 1 : @chip.id = Chip.last.id.next
-      @chip.package_id = import_package
-      @chip.chip_type_id = import_type
-      @chip.manufacturer_id = import_manufacturer
-      @chip.save
-      import_parallel_settings unless @data_chip['settings']['parallel'].nil?
-      import_spi_settings unless @data_chip['settings']['spi'].empty?
-      import_i2c_settings unless @data_chip['settings']['i2c'].empty?
-      import_swd_settings unless @data_chip['settings']['swd'].empty?
-      import_uart_settings unless @data_chip['settings']['uart'].empty?
-      import_pins
-    end
-  end
-
-  def import_pins
-    @data_chip['pins'].each do |data_pin|
-      pin = Pin.new
-      pin.from_json(data_pin.to_json)
-      pin.chip_id = @chip.id
-      Pin.last.nil? ? pin.id = 1 : pin.id = Pin.last.id.next
-      pin.save
-    end
-  end
-
-  def import_parallel_settings
-    parallel_settings = ParallelSetting.new
-    parallel_settings.from_json(@data_chip['settings']['parallel'].to_json)
-    parallel_settings.chip_id = @chip.id
-    ParallelSetting.last.nil? ? parallel_settings.id = 1 : parallel_settings.id = ParallelSetting.last.id.next
-    parallel_settings.save
-  end
-
-  def import_spi_settings
-    spi_settings = SpiSetting.new
-    spi_settings.from_json(@data_chip['settings']['spi'].to_json)
-    spi_settings.chip_id = @chip.id
-    SpiSetting.last.nil? ? spi_settings.id = 1 : spi_settings.id = SpiSetting.last.id.next
-    spi_settings.save
-  end
-
-  def import_i2c_settings
-    i2c_settings = I2cSetting.new
-    i2c_settings.from_json(@data_chip['settings']['i2c'].to_json)
-    i2c_settings.chip_id = @chip.id
-    I2cSetting.last.nil? ? i2c_settings.id = 1 : i2c_settings.id = I2cSetting.last.id.next
-    i2c_settings.save
-  end
-
-  def import_swd_settings
-    swd_settings = SwdSetting.new
-    swd_settings.from_json(@data_chip['settings']['swd'].to_json)
-    swd_settings.chip_id = @chip.id
-    SwdSetting.last.nil? ? swd_settings.id = 1 : swd_settings.id = SwdSetting.last.id.next
-    swd_settings.save
-  end
-
-  def import_uart_settings
-    uart_settings = UartSetting.new
-    uart_settings.from_json(@data_chip['settings']['uart'].to_json)
-    uart_settings.chip_id = @chip.id
-    UartSetting.last.nil? ? uart_settings.id = 1 : uart_settings.id = UartSetting.last.id.next
-    uart_settings.save
-  end
-
-  def import_commands
-    return 0 if @data_commands.empty?
-    @data_commands.each do |cmd_element|
-      cmd = Command.new
-      cmd.from_json(cmd_element[1]['characteristics'].to_json)
-      Command.last.nil? ? cmd.id = 1 : cmd.id = Command.last.id.next
-      cmd.chip_id = @chip.id
-      cmd.save
-      cmd_element[1]['bytes'].each do |byte_element|
-        byte = Byte.new
-        byte.from_json(byte_element.to_json)
-        Byte.last.nil? ? byte.id = 1 : byte.id = Byte.last.id.next
-        byte.command_id = cmd.id
-        byte.save
-      end
-    end
-  end
-end
+#===================================================
+#  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 'json'
+require_relative '../gui/gui_import'
+class Import < Qt::Widget
+  slots 'import()'
+  slots 'import_file()'
+
+  def initialize(cm, chip = 'none')
+    super()
+    @import_ui = Ui_Import.new
+    centerWindow(self)
+    @import_ui.setupUi(self)
+    @chip = chip
+    @cm = cm
+    @import_ui.lbl_import.setText("Import:")
+  end
+
+  def import_file
+    @filepath = Qt::FileDialog.getOpenFileName(
+      self,
+      tr('Select a file'),
+      '/',
+      tr('JSON file (*.json)')
+    )
+  end
+
+  def import
+    unless @filepath.nil?
+      @file = File.read("#{@filepath}")
+      data_hash = JSON.parse(@file)
+      data_hash.each_with_index do |element, index|
+        @data_chip = element[1]['chip']
+        @data_commands =  element[1]['commands']
+        if @import_ui.rbn_both.isChecked
+            import_chip
+            import_commands
+        elsif @import_ui.rbn_comp.isChecked
+            import_chip
+        else
+            import_commands
+        end
+      end
+      @cm.feed_chip_array
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Import status',
+        'Import complete'
+      ).exec
+      self.close
+    else
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'File missing',
+        'Choose a file before importing'
+      ).exec
+    end
+  end
+
+  def import_package
+    package = Package.new
+    package.from_json(@data_chip['package'].to_json)
+    if Package.exists?(name: package.name)
+      return Package.find_by(name: package.name).id
+    else
+      Package.last.nil? ? package.id = 1 : package.id = Package.last.id.next
+      package.save
+      return package.id
+    end
+  end
+
+  def import_type
+    chip_type = ChipType.new
+    chip_type.from_json(@data_chip['chip_type'].to_json)
+    if ChipType.exists?(name: chip_type.name)
+      return ChipType.find_by(name: chip_type.name).id
+    else
+      ChipType.last.nil? ? chip_type.id = 1 : chip_type.id = ChipType.last.id.next
+      chip_type.save
+      return chip_type.id
+    end
+  end
+
+  def import_manufacturer
+    manufacturer = Manufacturer.new
+    manufacturer.from_json(@data_chip['manufacturer'].to_json)
+    if Manufacturer.exists?(name: manufacturer.name)
+      return Manufacturer.find_by(name: manufacturer.name).id
+    else
+      Manufacturer.last.nil? ? manufacturer.id = 1 : manufacturer.id = Manufacturer.last.id.next
+      manufacturer.save
+      return manufacturer.id
+    end
+  end
+
+  def import_chip
+    @chip = Chip.new
+    @chip.from_json(@data_chip['characteristics'].to_json)
+    if Chip.exists?(reference: @chip.reference)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Import aborted',
+        'This component already exists'
+      ).exec
+      return 0
+    else
+      Chip.last.nil? ? @chip.id = 1 : @chip.id = Chip.last.id.next
+      @chip.package_id = import_package
+      @chip.chip_type_id = import_type
+      @chip.manufacturer_id = import_manufacturer
+      @chip.save
+      import_parallel_settings unless @data_chip['settings']['parallel'].nil?
+      import_spi_settings unless @data_chip['settings']['spi'].empty?
+      import_i2c_settings unless @data_chip['settings']['i2c'].empty?
+      import_swd_settings unless @data_chip['settings']['swd'].empty?
+      import_uart_settings unless @data_chip['settings']['uart'].empty?
+      import_pins
+    end
+  end
+
+  def import_pins
+    @data_chip['pins'].each do |data_pin|
+      pin = Pin.new
+      pin.from_json(data_pin.to_json)
+      pin.chip_id = @chip.id
+      Pin.last.nil? ? pin.id = 1 : pin.id = Pin.last.id.next
+      pin.save
+    end
+  end
+
+  def import_parallel_settings
+    parallel_settings = ParallelSetting.new
+    parallel_settings.from_json(@data_chip['settings']['parallel'].to_json)
+    parallel_settings.chip_id = @chip.id
+    ParallelSetting.last.nil? ? parallel_settings.id = 1 : parallel_settings.id = ParallelSetting.last.id.next
+    parallel_settings.save
+  end
+
+  def import_spi_settings
+    spi_settings = SpiSetting.new
+    spi_settings.from_json(@data_chip['settings']['spi'].to_json)
+    spi_settings.chip_id = @chip.id
+    SpiSetting.last.nil? ? spi_settings.id = 1 : spi_settings.id = SpiSetting.last.id.next
+    spi_settings.save
+  end
+
+  def import_i2c_settings
+    i2c_settings = I2cSetting.new
+    i2c_settings.from_json(@data_chip['settings']['i2c'].to_json)
+    i2c_settings.chip_id = @chip.id
+    I2cSetting.last.nil? ? i2c_settings.id = 1 : i2c_settings.id = I2cSetting.last.id.next
+    i2c_settings.save
+  end
+
+  def import_swd_settings
+    swd_settings = SwdSetting.new
+    swd_settings.from_json(@data_chip['settings']['swd'].to_json)
+    swd_settings.chip_id = @chip.id
+    SwdSetting.last.nil? ? swd_settings.id = 1 : swd_settings.id = SwdSetting.last.id.next
+    swd_settings.save
+  end
+
+  def import_uart_settings
+    uart_settings = UartSetting.new
+    uart_settings.from_json(@data_chip['settings']['uart'].to_json)
+    uart_settings.chip_id = @chip.id
+    UartSetting.last.nil? ? uart_settings.id = 1 : uart_settings.id = UartSetting.last.id.next
+    uart_settings.save
+  end
+
+  def import_commands
+    return 0 if @data_commands.empty?
+    @data_commands.each do |cmd_element|
+      cmd = Command.new
+      cmd.from_json(cmd_element[1]['characteristics'].to_json)
+      Command.last.nil? ? cmd.id = 1 : cmd.id = Command.last.id.next
+      cmd.chip_id = @chip.id
+      cmd.save
+      cmd_element[1]['bytes'].each do |byte_element|
+        byte = Byte.new
+        byte.from_json(byte_element.to_json)
+        Byte.last.nil? ? byte.id = 1 : byte.id = Byte.last.id.next
+        byte.command_id = cmd.id
+        byte.save
+      end
+    end
+  end
+end

data/lib/class/{parallel → PARALLEL}/Parallel_export.rb

@@ -1,118 +1,118 @@
-#===================================================
-#  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/NO_MUX_PARALLEL_MEMORY/HardsploitAPI_NO_MUX_PARALLEL_MEMORY'
-
-class Parallel_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('Bin file (*.bin)'))
-    unless @filepath.nil?
-      @view.btn_export.setEnabled(true)
-      @view.btn_full_export.setEnabled(true)
-    end
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def export
-    if sender.objectName == 'btn_full_export'
-      return false unless control_export_settings('full')
-      start = 0
-      stop  = @chip.parallel_setting.total_size - 1
-    else
-      return false unless control_export_settings('partial')
-      start = @view.lie_start.text.to_i
-      stop  = @view.lie_stop.text.to_i
-    end
-    @chip.parallel_setting.word_size == 0 ? word_size = true : word_size = false
-    Firmware.new('PARALLEL')
-    parallel = HardsploitAPI_PARALLEL.new
-    time = Time.new
-
-    parallel.read_Memory_WithoutMultiplexing(
-      path:                     @filepath,
-      addressStart:             start,
-      addressStop:              stop,
-      bits8_or_bits16_DataSize: word_size,
-      latency:                  @chip.parallel_setting.read_latency
-    )
-
-    control_export_result(stop, time)
-    @view.btn_export.setEnabled(false)
-    @view.btn_full_export.setEnabled(false)
-    @filepath = nil
-  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_result(stop, time)
-    time = Time.new - time
-    file_size = File.size("#{@filepath}")
-    # 8 bits test
-    if @chip.parallel_setting.word_size.zero?
-      if (stop - @view.lie_start.text.to_i + 1) == file_size
-        Qt::MessageBox.new(
-          Qt::MessageBox::Information,
-          "Information",
-          "Dump finished at #{((file_size / time)).round(2)}Bytes/s (#{(file_size)} Bytes in  #{time.round(4)} s)"
-        ).exec
-      else
-        ErrorMsg.new.filesize_error
-      end
-    else
-      if (stop - @view.lie_start.text.to_i + 1) == (file_size / 2)
-        Qt::MessageBox.new(
-          Qt::MessageBox::Information,
-          "Information",
-          "Dump finished at #{((file_size / time)).round(2)}Bytes/s (#{(file_size)} Bytes in  #{time.round(4)} s)"
-        ).exec
-      else
-        ErrorMsg.new.filesize_error
-      end
-    end
-    p "DUMP #{((file_size/time)).round(2)}Bytes/s (#{(file_size)}Bytes in  #{time.round(4)} s)"
-  end
-
-  def control_export_settings(type)
-    return ErrorMsg.new.settings_missing  if @chip.parallel_setting.nil?
-    return ErrorMsg.new.para_read_latency if @chip.parallel_setting.read_latency.nil?
-    return ErrorMsg.new.para_word_size    if @chip.parallel_setting.word_size.nil?
-    return ErrorMsg.new.full_size_error   if @chip.parallel_setting.total_size.nil?
-    return ErrorMsg.new.full_size_error   if @chip.parallel_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.parallel_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
+#===================================================
+#  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/NO_MUX_PARALLEL_MEMORY/HardsploitAPI_NO_MUX_PARALLEL_MEMORY'
+
+class Parallel_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('Bin file (*.bin)'))
+    unless @filepath.nil?
+      @view.btn_export.setEnabled(true)
+      @view.btn_full_export.setEnabled(true)
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def export
+    if sender.objectName == 'btn_full_export'
+      return false unless control_export_settings('full')
+      start = 0
+      stop  = @chip.parallel_setting.total_size - 1
+    else
+      return false unless control_export_settings('partial')
+      start = @view.lie_start.text.to_i
+      stop  = @view.lie_stop.text.to_i
+    end
+    @chip.parallel_setting.word_size == 0 ? word_size = true : word_size = false
+    Firmware.new('PARALLEL')
+    parallel = HardsploitAPI_PARALLEL.new
+    time = Time.new
+
+    parallel.read_Memory_WithoutMultiplexing(
+      path:                     @filepath,
+      addressStart:             start,
+      addressStop:              stop,
+      bits8_or_bits16_DataSize: word_size,
+      latency:                  @chip.parallel_setting.read_latency
+    )
+
+    control_export_result(stop, time)
+    @view.btn_export.setEnabled(false)
+    @view.btn_full_export.setEnabled(false)
+    @filepath = nil
+  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_result(stop, time)
+    time = Time.new - time
+    file_size = File.size("#{@filepath}")
+    # 8 bits test
+    if @chip.parallel_setting.word_size.zero?
+      if (stop - @view.lie_start.text.to_i + 1) == file_size
+        Qt::MessageBox.new(
+          Qt::MessageBox::Information,
+          "Information",
+          "Dump finished at #{((file_size / time)).round(2)}Bytes/s (#{(file_size)} Bytes in  #{time.round(4)} s)"
+        ).exec
+      else
+        ErrorMsg.new.filesize_error
+      end
+    else
+      if (stop - @view.lie_start.text.to_i + 1) == (file_size / 2)
+        Qt::MessageBox.new(
+          Qt::MessageBox::Information,
+          "Information",
+          "Dump finished at #{((file_size / time)).round(2)}Bytes/s (#{(file_size)} Bytes in  #{time.round(4)} s)"
+        ).exec
+      else
+        ErrorMsg.new.filesize_error
+      end
+    end
+    p "DUMP #{((file_size/time)).round(2)}Bytes/s (#{(file_size)}Bytes in  #{time.round(4)} s)"
+  end
+
+  def control_export_settings(type)
+    return ErrorMsg.new.settings_missing  if @chip.parallel_setting.nil?
+    return ErrorMsg.new.para_read_latency if @chip.parallel_setting.read_latency.nil?
+    return ErrorMsg.new.para_word_size    if @chip.parallel_setting.word_size.nil?
+    return ErrorMsg.new.full_size_error   if @chip.parallel_setting.total_size.nil?
+    return ErrorMsg.new.full_size_error   if @chip.parallel_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.parallel_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