hardsploit_gui 2.3 → 2.4.0

data/lib/class/Export.rb

@@ -1,140 +1,140 @@
-#===================================================
-#  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_export'
-class Export < Qt::Widget
-  slots 'export()'
-  slots 'export_file()'
-
-  def initialize(chip_list)
-    super()
-    @export_ui = Ui_Export.new
-    centerWindow(self)
-    @export_ui.setupUi(self)
-    @chip_list = chip_list
-    @export_ui.rbn_cmds.setEnabled(false) unless @chip_list.count == 1
-  end
-
-  def export_file
-    @filepath = Qt::FileDialog.getSaveFileName(self, tr('Select a file'), '/')
-  end
-
-  def export
-    unless @filepath.nil?
-      @file = File.open("#{@filepath}.json", 'w')
-      unless @file.nil?
-        @file.write("{\n")
-        @chip_list.each_with_index do |current_chip, index|
-          @file.write("\t\"#{index}\": {\n")
-          @chip = Chip.find_by(reference: current_chip.text)
-          if @export_ui.rbn_both.isChecked
-              export_chip
-              export_commands
-          elsif @export_ui.rbn_comp.isChecked
-            export_chip
-          else
-            export_commands
-          end
-          if index.next < @chip_list.count
-            @file.write("\t},\n")
-          else
-            @file.write("\t}\n")
-          end
-        end
-        @file.write("}")
-        @file.close
-      end
-      Qt::MessageBox.new(Qt::MessageBox::Information, 'Export status', 'Export finished').exec
-    else
-      Qt::MessageBox.new(Qt::MessageBox::Warning, 'File missing', 'Choose a file before exporting').exec
-    end
-  end
-
-  def export_chip
-    @file.write("\t\t\"chip\":{\n")
-    write_commons
-    write_settings
-    write_pins
-    if @export_ui.rbn_both.isChecked
-      @file.write("\t\t},\n")
-    else
-      @file.write("\t\t}\n")
-    end
-  end
-
-  def write_commons
-    @file.write("\t\t\t\t\"package\": #{@chip.package.to_json},\n")
-    @file.write("\t\t\t\t\"chip_type\": #{@chip.chip_type.to_json},\n")
-    @file.write("\t\t\t\t\"manufacturer\": #{@chip.manufacturer.to_json},\n")
-    @file.write("\t\t\t\t\"characteristics\": #{@chip.to_json},\n")
-  end
-
-  def write_settings
-    @file.write("\t\t\t\t\"settings\":{\n")
-    if @chip.parallel_setting.nil?
-      @file.write("\t\t\t\t\t\t\"parralel\": {},\n")
-    else
-      @file.write("\t\t\t\t\t\t\"parralel\": #{@chip.parallel_setting.to_json},\n")
-    end
-    if @chip.spi_setting.nil?
-      @file.write("\t\t\t\t\t\t\"spi\": {},\n")
-    else
-      @file.write("\t\t\t\t\t\t\"spi\":#{@chip.spi_setting.to_json},\n")
-    end
-    if @chip.i2c_setting.nil?
-      @file.write("\t\t\t\t\t\t\"i2c\": {},\n")
-    else
-      @file.write("\t\t\t\t\t\t\"i2c\": #{@chip.i2c_setting.to_json},\n")
-    end
-    if @chip.swd_setting.nil?
-      @file.write("\t\t\t\t\t\t\"swd\": {},\n")
-    else
-      @file.write("\t\t\t\t\t\t\"swd\": #{@chip.swd_setting.to_json},\n")
-    end
-    if @chip.uart_setting.nil?
-      @file.write("\t\t\t\t\t\t\"uart\": {}\n")
-    else
-      @file.write("\t\t\t\t\t\t\"uart\": #{@chip.uart_setting.to_json}\n")
-    end
-    @file.write("\t\t\t\t},\n")
-  end
-
-  def write_pins
-    @file.write("\t\t\t\t\"pins\":[\n")
-    @chip.pins.each_with_index do |pin, index|
-      if index + 1 < @chip.pins.count
-        @file.write("\t\t\t\t\t\t#{pin.to_json},\n")
-      else
-        @file.write("\t\t\t\t\t\t#{pin.to_json}\n")
-      end
-    end
-    @file.write("\t\t\t\t]\n")
-  end
-
-  def export_commands
-    @file.write("\t\t\"commands\":{\n")
-      @chip.commands.each_with_index do |cmd, index|
-        @file.write("\t\t\t\t\"#{index}\":{\n")
-        @file.write("\t\t\t\t\t\t\"characteristics\": #{cmd.to_json},\n")
-        @file.write("\t\t\t\t\t\t\"bytes\":[\n")
-        cmd.bytes.each_with_index do |bytes, index|
-          if index + 1 < cmd.bytes.count
-            @file.write("\t\t\t\t\t\t\t\t#{bytes.to_json},\n")
-          else
-            @file.write("\t\t\t\t\t\t\t\t#{bytes.to_json}\n")
-          end
-        end
-        @file.write("\t\t\t\t\t\t]\n")
-        if index.next < @chip.commands.count
-          @file.write("\t\t\t\t},\n")
-        else
-          @file.write("\t\t\t\t}\n")
-        end
-      end
-    @file.write("\t\t}\n")
-  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_export'
+class Export < Qt::Widget
+  slots 'export()'
+  slots 'export_file()'
+
+  def initialize(chip_list)
+    super()
+    @export_ui = Ui_Export.new
+    centerWindow(self)
+    @export_ui.setupUi(self)
+    @chip_list = chip_list
+    @export_ui.rbn_cmds.setEnabled(false) unless @chip_list.count == 1
+  end
+
+  def export_file
+    @filepath = Qt::FileDialog.getSaveFileName(self, tr('Select a file'), '/')
+  end
+
+  def export
+    unless @filepath.nil?
+      @file = File.open("#{@filepath}.json", 'w')
+      unless @file.nil?
+        @file.write("{\n")
+        @chip_list.each_with_index do |current_chip, index|
+          @file.write("\t\"#{index}\": {\n")
+          @chip = Chip.find_by(reference: current_chip.text)
+          if @export_ui.rbn_both.isChecked
+              export_chip
+              export_commands
+          elsif @export_ui.rbn_comp.isChecked
+            export_chip
+          else
+            export_commands
+          end
+          if index.next < @chip_list.count
+            @file.write("\t},\n")
+          else
+            @file.write("\t}\n")
+          end
+        end
+        @file.write("}")
+        @file.close
+      end
+      Qt::MessageBox.new(Qt::MessageBox::Information, 'Export status', 'Export finished').exec
+    else
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'File missing', 'Choose a file before exporting').exec
+    end
+  end
+
+  def export_chip
+    @file.write("\t\t\"chip\":{\n")
+    write_commons
+    write_settings
+    write_pins
+    if @export_ui.rbn_both.isChecked
+      @file.write("\t\t},\n")
+    else
+      @file.write("\t\t}\n")
+    end
+  end
+
+  def write_commons
+    @file.write("\t\t\t\t\"package\": #{@chip.package.to_json},\n")
+    @file.write("\t\t\t\t\"chip_type\": #{@chip.chip_type.to_json},\n")
+    @file.write("\t\t\t\t\"manufacturer\": #{@chip.manufacturer.to_json},\n")
+    @file.write("\t\t\t\t\"characteristics\": #{@chip.to_json},\n")
+  end
+
+  def write_settings
+    @file.write("\t\t\t\t\"settings\":{\n")
+    if @chip.parallel_setting.nil?
+      @file.write("\t\t\t\t\t\t\"parralel\": {},\n")
+    else
+      @file.write("\t\t\t\t\t\t\"parralel\": #{@chip.parallel_setting.to_json},\n")
+    end
+    if @chip.spi_setting.nil?
+      @file.write("\t\t\t\t\t\t\"spi\": {},\n")
+    else
+      @file.write("\t\t\t\t\t\t\"spi\":#{@chip.spi_setting.to_json},\n")
+    end
+    if @chip.i2c_setting.nil?
+      @file.write("\t\t\t\t\t\t\"i2c\": {},\n")
+    else
+      @file.write("\t\t\t\t\t\t\"i2c\": #{@chip.i2c_setting.to_json},\n")
+    end
+    if @chip.swd_setting.nil?
+      @file.write("\t\t\t\t\t\t\"swd\": {},\n")
+    else
+      @file.write("\t\t\t\t\t\t\"swd\": #{@chip.swd_setting.to_json},\n")
+    end
+    if @chip.uart_setting.nil?
+      @file.write("\t\t\t\t\t\t\"uart\": {}\n")
+    else
+      @file.write("\t\t\t\t\t\t\"uart\": #{@chip.uart_setting.to_json}\n")
+    end
+    @file.write("\t\t\t\t},\n")
+  end
+
+  def write_pins
+    @file.write("\t\t\t\t\"pins\":[\n")
+    @chip.pins.each_with_index do |pin, index|
+      if index + 1 < @chip.pins.count
+        @file.write("\t\t\t\t\t\t#{pin.to_json},\n")
+      else
+        @file.write("\t\t\t\t\t\t#{pin.to_json}\n")
+      end
+    end
+    @file.write("\t\t\t\t]\n")
+  end
+
+  def export_commands
+    @file.write("\t\t\"commands\":{\n")
+      @chip.commands.each_with_index do |cmd, index|
+        @file.write("\t\t\t\t\"#{index}\":{\n")
+        @file.write("\t\t\t\t\t\t\"characteristics\": #{cmd.to_json},\n")
+        @file.write("\t\t\t\t\t\t\"bytes\":[\n")
+        cmd.bytes.each_with_index do |bytes, index|
+          if index + 1 < cmd.bytes.count
+            @file.write("\t\t\t\t\t\t\t\t#{bytes.to_json},\n")
+          else
+            @file.write("\t\t\t\t\t\t\t\t#{bytes.to_json}\n")
+          end
+        end
+        @file.write("\t\t\t\t\t\t]\n")
+        if index.next < @chip.commands.count
+          @file.write("\t\t\t\t},\n")
+        else
+          @file.write("\t\t\t\t}\n")
+        end
+      end
+    @file.write("\t\t}\n")
+  end
+end

data/lib/class/Export_manager.rb

@@ -1,124 +1,124 @@
-#===================================================
-#  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 'csv'
-class Export_manager < Qt::Widget
-  slots 'save_result()'
-
-  def initialize(bus, result, spiDataSended = [])
-    super()
-    @view = Ui_Export_manager.new
-    centerWindow(self)
-    @view.setupUi(self)
-    @bus = bus
-    # Check the bus type to adapt the table column
-    if bus == 'SPI'
-      create_result_table('SPI')
-      @view.cbx_export.hide
-      @view.tbl_result.setRowCount(result.length)
-      result.length.times do |i|
-        @view.tbl_result.setItem(i, 0, Qt::TableWidgetItem.new(spiDataSended[i].to_s))
-        @view.tbl_result.setItem(i, 1, Qt::TableWidgetItem.new(result[i].to_s))
-      end
-      @view.tbl_result.resizeColumnsToContents
-      @view.tbl_result.resizeRowsToContents
-      @view.tbl_result.horizontalHeader.stretchLastSection = true
-    else
-      create_result_table('I2C')
-      @view.tbl_result.setRowCount(result.length / 2)
-      (0..result.length - 1).step(2).each_with_index do |i, v|
-        case result[i]
-        when 0
-          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Write'))
-          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('ACK'))
-          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
-        when 1
-          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Read'))
-          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('ACK'))
-          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
-        when 2
-          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Write'))
-          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('NACK'))
-          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
-        else
-          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Write'))
-          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('NACK'))
-          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
-        end
-      end
-      @view.tbl_result.resizeColumnsToContents
-      @view.tbl_result.resizeRowsToContents
-      @view.tbl_result.horizontalHeader.stretchLastSection = true
-    end
-  end
-
-  def create_result_table(bus)
-    if bus == 'SPI'
-      @view.tbl_result.insertColumn(0)
-      @view.tbl_result.setHorizontalHeaderItem(0, Qt::TableWidgetItem.new('Data send'))
-      @view.tbl_result.insertColumn(1)
-      @view.tbl_result.setHorizontalHeaderItem(1, Qt::TableWidgetItem.new('Data receive'))
-    else
-      @view.tbl_result.insertColumn(0)
-      @view.tbl_result.setHorizontalHeaderItem(0, Qt::TableWidgetItem.new('R/W'))
-      @view.tbl_result.insertColumn(1)
-      @view.tbl_result.setHorizontalHeaderItem(1, Qt::TableWidgetItem.new('(N)ACK)'))
-      @view.tbl_result.insertColumn(2)
-      @view.tbl_result.setHorizontalHeaderItem(2, Qt::TableWidgetItem.new('DATA'))
-    end
-  end
-
-  def save_result
-    result_file = Qt::FileDialog.getSaveFileName(self, tr('Create a file'), '/', tr('All files (*)'))
-    unless result_file.nil?
-      if @bus == 'I2C'
-        if @view.cbx_export.currentIndex == 0
-          save_i2c_csv(result_file)
-        else
-          save_i2c(result_file)
-        end
-      else
-        save_spi(result_file)
-      end
-    end
-  rescue Exception => msg
-    logger = Logger.new($logFilePath)
-    logger.error msg
-    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while saving the results. Consult the logs for more details').exec
-  end
-
-  def save_i2c_csv(result_file)
-    CSV.open("#{result_file}.csv", 'wb', :col_sep => ';') do |csv|
-      csv << ['R/W', '(N)ACK', 'DATA']
-      @view.tbl_result.rowCount.times do |i|
-        csv << [@view.tbl_result.item(i, 0).text, @view.tbl_result.item(i, 1).text, @view.tbl_result.item(i, 2).text]
-      end
-    end
-  end
-
-  def save_i2c(result_file)
-    file = File.open("#{result_file}", 'w')
-    result = []
-    @view.tbl_result.rowCount.times do |i|
-      if @view.tbl_result.item(i, 0).text == 'Read' && @view.tbl_result.item(i, 1).text == 'ACK'
-        result.push(@view.tbl_result.item(i, 2).text.to_i)
-      end
-    end
-    file.write(result.pack('C*'))
-    file.close
-  end
-
-  def save_spi(result_file)
-    file = File.open("#{result_file}", 'w')
-    result = []
-    @view.tbl_result.rowCount.times do |i|
-      result.push(@view.tbl_result.item(i, 1).text.to_i)
-    end
-    file.write(result.pack('C*'))
-    file.close
-  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 'csv'
+class Export_manager < Qt::Widget
+  slots 'save_result()'
+
+  def initialize(bus, result, spiDataSended = [])
+    super()
+    @view = Ui_Export_manager.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @bus = bus
+    # Check the bus type to adapt the table column
+    if bus == 'SPI'
+      create_result_table('SPI')
+      @view.cbx_export.hide
+      @view.tbl_result.setRowCount(result.length)
+      result.length.times do |i|
+        @view.tbl_result.setItem(i, 0, Qt::TableWidgetItem.new(spiDataSended[i].to_s))
+        @view.tbl_result.setItem(i, 1, Qt::TableWidgetItem.new(result[i].to_s))
+      end
+      @view.tbl_result.resizeColumnsToContents
+      @view.tbl_result.resizeRowsToContents
+      @view.tbl_result.horizontalHeader.stretchLastSection = true
+    else
+      create_result_table('I2C')
+      @view.tbl_result.setRowCount(result.length / 2)
+      (0..result.length - 1).step(2).each_with_index do |i, v|
+        case result[i]
+        when 0
+          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Write'))
+          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('ACK'))
+          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
+        when 1
+          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Read'))
+          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('ACK'))
+          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
+        when 2
+          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Write'))
+          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('NACK'))
+          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
+        else
+          @view.tbl_result.setItem(v, 0, Qt::TableWidgetItem.new('Write'))
+          @view.tbl_result.setItem(v, 1, Qt::TableWidgetItem.new('NACK'))
+          @view.tbl_result.setItem(v, 2, Qt::TableWidgetItem.new(result[i + 1].to_s))
+        end
+      end
+      @view.tbl_result.resizeColumnsToContents
+      @view.tbl_result.resizeRowsToContents
+      @view.tbl_result.horizontalHeader.stretchLastSection = true
+    end
+  end
+
+  def create_result_table(bus)
+    if bus == 'SPI'
+      @view.tbl_result.insertColumn(0)
+      @view.tbl_result.setHorizontalHeaderItem(0, Qt::TableWidgetItem.new('Data send'))
+      @view.tbl_result.insertColumn(1)
+      @view.tbl_result.setHorizontalHeaderItem(1, Qt::TableWidgetItem.new('Data receive'))
+    else
+      @view.tbl_result.insertColumn(0)
+      @view.tbl_result.setHorizontalHeaderItem(0, Qt::TableWidgetItem.new('R/W'))
+      @view.tbl_result.insertColumn(1)
+      @view.tbl_result.setHorizontalHeaderItem(1, Qt::TableWidgetItem.new('(N)ACK)'))
+      @view.tbl_result.insertColumn(2)
+      @view.tbl_result.setHorizontalHeaderItem(2, Qt::TableWidgetItem.new('DATA'))
+    end
+  end
+
+  def save_result
+    result_file = Qt::FileDialog.getSaveFileName(self, tr('Create a file'), '/', tr('All files (*)'))
+    unless result_file.nil?
+      if @bus == 'I2C'
+        if @view.cbx_export.currentIndex == 0
+          save_i2c_csv(result_file)
+        else
+          save_i2c(result_file)
+        end
+      else
+        save_spi(result_file)
+      end
+    end
+  rescue Exception => msg
+    logger = Logger.new($logFilePath)
+    logger.error msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while saving the results. Consult the logs for more details').exec
+  end
+
+  def save_i2c_csv(result_file)
+    CSV.open("#{result_file}.csv", 'wb', :col_sep => ';') do |csv|
+      csv << ['R/W', '(N)ACK', 'DATA']
+      @view.tbl_result.rowCount.times do |i|
+        csv << [@view.tbl_result.item(i, 0).text, @view.tbl_result.item(i, 1).text, @view.tbl_result.item(i, 2).text]
+      end
+    end
+  end
+
+  def save_i2c(result_file)
+    file = File.open("#{result_file}", 'w')
+    result = []
+    @view.tbl_result.rowCount.times do |i|
+      if @view.tbl_result.item(i, 0).text == 'Read' && @view.tbl_result.item(i, 1).text == 'ACK'
+        result.push(@view.tbl_result.item(i, 2).text.to_i)
+      end
+    end
+    file.write(result.pack('C*'))
+    file.close
+  end
+
+  def save_spi(result_file)
+    file = File.open("#{result_file}", 'w')
+    result = []
+    @view.tbl_result.rowCount.times do |i|
+      result.push(@view.tbl_result.item(i, 1).text.to_i)
+    end
+    file.write(result.pack('C*'))
+    file.close
+  end
+end