gem5/base/loader/elf_object.cc
Ali Saidi 54782bfb30 Elf loader now conforms to coding style and config files for linux
use /m5/system/linux for their binaries

base/loader/elf_object.cc:
    Elf loader now conforms to coding style

--HG--
extra : convert_revision : 558e587e969535f31987f2810ee17ec72006de0a
2004-01-14 16:12:00 -05:00

331 lines
9.2 KiB
C++

/*
* Copyright (c) 2003 The Regents of The University of Michigan
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met: redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer;
* redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution;
* neither the name of the copyright holders nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <string>
#include "base/loader/elf_object.hh"
#include "mem/functional_mem/functional_memory.hh"
#include "base/loader/symtab.hh"
#include "base/trace.hh" // for DPRINTF
using namespace std;
ObjectFile *
ElfObject::tryFile(const string &fname, int fd, size_t len, uint8_t *data)
{
Elf *elf;
GElf_Ehdr ehdr;
/* check that header matches library version */
assert(elf_version(EV_CURRENT) != EV_NONE);
/* get a pointer to elf structure */
elf = elf_memory((char*)data,len);
/* will only fail if fd is invalid */
assert(elf != NULL);
/* Check that we actually have a elf file */
if(gelf_getehdr(elf, &ehdr) ==0)
{
DPRINTFR(Loader, "Not ELF\n");
elf_end(elf);
return NULL;
}
else
{
if (ehdr.e_ident[EI_CLASS] == ELFCLASS32)
panic("32 bit ELF Binary, Not Supported");
if (ehdr.e_machine != EM_ALPHA)
panic("Non Alpha Binary, Not Supported");
elf_end(elf);
return new ElfObject(fname, fd, len, data,
ObjectFile::Alpha, ObjectFile::Linux);
}
}
ElfObject::ElfObject(const string &_filename, int _fd,
size_t _len, uint8_t *_data,
Arch _arch, OpSys _opSys)
: ObjectFile(_filename, _fd, _len, _data, _arch, _opSys)
{
Elf *elf;
GElf_Ehdr ehdr;
/* check that header matches library version */
assert(elf_version(EV_CURRENT) != EV_NONE);
/* get a pointer to elf structure */
elf = elf_memory((char*)fileData,len);
/* will only fail if fd is invalid */
assert(elf != NULL);
/* Check that we actually have a elf file */
if(gelf_getehdr(elf, &ehdr) ==0)
{
panic("Not ELF, shouldn't be here");
}
entry = ehdr.e_entry;
elf_end(elf);
/* We will actually read the sections when we need to load them*/
}
bool
ElfObject::loadSections(FunctionalMemory *mem, bool loadPhys)
{
Elf *elf;
int sec_idx = 1; /* there is a 0 but it is nothing, go figure*/
Elf_Scn *section;
GElf_Shdr shdr;
GElf_Ehdr ehdr;
uint8_t *zero_mem;
uint8_t *section_data;
Addr address;
char *sec_name;
/* check that header matches library version */
assert(elf_version(EV_CURRENT) != EV_NONE);
/* get a pointer to elf structure */
elf = elf_memory((char*)fileData,len);
assert(elf != NULL);
/* Check that we actually have a elf file */
if(gelf_getehdr(elf, &ehdr) ==0)
{
panic("Not ELF, shouldn't be here");
}
/* Get the first section */
section = elf_getscn(elf, sec_idx);
/* While there are no more sections */
while (section != NULL)
{
gelf_getshdr(section, &shdr);
if (shdr.sh_flags & SHF_ALLOC)
{
/* we should load this */
DPRINTF(Loader,"Name: %20s Address: 0x%016llx Size: 0x%08llx Offset: 0x%08llx Flags:0x%08llx %s\n",
elf_strptr(elf, ehdr.e_shstrndx, shdr.sh_name), shdr.sh_addr,
shdr.sh_size, shdr.sh_offset, shdr.sh_flags, shdr.sh_flags & SHF_ALLOC ? "ALLOC" : "");
sec_name = elf_strptr(elf, ehdr.e_shstrndx, shdr.sh_name);
section_data = fileData + shdr.sh_offset;
if(sec_name)
{
if (strcmp(sec_name, ".text")==0)
{
text.baseAddr = shdr.sh_addr;
text.size = shdr.sh_size;
}
if (strcmp(sec_name, ".data")==0)
{
data.baseAddr = shdr.sh_addr;
data.size = shdr.sh_size;
}
if (strcmp(sec_name, ".bss")==0)
{
bss.baseAddr = shdr.sh_addr;
bss.size = shdr.sh_size;
/* If this is the .bss section it must be 0, so just
to be extra causious, lets allocate some memory
bzero it, and write that */
zero_mem = (uint8_t*)malloc(shdr.sh_size);
memset(zero_mem, 0, shdr.sh_size);
section_data = zero_mem;
}
}
if(shdr.sh_size != 0)
{
if (loadPhys)
{
address = shdr.sh_addr &= (ULL(1) << 40) - 1;
mem->prot_write(address, section_data, shdr.sh_size);
}
else
{
mem->prot_write(shdr.sh_addr, section_data, shdr.sh_size);
}
}
}
++sec_idx;
section = elf_getscn(elf, sec_idx);
}
free(zero_mem);
elf_end(elf);
return true;
}
bool
ElfObject::loadGlobalSymbols(SymbolTable *symtab)
{
Elf *elf;
int sec_idx = 1; /* there is a 0 but it is nothing, go figure*/
Elf_Scn *section;
GElf_Shdr shdr;
Elf_Data *data;
int count, ii;
bool found = false;
GElf_Sym sym;
if (!symtab)
return false;
/* check that header matches library version */
assert(elf_version(EV_CURRENT) != EV_NONE);
/* get a pointer to elf structure */
elf = elf_memory((char*)fileData,len);
assert(elf != NULL);
/* Get the first section */
section = elf_getscn(elf, sec_idx);
/* While there are no more sections */
while (section != NULL)
{
gelf_getshdr(section, &shdr);
if(shdr.sh_type == SHT_SYMTAB)
{
found = true;
data = elf_getdata(section, NULL);
count = shdr.sh_size / shdr.sh_entsize;
DPRINTF(Loader, "Found Symbol Table, %d symbols present\n", count);
/* loop through all the symbols, only loading global ones*/
for (ii = 0; ii < count; ++ii)
{
gelf_getsym(data, ii, &sym);
if ((GELF_ST_BIND(sym.st_info) & STB_GLOBAL) &&
((GELF_ST_TYPE(sym.st_info) == STT_FUNC) || (GELF_ST_TYPE(sym.st_info) == STT_NOTYPE)))
{
symtab->insert(sym.st_value, elf_strptr(elf, shdr.sh_link, sym.st_name));
}
}
}
++sec_idx;
section = elf_getscn(elf, sec_idx);
}
elf_end(elf);
return found;
}
bool
ElfObject::loadLocalSymbols(SymbolTable *symtab)
{
Elf *elf;
int sec_idx = 1; /* there is a 0 but it is nothing, go figure*/
Elf_Scn *section;
GElf_Shdr shdr;
Elf_Data *data;
int count, ii;
bool found = false;
GElf_Sym sym;
if (!symtab)
return false;
/* check that header matches library version */
assert(elf_version(EV_CURRENT) != EV_NONE);
/* get a pointer to elf structure */
elf = elf_memory((char*)fileData,len);
assert(elf != NULL);
/* Get the first section */
section = elf_getscn(elf, sec_idx);
/* While there are no more sections */
while (section != NULL)
{
gelf_getshdr(section, &shdr);
if(shdr.sh_type == SHT_SYMTAB)
{
found = true;
data = elf_getdata(section, NULL);
count = shdr.sh_size / shdr.sh_entsize;
DPRINTF(Loader, "Found Symbol Table, %d symbols present\n", count);
/* loop through all the symbols, only loading global ones*/
for (ii = 0; ii < count; ++ii)
{
gelf_getsym(data, ii, &sym);
if (GELF_ST_BIND(sym.st_info) & STB_LOCAL)
{
symtab->insert(sym.st_value, elf_strptr(elf, shdr.sh_link, sym.st_name));
}
}
}
++sec_idx;
section = elf_getscn(elf, sec_idx);
}
elf_end(elf);
return found;
}