hardsploit_gui 2.0

data/lib/class/Command_editor.rb

@@ -0,0 +1,268 @@
+#===================================================
+#  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 '../HardsploitAPI/HardsploitAPI'
+require_relative '../class/Command_table'
+require_relative '../gui/gui_command_editor'
+class Command_editor < Qt::Widget
+  slots 'add_cmd()'
+  slots 'edit_cmd()'
+  slots 'add_row()'
+  slots 'clone_row()'
+  slots 'remove_row()'
+  slots 'check_cell_content(QTableWidgetItem*)'
+
+  def initialize(action, cmd_name, chip, bus_id, parent, api, bus_param={})
+    super()
+    @cmd_editor_gui = Ui_Command_editor.new
+    centerWindow(self)
+    @cmd_editor_gui.setupUi(self)
+    @bus = Bus.find_by(bus_id: bus_id)
+    @chip_id = chip.chip_id
+    @cmd_name = cmd_name
+    @bus_id = bus_id
+    @parent = parent
+    @api = api
+    @cmd_table = Command_table.new(@api, @cmd_editor_gui.tbl_bytes, @bus.bus_name)
+
+    inputRestrict(@cmd_editor_gui.lie_name, 2)
+    inputRestrict(@cmd_editor_gui.lie_description, 2)
+    @cmd_editor_gui.lbl_chip_val.setText(chip.chip_reference)
+
+    @cmd_table.resize_to_content
+    case @bus.bus_name
+    when 'SPI'
+      @cmd_table.build_spi
+    end
+    @cmd_editor_gui.lie_text_2_bytes.hide if @bus.bus_name != 'SPI'
+
+    case action
+    when 0
+      # Add
+      @cmd_editor_gui.lbl_cmd_val.text = '-'
+      Qt::Object.connect(@cmd_editor_gui.btn_validate, SIGNAL('clicked()'), self, SLOT('add_cmd()'))
+    when 1
+      # Temp
+      @cmd_editor_gui.lbl_cmd_val.text = cmd_name
+      feed_editor_form
+      @cmd_editor_gui.lie_name.setText('')
+      Qt::Object.connect(@cmd_editor_gui.btn_validate, SIGNAL('clicked()'), self, SLOT('add_cmd()'))
+    else
+      # Edit
+      @cmd_editor_gui.lbl_cmd_val.text = "#{cmd_name}"
+      feed_editor_form
+      @cmd_editor_gui.btn_validate.setText('Edit')
+      Qt::Object.connect(@cmd_editor_gui.btn_validate, SIGNAL('clicked()'), self, SLOT('edit_cmd()'))
+    end
+    unless bus_param.empty?
+      feed_i2c_cmd_array(bus_param)
+    end
+  rescue Exception => msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while loading the GUI. Consult the logs for more details').exec
+    logger = Logger.new($logFilePath)
+    logger.error msg
+  end
+
+  def check_cell_content(item)
+    case @cmd_editor_gui.tbl_bytes.horizontalHeaderItem(item.column).text
+    when 'Order', 'Repetition'
+      if item.text.to_i < 0
+        item.setData(0, Qt::Variant.new(0))
+        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Wrong data', 'Only positive values are accepted in this cell').exec
+      end
+    when 'Byte (Hexa)'
+      reg = Qt::RegExp.new("^[A-Fa-f0-9]{2}")
+      reg_val = Qt::RegExpValidator.new(reg, self)
+      if reg_val.validate(item.text, item.text.length) == 0
+        item.setText('')
+        Qt::MessageBox.new(Qt::MessageBox::Warning, 'Wrong data', 'Only hexadecimal values are accepted in this cell').exec
+      end
+    when 'Description'
+      unless item.text.nil?
+        reg = Qt::RegExp.new("^[a-zA-Z0-9_@-\s]+$")
+        reg_val = Qt::RegExpValidator.new(reg, self)
+        if reg_val.validate(item.text, item.text.length) == 0
+          item.setText('')
+          Qt::MessageBox.new(Qt::MessageBox::Warning, 'Wrong data', 'Wrong characters in this cell').exec
+        end
+      end
+    end
+  rescue Exception => msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while verifying the command byte column. Consult the logs for more details').exec
+    logger = Logger.new($logFilePath)
+    logger.error msg
+  end
+
+  def is_cmd_size_valid?
+    # SPI
+    if @bus.bus_name == 'SPI'
+      if (@cmd_editor_gui.tbl_bytes.rowCount + @cmd_table.count_total_repetition) > 4000
+			  Qt::MessageBox.new(Qt::MessageBox::Warning, 'SPI command invalid', 'SPI command size is to big (> 4000)').exec
+        return false
+      else
+        return true
+      end
+    # I2C
+    elsif @bus.bus_name == 'I2C'
+      count = 0
+      i = 0
+      while i <= (@cmd_editor_gui.tbl_bytes.rowCount) - 1 do
+        if @cmd_editor_gui.tbl_bytes.item(i, 1).nil?
+          Qt::MessageBox.new(Qt::MessageBox::Critical, 'Command error', 'Payload size invalid or payload size (low) missing').exec
+          return false
+        else
+          lowByte = @cmd_editor_gui.tbl_bytes.item(i, 1).text
+        end
+        if @cmd_editor_gui.tbl_bytes.item(i + 1, 1).nil?
+          Qt::MessageBox.new(Qt::MessageBox::Critical, 'Command error', 'Payload size invalid or payload size (high) missing').exec
+          return false
+        else
+          highByte = @cmd_editor_gui.tbl_bytes.item(i + 1, 1).text
+        end
+        if @cmd_editor_gui.tbl_bytes.item(i + 2, 1).nil?
+          Qt::MessageBox.new(Qt::MessageBox::Critical, 'Command error', 'Payload size invalid or Read / Write address missing').exec
+          return false
+        else
+          commandType = @cmd_editor_gui.tbl_bytes.item(i + 2, 1).text
+        end
+        count = count + (HardsploitAPI.BytesToInt(lowByte.to_i(16), highByte.to_i(16)))
+        if commandType.to_i(16) % 2 == 0 #WRITE
+          i = (i + ((HardsploitAPI.BytesToInt(lowByte.to_i(16), highByte.to_i(16))) + 3))
+        else #READ
+          i = (i + 3)
+        end
+      end
+      if i != @cmd_editor_gui.tbl_bytes.rowCount
+        Qt::MessageBox.new(Qt::MessageBox::Critical, 'Command error', 'I2C command incorrectly formated: The payload size does not match with the row number').exec
+        return false
+      end
+      if count > 2000
+        Qt::MessageBox.new(Qt::MessageBox::Critical, 'I2C command invalid', 'Your payload is too big (> 2000)').exec
+        return false
+      else
+        return true
+      end
+    else
+      return false
+    end
+  rescue Exception => msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while verifying the command size. Consult the logs for more details').exec
+    logger = Logger.new($logFilePath)
+    logger.error msg
+  end
+
+  def feed_i2c_cmd_array(bus_param)
+    if bus_param[:mode] == 'w'
+      @cmd_table.i2c_write_cmd(bus_param[:addr], bus_param[:size].to_i)
+    else
+      @cmd_table.i2c_read_cmd(bus_param[:addr], bus_param[:size].to_i)
+    end
+  end
+
+  def add_cmd
+    if @cmd_table.empty_data_exist? || is_cmd_size_valid? == false
+			return false
+    end
+    if @cmd_editor_gui.lie_name.text.empty? || @cmd_editor_gui.lie_description.text.empty?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Information missing', 'Name or description empty').exec
+      return false
+    end
+    cmd = Cmd.new
+    cmd.cmd_name = @cmd_editor_gui.lie_name.text
+    cmd.cmd_desc = @cmd_editor_gui.lie_description.text
+    cmd.cmd_bus = @bus_id
+    cmd.cmd_chip = @chip_id
+    cmd.save
+    @cmd_editor_gui.tbl_bytes.sortItems(0, Qt::AscendingOrder)
+    @cmd_editor_gui.tbl_bytes.rowCount.times do |i|
+      byte = Byte.new
+      byte.byte_index = i + 1
+      byte.byte_value = @cmd_editor_gui.tbl_bytes.item(i, 1).text
+      if @bus.bus_name == 'SPI'
+        byte.byte_iteration = @cmd_editor_gui.tbl_bytes.item(i, 2).text
+        byte.byte_description = @cmd_editor_gui.tbl_bytes.item(i, 3).text
+      else
+        byte.byte_description = @cmd_editor_gui.tbl_bytes.item(i, 2).text
+      end
+      byte.byte_cmd = Cmd.ids.last
+      byte.save
+    end
+    @parent.feed_cmd_array
+    close
+  rescue Exception => msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while adding the command. Consult the logs for more details').exec
+    logger = Logger.new($logFilePath)
+    logger.error msg
+  end
+
+  def edit_cmd
+    if @cmd_table.empty_data_exist? && !is_cmd_size_valid?
+			return false
+    end
+    if @cmd_editor_gui.lie_name.text.empty? || @cmd_editor_gui.lie_description.text.empty?
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'Information missing', 'Name or description empty').exec
+      return false
+    end
+    cmd = Cmd.find_by(cmd_name: @cmd_name, cmd_chip: @chip_id)
+    if cmd.cmd_name != @cmd_editor_gui.lie_name.text
+      cmd.update(cmd_name: @cmd_editor_gui.lie_name.text)
+    end
+    if cmd.cmd_desc != @cmd_editor_gui.lie_description.text
+      cmd.update(cmd_desc: @cmd_editor_gui.lie_description.text)
+    end
+    @cmd_editor_gui.tbl_bytes.sortItems(0, Qt::AscendingOrder)
+    bytes = Byte.where(byte_cmd: cmd.cmd_id)
+    bytes.destroy_all
+    @cmd_editor_gui.tbl_bytes.rowCount.times do |i|
+      byte = Byte.new
+      byte.byte_index = i + 1
+      byte.byte_value = @cmd_editor_gui.tbl_bytes.item(i, 1).text
+      if @bus.bus_name == 'SPI'
+        byte.byte_iteration = @cmd_editor_gui.tbl_bytes.item(i, 2).text
+        byte.byte_description = @cmd_editor_gui.tbl_bytes.item(i, 3).text
+      else
+        byte.byte_description = @cmd_editor_gui.tbl_bytes.item(i, 2).text
+      end
+      byte.byte_cmd = cmd.cmd_id
+      byte.save
+    end
+    @parent.feed_cmd_array
+    close
+  rescue Exception => msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured while editing the command. Consult the logs for more details').exec
+    logger = Logger.new($logFilePath)
+    logger.error msg
+  end
+
+  def feed_editor_form
+    cmd = Cmd.find_by(cmd_name: @cmd_name)
+    @cmd_editor_gui.lie_name.setText(@cmd_name)
+    @cmd_editor_gui.lie_description.setText(cmd.cmd_desc)
+    @cmd_table.fill_byte_table(cmd.byte)
+  rescue Exception => msg
+    Qt::MessageBox.new(Qt::MessageBox::Critical, 'Critical error', 'Error occured when loading the command. Consult the logs for more details').exec
+    logger = Logger.new($logFilePath)
+    logger.error msg
+  end
+
+  def add_row
+    if @cmd_editor_gui.lie_text_2_bytes.text.empty?
+      @cmd_table.add_row
+    else
+      @cmd_table.add_text_rows(@cmd_editor_gui.lie_text_2_bytes.text)
+      @cmd_editor_gui.lie_text_2_bytes.setText('')
+    end
+  end
+
+  def clone_row
+    @cmd_table.clone_rows
+  end
+
+  def remove_row
+    @cmd_table.remove_rows
+  end
+end

data/lib/class/Command_table.rb

@@ -0,0 +1,239 @@
+#===================================================
+#  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 '../HardsploitAPI/HardsploitAPI'
+
+class Command_table < Qt::Widget
+
+  def initialize(api, cmd_table, bus)
+    super()
+    @cmd_table = cmd_table
+    @bus = bus
+    @api = api
+    cmd_table.insertColumn(0)
+    cmd_table.setHorizontalHeaderItem(0, Qt::TableWidgetItem.new('Order'))
+    cmd_table.insertColumn(1)
+    cmd_table.setHorizontalHeaderItem(1, Qt::TableWidgetItem.new('Byte (Hexa)'))
+    cmd_table.insertColumn(2)
+    cmd_table.setHorizontalHeaderItem(2, Qt::TableWidgetItem.new('Description'))
+  end
+
+  def build_spi
+    @cmd_table.insertColumn(2)
+    @cmd_table.setHorizontalHeaderItem(2, Qt::TableWidgetItem.new('Repetition'))
+  end
+
+  def fill_byte_table(byte_list)
+    byte_list.to_enum.with_index(0).each do |b, i|
+      @cmd_table.insertRow(@cmd_table.rowCount)
+      @cmd_table.setItem(i, 1, Qt::TableWidgetItem.new(b.byte_value))
+      if @bus == 'SPI'
+        item =  Qt::TableWidgetItem.new
+        item.setData(0, Qt::Variant.new(b.byte_iteration))
+        @cmd_table.setItem(i, 2, item)
+        @cmd_table.setItem(i, 3, Qt::TableWidgetItem.new(b.byte_description))
+      else
+        @cmd_table.setItem(i, 2, Qt::TableWidgetItem.new(b.byte_description))
+      end
+      item =  Qt::TableWidgetItem.new
+      item.setData(0, Qt::Variant.new(b.byte_index))
+      @cmd_table.setItem(i, 0, item)
+    end
+  end
+
+  def add_row
+    @cmd_table.insertRow(@cmd_table.rowCount)
+    @cmd_table.setItem(@cmd_table.rowCount - 1, 1, Qt::TableWidgetItem.new('00'))
+    if @bus == 'SPI'
+      item =  Qt::TableWidgetItem.new
+      item.setData(0, Qt::Variant.new(0))
+      @cmd_table.setItem(@cmd_table.rowCount - 1, 2, item)
+      @cmd_table.setItem(@cmd_table.rowCount - 1, 3, Qt::TableWidgetItem.new)
+    else
+      @cmd_table.setItem(@cmd_table.rowCount - 1, 2, Qt::TableWidgetItem.new)
+    end
+    item =  Qt::TableWidgetItem.new
+    item.setData(0, Qt::Variant.new(@cmd_table.rowCount))
+    @cmd_table.setItem(@cmd_table.rowCount - 1, 0, item)
+  end
+
+  def add_text_rows(txt)
+    if txt.ascii_only?
+      txt.each_byte do |x|
+        @cmd_table.insertRow(@cmd_table.rowCount)
+        @cmd_table.setItem(@cmd_table.rowCount - 1, 1, Qt::TableWidgetItem.new(x.to_s(16)))
+        item =  Qt::TableWidgetItem.new
+        item.setData(0, Qt::Variant.new(0))
+        @cmd_table.setItem(@cmd_table.rowCount - 1, 2, item)
+        @cmd_table.setItem(@cmd_table.rowCount - 1, 3, Qt::TableWidgetItem.new(x.chr))
+        item =  Qt::TableWidgetItem.new
+        item.setData(0, Qt::Variant.new(@cmd_table.rowCount))
+        @cmd_table.setItem(@cmd_table.rowCount - 1, 0, item)
+      end
+    else
+      Qt::MessageBox.new(Qt::MessageBox::Warning, 'String error', 'Only ASCII characters can be specified').exec
+    end
+  end
+
+  def clone_rows
+    return 0 if @cmd_table.currentItem.nil?
+    rows_to_delete = []
+    @cmd_table.selectedItems.each do |item|
+      rows_to_delete.push item.row
+    end
+    rows_to_delete.uniq!
+    rows_to_delete.each_with_index do |row, i|
+      @cmd_table.insertRow(@cmd_table.rowCount)
+      unless @cmd_table.item(row, 1).nil?
+        @cmd_table.setItem(@cmd_table.rowCount - 1, 1, Qt::TableWidgetItem.new(@cmd_table.item(row, 1).text))
+      end
+      if @bus == 'SPI'
+        unless @cmd_table.item(row, 2).nil?
+          repetition_item =  Qt::TableWidgetItem.new
+          repetition_item.setData(0, Qt::Variant.new(@cmd_table.item(row, 2).text.to_i))
+          @cmd_table.setItem(@cmd_table.rowCount - 1, 2, repetition_item)
+        end
+        unless @cmd_table.item(row, 3).nil?
+          @cmd_table.setItem(@cmd_table.rowCount - 1, 3, Qt::TableWidgetItem.new(@cmd_table.item(row, 3).text))
+        end
+      else
+        unless @cmd_table.item(row, 2).nil?
+          @cmd_table.setItem(@cmd_table.rowCount - 1, 2, Qt::TableWidgetItem.new(@cmd_table.item(row, 3).text))
+        end
+      end
+      unless @cmd_table.item(row, 0).nil?
+        order_item =  Qt::TableWidgetItem.new
+        order_item.setData(0, Qt::Variant.new(@cmd_table.item(row, 0).text.to_i))
+        @cmd_table.setItem(@cmd_table.rowCount - 1, 0, order_item)
+      end
+    end
+  end
+
+  def remove_rows
+    return 0 if @cmd_table.currentItem.nil?
+    rows_to_delete = []
+    @cmd_table.selectedItems.each do |item|
+      rows_to_delete.push item.row
+    end
+    rows_to_delete.uniq!
+    rows_to_delete.each_with_index do |row, i|
+      @cmd_table.removeRow(row - i)
+    end
+  end
+
+  def i2c_read_cmd(write_address, cmd_size)
+    # Byte array size
+    @cmd_table.setRowCount(3)
+    # Size 1
+    @cmd_table.setItem(0, 1, Qt::TableWidgetItem.new(HardsploitAPI.lowByte(cmd_size).to_s(16).upcase))
+    @cmd_table.setItem(0, 2, Qt::TableWidgetItem.new('Payload size - low'))
+    # Index
+    item =  Qt::TableWidgetItem.new
+    item.setData(0, Qt::Variant.new(1))
+    @cmd_table.setItem(0, 0, item)
+    # Size 2
+    @cmd_table.setItem(1, 1, Qt::TableWidgetItem.new(HardsploitAPI.highByte(cmd_size).to_s(16).upcase))
+    @cmd_table.setItem(1, 2, Qt::TableWidgetItem.new('Payload size - high'))
+    # Index
+    item =  Qt::TableWidgetItem.new
+    item.setData(0, Qt::Variant.new(2))
+    @cmd_table.setItem(1, 0, item)
+    # Address
+    @cmd_table.setItem(2, 1, Qt::TableWidgetItem.new(write_address))
+    @cmd_table.setItem(2, 2, Qt::TableWidgetItem.new('Read address (control byte)'))
+    # Index
+    item =  Qt::TableWidgetItem.new
+    item.setData(0, Qt::Variant.new(3))
+    @cmd_table.setItem(2, 0, item)
+  end
+
+  def i2c_write_cmd(write_address, cmd_size)
+    # Byte array size
+    @cmd_table.setRowCount(3 + cmd_size)
+    # Payload size low
+    # -byte
+    itemLB1 = Qt::TableWidgetItem.new(HardsploitAPI.lowByte(cmd_size).to_s(16).upcase)
+    @cmd_table.setItem(0, 1, itemLB1)
+    # -description
+    itemLB3 = Qt::TableWidgetItem.new('Payload size - low')
+    @cmd_table.setItem(0, 2, itemLB3)
+    # -order
+    itemLB0 =  Qt::TableWidgetItem.new
+    itemLB0.setData(0, Qt::Variant.new(1))
+    @cmd_table.setItem(0, 0, itemLB0)
+    # Payload size high
+    # -byte
+    itemHB1 = Qt::TableWidgetItem.new(HardsploitAPI.highByte(cmd_size).to_s(16).upcase)
+    @cmd_table.setItem(1, 1, itemHB1)
+    # -description
+    itemHB3 = Qt::TableWidgetItem.new('Payload size - high')
+    @cmd_table.setItem(1, 2, itemHB3)
+    # -order
+    itemHB0 = Qt::TableWidgetItem.new
+    itemHB0.setData(0, Qt::Variant.new(2))
+    @cmd_table.setItem(1, 0, itemHB0)
+    # Address
+    # -byte
+    itemA1 = Qt::TableWidgetItem.new(write_address)
+    @cmd_table.setItem(2, 1, itemA1)
+    # -description
+    itemA3 = Qt::TableWidgetItem.new('Write address (control byte)')
+    @cmd_table.setItem(2, 2, itemA3)
+    # -order
+    itemA0 = Qt::TableWidgetItem.new
+    itemA0.setData(0, Qt::Variant.new(3))
+    @cmd_table.setItem(2, 0, itemA0)
+    # Payload bytes
+    for i in 3...(cmd_size + 3) do
+      @cmd_table.setItem(i, 1, Qt::TableWidgetItem.new(''))
+      @cmd_table.setItem(i, 2, Qt::TableWidgetItem.new('Payload byte'))
+      # Index
+      item =  Qt::TableWidgetItem.new
+      item.setData(0, Qt::Variant.new((i + 1)))
+      @cmd_table.setItem(i, 0, item)
+    end
+  end
+
+  def resize_to_content
+    @cmd_table.resizeColumnsToContents
+    @cmd_table.resizeRowsToContents
+    @cmd_table.horizontalHeader.stretchLastSection = true
+  end
+
+  def count_total_repetition
+	  repetitionValue = 0
+	  @cmd_table.rowCount.times do |i|
+			repetitionValue += @cmd_table.item(i, 2).text.to_i
+		end
+		return repetitionValue
+	end
+
+  def empty_data_exist?
+    # ARRAY SIZE
+		if @cmd_table.rowCount == 0
+		  Qt::MessageBox.new(Qt::MessageBox::Warning, 'Bytes array empty', 'The bytes array is empty').exec
+			return true
+    end
+		@cmd_table.rowCount.times do |i|
+			# EMPTY BYTE CELL
+			if @cmd_table.item(i, 1).text == ''
+				@cmd_table.setCurrentItem(@cmd_table.item(i, 1))
+				Qt::MessageBox.new(Qt::MessageBox::Warning, 'Empty byte', 'Empty byte cell detected').exec
+				return true
+			end
+      if @bus == 'SPI'
+  			# EMPTY REPETITION CELL
+  			if @cmd_table.item(i, 2).text == ''
+  				@cmd_table.setCurrentItem(@cmd_table.item(i, 2))
+  				Qt::MessageBox.new(Qt::MessageBox::Warning, 'Empty repetition', 'Empty repetition cell detected').exec
+  				return true
+  			end
+      end
+		end
+		return false
+	end
+end