hardsploit_gui 2.2 → 2.3

data/lib/class/I2C/I2c_export.rb

@@ -1,118 +0,0 @@
-#===================================================
-#  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 I2c_export < Qt::Widget
-  slots 'export()'
-  slots 'select_export_file()'
-
-  def initialize(api, chip)
-    super()
-    @i2c_export_gui = Ui_Generic_export.new
-    centerWindow(self)
-    @i2c_export_gui.setupUi(self)
-    @i2c_export_gui.lbl_chip.setText(chip.chip_reference)
-    inputRestrict(@i2c_export_gui.lie_start, 0)
-    inputRestrict(@i2c_export_gui.lie_stop, 0)
-    @api = api
-    @chip_settings = I2C.find_by(i2c_chip: chip.chip_id)
-  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')
-      @i2c_export_gui.btn_export.setEnabled(true)
-      @i2c_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, 'I2C')
-    $pgb = Progress_bar.new("I²C: Exporting...")
-    $pgb.show
-    if sender.objectName == 'btn_full_export'
-      @api.i2c_Generic_Dump(@chip_settings.i2c_frequency, @chip_settings.i2c_address_w.to_i(16), 0, @chip_settings.i2c_total_size - 1, @chip_settings.i2c_total_size)
-      close_file
-      control_export_result('full', @chip_settings.i2c_total_size - 1)
-    else
-      @api.i2c_Generic_Dump(@chip_settings.i2c_frequency, @chip_settings.i2c_address_w.to_i(16), @i2c_export_gui.lie_start.text.to_i, @i2c_export_gui.lie_stop.text.to_i, @chip_settings.i2c_total_size)
-      close_file
-      control_export_result('partial', @i2c_export_gui.lie_stop.text.to_i)
-    end
-    @i2c_export_gui.btn_export.setEnabled(false)
-    @i2c_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)
-    if type == 'partial'
-      toCompare = ((stop - @i2c_export_gui.lie_start.text.to_i) + 1)
-    else
-      toCompare = @chip_settings.i2c_total_size
-    end
-    file_size = File.size(@filepath)
-    if toCompare != file_size
-      Qt::MessageBox.new(Qt::MessageBox::Critical, 'Error', 'Dump error: Size does not match').exec
-    end
-  end
-
-  def control_export_settings(type)
-    if @chip_settings.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C settings', 'No settings saved for this chip').exec
-      return 0
-    end
-    if @chip_settings.i2c_frequency.nil? || @chip_settings.i2c_address_w.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C settings', 'Write base address or frequency settings missing').exec
-      return 0
-    end
-    if type == 'full'
-      if @chip_settings.i2c_total_size.zero? || @chip_settings.i2c_total_size.nil?
-        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Empty field', 'Full size setting missing or equal 0').exec
-        return 0
-      end
-    else
-      if @i2c_export_gui.lie_start.text.empty? || @i2c_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 @i2c_export_gui.lie_start.text.to_i > @i2c_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 @i2c_export_gui.lie_start.text.to_i > (@chip_settings.i2c_total_size - 1) || @i2c_export_gui.lie_stop.text.to_i > (@chip_settings.i2c_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/I2C/I2c_import.rb

@@ -1,79 +0,0 @@
-#===================================================
-#  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 I2c_import < Qt::Widget
-  slots 'import()'
-  slots 'select_import_file()'
-
-  def initialize(api, chip)
-    super()
-    @i2c_import_gui = Ui_Generic_import.new
-    centerWindow(self)
-    @i2c_import_gui.setupUi(self)
-    @i2c_import_gui.lbl_chip.setText(chip.chip_reference)
-    inputRestrict(@i2c_import_gui.lie_start, 0)
-    @api = api
-    @chip = chip
-    @chip_settings = I2C.find_by(i2c_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?
-      @i2c_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 import
-    return 0 if control_import_settings.zero?
-    start = @i2c_import_gui.lie_start.text.to_i
-    Firmware.new(@api, 'I2C')
-    $pgb = Progress_bar.new("I2C: Importing...")
-    $pgb.show
-    @api.i2c_Generic_Import(@chip_settings.i2c_frequency, @chip_settings.i2c_address_w.to_i(16), start, @chip_settings.i2c_page_size, @chip_settings.i2c_total_size, @filepath, @chip_settings.i2c_write_page_latency/1000.0)
-  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
-    @chip_settings = I2C.find_by(i2c_chip: @chip.chip_id)
-    if @chip_settings.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C setting', 'No settings saved for this chip').exec
-      return 0
-    end
-    if @chip_settings.i2c_total_size.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C setting', 'Total size setting missing').exec
-      return 0
-    end
-    if @chip_settings.i2c_page_size.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C setting', 'Page size setting missing').exec
-      return 0
-    end
-    if @chip_settings.i2c_write_page_latency.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C setting', 'Write page latency setting missing').exec
-      return 0
-    end
-    if @chip_settings.i2c_frequency.nil? || @chip_settings.i2c_address_w.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing I2C setting', 'Write base address or frequency settings missing').exec
-      return 0
-    end
-    if @i2c_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
-    return 1
-  end
-end

data/lib/class/I2C/I2c_settings.rb

@@ -1,129 +0,0 @@
-#===================================================
-#  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_i2c_settings'
-class I2c_settings < Qt::Widget
-  slots 'save_settings()'
-  slots 'bus_scan()'
-
-  def initialize(api, chip)
-    super()
-    @i2c_settings_gui = Ui_I2c_settings.new
-    centerWindow(self)
-    @i2c_settings_gui.setupUi(self)
-    @i2c_settings_gui.lbl_chip.setText(chip.chip_reference)
-    inputRestrict(@i2c_settings_gui.lie_total_size, 0)
-    inputRestrict(@i2c_settings_gui.lie_page_size, 0)
-    inputRestrict(@i2c_settings_gui.lie_write_page_latency, 0)
-    inputRestrict(@i2c_settings_gui.lie_address_r, 3)
-    inputRestrict(@i2c_settings_gui.lie_address_w, 3)
-    @i2c_settings_gui.tbl_bus_scan.resizeColumnsToContents
-    @i2c_settings_gui.tbl_bus_scan.resizeRowsToContents
-    @i2c_settings_gui.tbl_bus_scan.horizontalHeader.stretchLastSection = true
-    @chip_settings = I2C.find_by(i2c_chip: chip.chip_id)
-    @chip = chip
-    @api = api
-    feed_settings_form
-  end
-
-  def save_settings
-    if @chip_settings.nil?
-      create
-    else
-      @chip_settings = I2C.find_by(i2c_chip: @chip.chip_id)
-      update
-    end
-  end
-
-  def feed_settings_form
-    unless @chip_settings.nil?
-      @i2c_settings_gui.lie_address_w.setText(@chip_settings.i2c_address_w)
-      @i2c_settings_gui.lie_address_r.setText(@chip_settings.i2c_address_r)
-      @i2c_settings_gui.cbx_frequency.setCurrentIndex(@i2c_settings_gui.cbx_frequency.findText(@chip_settings.i2c_frequency.to_s))
-      @i2c_settings_gui.lie_write_page_latency.setText(@chip_settings.i2c_write_page_latency.to_s)
-      @i2c_settings_gui.lie_page_size.setText(@chip_settings.i2c_page_size.to_s)
-      @i2c_settings_gui.lie_total_size.setText(@chip_settings.i2c_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 = I2C.create(
-      i2c_address_w: @i2c_settings_gui.lie_address_w.text,
-      i2c_address_r: @i2c_settings_gui.lie_address_r.text,
-      i2c_frequency: @i2c_settings_gui.cbx_frequency.currentText,
-      i2c_write_page_latency: @i2c_settings_gui.lie_write_page_latency.text,
-      i2c_page_size: @i2c_settings_gui.lie_page_size.text,
-      i2c_total_size: @i2c_settings_gui.lie_total_size.text,
-      i2c_chip: @chip.chip_id
-    )
-    Qt::MessageBox.new(Qt::MessageBox::Information, 'Succes', 'I2C 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 creating the I2C parameters. Consult the log for more details').exec
-  end
-
-  def update
-    if @chip_settings.i2c_address_w != @i2c_settings_gui.lie_address_w.text
-      @chip_settings.update(i2c_address_w: @i2c_settings_gui.lie_address_w.text)
-    end
-    if @chip_settings.i2c_address_r != @i2c_settings_gui.lie_address_r.text
-      @chip_settings.update(i2c_address_r: @i2c_settings_gui.lie_address_r.text)
-    end
-    if @chip_settings.i2c_frequency != @i2c_settings_gui.cbx_frequency.currentText.to_i
-      @chip_settings.update(i2c_frequency: @i2c_settings_gui.cbx_frequency.currentText.to_i)
-    end
-    if @chip_settings.i2c_write_page_latency != @i2c_settings_gui.lie_write_page_latency.text
-      @chip_settings.update(i2c_write_page_latency: @i2c_settings_gui.lie_write_page_latency.text)
-    end
-    if @chip_settings.i2c_page_size != @i2c_settings_gui.lie_page_size.text
-      @chip_settings.update(i2c_page_size: @i2c_settings_gui.lie_page_size.text)
-    end
-    if @chip_settings.i2c_total_size != @i2c_settings_gui.lie_total_size.text
-      @chip_settings.update(i2c_total_size: @i2c_settings_gui.lie_total_size.text)
-    end
-    Qt::MessageBox.new(Qt::MessageBox::Information, 'Succes', 'I2C 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 I2C parameters. Consult the log for more details').exec
-  end
-
-  def bus_scan
-    @i2c_settings_gui.tbl_bus_scan.setRowCount(0)
-    Firmware.new(@api, 'I2C')
-    scan_result = @api.i2c_Scan(0)
-    if scan_result.include?(1)
-      scan_result.each_with_index do |v, i|
-        if v == 1
-          @i2c_settings_gui.tbl_bus_scan.insertRow(@i2c_settings_gui.tbl_bus_scan.rowCount)
-          @i2c_settings_gui.tbl_bus_scan.setItem(@i2c_settings_gui.tbl_bus_scan.rowCount - 1, 0, Qt::TableWidgetItem.new(i.to_s(16).upcase))
-          if i % 2 == 0
-            @i2c_settings_gui.tbl_bus_scan.setItem(@i2c_settings_gui.tbl_bus_scan.rowCount - 1, 1, Qt::TableWidgetItem.new('Write'))
-          else
-            @i2c_settings_gui.tbl_bus_scan.setItem(@i2c_settings_gui.tbl_bus_scan.rowCount - 1, 1, Qt::TableWidgetItem.new('Read'))
-          end
-        end
-      end
-      @i2c_settings_gui.tbl_bus_scan.resizeColumnsToContents
-      @i2c_settings_gui.tbl_bus_scan.resizeRowsToContents
-      @i2c_settings_gui.tbl_bus_scan.horizontalHeader.stretchLastSection = true
-      Qt::MessageBox.new(Qt::MessageBox::Information, "Bus Scan", "Bus scan ended correctly: #{@i2c_settings_gui.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 Exception => msg
-    logger = Logger.new($logFilePath)
-    logger.error msg
-    Qt::MessageBox.new(Qt::MessageBox::Critical, "Critical error", "Error occured when scanning I2C. Consult the log for more details").exec
-  end
-end

data/lib/class/PARALLEL/Parallel_export.rb

@@ -1,146 +0,0 @@
-#===================================================
-#  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 Parallel_export < Qt::Widget
-  slots 'export()'
-  slots 'select_export_file()'
-
-  def initialize(api, chip)
-    super()
-    @parallel_export_gui = Ui_Generic_export.new
-    centerWindow(self)
-    @parallel_export_gui.setupUi(self)
-    @parallel_export_gui.lbl_chip.setText(chip.chip_reference)
-    inputRestrict(@parallel_export_gui.lie_start, 0)
-    inputRestrict(@parallel_export_gui.lie_stop, 0)
-    @api = api
-    @chip_settings = Parallel.find_by(parallel_chip: chip.chip_id)
-  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')
-      @parallel_export_gui.btn_export.setEnabled(true)
-      @parallel_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, 'PARALLEL')
-    time = Time.new
-    if sender.objectName == 'btn_full_export'
-      if @chip_settings.parallel_word_size.zero?
-        check_SendAndReceivedData(@api.read_Memory_WithoutMultiplexing(0, @chip_settings.parallel_total_size - 1, true, @chip_settings.parallel_read_latency))
-      else
-        check_SendAndReceivedData(@api.read_Memory_WithoutMultiplexing(0, @chip_settings.parallel_total_size - 1, false, @chip_settings.parallel_read_latency))
-      end
-      close_file
-      control_export_result(@chip_settings.parallel_total_size - 1, time)
-    else
-      if @chip_settings.parallel_word_size.zero?
-        check_SendAndReceivedData(@api.read_Memory_WithoutMultiplexing(@parallel_export_gui.lie_start.text.to_i, @parallel_export_gui.lie_stop.text.to_i, true, @chip_settings.parallel_read_latency))
-      else
-        check_SendAndReceivedData(@api.read_Memory_WithoutMultiplexing(@parallel_export_gui.lie_start.text.to_i, @parallel_export_gui.lie_stop.text.to_i, true, @chip_settings.parallel_read_latency))
-      end
-      close_file
-      control_export_result(@parallel_export_gui.lie_stop.text.to_i, time)
-    end
-    @parallel_export_gui.btn_export.setEnabled(false)
-    @parallel_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(stop, time)
-    time = Time.new - time
-    file_size = File.size("#{@filepath}")
-    # 8 bits test
-    if @chip_settings.parallel_word_size.zero?
-      if (stop - @cw.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
-        Qt::MessageBox.new(Qt::MessageBox::Critical, "Error", "Dump error: Size does not match").exec
-      end
-    else
-      if (stop - @cw.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
-        Qt::MessageBox.new(Qt::MessageBox::Critical, "Error", "Dump error: File size and dump size does not match").exec
-      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)
-    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_read_latency.nil? || @chip_settings.parallel_word_size.nil?
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Missing parallel settings', 'Read latency or word size settings missing').exec
-      return 0
-    end
-    if type == 'full'
-      if @chip_settings.parallel_total_size.zero? || @chip_settings.parallel_total_size.nil?
-        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Empty field', 'Full size setting missing or equal 0').exec
-        return 0
-      end
-    else
-      if @parallel_export_gui.lie_start.text.empty? || @parallel_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 @parallel_export_gui.lie_start.text.to_i > @parallel_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 @parallel_export_gui.lie_start.text.to_i > (@chip_settings.parallel_total_size - 1) || @parallel_export_gui.lie_stop.text.to_i > (@chip_settings.parallel_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
-
-  def check_SendAndReceivedData(value)
-    case value
-      when HardsploitAPI::USB_STATE::PACKET_IS_TOO_LARGE
-        Qt::MessageBox.new(Qt::MessageBox::Critical, "Critical error", "PACKET_IS_TOO_LARGE max: #{HardsploitAPI::USB::USB_TRAME_SIZE}").exec
-      when HardsploitAPI::USB_STATE::ERROR_SEND
-        Qt::MessageBox.new(Qt::MessageBox::Critical, "Critical error", "ERROR_SEND").exec
-      when HardsploitAPI::USB_STATE::BUSY
-        Qt::MessageBox.new(Qt::MessageBox::Warning, "BUSY", "Device busy").exec
-      else
-        return value
-    end
-  end
-end