hardsploit_gui 2.3 → 2.4.0

data/lib/class/Command_editor.rb

@@ -1,216 +1,216 @@
-#===================================================
-#  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_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, bus_param={})
-    super()
-    @view = Ui_Command_editor.new
-    centerWindow(self)
-    @view.setupUi(self)
-    @bus = Bus.find(bus_id)
-    @chip_id = chip.id
-    @cmd_name = cmd_name
-    @bus_id = bus_id
-    @parent = parent
-    @cmd_table = Command_table.new(@view.tbl_bytes, @bus.name)
-
-    inputRestrict(@view.lie_name, 2)
-    inputRestrict(@view.lie_description, 2)
-    @view.lbl_chip_val.setText(chip.reference)
-    @cmd_table.resize_to_content
-    @cmd_table.build_spi if @bus.name == 'SPI'
-    @view.lie_text_2_bytes.hide unless @bus.name == 'SPI'
-
-    case action
-    when 0
-      # Add
-      @view.lbl_cmd_val.setText('-')
-      Qt::Object.connect(@view.btn_validate, SIGNAL('clicked()'), self, SLOT('add_cmd()'))
-    when 1
-      # Template
-      @view.lbl_cmd_val.setText(cmd_name)
-      feed_editor_form
-      @view.lie_name.setText('')
-      Qt::Object.connect(@view.btn_validate, SIGNAL('clicked()'), self, SLOT('add_cmd()'))
-    else
-      # Edit
-      @view.lbl_cmd_val.setText(cmd_name)
-      feed_editor_form
-      @view.btn_validate.setText('Edit')
-      Qt::Object.connect(@view.btn_validate, SIGNAL('clicked()'), self, SLOT('edit_cmd()'))
-    end
-    unless bus_param.empty?
-      feed_i2c_cmd_array(bus_param)
-    end
-  rescue Exception => msg
-    ErrorMsg.new.unknown(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
-    return false if @cmd_table.empty_data_exist? || is_cmd_size_valid? == false
-    @cmd = Command.create(
-      name:        @view.lie_name.text,
-      description: @view.lie_description.text,
-      bus_id:      @bus_id,
-      chip_id:     @chip_id
-    )
-    unless check_for_errors(@cmd)
-      add_bytes
-      @parent.feed_cmd_array
-      close
-    end
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def edit_cmd
-    return false if @cmd_table.empty_data_exist? && !is_cmd_size_valid?
-    @cmd = Command.find_by(name: @cmd_name, chip_id: @chip_id)
-    @cmd.update(
-      name: @view.lie_name.text,
-      description: @view.lie_description.text
-    )
-    @view.tbl_bytes.sortItems(0, Qt::AscendingOrder)
-    unless check_for_errors(@cmd)
-      Byte.where(command_id: @cmd.id).destroy_all
-      add_bytes
-      @parent.feed_cmd_array
-      close
-    end
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def add_bytes
-    @view.tbl_bytes.sortItems(0, Qt::AscendingOrder)
-    @view.tbl_bytes.rowCount.times do |i|
-      if @bus.name == 'SPI'
-        iteration = @view.tbl_bytes.item(i, 2).text
-        description = @view.tbl_bytes.item(i, 3).text
-      else
-        iteration = nil
-        description = @view.tbl_bytes.item(i, 2).text
-      end
-      byte = Byte.create(
-        index:       i.next,
-        value:       @view.tbl_bytes.item(i, 1).text,
-        iteration:   iteration,
-        description: description,
-        command_id:  @cmd.id
-      )
-      return false if check_for_errors(byte)
-    end
-    Qt::MessageBox.new(
-      Qt::MessageBox::Information,
-      'Command status',
-      'Command saved'
-    ).exec
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def check_cell_content(item)
-    case @view.tbl_bytes.horizontalHeaderItem(item.column).text
-    when 'Order', 'Repetition'
-      if item.text.to_i < 0
-        item.setData(0, Qt::Variant.new(0))
-        ErrorMsg.new.positive_cell_value
-      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('')
-        ErrorMsg.new.hexa_cell_value
-      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('')
-          ErrorMsg.new.char_cell_value
-        end
-      end
-    end
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def is_cmd_size_valid?
-    byte_table_size = @view.tbl_bytes.rowCount
-    case @bus_name
-    when 'SPI'
-      cmd_size =  byte_table_size + @cmd_table.count_total_repetition
-      return true if cmd_size < 4000
-      return ErrorMsg.new.spi_cmd_too_long
-    when 'I2C'
-      count = i = 0
-      while i <= byte_table_size - 1 do
-        low_byte     = @view.tbl_bytes.item(i, 1)
-        high_byte    = @view.tbl_bytes.item(i + 1, 1)
-        command_byte = @view.tbl_bytes.item(i + 2, 1)
-        return ErrorMsg.new.lowbyte_missing  if low_byte.nil?
-        return ErrorMsg.new.highbyte_missing if high_byte.nil?
-        return ErrorMsg.new.mode_missing     if command_byte.nil?
-        current_count = HardsploitAPI.BytesToInt(
-          lByte: low_byte.text.to_i(16),
-          hByte: high_byte.text.to_i(16)
-        )
-        count += current_count
-        (command_type.to_i(16) % 2).zero? ? i += current_count + 3 : i += 3
-      end
-      return ErrorMsg.new.size_neq_row_number unless i == byte_table_size
-      return ErrorMsg.new.i2c_cmd_too_long    if count > 2000
-      return true
-    end
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def feed_editor_form
-    cmd = Command.find_by(name: @cmd_name)
-    @view.lie_name.setText(@cmd_name)
-    @view.lie_description.setText(cmd.description)
-    @cmd_table.fill_byte_table(cmd.bytes)
-  rescue Exception => msg
-    ErrorMsg.new.unknown(msg)
-  end
-
-  def add_row
-    if @view.lie_text_2_bytes.text.empty?
-      @cmd_table.add_row
-    else
-      @cmd_table.add_text_rows(@view.lie_text_2_bytes.text)
-      @view.lie_text_2_bytes.setText('')
-    end
-  end
-
-  def clone_row
-    @cmd_table.clone_rows
-  end
-
-  def remove_row
-    @cmd_table.remove_rows
-  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_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, bus_param={})
+    super()
+    @view = Ui_Command_editor.new
+    centerWindow(self)
+    @view.setupUi(self)
+    @bus = Bus.find(bus_id)
+    @chip_id = chip.id
+    @cmd_name = cmd_name
+    @bus_id = bus_id
+    @parent = parent
+    @cmd_table = Command_table.new(@view.tbl_bytes, @bus.name)
+
+    inputRestrict(@view.lie_name, 2)
+    inputRestrict(@view.lie_description, 2)
+    @view.lbl_chip_val.setText(chip.reference)
+    @cmd_table.resize_to_content
+    @cmd_table.build_spi if @bus.name == 'SPI'
+    @view.lie_text_2_bytes.hide unless @bus.name == 'SPI'
+
+    case action
+    when 0
+      # Add
+      @view.lbl_cmd_val.setText('-')
+      Qt::Object.connect(@view.btn_validate, SIGNAL('clicked()'), self, SLOT('add_cmd()'))
+    when 1
+      # Template
+      @view.lbl_cmd_val.setText(cmd_name)
+      feed_editor_form
+      @view.lie_name.setText('')
+      Qt::Object.connect(@view.btn_validate, SIGNAL('clicked()'), self, SLOT('add_cmd()'))
+    else
+      # Edit
+      @view.lbl_cmd_val.setText(cmd_name)
+      feed_editor_form
+      @view.btn_validate.setText('Edit')
+      Qt::Object.connect(@view.btn_validate, SIGNAL('clicked()'), self, SLOT('edit_cmd()'))
+    end
+    unless bus_param.empty?
+      feed_i2c_cmd_array(bus_param)
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(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
+    return false if @cmd_table.empty_data_exist? || is_cmd_size_valid? == false
+    @cmd = Command.create(
+      name:        @view.lie_name.text,
+      description: @view.lie_description.text,
+      bus_id:      @bus_id,
+      chip_id:     @chip_id
+    )
+    unless check_for_errors(@cmd)
+      add_bytes
+      @parent.feed_cmd_array
+      close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def edit_cmd
+    return false if @cmd_table.empty_data_exist? && !is_cmd_size_valid?
+    @cmd = Command.find_by(name: @cmd_name, chip_id: @chip_id)
+    @cmd.update(
+      name: @view.lie_name.text,
+      description: @view.lie_description.text
+    )
+    @view.tbl_bytes.sortItems(0, Qt::AscendingOrder)
+    unless check_for_errors(@cmd)
+      Byte.where(command_id: @cmd.id).destroy_all
+      add_bytes
+      @parent.feed_cmd_array
+      close
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def add_bytes
+    @view.tbl_bytes.sortItems(0, Qt::AscendingOrder)
+    @view.tbl_bytes.rowCount.times do |i|
+      if @bus.name == 'SPI'
+        iteration = @view.tbl_bytes.item(i, 2).text
+        description = @view.tbl_bytes.item(i, 3).text
+      else
+        iteration = nil
+        description = @view.tbl_bytes.item(i, 2).text
+      end
+      byte = Byte.create(
+        index:       i.next,
+        value:       @view.tbl_bytes.item(i, 1).text,
+        iteration:   iteration,
+        description: description,
+        command_id:  @cmd.id
+      )
+      return false if check_for_errors(byte)
+    end
+    Qt::MessageBox.new(
+      Qt::MessageBox::Information,
+      'Command status',
+      'Command saved'
+    ).exec
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def check_cell_content(item)
+    case @view.tbl_bytes.horizontalHeaderItem(item.column).text
+    when 'Order', 'Repetition'
+      if item.text.to_i < 0
+        item.setData(0, Qt::Variant.new(0))
+        ErrorMsg.new.positive_cell_value
+      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('')
+        ErrorMsg.new.hexa_cell_value
+      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('')
+          ErrorMsg.new.char_cell_value
+        end
+      end
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def is_cmd_size_valid?
+    byte_table_size = @view.tbl_bytes.rowCount
+    case @bus_name
+    when 'SPI'
+      cmd_size =  byte_table_size + @cmd_table.count_total_repetition
+      return true if cmd_size < 4000
+      return ErrorMsg.new.spi_cmd_too_long
+    when 'I2C'
+      count = i = 0
+      while i <= byte_table_size - 1 do
+        low_byte     = @view.tbl_bytes.item(i, 1)
+        high_byte    = @view.tbl_bytes.item(i + 1, 1)
+        command_byte = @view.tbl_bytes.item(i + 2, 1)
+        return ErrorMsg.new.lowbyte_missing  if low_byte.nil?
+        return ErrorMsg.new.highbyte_missing if high_byte.nil?
+        return ErrorMsg.new.mode_missing     if command_byte.nil?
+        current_count = HardsploitAPI.BytesToInt(
+          lByte: low_byte.text.to_i(16),
+          hByte: high_byte.text.to_i(16)
+        )
+        count += current_count
+        (command_type.to_i(16) % 2).zero? ? i += current_count + 3 : i += 3
+      end
+      return ErrorMsg.new.size_neq_row_number unless i == byte_table_size
+      return ErrorMsg.new.i2c_cmd_too_long    if count > 2000
+      return true
+    end
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def feed_editor_form
+    cmd = Command.find_by(name: @cmd_name)
+    @view.lie_name.setText(@cmd_name)
+    @view.lie_description.setText(cmd.description)
+    @cmd_table.fill_byte_table(cmd.bytes)
+  rescue Exception => msg
+    ErrorMsg.new.unknown(msg)
+  end
+
+  def add_row
+    if @view.lie_text_2_bytes.text.empty?
+      @cmd_table.add_row
+    else
+      @cmd_table.add_text_rows(@view.lie_text_2_bytes.text)
+      @view.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

@@ -1,233 +1,233 @@
-#===================================================
-#  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
-#===================================================
-
-class Command_table < Qt::Widget
-
-  def initialize(cmd_table, bus)
-    super()
-    @cmd_table = cmd_table
-    @bus = bus
-    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.value))
-      if @bus == 'SPI'
-        item =  Qt::TableWidgetItem.new
-        item.setData(0, Qt::Variant.new(b.iteration))
-        @cmd_table.setItem(i, 2, item)
-        @cmd_table.setItem(i, 3, Qt::TableWidgetItem.new(b.description))
-      else
-        @cmd_table.setItem(i, 2, Qt::TableWidgetItem.new(b.description))
-      end
-      item =  Qt::TableWidgetItem.new
-      item.setData(0, Qt::Variant.new(b.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)
-      return ErrorMsg.new.ascii_only unless 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
-  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, 2).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(word: 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(word: 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(word: 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(word: 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.next))
-      @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
-	  repetition_value = 0
-	  @cmd_table.rowCount.times do |i|
-			repetition_value += @cmd_table.item(i, 2).text.to_i
-		end
-		return repetition_value
-	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
+#===================================================
+#  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
+#===================================================
+
+class Command_table < Qt::Widget
+
+  def initialize(cmd_table, bus)
+    super()
+    @cmd_table = cmd_table
+    @bus = bus
+    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.value))
+      if @bus == 'SPI'
+        item =  Qt::TableWidgetItem.new
+        item.setData(0, Qt::Variant.new(b.iteration))
+        @cmd_table.setItem(i, 2, item)
+        @cmd_table.setItem(i, 3, Qt::TableWidgetItem.new(b.description))
+      else
+        @cmd_table.setItem(i, 2, Qt::TableWidgetItem.new(b.description))
+      end
+      item =  Qt::TableWidgetItem.new
+      item.setData(0, Qt::Variant.new(b.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)
+      return ErrorMsg.new.ascii_only unless 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
+  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, 2).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(word: 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(word: 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(word: 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(word: 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.next))
+      @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
+	  repetition_value = 0
+	  @cmd_table.rowCount.times do |i|
+			repetition_value += @cmd_table.item(i, 2).text.to_i
+		end
+		return repetition_value
+	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