ElfReader: Constructor tries to get ELF from ENV FAIL_ELF_PATH
git-svn-id: https://www4.informatik.uni-erlangen.de/i4svn/danceos/trunk/devel/fail@2002 8c4709b5-6ec9-48aa-a5cd-a96041d1645a
This commit is contained in:
@ -10,157 +10,172 @@ namespace fail {
|
||||
|
||||
const std::string ElfReader::NOTFOUND = "[ELFReader] Function not found.";
|
||||
|
||||
ElfReader::ElfReader(const char* path) : m_log("Fail*Elfinfo", false){
|
||||
// Try to open the ELF file
|
||||
FILE * fp = fopen(path, "r");
|
||||
if (!fp) {
|
||||
m_log << "Error: Could not open " << path << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
// Evaluate headers
|
||||
Elf32_Ehdr ehdr;
|
||||
Elf32_Shdr sec_hdr;
|
||||
int num_hdrs,i;
|
||||
fseek(fp,(off_t)0,SEEK_SET);
|
||||
read_ELF_file_header(fp, &ehdr);
|
||||
num_hdrs=ehdr.e_shnum;
|
||||
m_log << "Evaluating ELF File: " << path << std::endl;
|
||||
// Parse symbol table and generate internal map
|
||||
for(i=0;i<num_hdrs;i++)
|
||||
void ElfReader::setup(const char* path) {
|
||||
// Try to open the ELF file
|
||||
FILE * fp = fopen(path, "r");
|
||||
if (!fp) {
|
||||
m_log << "Error: Could not open " << path << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
// Evaluate headers
|
||||
Elf32_Ehdr ehdr;
|
||||
Elf32_Shdr sec_hdr;
|
||||
int num_hdrs,i;
|
||||
fseek(fp,(off_t)0,SEEK_SET);
|
||||
read_ELF_file_header(fp, &ehdr);
|
||||
num_hdrs=ehdr.e_shnum;
|
||||
m_log << "Evaluating ELF File: " << path << std::endl;
|
||||
// Parse symbol table and generate internal map
|
||||
for(i=0;i<num_hdrs;i++)
|
||||
{
|
||||
if(read_ELF_section_header(i,&sec_hdr,fp)==-1)
|
||||
{
|
||||
if(read_ELF_section_header(i,&sec_hdr,fp)==-1)
|
||||
m_log << "Wrong Section to read" << std::endl;
|
||||
}
|
||||
else
|
||||
{
|
||||
if((sec_hdr.sh_type==SHT_SYMTAB)||(sec_hdr.sh_type==SHT_DYNSYM))
|
||||
{
|
||||
m_log << "Wrong Section to read" << std::endl;
|
||||
process_symboltable(i,fp);
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
if((sec_hdr.sh_type==SHT_SYMTAB)||(sec_hdr.sh_type==SHT_DYNSYM))
|
||||
{
|
||||
process_symboltable(i,fp);
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
continue;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
fclose(fp);
|
||||
}
|
||||
|
||||
int ElfReader::process_symboltable(int sect_num, FILE* fp){
|
||||
fclose(fp);
|
||||
}
|
||||
|
||||
Elf32_Shdr sect_hdr;
|
||||
Elf32_Sym mysym;
|
||||
char *name_buf=NULL;
|
||||
int num_sym,link,i,idx;
|
||||
off_t sym_data_offset;
|
||||
int sym_data_size;
|
||||
if(read_ELF_section_header(sect_num,§_hdr,fp)==-1)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
//we have to check to which strtab it is linked
|
||||
link=sect_hdr.sh_link;
|
||||
sym_data_offset=sect_hdr.sh_offset;
|
||||
sym_data_size=sect_hdr.sh_size;
|
||||
num_sym=sym_data_size/sizeof(Elf32_Sym);
|
||||
ElfReader::ElfReader() : m_log("Fail*Elfinfo", false){
|
||||
// try to open elf file from environment variable
|
||||
char * elfpath = getenv("FAIL_ELF_PATH");
|
||||
if(elfpath == NULL){
|
||||
m_log << "FAIL_ELF_PATH not set :(" << std::endl;
|
||||
}else{
|
||||
setup(elfpath);
|
||||
}
|
||||
}
|
||||
|
||||
//read the coresponding strtab
|
||||
if(read_ELF_section_header(link,§_hdr,fp)==-1)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
//get the size of strtab in file and allocate a buffer
|
||||
name_buf=(char*)malloc(sect_hdr.sh_size);
|
||||
if(!name_buf)
|
||||
return -1;
|
||||
//get the offset of strtab in file and seek to it
|
||||
fseek(fp,sect_hdr.sh_offset,SEEK_SET);
|
||||
//read all data from the section to the buffer.
|
||||
fread(name_buf,sect_hdr.sh_size,1,fp);
|
||||
//so we have the namebuf now seek to symtab data
|
||||
fseek(fp,sym_data_offset,SEEK_SET);
|
||||
ElfReader::ElfReader(const char* path) : m_log("Fail*Elfinfo", false){
|
||||
setup(path);
|
||||
}
|
||||
|
||||
for(i=0;i<num_sym;i++)
|
||||
{
|
||||
int ElfReader::process_symboltable(int sect_num, FILE* fp){
|
||||
|
||||
fread(&mysym,sizeof(Elf32_Sym),1,fp);
|
||||
idx=mysym.st_name;
|
||||
Elf32_Shdr sect_hdr;
|
||||
Elf32_Sym mysym;
|
||||
char *name_buf=NULL;
|
||||
int num_sym,link,i,idx;
|
||||
off_t sym_data_offset;
|
||||
int sym_data_size;
|
||||
if(read_ELF_section_header(sect_num,§_hdr,fp)==-1)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
//we have to check to which strtab it is linked
|
||||
link=sect_hdr.sh_link;
|
||||
sym_data_offset=sect_hdr.sh_offset;
|
||||
sym_data_size=sect_hdr.sh_size;
|
||||
num_sym=sym_data_size/sizeof(Elf32_Sym);
|
||||
|
||||
int type = ELF32_ST_TYPE(mysym.st_info);
|
||||
if((type != STT_SECTION) && (type != STT_FILE)){
|
||||
//read the coresponding strtab
|
||||
if(read_ELF_section_header(link,§_hdr,fp)==-1)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
//get the size of strtab in file and allocate a buffer
|
||||
name_buf=(char*)malloc(sect_hdr.sh_size);
|
||||
if(!name_buf)
|
||||
return -1;
|
||||
//get the offset of strtab in file and seek to it
|
||||
fseek(fp,sect_hdr.sh_offset,SEEK_SET);
|
||||
//read all data from the section to the buffer.
|
||||
fread(name_buf,sect_hdr.sh_size,1,fp);
|
||||
//so we have the namebuf now seek to symtab data
|
||||
fseek(fp,sym_data_offset,SEEK_SET);
|
||||
|
||||
for(i=0;i<num_sym;i++)
|
||||
{
|
||||
|
||||
fread(&mysym,sizeof(Elf32_Sym),1,fp);
|
||||
idx=mysym.st_name;
|
||||
|
||||
int type = ELF32_ST_TYPE(mysym.st_info);
|
||||
if((type != STT_SECTION) && (type != STT_FILE)){
|
||||
#ifndef __puma
|
||||
m_bimap_mangled.insert( entry(name_buf+idx, mysym.st_value) );
|
||||
m_bimap_demangled.insert( entry ( Demangler::demangle(name_buf+idx), mysym.st_value) );
|
||||
m_bimap_mangled.insert( entry(name_buf+idx, mysym.st_value) );
|
||||
m_bimap_demangled.insert( entry ( Demangler::demangle(name_buf+idx), mysym.st_value) );
|
||||
|
||||
#endif
|
||||
}
|
||||
}
|
||||
free (name_buf);
|
||||
return 0;
|
||||
}
|
||||
free (name_buf);
|
||||
return 0;
|
||||
}
|
||||
|
||||
guest_address_t ElfReader::getAddressByName(const std::string& name) {
|
||||
guest_address_t ElfReader::getAddressByName(const std::string& name) {
|
||||
#ifndef __puma
|
||||
guest_address_t res = getAddress(m_bimap_demangled, name);
|
||||
if(res == ADDR_INV){
|
||||
res = getAddress(m_bimap_mangled, name);
|
||||
}
|
||||
return res;
|
||||
guest_address_t res = getAddress(m_bimap_demangled, name);
|
||||
if(res == ADDR_INV){
|
||||
res = getAddress(m_bimap_mangled, name);
|
||||
}
|
||||
return res;
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
#ifndef __puma
|
||||
guest_address_t ElfReader::getAddress(const bimap_t& map, const std::string& name){
|
||||
typedef bimap_t::left_map::const_iterator const_iterator_t;
|
||||
guest_address_t ElfReader::getAddress(const bimap_t& map, const std::string& name){
|
||||
typedef bimap_t::left_map::const_iterator const_iterator_t;
|
||||
|
||||
const_iterator_t iterator = map.left.find(name);
|
||||
if(iterator == map.left.end()){
|
||||
return ADDR_INV;
|
||||
}else{
|
||||
return iterator->second;
|
||||
}
|
||||
const_iterator_t iterator = map.left.find(name);
|
||||
if(iterator == map.left.end()){
|
||||
return ADDR_INV;
|
||||
}else{
|
||||
return iterator->second;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
#ifndef __puma
|
||||
std::string ElfReader::getName(const bimap_t& map, guest_address_t address){
|
||||
// .right switches key/value
|
||||
typedef bimap_t::right_map::const_iterator const_iterator_t;
|
||||
std::string ElfReader::getName(const bimap_t& map, guest_address_t address){
|
||||
// .right switches key/value
|
||||
typedef bimap_t::right_map::const_iterator const_iterator_t;
|
||||
|
||||
const_iterator_t iterator = map.right.find(address);
|
||||
if(iterator != map.right.end()){
|
||||
return iterator->second;
|
||||
}
|
||||
return NOTFOUND;
|
||||
const_iterator_t iterator = map.right.find(address);
|
||||
if(iterator != map.right.end()){
|
||||
return iterator->second;
|
||||
}
|
||||
return NOTFOUND;
|
||||
}
|
||||
|
||||
std::string ElfReader::getNameByAddress(guest_address_t address) {
|
||||
std::string res = getName(m_bimap_demangled, address);
|
||||
if(res == NOTFOUND){
|
||||
return getName(m_bimap_mangled, address);
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
std::string ElfReader::getMangledNameByAddress(guest_address_t address) {
|
||||
std::string ElfReader::getNameByAddress(guest_address_t address) {
|
||||
std::string res = getName(m_bimap_demangled, address);
|
||||
if(res == NOTFOUND){
|
||||
return getName(m_bimap_mangled, address);
|
||||
}
|
||||
return res;
|
||||
}
|
||||
|
||||
std::string ElfReader::getDemangledNameByAddress(guest_address_t address) {
|
||||
return getName(m_bimap_demangled, address);
|
||||
}
|
||||
std::string ElfReader::getMangledNameByAddress(guest_address_t address) {
|
||||
return getName(m_bimap_mangled, address);
|
||||
}
|
||||
|
||||
void ElfReader::printDemangled(){
|
||||
print_map(m_bimap_demangled.right); // print Address as first element
|
||||
}
|
||||
std::string ElfReader::getDemangledNameByAddress(guest_address_t address) {
|
||||
return getName(m_bimap_demangled, address);
|
||||
}
|
||||
|
||||
void ElfReader::printMangled(){
|
||||
print_map(m_bimap_mangled.right); // print Address as first element
|
||||
}
|
||||
void ElfReader::printDemangled(){
|
||||
print_map(m_bimap_demangled.right); // print Address as first element
|
||||
}
|
||||
|
||||
void ElfReader::printMangled(){
|
||||
print_map(m_bimap_mangled.right); // print Address as first element
|
||||
}
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
@ -22,13 +22,19 @@ namespace fail {
|
||||
|
||||
public:
|
||||
|
||||
/**
|
||||
/**
|
||||
* Constructor.
|
||||
* @param path Path to the ELF file.
|
||||
*/
|
||||
ElfReader(const char* path);
|
||||
|
||||
/**
|
||||
* Constructor.
|
||||
* @note The path is guessed from a FAIL_ELF_PATH environment variable
|
||||
*/
|
||||
ElfReader();
|
||||
|
||||
/**
|
||||
* Get guest address by symbol name.
|
||||
* Both mangled an demangled symbols are searched.
|
||||
* @param name The symbol name as string
|
||||
@ -79,11 +85,12 @@ namespace fail {
|
||||
Logger m_log;
|
||||
|
||||
|
||||
void setup(const char*);
|
||||
int process_symboltable(int sect_num, FILE* fp);
|
||||
#ifndef __puma
|
||||
typedef boost::bimap< std::string, guest_address_t > bimap_t;
|
||||
typedef bimap_t::value_type entry;
|
||||
|
||||
|
||||
bimap_t m_bimap_mangled;
|
||||
bimap_t m_bimap_demangled;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user