hardsploit_gui 2.0

data/lib/class/PARALLEL/Parallel_import.rb

@@ -0,0 +1,88 @@
+#===================================================
+#  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 '../../HardsploitAPI/HardsploitAPI'
+require_relative '../../gui/gui_generic_import'
+
+class Parallel_import < Qt::Widget
+  slots 'import()'
+  slots 'select_import_file()'
+
+  def initialize(api, chip)
+    super()
+    @parallel_import_gui = Ui_Generic_import.new
+    centerWindow(self)
+    @parallel_import_gui.setupUi(self)
+    @parallel_import_gui.lbl_chip.setText(chip.chip_reference)
+    inputRestrict(@parallel_import_gui.lie_start, 0)
+    @api = api
+    @chip_settings = Parallel.find_by(parallel_chip: chip.chip_id)
+  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')
+      @parallel_import_gui.btn_import.setEnabled(true)
+    end
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while openning the export file. Consult the logs for more details').exec
+  end
+
+  def close_file
+    unless $file.nil?
+      $file.close
+    end
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while closing the export file. Consult the logs for more details').exec
+  end
+
+  def import
+    return 0 if control_import_settings.zero?
+    start = @parallel_import_gui.lie_start.text.to_i
+    Firmware.new(@api, 'PARALLEL')
+    time = Time.new
+    # API COMMAND GOES HERE
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while partial import operation. Consult the logs for more details').exec
+  end
+
+  def control_import_settings
+    file_size = File.size("#{@filepath}")
+    if @chip_settings.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing parallel settings', 'No settings saved for this chip').exec
+      return 0
+    end
+    if @chip_settings.parallel_total_size.nil? || @chip_settings.parallel_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_settings.parallel_page_size.nil? || @chip_settings.parallel_write_latency.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing parallel settings', 'Page size or write latency settings missing').exec
+      return 0
+    end
+    if @parallel_import_gui.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_settings.parallel_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_settings.parallel_total_size - @parallel_import_gui.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,102 @@
+#===================================================
+#  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()
+    @parallel_settings_gui = Ui_Parallel_settings.new
+    centerWindow(self)
+    @parallel_settings_gui.setupUi(self)
+    @parallel_settings_gui.lbl_chip.setText(chip.chip_reference)
+    inputRestrict(@parallel_settings_gui.lie_total_size, 0)
+    inputRestrict(@parallel_settings_gui.lie_write_latency, 0)
+    inputRestrict(@parallel_settings_gui.lie_read_latency, 0)
+    inputRestrict(@parallel_settings_gui.lie_page_size, 0)
+    @chip_settings = Parallel.find_by(parallel_chip: chip.chip_id)
+    @chip = chip
+    feed_settings_form
+  end
+
+  def save_settings
+    if @chip_settings.nil?
+      create
+    else
+      @chip_settings = Parallel.find_by(parallel_chip: @chip.chip_id)
+      update
+    end
+  end
+
+  def feed_settings_form
+    unless @chip_settings.nil?
+      @parallel_settings_gui.lie_read_latency.setText(@chip_settings.parallel_read_latency.to_s)
+      @parallel_settings_gui.lie_write_latency.setText(@chip_settings.parallel_write_latency.to_s)
+      @parallel_settings_gui.lie_page_size.setText(@chip_settings.parallel_page_size.to_s)
+      @parallel_settings_gui.lie_total_size.setText(@chip_settings.parallel_total_size.to_s)
+      if @chip_settings.parallel_word_size.zero?
+        @parallel_settings_gui.rbn_8b.setChecked(true)
+      else
+        @parallel_settings_gui.rbn_16b.setChecked(true)
+      end
+    end
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error while loading the form. Consult the log for more details').exec
+  end
+
+  def create
+    if @parallel_settings_gui.rbn_8b.isChecked
+      word_size = 0
+    else
+      word_size = 1
+    end
+    @chip_settings = Parallel.create(
+      parallel_word_size: word_size,
+      parallel_write_latency: @parallel_settings_gui.lie_write_latency,
+      parallel_read_latency: @parallel_settings_gui.lie_read_latency.text,
+      parallel_page_size: @parallel_settings_gui.lie_page_size.text,
+      parallel_total_size: @parallel_settings_gui.lie_total_size.text,
+      parallel_chip: @chip.chip_id
+    )
+    Qt::MessageBox.new(Qt::MessageBox::Information, 'Succes', 'Parallel parameters created successfully').exec
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured when saving the parallel parameters. Consult the log for more details').exec
+  end
+
+  def update
+    if @parallel_settings_gui.rbn_8b.isChecked
+      word_size = 0
+    else
+      word_size = 1
+    end
+    if @chip_settings.parallel_page_size != @parallel_settings_gui.lie_page_size
+      @chip_settings.update(parallel_page_size: @parallel_settings_gui.lie_page_size.text)
+    end
+    if @chip_settings.parallel_total_size != @parallel_settings_gui.lie_total_size
+      @chip_settings.update(parallel_total_size: @parallel_settings_gui.lie_total_size.text)
+    end
+    if @chip_settings.parallel_read_latency != @parallel_settings_gui.lie_read_latency
+      @chip_settings.update(parallel_read_latency: @parallel_settings_gui.lie_read_latency.text)
+    end
+    if @chip_settings.parallel_write_latency != @parallel_settings_gui.lie_write_latency
+      @chip_settings.update(parallel_write_latency: @parallel_settings_gui.lie_write_latency.text)
+    end
+    if @chip_settings.parallel_word_size != word_size
+      @chip_settings.update(parallel_word_size: word_size)
+    end
+    Qt::MessageBox.new(Qt::MessageBox::Information, 'Succes', 'Parallel parameters saved successfully').exec
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured when saving the parallel parameters. Consult the log for more details').exec
+  end
+end

data/lib/class/SPI/Spi_export.rb

@@ -0,0 +1,141 @@
+#===================================================
+#  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 '../../HardsploitAPI/HardsploitAPI'
+require_relative '../../gui/gui_generic_export'
+
+class Spi_export < Qt::Widget
+  slots 'export()'
+  slots 'select_export_file()'
+
+  def initialize(api, chip)
+    super()
+    @spi_export_gui = Ui_Generic_export.new
+    centerWindow(self)
+    @spi_export_gui.setupUi(self)
+    @spi_export_gui.lbl_chip.setText(chip.chip_reference)
+    inputRestrict(@spi_export_gui.lie_start, 0)
+    inputRestrict(@spi_export_gui.lie_stop, 0)
+    @api = api
+    @chip_settings = Spi.find_by(spi_chip: chip.chip_id)
+    @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('Bin file (*.bin)'))
+    unless @filepath.nil?
+      $file = File.open("#{@filepath}", 'w')
+      @spi_export_gui.btn_export.setEnabled(true)
+      @spi_export_gui.btn_full_export.setEnabled(true)
+    end
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while openning the export file. Consult the logs for more details').exec
+  end
+
+  def close_file
+    unless $file.nil?
+      $file.close
+    end
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while closing the export file. Consult the logs for more details').exec
+  end
+
+  def export
+    if sender.objectName == 'btn_full_export'
+      return 0 if control_export_settings('full').zero?
+    else
+      return 0 if control_export_settings('partial').zero?
+    end
+    Firmware.new(@api, 'SPI')
+    time = Time.new
+    if sender.objectName == 'btn_full_export'
+      @api.spi_Generic_Dump(@chip_settings.spi_mode, @speeds[@chip_settings.spi_frequency], @chip_settings.spi_command_read, 0, @chip_settings.spi_total_size - 1, @chip_settings.spi_total_size.to_i)
+      close_file
+      control_export_result('full', @chip_settings.spi_total_size - 1, time)
+    else
+      @api.spi_Generic_Dump(@chip_settings.spi_mode, @speeds[@chip_settings.spi_frequency], @chip_settings.spi_command_read, @spi_export_gui.lie_start.text.to_i, @spi_export_gui.lie_stop.text.to_i, @chip_settings.spi_total_size.to_i)
+      close_file
+      control_export_result('partial', @spi_export_gui.lie_stop.text.to_i, time)
+    end
+    @spi_export_gui.btn_export.setEnabled(false)
+    @spi_export_gui.btn_full_export.setEnabled(false)
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while full export operation. Consult the logs for more details').exec
+  end
+
+  def control_export_result(type, stop, time)
+    time = Time.new - time
+    if type == 'partial'
+      toCompare = ((stop - @spi_export_gui.lie_start.text.to_i) + 1)
+    else
+      toCompare = @chip_settings.spi_total_size
+    end
+    file_size = File.size(@filepath)
+    if toCompare == 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
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Error', 'Dump error: Size does not match').exec
+    end
+    p "DUMP #{((file_size / time)).round(2)}Bytes/s (#{(file_size)}Bytes in  #{time.round(4)} s)"
+  end
+
+  def control_export_settings(type)
+    if @chip_settings.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI settings', 'No settings saved for this chip').exec
+      return 0
+    end
+    if @chip_settings.spi_command_read.nil? || @chip_settings.spi_frequency.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI settings', 'Read command or frequency settings missing').exec
+      return 0
+    end
+    if @chip_settings.spi_mode.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI setting', 'Mode setting missing').exec
+      return 0
+    end
+    if type == 'full'
+      if @chip_settings.spi_total_size.zero? || @chip_settings.spi_total_size.nil?
+        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Empty field', 'Full size setting missing or equal 0').exec
+        return 0
+      end
+    else
+      if @spi_export_gui.lie_start.text.empty? || @spi_export_gui.lie_stop.text.empty?
+        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Empty field', 'Start and stop address must be filled').exec
+        return 0
+      end
+      if @spi_export_gui.lie_start.text.to_i > @spi_export_gui.lie_stop.text.to_i
+        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Wrong value', 'Start address must be inforior to the stop address').exec
+        return 0
+      end
+      if @spi_export_gui.lie_start.text.to_i > (@chip_settings.spi_total_size - 1) || @spi_export_gui.lie_stop.text.to_i > (@chip_settings.spi_total_size - 1)
+        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Wrong value', 'Start and stop address must be inforior to the chip total size').exec
+        return 0
+      end
+    end
+    return 1
+  end
+end

data/lib/class/SPI/Spi_import.rb

@@ -0,0 +1,112 @@
+#===================================================
+#  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 '../../HardsploitAPI/HardsploitAPI'
+require_relative '../../gui/gui_generic_import'
+
+class Spi_import < Qt::Widget
+  slots 'import()'
+  slots 'select_import_file()'
+
+  def initialize(api, chip)
+    super()
+    @spi_import_gui = Ui_Generic_import.new
+    centerWindow(self)
+    @spi_import_gui.setupUi(self)
+    @spi_import_gui.lbl_chip.setText(chip.chip_reference)
+    inputRestrict(@spi_import_gui.lie_start, 0)
+    @api = api
+    @chip_settings = Spi.find_by(spi_chip: chip.chip_id)
+    @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('Bin file (*.bin)'))
+    unless @filepath.nil?
+      $file = File.open("#{@filepath}", 'w')
+      @spi_import_gui.btn_import.setEnabled(true)
+    end
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while openning the export file. Consult the logs for more details').exec
+  end
+
+  def close_file
+    unless $file.nil?
+      $file.close
+    end
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while closing the export file. Consult the logs for more details').exec
+  end
+
+  def import
+    return 0 if control_import_settings.zero?
+    start = @spi_import_gui.lie_start.text.to_i
+    Firmware.new(@api, 'SPI')
+    time = Time.new
+    @api.spi_Generic_Dump(@chip_settings.spi_mode, @speeds[@chip_settings.spi_frequency], @chip_settings.spi_command_read, @spi_import_gui.lie_start.text.to_i, @chip_settings.spi_total_size.to_i)
+  rescue Exception => msg
+      logger = Logger.new($logFilePath)
+      logger.error msg
+      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while partial import operation. Consult the logs for more details').exec
+  end
+
+  def control_import_result(type, stop, pathfile, time)
+    # TO COME
+  end
+
+  def control_import_settings
+    file_size = File.size("#{@filepath}")
+    if @chip_settings.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI settings', 'No settings saved for this chip').exec
+      return 0
+    end
+    if @chip_settings.spi_total_size.nil? || @chip_settings.spi_page_size.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI settings', 'Total size or page size settings missing').exec
+      return 0
+    end
+    if @chip_settings.spi_read_command.nil? || @chip_settings.spi_frequency.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI settings', 'Read command or frequency settings missing').exec
+      return 0
+    end
+    if @chip_settings.spi_mode.nil?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing SPI setting', 'Mode setting missing').exec
+      return 0
+    end
+    if @spi_import_gui.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_settings.spi_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_settings.spi_total_size - @spi_import_gui.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/SPI/Spi_settings.rb

@@ -0,0 +1,90 @@
+#===================================================
+#  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_spi_settings'
+class Spi_settings < Qt::Widget
+  slots 'save_settings()'
+
+  def initialize(chip)
+    super()
+    @spi_settings_gui = Ui_Spi_settings.new
+    centerWindow(self)
+    @spi_settings_gui.setupUi(self)
+    @spi_settings_gui.lbl_chip.setText(chip.chip_reference)
+    inputRestrict(@spi_settings_gui.lie_total_size, 0)
+    inputRestrict(@spi_settings_gui.lie_cmd_read, 0)
+    inputRestrict(@spi_settings_gui.lie_page_size, 0)
+    @chip_settings = Spi.find_by(spi_chip: chip.chip_id)
+    @chip = chip
+    feed_settings_form
+  end
+
+  def save_settings
+    if @chip_settings.nil?
+      create
+    else
+      @chip_settings = Spi.find_by(spi_chip: @chip.chip_id)
+      update
+    end
+  end
+
+  def feed_settings_form
+    unless @chip_settings.nil?
+      @spi_settings_gui.cbx_mode.setCurrentIndex(@spi_settings_gui.cbx_mode.findText(@chip_settings.spi_mode.to_s))
+      @spi_settings_gui.cbx_frequency.setCurrentIndex(@spi_settings_gui.cbx_frequency.findText(@chip_settings.spi_frequency))
+      @spi_settings_gui.lie_cmd_read.setText(@chip_settings.spi_command_read.to_s)
+      @spi_settings_gui.lie_page_size.setText(@chip_settings.spi_page_size.to_s)
+      @spi_settings_gui.lie_total_size.setText(@chip_settings.spi_total_size.to_s)
+    end
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error while loading the form. Consult the log for more details').exec
+  end
+
+  def create
+    @chip_settings = Spi.create(
+      spi_mode: @spi_settings_gui.cbx_mode.currentText,
+      spi_frequency: @spi_settings_gui.cbx_frequency.currentText,
+      spi_command_read: @spi_settings_gui.lie_cmd_read.text,
+      spi_page_size: @spi_settings_gui.lie_page_size.text,
+      spi_total_size: @spi_settings_gui.lie_total_size.text,
+      spi_chip: @chip.chip_id
+    )
+    Qt::MessageBox.new(Qt::MessageBox::Information, 'Succes', 'SPI parameters created successfully').exec
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured when saving the SPI parameters. Consult the log for more details').exec
+  end
+
+  def update
+    if @chip_settings.spi_mode != @spi_settings_gui.cbx_mode.currentText.to_i
+      @chip_settings.update(spi_mode: @spi_settings_gui.cbx_mode.currentText.to_i)
+    end
+    if @chip_settings.spi_frequency != @spi_settings_gui.cbx_frequency.currentText
+      @chip_settings.update(spi_frequency: @spi_settings_gui.cbx_frequency.currentText)
+    end
+    if @chip_settings.spi_command_read != @spi_settings_gui.lie_cmd_read.text
+      if @spi_settings_gui.lie_cmd_read.text.empty?
+        @spi_settings_gui.lie_cmd_read.setText('3')
+      end
+      @chip_settings.update(spi_command_read: @spi_settings_gui.lie_cmd_read.text)
+    end
+    if @chip_settings.spi_page_size != @spi_settings_gui.lie_page_size.text
+      @chip_settings.update(spi_page_size: @spi_settings_gui.lie_page_size.text)
+    end
+    if @chip_settings.spi_total_size != @spi_settings_gui.lie_total_size.text
+      @chip_settings.update(spi_total_size: @spi_settings_gui.lie_total_size.text)
+    end
+    Qt::MessageBox.new(Qt::MessageBox::Information, 'Succes', 'SPI parameters saved successfully').exec
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured when saving the SPI parameters. Consult the log for more details').exec
+  end
+end