Subversion Repositories c0or1k

import mmap

import os

class ELF:

    EI_MAGIC = "\x7fELF"

    def __init__(self, name):

        f = file(name, "rb")

        size = os.stat(name).st_size

        self.data = mmap.mmap(f.fileno(), size, mmap.MAP_PRIVATE, mmap.PROT_READ)

        if self.magic != self.EI_MAGIC:

            raise "Not an elf"

    def get_magic(self):

        return self.data[:4]

    magic = property(get_magic)

    def get_class(self):

        return self.data[4]

    elf_class = property(get_class)

"Test suite"

x = ELF("a.out")

# Check can load an elf

success = 1

try:

    x = ELF("a.out")

except:

    success = 0

assert success

# CHeck can't load not and elf

success = 0

try:

    x = ELF("pyelf.py")

except:

    success = 1

assert success