class Rex::PeScan::Scanner::JmpRegScanner

Public Instance Methods

_build_byte_list(base, regnums) click to toggle source

build a list for regex of the possible bytes, based on a base byte and a list of register numbers..

# File lib/rex/pescan/scanner.rb, line 89
def _build_byte_list(base, regnums)
  regnums.collect { |regnum| Regexp.escape((base | regnum).chr) }.join('')
end
_parse_ret(data) click to toggle source
# File lib/rex/pescan/scanner.rb, line 105
def _parse_ret(data)
  if data.length == 1
    return "ret"
  else
    return "retn 0x%04x" % data[1, 2].unpack('v')[0]
  end
end
_ret_size(section, index) click to toggle source
# File lib/rex/pescan/scanner.rb, line 93
def _ret_size(section, index)
  d = section.read(index, 1)
  case d
    when "\xc3"
      return 1
    when "\xc2"
      return 3
  end

  raise RuntimeError, "invalid return opcode"
end
config(param) click to toggle source
# File lib/rex/pescan/scanner.rb, line 68
def config(param)
  regnums = param['args']

  # build a list of the call bytes
  calls  = _build_byte_list(0xd0, regnums - [4]) # note call esp's don't work..
  jmps   = _build_byte_list(0xe0, regnums)
  pushs1 = _build_byte_list(0x50, regnums)
  pushs2 = _build_byte_list(0xf0, regnums)

  regexstr = '('
  if !calls.empty?
    regexstr += "\xff[#{calls}]|"
  end

  regexstr += "\xff[#{jmps}]|([#{pushs1}]|\xff[#{pushs2}])(\xc3|\xc2..))"

  self.regex = Regexp.new(regexstr, nil, 'n')
end
scan_section(section, param={}) click to toggle source
# File lib/rex/pescan/scanner.rb, line 114
def scan_section(section, param={})
  index = 0

  hits  = [ ]

  while (index = section.index(regex, index)) != nil
    rva     = section.offset_to_rva(index)
    message = ''

    parse_ret = false

    byte1 = section.read(index, 1).unpack("C*")[0]

    if byte1 == 0xff
      byte2   = section.read(index+1, 1).unpack("C*")[0]
      regname = Rex::Arch::X86.reg_name32(byte2 & 0x7)

      case byte2 & 0xf8
      when 0xd0
        message = "call #{regname}"
        index += 2
      when 0xe0
        message = "jmp #{regname}"
        index += 2
      when 0xf0
        retsize = _ret_size(section, index+2)
        message = "push #{regname}; " + _parse_ret(section.read(index+2, retsize))
        index += 2 + retsize
      else
        raise "wtf"
      end
    else
      regname = Rex::Arch::X86.reg_name32(byte1 & 0x7)
      retsize = _ret_size(section, index+1)
      message = "push #{regname}; " + _parse_ret(section.read(index+1, retsize))
      index += 1 + retsize
    end

    hits << [ rva, message ]
  end

  return hits
end