hardsploit_gui 2.2 → 2.3

data/lib/class/swd/Swd_settings.rb

@@ -0,0 +1,76 @@
+#===================================================
+#  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_swd_settings'
+class Swd_settings < Qt::Widget
+  slots 'save_settings()'
+
+  def initialize(chip)
+    super()
+    @view = Ui_Swd_settings.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    inputRestrict(@view.lie_cpu_id_address,     4)
+    inputRestrict(@view.lie_device_id_address,  4)
+    inputRestrict(@view.lie_size_address,       4)
+    inputRestrict(@view.lie_start_address,      4)
+    @chip = chip
+    feed_settings_form unless chip.swd_setting.nil?
+  end
+
+  def save_settings
+    @chip.swd_setting.nil? ? create : update
+    @chip.reload
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def feed_settings_form
+    @view.lie_cpu_id_address.setText(@chip.swd_setting.cpu_id_address)
+    @view.lie_device_id_address.setText(@chip.swd_setting.device_id_address)
+    @view.lie_size_address.setText(@chip.swd_setting.memory_size_address)
+    @view.lie_start_address.setText(@chip.swd_setting.memory_start_address)
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def create
+    chip_settings = SwdSetting.create(
+      cpu_id_address:       @view.lie_cpu_id_address.text,
+      device_id_address:    @view.lie_device_id_address.text,
+      memory_size_address:  @view.lie_size_address.text,
+      memory_start_address: @view.lie_start_address.text,
+      chip_id:              @chip.id
+    )
+    unless check_for_errors(chip_settings)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'SWD settings saved'
+      ).exec
+      close
+    end
+  end
+
+  def update
+    @chip.swd_setting.update(
+      cpu_id_address:       @view.lie_cpu_id_address.text,
+      device_id_address:    @view.lie_device_id_address.text,
+      memory_size_address:  @view.lie_size_address.text,
+      memory_start_address: @view.lie_start_address.text,
+    )
+    unless check_for_errors(@chip.swd_setting)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'SWD settings updated'
+      ).exec
+      close
+    end
+  end
+end

data/lib/class/uart/Uart_baudrate.rb

@@ -0,0 +1,62 @@
+#===================================================
+#  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_uart_baudrate'
+require_relative '../../HardsploitAPI/Modules/UART/HardsploitAPI_UART'
+
+class Uart_baudrate < Qt::Widget
+  slots 'start_detect()'
+  slots 'stop_detect()'
+  slots 'copy()'
+
+  def initialize(parent_view)
+    super()
+    Firmware.new('UART')
+    @view = Ui_Uart_baudrate.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @parent_view = parent_view
+  end
+
+  def start_detect
+    @view.btn_start.setEnabled(false)
+    $app.processEvents
+    @uart = HardsploitAPI_UART.new(
+      baud_rate:        115200,
+      word_width:       8,
+      use_parity_bit:   0,
+      parity_type:      0,
+      nb_stop_bits:     1,
+      idle_line_level:  1
+    )
+    @uart.enableMeasureBaudRate
+    @view.btn_stop.setEnabled(true)
+  end
+
+  def stop_detect
+    @view.btn_stop.setEnabled(false)
+    @baudrate = @uart.measureBaudRate
+    @uart.disableMeasureBaudRate
+    @view.btn_start.setEnabled(true)
+    unless @baudrate.zero?
+      @view.lbl_baudrate.setText("Baud rate detected: #{@baudrate} Hz")
+    else
+      @view.lbl_baudrate.setText("Baud rate detected: None.")
+    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
+
+  def copy
+    @parent_view.lie_baud_rate.setText(@baudrate.to_s)
+    close
+  end
+end

data/lib/class/uart/Uart_console.rb

@@ -0,0 +1,115 @@
+#===================================================
+#  Hardsploit API - 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_uart_console'
+require_relative '../../gui/gui_uart_settings'
+require_relative '../../HardsploitAPI/Modules/UART/HardsploitAPI_UART'
+class Uart_console < Qt::Widget
+  slots 'send()'
+  slots 'update()'
+  slots 'connect()'
+  slots 'disconnect()'
+  slots 'clear_console()'
+  slots 'open_settings()'
+
+  def initialize(chip)
+    super()
+    Firmware.new('UART')
+    @view = Ui_Uart_console.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    @chip = chip
+    @data = ''
+  end
+
+  def closeEvent(event)
+    @timer.killTimer(@timer.timerId) unless @timer.nil?
+  end
+
+  def keyPressEvent(event)
+    send if event.key() == Qt::Key_Enter or event.key() == Qt::Key_Return
+  end
+
+  def connect
+    @uart = HardsploitAPI_UART.new(
+      baud_rate:        @chip.uart_setting.baud_rate,
+      word_width:       @chip.uart_setting.word_size,
+      use_parity_bit:   @chip.uart_setting.parity_bit,
+      parity_type:      @chip.uart_setting.parity_type,
+      nb_stop_bits:     @chip.uart_setting.stop_bits_nbr,
+      idle_line_level:  @chip.uart_setting.idle_line
+    )
+    @timer = Qt::Timer.new
+    Qt::Object.connect(@timer, SIGNAL('timeout()'), self, SLOT('update()'))
+    @timer.start(100)
+    @data << 'Listening and ready to go !' + 0x0d.chr
+    @view.console.setPlainText(@data)
+    scroll_down
+    sender.setEnabled(false)
+    @view.btn_disconnect.setEnabled(true)
+  end
+
+  def disconnect
+    @timer.killTimer(@timer.timerId)
+    @data << 'Disconnected' + 0x0d.chr
+    @view.console.setPlainText(@data)
+    scroll_down
+    sender.setEnabled(false)
+    @view.btn_connect.setEnabled(true)
+  end
+
+  def send
+    return false if @uart.nil? or @timer.nil?
+    return ErrorMsg.new.ascii_only unless @view.lie_cmd.text.ascii_only?
+    packet = []
+    @view.lie_cmd.text.each_byte do |x|
+      packet.push x
+    end
+    packet.push 13 if @chip.uart_setting.return_type == 0
+    packet.push 10 if @chip.uart_setting.return_type == 1
+    if @chip.uart_setting.return_type == 2
+      packet.push 13
+      packet.push 10
+    end
+    @view.lie_cmd.clear
+		@uart.write(payload: packet)
+  #Rescue
+  end
+
+  def update
+    received = @uart.sendAndReceived.collect{|i| i.chr}
+    unless received.empty?
+      @data << "#{received.join('')}"
+      @view.console.setPlainText(@data)
+    end
+    scroll_down
+  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
+
+  def clear_console
+    @view.console.clear
+    @data = ''
+  end
+
+  def open_settings
+    settingsUART = Uart_settings.new(@chip)
+    settingsUART.setWindowModality(Qt::ApplicationModal)
+    settingsUART.show
+  end
+
+  def scroll_down
+    sb = Qt::ScrollBar.new
+    sb = @view.console.verticalScrollBar
+    sb.setValue(sb.maximum())
+  end
+end

data/lib/class/uart/Uart_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_uart_settings'
+class Uart_settings < Qt::Widget
+  slots 'save_settings()'
+  slots 'autodetect()'
+
+  def initialize(chip)
+    super()
+    @view = Ui_Uart_settings.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @view.lbl_chip.setText(chip.reference)
+    inputRestrict(@view.lie_baud_rate,     0)
+    inputRestrict(@view.lie_idle_line_lvl, 0)
+    inputRestrict(@view.lie_parity_bit,    0)
+    inputRestrict(@view.lie_parity_type,   0)
+    inputRestrict(@view.lie_stop_bits_nbr, 0)
+    inputRestrict(@view.lie_word_size,     0)
+    @chip = chip
+    feed_settings_form unless chip.uart_setting.nil?
+  end
+
+  def save_settings
+    @chip.uart_setting.nil? ? create : update
+    @chip.reload
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def feed_settings_form
+    @view.lie_baud_rate.setText(@chip.uart_setting.baud_rate.to_s)
+    @view.lie_idle_line_lvl.setText(@chip.uart_setting.idle_line.to_s)
+    @view.lie_parity_bit.setText(@chip.uart_setting.parity_bit.to_s)
+    @view.lie_parity_type.setText(@chip.uart_setting.parity_type.to_s)
+    @view.lie_stop_bits_nbr.setText(@chip.uart_setting.stop_bits_nbr.to_s)
+    @view.lie_word_size.setText(@chip.uart_setting.word_size.to_s)
+    @view.rbn_cr.setChecked(true) if @chip.uart_setting.return_type == 0
+    @view.rbn_lf.setChecked(true) if @chip.uart_setting.return_type == 1
+    @view.rbn_both.setChecked(true) if @chip.uart_setting.return_type == 2
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def create
+    chip_settings = UartSetting.create(
+      baud_rate:     @view.lie_baud_rate.text.to_i,
+      idle_line:     @view.lie_idle_line_lvl.text.to_i,
+      parity_bit:    @view.lie_parity_bit.text.to_i,
+      parity_type:   @view.lie_parity_type.text.to_i,
+      stop_bits_nbr: @view.lie_stop_bits_nbr.text.to_i,
+      word_size:     @view.lie_word_size.text.to_i,
+      return_type:   get_return_type,
+      chip_id:       @chip.id
+    )
+    unless check_for_errors(chip_settings)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'UART settings saved'
+      ).exec
+      close
+    end
+  end
+
+  def update
+    @chip.uart_setting.update(
+      baud_rate:     @view.lie_baud_rate.text.to_i,
+      idle_line:     @view.lie_idle_line_lvl.text.to_i,
+      parity_bit:    @view.lie_parity_bit.text.to_i,
+      parity_type:   @view.lie_parity_type.text.to_i,
+      stop_bits_nbr: @view.lie_stop_bits_nbr.text.to_i,
+      word_size:     @view.lie_word_size.text.to_i,
+      return_type:   get_return_type
+    )
+    unless check_for_errors(@chip.uart_setting)
+      Qt::MessageBox.new(
+        Qt::MessageBox::Information,
+        'Succes',
+        'UART settings updated'
+      ).exec
+      close
+    end
+  end
+
+  def get_return_type
+    return 0 if @view.rbn_cr.isChecked
+    return 1 if @view.rbn_lf.isChecked
+    return 2 if @view.rbn_both.isChecked
+  end
+
+  def autodetect
+    baudUART = Uart_baudrate.new(@view)
+    baudUART.setWindowModality(Qt::ApplicationModal)
+    baudUART.show
+  end
+end

data/lib/db/associations.rb

@@ -14,67 +14,78 @@ ActiveRecord::Base.establish_connection(
 )
 
 class Package < ActiveRecord::Base
-	self.table_name = "HS_PACKAGE"
+	self.table_name = "HS_PACKAGES"
+  validates :package_name, uniqueness: true
 	has_many :chips,
 		foreign_key: "chip_package",
-		:dependent => :delete_all
+		:dependent => :destroy
 end
 
 class Manufacturer < ActiveRecord::Base
-	self.table_name = "HS_MANUFACTURER"
+	self.table_name = "HS_MANUFACTURERS"
 	validates :manufacturer_name, uniqueness: true
 	has_many :chips,
 		foreign_key: "chip_manufacturer",
-		:dependent => :delete_all
+		:dependent => :destroy
 end
 
 class CType < ActiveRecord::Base
-	self.table_name = "HS_CHIP_TYPE"
+	self.table_name = "HS_CHIP_TYPES"
 	validates :cType_name, uniqueness: true
 	has_many :chips,
 		foreign_key: "chip_type",
-		:dependent => :delete_all
+		:dependent => :destroy
 end
 
 class Chip < ActiveRecord::Base
-	self.table_name = "HS_CHIP"
-	belongs_to	:package
+	self.table_name = "HS_CHIPS"
+	belongs_to	:package,
+    foreign_key: "chip_package"
 	belongs_to	:manufacturer,
 		foreign_key: "chip_manufacturer"
 	belongs_to	:ctype,
 		class_name:		"CType",
 		foreign_key:	"chip_type"
-	has_many	:pin,
+	has_many	:pins,
 		foreign_key:	"pin_chip",
-		:dependent => :delete_all
-	has_many	:cmd,
+		:dependent => :destroy
+	has_many	:cmds,
 		foreign_key:	"cmd_chip",
-		:dependent => :delete_all
+		:dependent => :destroy
+  has_many	:parallel,
+    foreign_key:	"parallel_chip",
+		:dependent => :destroy
+  has_many	:spi,
+    foreign_key:	"spi_chip",
+		:dependent => :destroy
+  has_many	:I2C,
+    foreign_key:	"i2c_chip",
+		:dependent => :destroy
 end
 
 class Parallel < ActiveRecord::Base
-	self.table_name = "HS_PARAM_PARA"
+	self.table_name = "HS_PARALLEL_SETTINGS"
 	belongs_to	:chip,
 		class_name:		"Chip",
 		foreign_key:	"parallel_chip"
 end
 
 class Spi < ActiveRecord::Base
-	self.table_name = "HS_PARAM_SPI"
+	self.table_name = "HS_SPI_SETTINGS"
 	belongs_to	:chip,
 		class_name:		"Chip",
 		foreign_key:	"spi_chip"
 end
 
 class I2C < ActiveRecord::Base
-	self.table_name = "HS_PARAM_I2C"
+	self.table_name = "HS_I2C_SETTINGS"
 	belongs_to	:chip,
 		class_name:		"Chip",
 		foreign_key:	"i2c_chip"
 end
 
 class Pin < ActiveRecord::Base
-	self.table_name = "HS_PIN"
+	self.table_name = "HS_PINS"
 	belongs_to	:chip
 	belongs_to	:signall,
 		class_name:		"Signall",
@@ -82,16 +93,16 @@ class Pin < ActiveRecord::Base
 end
 
 class Signall < ActiveRecord::Base
-	self.table_name = "HS_SIGNAL"
-	has_many :use,
+	self.table_name = "HS_SIGNALS"
+	has_many :uses,
      class_name:  "Use",
      foreign_key: "signal_id"
-	has_many :bus, through: :use
-	has_many :pin
+	has_many :buses, through: :uses
+	has_many :pins
 end
 
 class Use < ActiveRecord::Base
-	self.table_name = "HS_USE"
+	self.table_name = "HS_USES"
 	belongs_to :signall,
 		class_name:		"Signall",
 		foreign_key:	"signal_id"
@@ -101,24 +112,26 @@ class Use < ActiveRecord::Base
 end
 
 class Bus < ActiveRecord::Base
-	self.table_name = "HS_BUS"
-	has_many :signall, through: :use
-	has_many :cmd
+	self.table_name = "HS_BUSES"
+	has_many :signalls, through: :uses
+	has_many :cmds
 end
 
 class Cmd < ActiveRecord::Base
-	self.table_name = "HS_CMD"
-	has_many	:byte,
+	self.table_name = "HS_COMMANDS"
+	has_many	:bytes,
 		foreign_key: "byte_cmd",
-		:dependent => :delete_all
+		:dependent => :destroy
 	belongs_to	:bus,
 		class_name:	"Bus",
 		foreign_key: "cmd_bus"
-	belongs_to	:chip
+	belongs_to	:chip,
+  class_name:	"Chip",
+  foreign_key: "cmd_chip"
 end
 
 class Byte < ActiveRecord::Base
-	self.table_name = "HS_CMD_BYTE"
+	self.table_name = "HS_CMD_BYTES"
 	belongs_to	:cmd,
 		class_name:		"Cmd",
 		foreign_key:	"byte_cmd"