research/hhe/gen-string-tables
Simon Howard c69b00ee9f Generate arrays to match coding style.
Subversion-branch: /research
Subversion-revision: 1905
2010-04-18 18:31:40 +00:00

127 lines
2.3 KiB
Ruby
Executable file

#!/usr/bin/env ruby
require "common.rb"
# Generate string -> offset lookup tables for each executable.
def load_string_lookups(exe_files)
string_lookups = []
CONFIGS.each_with_index do |config, i|
strings = find_strings(exe_files[i], config)
string_lookups.push(strings)
end
string_lookups
end
# Load list of "good" strings.
def load_good_strings(filename)
File.readlines(filename).map do |string|
JSON::load("["+ string.chomp + "]")[0]
end
end
# Look up a string's offset, removing it from the specified lookup table.
# This is done because a string can appear multiple times at different offsets.
def lookup_string(lookup_table, string)
lookup_table.each_with_index do |entry, i|
if entry[1] == string
lookup_table.delete_at(i)
return entry[0]
end
end
nil
end
# Find offsets for the specified string.
def get_string_offsets(string, string_lookups)
string_lookups.map do |lookup|
offset = lookup_string(lookup, string)
if offset != nil
offset
else
0
end
end
end
def format_offsets(offsets)
formatted_offsets = offsets.map do |offset|
sprintf("%5i", offset)
end
"{ " + formatted_offsets.join(", ") + " }"
end
# Print strings left over in the specified lookup table
# that were not in the good strings list.
def print_unsupported_strings(name, strings)
unsupported_offsets = strings.map do |entry|
entry[0]
end
puts "static int #{name}[] ="
puts "{"
numbers = 0
for offset in unsupported_offsets.sort + [ -1 ]
if numbers == 0
print " "
else
print " "
end
printf "%5i,", offset
numbers += 1
if numbers >= 8
print "\n"
numbers = 0
end
end
if numbers > 0
print "\n"
end
puts "};"
puts
end
string_lookups = load_string_lookups(ARGV)
good_strings = load_good_strings("good-strings.txt")
# Output list of supported strings with their offset for each version
puts "static hhe_string_t strings[] ="
puts "{"
for string in good_strings
offsets = get_string_offsets(string, string_lookups)
print " { " + format_offsets(offsets) + ", "
# print "\n "
print string.to_json + " },\n"
end
puts "};"
puts
# Output strings that appear in HHE but are not supported (ie. in good_strings)
CONFIGS.each_with_index do |config, i|
name = "unsupported_strings_#{config::SUFFIX}"
print_unsupported_strings(name, string_lookups[i])
end