generate_offset_extractor.rb [plain text]
$: << File.dirname(__FILE__)
require "backends"
require "digest/sha1"
require "offsets"
require "parser"
require "self_hash"
require "settings"
require "transform"
inputFlnm = ARGV.shift
outputFlnm = ARGV.shift
$stderr.puts "offlineasm: Parsing #{inputFlnm} and creating offset extractor #{outputFlnm}."
def emitMagicNumber
OFFSET_MAGIC_NUMBERS.each {
| number |
$output.puts "#{number},"
}
end
inputHash = "// offlineasm input hash: #{parseHash(inputFlnm)} #{selfHash}"
if FileTest.exist? outputFlnm
File.open(outputFlnm, "r") {
| inp |
firstLine = inp.gets
if firstLine and firstLine.chomp == inputHash
$stderr.puts "offlineasm: Nothing changed."
exit 0
end
}
end
originalAST = parse(inputFlnm)
class Node
def offsetsPruneTo(sequence)
children.each {
| child |
child.offsetsPruneTo(sequence)
}
end
def offsetsPrune
result = Sequence.new(codeOrigin, [])
offsetsPruneTo(result)
result
end
end
class IfThenElse
def offsetsPruneTo(sequence)
ifThenElse = IfThenElse.new(codeOrigin, predicate, thenCase.offsetsPrune)
ifThenElse.elseCase = elseCase.offsetsPrune
sequence.list << ifThenElse
end
end
class StructOffset
def offsetsPruneTo(sequence)
sequence.list << self
end
end
class Sizeof
def offsetsPruneTo(sequence)
sequence.list << self
end
end
prunedAST = originalAST.offsetsPrune
File.open(outputFlnm, "w") {
| outp |
$output = outp
outp.puts inputHash
length = 0
emitCodeInAllConfigurations(prunedAST) {
| settings, ast, backend, index |
offsetsList = ast.filter(StructOffset).uniq.sort
sizesList = ast.filter(Sizeof).uniq.sort
length += OFFSET_HEADER_MAGIC_NUMBERS.size + (OFFSET_MAGIC_NUMBERS.size + 1) * (1 + offsetsList.size + sizesList.size)
}
outp.puts "static const unsigned extractorTable[#{length}] = {"
emitCodeInAllConfigurations(prunedAST) {
| settings, ast, backend, index |
OFFSET_HEADER_MAGIC_NUMBERS.each {
| number |
$output.puts "#{number},"
}
offsetsList = ast.filter(StructOffset).uniq.sort
sizesList = ast.filter(Sizeof).uniq.sort
emitMagicNumber
outp.puts "#{index},"
offsetsList.each {
| offset |
emitMagicNumber
outp.puts "OFFLINE_ASM_OFFSETOF(#{offset.struct}, #{offset.field}),"
}
sizesList.each {
| offset |
emitMagicNumber
outp.puts "sizeof(#{offset.struct}),"
}
}
outp.puts "};"
}
$stderr.puts "offlineasm: offset extractor #{outputFlnm} successfully generated."