require "config"
require "ast"
OFFSET_HEADER_MAGIC_NUMBERS = [ 0x2e43fd66, 0x4379bfba ]
OFFSET_MAGIC_NUMBERS = [ 0x5c577ac7, 0x0ff5e755 ]
class MissingMagicValuesException < Exception
end
def offsetsList(ast)
ast.filter(StructOffset).uniq.sort
end
def sizesList(ast)
ast.filter(Sizeof).uniq.sort
end
def constsList(ast)
ast.filter(ConstExpr).uniq.sort
end
def offsetsAndConfigurationIndex(file)
endiannessMarkerBytes = nil
result = {}
def readInt(endianness, bytes)
if endianness == :little
number = (bytes[0] << 0 |
bytes[1] << 8 |
bytes[2] << 16 |
bytes[3] << 24 |
bytes[4] << 32 |
bytes[5] << 40 |
bytes[6] << 48 |
bytes[7] << 56)
else
number = (bytes[0] << 56 |
bytes[1] << 48 |
bytes[2] << 40 |
bytes[3] << 32 |
bytes[4] << 24 |
bytes[5] << 16 |
bytes[6] << 8 |
bytes[7] << 0)
end
if number > 0x7fffffff_ffffffff
number -= 1 << 64
end
number
end
def prepareMagic(endianness, numbers)
magicBytes = []
numbers.each {
| number |
currentBytes = []
8.times {
currentBytes << (number & 0xff)
number >>= 8
}
if endianness == :big
currentBytes.reverse!
end
magicBytes += currentBytes
}
magicBytes
end
fileBytes = []
File.open(file, "rb") {
| inp |
loop {
byte = inp.getbyte
break unless byte
fileBytes << byte
}
}
def sliceByteArrays(byteArray, pattern)
result = []
lastSlicePoint = 0
(byteArray.length - pattern.length + 1).times {
| index |
foundOne = true
pattern.length.times {
| subIndex |
if byteArray[index + subIndex] != pattern[subIndex]
foundOne = false
break
end
}
if foundOne
result << byteArray[lastSlicePoint...index]
lastSlicePoint = index + pattern.length
end
}
result << byteArray[lastSlicePoint...(byteArray.length)]
result
end
[:little, :big].each {
| endianness |
headerMagicBytes = prepareMagic(endianness, OFFSET_HEADER_MAGIC_NUMBERS)
magicBytes = prepareMagic(endianness, OFFSET_MAGIC_NUMBERS)
bigArray = sliceByteArrays(fileBytes, headerMagicBytes)
unless bigArray.size <= 1
bigArray[1..-1].each {
| configArray |
array = sliceByteArrays(configArray, magicBytes)
index = readInt(endianness, array[1])
offsets = []
array[2..-1].each {
| data |
offsets << readInt(endianness, data)
}
result[index] = offsets
}
end
}
raise MissingMagicValuesException unless result.length >= 1
return result.map {
| pair |
pair.reverse
}
end
def buildOffsetsMap(ast, extractedConstants)
map = {}
astOffsetsList = offsetsList(ast)
astSizesList = sizesList(ast)
astConstsList = constsList(ast)
raise unless astOffsetsList.size + astSizesList.size + astConstsList.size == extractedConstants.size
astOffsetsList.each_with_index {
| structOffset, index |
map[structOffset] = extractedConstants.shift
}
astSizesList.each_with_index {
| sizeof, index |
map[sizeof] = extractedConstants.shift
}
astConstsList.each_with_index {
| const, index |
map[const] = extractedConstants.shift
}
map
end