hardsploit_gui 2.2 → 2.3

data/lib/class/parallel/Parallel_export.rb

@@ -0,0 +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

data/lib/class/parallel/Parallel_import.rb

@@ -0,0 +1,113 @@
+#===================================================
+#  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'
+
+class Parallel_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)
+  end
+
+  def select_import_file
+    @filepath = Qt::FileDialog.getOpenFileName(self, tr('Select a file'), '/', tr('Bin file (*.bin)'))
+    unless @filepath.nil?
+      $file = File.open("#{@filepath}", 'w')
+      @view.btn_import.setEnabled(true)
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknow(msg)
+  end
+
+  def close_file
+    unless $file.nil?
+      $file.close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknow(msg)
+  end
+
+  def import
+    return 0 if control_import_settings.zero?
+    start = @view.lie_start.text.to_i
+    Firmware.new('PARALLEL')
+    time = Time.new
+    # API COMMAND GOES HERE
+  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 HardsploitAPI::ERROR::PARALLEL_ERROR
+    ErrorMsg.new.swd_error
+    return false
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+    return false
+  end
+
+  def control_import_settings
+    file_size = File.size("#{@filepath}")
+    if @chip.parallel_setting.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing parallel settings',
+        'No settings saved for this chip'
+      ).exec
+      return 0
+    end
+    if @chip.parallel_setting.total_size.nil? || @chip.parallel_setting.word_size.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing parallel settings',
+        'Total size or word size settings missing'
+      ).exec
+      return 0
+    end
+    if @chip.parallel_setting.page_size.nil? || @chip.parallel_setting.write_latency.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing parallel settings',
+        'Page size or write latency 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
+    if file_size > @chip.parallel_setting.total_size
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Incorrect file size',
+        'The file size is superior to the chip capacity'
+      ).exec
+      return 0
+    end
+    if file_size > (@chip.parallel_setting.total_size - @view.lie_start)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Incorrect file size',
+        'Starting at this address, the file size is superior to the chip capacity'
+      ).exec
+      return 0
+    end
+    return 1
+  end
+end

data/lib/class/parallel/Parallel_settings.rb

@@ -0,0 +1,81 @@
+#===================================================
+#  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_parallel_settings'
+class Parallel_settings < Qt::Widget
+  slots 'save_settings()'
+
+  def initialize(chip)
+    super()
+    @view = Ui_Parallel_settings.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    inputRestrict(@view.lie_total_size,     0)
+    inputRestrict(@view.lie_write_latency,  0)
+    inputRestrict(@view.lie_read_latency,   0)
+    inputRestrict(@view.lie_page_size,      0)
+    @chip = chip
+    feed_settings_form unless @chip.parallel_setting.nil?
+  end
+
+  def save_settings
+    @chip.parallel_setting.nil? ? create : update
+    @chip.reload
+  end
+
+  def feed_settings_form
+    @view.lie_read_latency.setText(@chip.parallel_setting.read_latency.to_s)
+    @view.lie_write_latency.setText(@chip.parallel_setting.write_latency.to_s)
+    @view.lie_page_size.setText(@chip.parallel_setting.page_size.to_s)
+    @view.lie_total_size.setText(@chip.parallel_setting.total_size.to_s)
+    @view.rbn_16b.setChecked(true) unless @chip.parallel_setting.word_size.zero?
+  end
+
+  def create
+    @view.rbn_8b.isChecked ? word_size = 0 : word_size = 1
+    chip_setting = ParallelSetting.create(
+      total_size:     @view.lie_total_size.text,
+      page_size:      @view.lie_page_size.text,
+      word_size:      word_size,
+      read_latency:   @view.lie_read_latency.text,
+      write_latency:  @view.lie_write_latency.text,
+      chip_id:        @chip.id
+    )
+    unless check_for_errors(chip_setting)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'Parallel settings saved'
+      ).exec
+      close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def update
+    @view.rbn_8b.isChecked ? word_size = 0 : word_size = 1
+    @chip.parallel_setting.update(
+    total_size:     @view.lie_total_size.text,
+    page_size:      @view.lie_page_size.text,
+    word_size:      word_size,
+    read_latency:   @view.lie_read_latency.text,
+    write_latency:  @view.lie_write_latency.text
+    )
+    unless check_for_errors(@chip.parallel_setting)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'Parallel settings updated'
+      ).exec
+      close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+end

data/lib/class/spi/Spi_export.rb

@@ -0,0 +1,108 @@
+#===================================================
+#  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/SPI/HardsploitAPI_SPI'
+
+class Spi_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
+    @speeds = {
+      '25.00' => 3,
+      '18.75' => 4,
+      '15.00' => 5,
+      '12.50' => 6,
+      '10.71' => 7,
+      '9.38'  => 8,
+      '7.50'  => 10,
+      '5.00'  => 15,
+      '3.95'  => 19,
+      '3.00'  => 25,
+      '2.03'  => 37,
+      '1.00'  => 75,
+      '0.50'  => 150,
+      '0.29'  => 255
+    }
+  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.spi_setting.total_size - 1
+      control = @chip.spi_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('SPI')
+    $pgb = Progress_bar.new("SPI: Exporting...")
+    $pgb.show
+    spi = HardsploitAPI_SPI.new(
+      speed: @speeds[@chip.spi_setting.frequency],
+      mode:  @chip.spi_setting.mode
+    )
+    spi.spi_Generic_Dump(
+      readSpiCommand: @chip.spi_setting.command_read,
+      startAddress:   start,
+      stopAddress:    stop,
+      sizeMax:        @chip.spi_setting.total_size.to_i
+    )
+    $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.spi_setting.nil?
+    return ErrorMsg.new.frequency_missing  if @chip.spi_setting.frequency.nil?
+    return ErrorMsg.new.mode_missing       if @chip.spi_setting.command_read.nil?
+    return ErrorMsg.new.full_size_error    if @chip.spi_setting.total_size.nil?
+    return ErrorMsg.new.full_size_error    if @chip.spi_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.spi_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/spi/Spi_import.rb

@@ -0,0 +1,159 @@
+#===================================================
+#  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/SPI/HardsploitAPI_SPI'
+
+class Spi_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
+    @speeds = {
+      '25.00' => 3,
+      '18.75' => 4,
+      '15.00' => 5,
+      '12.50' => 6,
+      '10.71' => 7,
+      '9.38' => 8,
+      '7.50' => 10,
+      '5.00' => 15,
+      '3.95' => 19,
+      '3.00' => 25,
+      '2.03' => 37,
+      '1.00' => 75,
+      '0.50' => 150,
+      '0.29' => 255
+    }
+  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?
+    spi = HardsploitAPI_SPI.new(
+      speed: @speeds[@chip.spi_setting.frequency],
+      mode:  @chip.spi_setting.mode
+    )
+    Firmware.new('SPI')
+    $pgb = Progress_bar.new("SPI: Importing...")
+    $pgb.show
+    @chip.spi_setting.is_flash.zero? ? flash = false : flash = true
+    spi.spi_Generic_Import(
+      startAddress:          @view.lie_start.text.to_i,
+      pageSize:              @chip.spi_setting.page_size,
+      memorySize:            @chip.spi_setting.total_size,
+      dataFile:              @filepath,
+      writeSpiCommand:       @chip.spi_setting.command_write,
+      writePageLatency:      @chip.spi_setting.write_page_latency / 1000.0,
+      enableWriteSpiCommand: @chip.spi_setting.command_write_enable,
+      clearSpiCommand:       @chip.spi_setting.command_erase,
+      clearChipTime:         @chip.spi_setting.erase_time,
+      isFLASH:               flash
+    )
+  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_import_settings
+    if @chip.spi_setting.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI settings',
+        'No settings saved for this chip'
+      ).exec
+      return 0
+    end
+    if @chip.spi_setting.total_size.nil? || @chip.spi_setting.page_size.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI settings',
+        'Total size or page size missing'
+    ).exec
+      return 0
+    end
+    if @chip.spi_setting.command_read.nil? || @chip.spi_setting.frequency.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI settings',
+        'Read command or frequency missing'
+      ).exec
+      return 0
+    end
+    if @chip.spi_setting.command_write.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI settings',
+        'Write command missing'
+      ).exec
+      return 0
+    end
+    if @chip.spi_setting.write_page_latency.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI setting',
+        'SPI write page latency missing'
+      ).exec
+      return 0
+    end
+    if @chip.spi_setting.is_flash.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI setting',
+        'SPI flash nature missing'
+      ).exec
+      return 0
+    else
+      if @chip.spi_setting.is_flash == 1
+        if @chip.spi_setting.erase_time.nil? || @chip.spi_setting.command_erase.nil?
+          Qt::MessageBox.new(
+            Qt::MessageBox::Warning,
+            'Missing SPI setting',
+            'SPI erase time or command missing'
+          ).exec
+          return 0
+        end
+      end
+    end
+    if @chip.spi_setting.mode.nil?
+      Qt::MessageBox.new(
+        Qt::MessageBox::Warning,
+        'Missing SPI setting',
+        'Mode 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
+  end
+end