hardsploit_gui 2.3 → 2.4.0

data/lib/class/swd/Swd_settings.rb

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

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

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