81 lines
2.6 KiB
Ruby
81 lines
2.6 KiB
Ruby
# This file is part of Metasm, the Ruby assembly manipulation suite
|
|
# Copyright (C) 2006-2009 Yoann GUILLOT
|
|
#
|
|
# Licence is LGPL, see LICENCE in the top-level directory
|
|
|
|
|
|
module Metasm
|
|
# root directory for metasm files
|
|
# used by some scripts, eg to find samples/dasm-plugin directory
|
|
Metasmdir = File.dirname(__FILE__)
|
|
# add it to the ruby library path
|
|
$: << Metasmdir
|
|
|
|
# constants defined in the same file as another
|
|
Const_autorequire_equiv = {
|
|
'X86' => 'Ia32', 'PPC' => 'PowerPC',
|
|
'X64' => 'X86_64', 'AMD64' => 'X86_64',
|
|
'UniversalBinary' => 'MachO', 'COFFArchive' => 'COFF',
|
|
'DEY' => 'DEX',
|
|
'PTrace' => 'LinOS', 'FatELF' => 'ELF',
|
|
'LoadedELF' => 'ELF', 'LoadedPE' => 'PE',
|
|
'LoadedAutoExe' => 'AutoExe',
|
|
'LinuxRemoteString' => 'LinOS',
|
|
'LinDebugger' => 'LinOS',
|
|
'WinAPI' => 'WinOS',
|
|
'WindowsRemoteString' => 'WinOS', 'WinDbgAPI' => 'WinOS',
|
|
'WinDebugger' => 'WinOS',
|
|
'GdbRemoteString' => 'GdbClient', 'GdbRemoteDebugger' => 'GdbClient',
|
|
'DecodedInstruction' => 'Disassembler', 'DecodedFunction' => 'Disassembler',
|
|
'InstructionBlock' => 'Disassembler',
|
|
}
|
|
|
|
# files to require to get the definition of those constants
|
|
Const_autorequire = {
|
|
'Ia32' => 'ia32', 'MIPS' => 'mips', 'PowerPC' => 'ppc', 'ARM' => 'arm',
|
|
'X86_64' => 'x86_64', 'Sh4' => 'sh4', 'Dalvik' => 'dalvik',
|
|
'C' => 'compile_c',
|
|
'MZ' => 'exe_format/mz', 'PE' => 'exe_format/pe',
|
|
'ELF' => 'exe_format/elf', 'COFF' => 'exe_format/coff',
|
|
'Shellcode' => 'exe_format/shellcode', 'AutoExe' => 'exe_format/autoexe',
|
|
'AOut' => 'exe_format/a_out', 'MachO' => 'exe_format/macho',
|
|
'DEX' => 'exe_format/dex',
|
|
'NDS' => 'exe_format/nds', 'XCoff' => 'exe_format/xcoff',
|
|
'Bflt' => 'exe_format/bflt', 'Dol' => 'exe_format/dol',
|
|
'Gui' => 'gui',
|
|
'WindowsExports' => 'os/windows_exports',
|
|
'GNUExports' => 'os/gnu_exports',
|
|
'LinOS' => 'os/linux', 'WinOS' => 'os/windows',
|
|
'GdbClient' => 'os/remote',
|
|
'Disassembler' => 'disassemble',
|
|
'Decompiler' => 'decompile',
|
|
'DynLdr' => 'dynldr',
|
|
}
|
|
|
|
# use the Module.autoload ruby functionnality to load framework components on demand
|
|
Const_autorequire.each { |cst, file|
|
|
autoload cst, File.join('metasm', file)
|
|
}
|
|
|
|
Const_autorequire_equiv.each { |cst, eqv|
|
|
file = Const_autorequire[eqv]
|
|
autoload cst, File.join('metasm', file)
|
|
}
|
|
end
|
|
|
|
# load Metasm core files
|
|
%w[main encode decode render exe_format/main os/main].each { |f|
|
|
require File.join('metasm', f)
|
|
}
|
|
|
|
|
|
# remove an 1.9 warning, couldn't find a compatible way...
|
|
if Hash.new.respond_to?(:key)
|
|
puts "using ruby1.9 workaround for Hash#index warning" if $DEBUG
|
|
class Hash
|
|
alias index_premetasm index rescue nil
|
|
undef index rescue nil
|
|
alias index key
|
|
end
|
|
end
|