hardsploit_gui 2.3 → 2.4.0

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

@@ -1,114 +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
+#===================================================
+#  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}/I2c_settings.rb

@@ -1,148 +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
+#===================================================
+#  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