Subversion Repositories c0or1k

/*

 * Australian Public Licence B (OZPLB)

 *

 * Version 1-0

 *

 * Copyright (c) 2004 University of New South Wales

 *

 * All rights reserved.

 *

 * Developed by: Operating Systems and Distributed Systems Group (DiSy)

 *               University of New South Wales

 *               http://www.disy.cse.unsw.edu.au

 *

 * Permission is granted by University of New South Wales, free of charge, to

 * any person obtaining a copy of this software and any associated

 * documentation files (the "Software") to deal with the Software without

 * restriction, including (without limitation) the rights to use, copy,

 * modify, adapt, merge, publish, distribute, communicate to the public,

 * sublicense, and/or sell, lend or rent out copies of the Software, and

 * to permit persons to whom the Software is furnished to do so, subject

 * to the following conditions:

 *

 *     * Redistributions of source code must retain the above copyright

 *       notice, this list of conditions and the following disclaimers.

 *

 *     * Redistributions in binary form must reproduce the above

 *       copyright notice, this list of conditions and the following

 *       disclaimers in the documentation and/or other materials provided

 *       with the distribution.

 *

 *     * Neither the name of University of New South Wales, nor the names of its

 *       contributors, may be used to endorse or promote products derived

 *       from this Software without specific prior written permission.

 *

 * EXCEPT AS EXPRESSLY STATED IN THIS LICENCE AND TO THE FULL EXTENT

 * PERMITTED BY APPLICABLE LAW, THE SOFTWARE IS PROVIDED "AS-IS", AND

 * NATIONAL ICT AUSTRALIA AND ITS CONTRIBUTORS MAKE NO REPRESENTATIONS,

 * WARRANTIES OR CONDITIONS OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING

 * BUT NOT LIMITED TO ANY REPRESENTATIONS, WARRANTIES OR CONDITIONS

 * REGARDING THE CONTENTS OR ACCURACY OF THE SOFTWARE, OR OF TITLE,

 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NONINFRINGEMENT,

 * THE ABSENCE OF LATENT OR OTHER DEFECTS, OR THE PRESENCE OR ABSENCE OF

 * ERRORS, WHETHER OR NOT DISCOVERABLE.

 *

 * TO THE FULL EXTENT PERMITTED BY APPLICABLE LAW, IN NO EVENT SHALL

 * NATIONAL ICT AUSTRALIA OR ITS CONTRIBUTORS BE LIABLE ON ANY LEGAL

 * THEORY (INCLUDING, WITHOUT LIMITATION, IN AN ACTION OF CONTRACT,

 * NEGLIGENCE OR OTHERWISE) FOR ANY CLAIM, LOSS, DAMAGES OR OTHER

 * LIABILITY, INCLUDING (WITHOUT LIMITATION) LOSS OF PRODUCTION OR

 * OPERATION TIME, LOSS, DAMAGE OR CORRUPTION OF DATA OR RECORDS; OR LOSS

 * OF ANTICIPATED SAVINGS, OPPORTUNITY, REVENUE, PROFIT OR GOODWILL, OR

 * OTHER ECONOMIC LOSS; OR ANY SPECIAL, INCIDENTAL, INDIRECT,

 * CONSEQUENTIAL, PUNITIVE OR EXEMPLARY DAMAGES, ARISING OUT OF OR IN

 * CONNECTION WITH THIS LICENCE, THE SOFTWARE OR THE USE OF OR OTHER

 * DEALINGS WITH THE SOFTWARE, EVEN IF NATIONAL ICT AUSTRALIA OR ITS

 * CONTRIBUTORS HAVE BEEN ADVISED OF THE POSSIBILITY OF SUCH CLAIM, LOSS,

 * DAMAGES OR OTHER LIABILITY.

 *

 * If applicable legislation implies representations, warranties, or

 * conditions, or imposes obligations or liability on University of New South

 * Wales or one of its contributors in respect of the Software that

 * cannot be wholly or partly excluded, restricted or modified, the

 * liability of University of New South Wales or the contributor is limited, to

 * the full extent permitted by the applicable legislation, at its

 * option, to:

 * a.  in the case of goods, any one or more of the following:

 * i.  the replacement of the goods or the supply of equivalent goods;

 * ii.  the repair of the goods;

 * iii. the payment of the cost of replacing the goods or of acquiring

 *  equivalent goods;

 * iv.  the payment of the cost of having the goods repaired; or

 * b.  in the case of services:

 * i.  the supplying of the services again; or

 * ii.  the payment of the cost of having the services supplied again.

 *

 * The construction, validity and performance of this licence is governed

 * by the laws in force in New South Wales, Australia.

 */

#include <elf/elf.h>

#include <string.h>

#include <inttypes.h>

int

elf32_checkFile(struct Elf32_Header *file)

{

        if (file->e_ident[EI_MAG0] != ELFMAG0

            || file->e_ident[EI_MAG1] != ELFMAG1

            || file->e_ident[EI_MAG2] != ELFMAG2

            || file->e_ident[EI_MAG3] != ELFMAG3)

                return -1;      /* not an elf file */

        if (file->e_ident[EI_CLASS] != ELFCLASS32)

                return -2;      /* not 32-bit file */

        return 0;                /* elf file looks OK */

}

/*

 * Returns the number of program segments in this elf file.

 */

unsigned

elf32_getNumSections(struct Elf32_Header *elfFile)

{

        return elfFile->e_shnum;

}

char *

elf32_getStringTable(struct Elf32_Header *elfFile)

{

        struct Elf32_Shdr *sections = elf32_getSectionTable(elfFile);

        return (char *)elfFile + sections[elfFile->e_shstrndx].sh_offset;

}

/* Returns a pointer to the program segment table, which is an array of

 * ELF32_Phdr_t structs.  The size of the array can be found by calling

 * getNumProgramSegments. */

struct Elf32_Phdr *

elf32_getProgramSegmentTable(struct Elf32_Header *elfFile)

{

        struct Elf32_Header *fileHdr = elfFile;

        return (struct Elf32_Phdr *) (fileHdr->e_phoff + (long) elfFile);

}

/* Returns the number of program segments in this elf file. */

uint16_t

elf32_getNumProgramHeaders(struct Elf32_Header *elfFile)

{

        struct Elf32_Header *fileHdr = elfFile;

        return fileHdr->e_phnum;

}

char *

elf32_getSectionName(struct Elf32_Header *elfFile, int i)

{

        struct Elf32_Shdr *sections = elf32_getSectionTable(elfFile);

        char *str_table = elf32_getSegmentStringTable(elfFile);

        if (str_table == NULL) {

                return "<corrupted>";

        } else {

                return str_table + sections[i].sh_name;

        }

}

uint32_t

elf32_getSectionSize(struct Elf32_Header *elfFile, int i)

{

        struct Elf32_Shdr *sections = elf32_getSectionTable(elfFile);

        return sections[i].sh_size;

}

uint32_t

elf32_getSectionAddr(struct Elf32_Header *elfFile, int i)

{

        struct Elf32_Shdr *sections = elf32_getSectionTable(elfFile);

        return sections[i].sh_addr;

}

void *

elf32_getSection(struct Elf32_Header *elfFile, int i)

{

        struct Elf32_Shdr *sections = elf32_getSectionTable(elfFile);

        return (char *)elfFile + sections[i].sh_offset;

}

void *

elf32_getSectionNamed(struct Elf32_Header *elfFile, char *str)

{

        int numSections = elf32_getNumSections(elfFile);

        int i;

        for (i = 0; i < numSections; i++) {

                if (strcmp(str, elf32_getSectionName(elfFile, i)) == 0) {

                        return elf32_getSection(elfFile, i);

                }

        }

        return NULL;

}

char *

elf32_getSegmentStringTable(struct Elf32_Header *elfFile)

{

        struct Elf32_Header *fileHdr = (struct Elf32_Header *) elfFile;

        if (fileHdr->e_shstrndx == 0) {

                return NULL;

        } else {

                return elf32_getStringTable(elfFile);

        }

}

#ifdef ELF_DEBUG

void

elf32_printStringTable(struct Elf32_Header *elfFile)

{

        int counter;

        struct Elf32_Shdr *sections = elf32_getSectionTable(elfFile);

        char * stringTable;

        if (!sections) {

                printf("No sections.\n");

                return;

        }

        stringTable = ((void *)elfFile) + sections[elfFile->e_shstrndx].sh_offset;

        printf("File is %p; sections is %p; string table is %p\n", elfFile, sections, stringTable);

        for (counter=0; counter < sections[elfFile->e_shstrndx].sh_size; counter++) {

                printf("%02x %c ", stringTable[counter],

                                stringTable[counter] >= 0x20 ? stringTable[counter] : '.');

        }

}

#endif

int

elf32_getSegmentType (struct Elf32_Header *elfFile, int segment)

{

        return elf32_getProgramSegmentTable(elfFile)[segment].p_type;

}

void

elf32_getSegmentInfo(struct Elf32_Header *elfFile, int segment, uint64_t *p_vaddr, uint64_t *p_addr, uint64_t *p_filesz, uint64_t *p_offset, uint64_t *p_memsz)

{

        struct Elf32_Phdr *segments;

        segments = elf32_getProgramSegmentTable(elfFile);

        *p_addr = segments[segment].p_paddr;

        *p_vaddr = segments[segment].p_vaddr;

        *p_filesz = segments[segment].p_filesz;

        *p_offset = segments[segment].p_offset;

        *p_memsz = segments[segment].p_memsz;

}

uint32_t

elf32_getEntryPoint (struct Elf32_Header *elfFile)

{

        return elfFile->e_entry;

}

/*

 * Debugging functions

 */

/*

 * prints out some details of one elf file

 */

void

elf32_fprintf(FILE *f, struct Elf32_Header *file, int size, const char *name, int flags)

{

        struct Elf32_Phdr *segments;

        unsigned numSegments;

        struct Elf32_Shdr *sections;

        unsigned numSections;

        int i, r;

        char *str_table;

        fprintf(f, "Found an elf32 file called \"%s\" located "

                "at address 0x%p\n", name, file);

        if ((r = elf32_checkFile(file)) != 0) {

                char *magic = (char*) file;

                fprintf(f, "Invalid elf file (%d)\n", r);

                fprintf(f, "Magic is: %2.2hhx %2.2hhx %2.2hhx %2.2hhx\n",

                        magic[0], magic[1], magic[2], magic[3]);

                return;

        }

        /*

         * get a pointer to the table of program segments

         */

        segments = elf32_getProgramHeaderTable(file);

        numSegments = elf32_getNumProgramHeaders(file);

        sections = elf32_getSectionTable(file);

        numSections = elf32_getNumSections(file);

        if ((uintptr_t) sections >  ((uintptr_t) file + size)) {

                fprintf(f, "Corrupted elfFile..\n");

                return;

        }

                /*

                 * print out info about each section

                 */

        if (flags & ELF_PRINT_PROGRAM_HEADERS) {

                /*

                 * print out info about each program segment

                 */

                fprintf(f, "Program Headers:\n");

                fprintf(f, "  Type           Offset   VirtAddr   PhysAddr   "

                        "FileSiz MemSiz  Flg Align\n");

                for (i = 0; i < numSegments; i++) {

                        if (segments[i].p_type != 1) {

                                fprintf(f, "segment %d is not loadable, "

                                        "skipping\n", i);

                        } else {

                                fprintf(f, "  LOAD           0x%06" PRIx32 " 0x%08" PRIx32 " 0x%08" PRIx32  \

                                       " 0x%05" PRIx32" 0x%05" PRIx32 " %c%c%c 0x%04" PRIx32 "\n",

                                //fprintf(f, "  LOAD           0x%" PRIxPTR " 0x%" PRIxPTR " 0x%" PRIxPTR 

                                //      " 0x%" PRIxPTR" 0x%" PRIxPTR " %c%c%c 0x%" PRIxPTR "\n",

                                        segments[i].p_offset, segments[i].p_vaddr,

                                        segments[i].p_paddr,

                                        segments[i].p_filesz, segments[i].p_memsz,

                                        segments[i].p_flags & PF_R ? 'R' : ' ',

                                        segments[i].p_flags & PF_W ? 'W' : ' ',

                                        segments[i].p_flags & PF_X ? 'E' : ' ',

                                        segments[i].p_align);

                        }

                }

        }

        if (flags & ELF_PRINT_SECTIONS) {

                str_table = elf32_getSegmentStringTable(file);

                printf("Section Headers:\n");

                printf("  [Nr] Name              Type            Addr     Off\n");

                for (i = 0; i < numSections; i++) {

                        //if (elf_checkSection(file, i) == 0) {

                        fprintf(f, "[%2d] %s %x %x\n", i, elf32_getSectionName(file, i),

                                //fprintf(f, "[%2d] %-17.17s %-15.15s %x %x\n", i, elf32_getSectionName(file, i), " ", 

                                ///fprintf(f, "%-17.17s %-15.15s %08x %06x\n", elf32_getSectionName(file, i), " "       /* sections[i].sh_type 

                                //                                                                       */ ,

                                sections[i].sh_addr, sections[i].sh_offset);

                        //}

                }

        }

}