hardsploit_gui 2.2 → 2.3

data/lib/class/Console.rb

@@ -5,8 +5,6 @@
 #  License URI: http://www.gnu.org/licenses/gpl.txt
 #===================================================
 require 'date'
-require_relative '../HardsploitAPI/HardsploitAPI'
-require_relative '../class/Command_table'
 require_relative '../gui/gui_command_editor'
 class Console < Qt::TableWidget
 

data/lib/class/ErrorMsg.rb

@@ -0,0 +1,312 @@
+class ErrorMsg
+
+	def hardsploit_not_found
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Hardsploit Connectivity',
+			'Hardsploit not detected. Please check the USB connection'
+		).exec
+		return true
+	end
+
+	def usb_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Hardsploit USB error',
+			'USB error occured'
+		).exec
+	end
+
+	def no_chip_loaded
+		Qt::MessageBox.new(
+			Qt::MessageBox::Information,
+			"Wire chip",
+			"You need to load a chip first"
+		).exec
+		return true
+	end
+
+	def invalid_pin_nbr
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Invalid pin number value',
+			'Pin number needs to be between 4 and 144'
+		).exec
+	end
+
+	def filesize_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Error',
+			'Dump error: The file size does not match with the given parameters'
+		).exec
+	end
+
+	def swd_not_found
+		Qt::MessageBox.new(
+			Qt::MessageBox::Information,
+			'SWD Action',
+			'No return from the SWD'
+		).exec
+	end
+
+	def swd_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'SWD Action',
+			'An error occured while processing the SWD scan'
+		).exec
+		return false
+	end
+
+	def spi_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'SPI Action',
+			'An error occured while processing the SPI command'
+		).exec
+	end
+
+	def spi_mode_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'SPI mode missing',
+			'Mode setting is missing for this chip'
+			).exec
+			return false
+	end
+
+	def spi_cmd_too_long
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'SPI command invalid',
+			'SPI command size is to big (> 4000)'
+			).exec
+			return false
+	end
+
+	def i2c_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'I²C Action',
+			'An error occured while processing the I²C command (I²C wrong speed)'
+		).exec
+		return false
+	end
+
+	def parallel_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Parallel Action',
+			'An error occured while processing the parallel command'
+		).exec
+	end
+# Commands
+
+	def no_cmd_selected
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Missing command',
+			'Select a command in the array first'
+		).exec
+		return true
+	end
+
+	def concat_nbr
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong selection',
+			'Select two commands in the table to concatenate them'
+		).exec
+		return true
+	end
+
+	def concat_disallow
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Concatenation option',
+			'This option can be used only with I2C bus commands'
+		).exec
+		return false
+	end
+
+	def lowbyte_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Command error',
+			'Payload size invalid or payload size (low) missing'
+		).exec
+		return false
+	end
+
+	def highbyte_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Command error',
+			'Payload size invalid or payload size (high) missing'
+		).exec
+		return false
+	end
+
+	def mode_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Command error',
+			'Payload size invalid or Read / Write address missing'
+		).exec
+		return false
+	end
+
+	def size_neq_row_number
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Command error',
+			'The payload size does not match with the row number'
+		).exec
+		return false
+	end
+
+	def i2c_cmd_too_long
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'I2C command invalid',
+			'Your payload is too big (> 2000)'
+		).exec
+		return false
+	end
+
+	def positive_cell_value
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong data',
+			'Only positive values are accepted in this cell'
+		).exec
+	end
+
+	def hexa_cell_value
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong data',
+			'Only hexadecimal values are accepted in this cell'
+		).exec
+	end
+
+	def char_cell_value
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong data',
+			'Wrong characters in this cell'
+		).exec
+	end
+
+	def ascii_only
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'String error',
+			'Only ASCII characters can be specified'
+		).exec
+		return false
+	end
+
+	# Global Settings
+
+	def settings_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Missing settings',
+			'No settings saved for this chip'
+		).exec
+		return false
+	end
+
+	def frequency_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Missing settings',
+			'Frequency setting missing'
+		).exec
+		return false
+	end
+
+	def mode_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Missing settings',
+			'Missing write base address (I²C) or read command (SPI)'
+		).exec
+		return false
+	end
+
+	def full_size_error
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Empty field',
+			'Full size setting missing or equal 0'
+		).exec
+		return false
+	end
+
+	def start_stop_missing
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Empty field',
+			'Start and stop address must be filled'
+		).exec
+		return false
+	end
+
+	def start_neq_stop
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong value',
+			'Start address must not be equal to the stop address'
+		).exec
+		return false
+	end
+
+	def start_inf_to_stop
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong value',
+			'Start address must be inforior to the stop address'
+		).exec
+		return false
+	end
+
+	def inf_to_total_size
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Wrong value',
+			'Start and stop address must be inforior to the chip total size'
+		).exec
+		return false
+	end
+
+	def para_read_latency
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Missing parallel settings',
+			'Read latency setting missing'
+		).exec
+		return false
+	end
+
+	def para_word_size
+		Qt::MessageBox.new(
+			Qt::MessageBox::Warning,
+			'Missing parallel settings',
+			'Word size setting missing'
+		).exec
+		return false
+	end
+
+# Unknown
+	def unknown(msg)
+		Logger.new($logFilePath).error(msg)
+		Qt::MessageBox.new(
+			Qt::MessageBox::Critical,
+			'Hardsploit unknown error',
+			'An unknown error has been detected. Check the log for more details.'
+		).exec
+	end
+end

data/lib/class/Export.rb

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