Method: Rex::PeParsey::PeBase#_load_relocations

Defined in:

lib/rex/peparsey/pebase.rb

#_load_relocations ⇒ Object

# File 'lib/rex/peparsey/pebase.rb', line 1455

def _load_relocations

  #
  # Get the data directory entry, size, etc
  #
  exports_entry = _optional_header['DataDirectory'][5]
  rva           = exports_entry.v['VirtualAddress']
  size          = exports_entry.v['Size']

  return nil if size == 0

  #
  # Ok, so we have the data directory, now lets parse it
  #

  dirdata = _isource.read(rva_to_file_offset(rva), size)

  relocdirs = [ ]

  while dirdata.length >= IMAGE_SIZEOF_BASE_RELOCATION
    header = IMAGE_BASE_RELOCATION.make_struct
    header.from_s(dirdata)
    dirdata = header.leftover

    numrelocs = (header.v['SizeOfBlock'] - IMAGE_SIZEOF_BASE_RELOCATION) / 2

    relocbase = header.v['VirtualAddress']

    relocdir = RelocationDirectory.new(relocbase, [ ])

    numrelocs.times do
      reloc = IMAGE_BASE_RELOCATION_TYPE_OFFSET.make_struct
      reloc.from_s(dirdata)
      dirdata = reloc.leftover

      typeoffset = reloc.v['TypeOffset']

      relocrva  = relocbase + (typeoffset & 0xfff)
      reloctype = (typeoffset >> 12) & 0xf

      relocdir.entries << RelocationEntry.new(relocrva, reloctype)
    end

    relocdirs << relocdir
  end

  return relocdirs
end