Subversion Repositories freemyipod

#!/usr/bin/env python

#

#

#    Copyright 2010 TheSeven, benedikt93, Farthen

#

#

#    This file is part of emCORE.

#

#    emCORE is free software: you can redistribute it and/or

#    modify it under the terms of the GNU General Public License as

#    published by the Free Software Foundation, either version 2 of the

#    License, or (at your option) any later version.

#

#    emCORE is distributed in the hope that it will be useful,

#    but WITHOUT ANY WARRANTY; without even the implied warranty of

#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

#    See the GNU General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with emCORE.  If not, see <http://www.gnu.org/licenses/>.

#

#

"""

    This file includes some reusable functions and classes that might be useful

    to all python scripts

"""

import sys

from ctypes import *

from _ctypes import _SimpleCData

class Error(Exception):

    def __init__(self, value=None):

        self.value = value

    def __str__(self):

        if self.value != None:

            return repr(self.value)

class ArgumentError(Error):

    pass

class ArgumentTypeError(Error):

    def __init__(self, expected, seen=False):

        self.expected = expected

        self.seen = seen

    def __str__(self):

        if self.seen:

            return "Expected %s but got %s" % (self.expected, self.seen)

        else:

            return "Expected %s, but saw something else" % self.expected

class Logger(object):

    """

        Simple stdout/stderr/file logger.

        Loglevel 3 is most verbose, Loglevel 0: Only log something if there is an error.

        Loglevel -1 means that nothing is logged.

        The write function doesn't care about the loglevel and always logs everything.

    """

    def __init__(self, loglevel = 2, target = "stderr", logfile = "tools.log"):

        """

            loglevel: Possible values: 0 (only errors), 1 (warnings), 2 (info,

                      recommended for production use), 3 and more (debug)

            logfile: File to log to if using the target = "file"

            target: Default logging target. Can be "stdout", "file" or "string"

        """

        self.loglevel = loglevel

        self.logfile = logfile

        self.target = target

    def write(self, text, indent = 0, target = None):

        if self.loglevel >= 0:

            if target is None: target = self.target

            text = (indent * " ") + text

            text = text.replace("\n", "\n" + (indent * " "), text.count("\n") - 1)

            if target == "stdout":

                if majorver() > 2:

                    sys.stdout.write(text)

                else:

                    sys.stdout.write(text.encode(sys.stdout.encoding or "latin1", "replace"))

            if target == "stderr":

                if majorver() > 2:

                    sys.stdout.write(text)

                else:

                    sys.stderr.write(text.encode(sys.stderr.encoding or "latin1", "replace"))

            elif target == "file":

                with open(self.logfile, 'a') as f:

                    f.write(text)

                    f.close()

            elif target == "string":

                return text

    def debug(self, text, indent = 0, target = None):

        if self.loglevel >= 3:

            self.write("DEBUG: " + text, indent, target)

    def info(self, text, indent = 0, target = None):

        if self.loglevel >= 2:

            self.write(text, indent, target)

    def warn(self, text, indent = 0, target = None):

        if self.loglevel >= 1:

            self.write("WARNING: " + text, indent, target)

    def error(self, text, indent = 0, target = None):

        if self.loglevel >= 0:

            self.write("ERROR: " + text, indent, target)

class Bunch(dict):

    """

        This is a dict whose items can also be accessed with

        bunchinstance.something.

    """

    def __init__(self, **kw):

        dict.__init__(self, kw)

        self.__dict__ = self

    def __getstate__(self):

        return self

    def __setstate__(self, state):

        self.update(state)

        self.__dict__ = self

class remote_pointer(dict):

    """

        This points to a (remote) location.

        Otherwise it behaves like a Bunch.

        The second argument must be a Bunch object

    """

    def __init__(self, address, bunch):

        dict.__init__(self, bunch.__dict__)

        self.__dict__ = self

        self._address_ = address

    def __getstate__(self):

        return self

    def __setstate__(self, state):

        self.update(state)

        self.__dict__ = self

    def __str__(self):

        return "<remote_pointer object with address 0x%X>" % (self._address_)

    def __int__(self):

        return self._address_

    def __repr__(self):

        return self.__str__()

class c_enum(_SimpleCData):

    """

        Resembles the enum datatype from C with an 8 bit size.

        Returns the associated string of a value with c_enum[i]

        Returns the current value of the associated value as c_enum.__repr__()

        Comparison operators work with strings and values at the same time.

        ATTENTION: You can not really see if this is initialized or not.

        If it is uninitialized it will return the first entry of the enum.

        While this may be circumvented by changing the default value to

        something else this will not work if the enum is placed inside a

        ctypes structure as the __init__() method will not be called then.

    """    

    _type_ = c_uint8._type_

    def __init__(self, value = 0):

        if type(value) == str:

            value = getattr(self, value)

        _SimpleCData.__init__(self, value)

        self[value]

    def __getattr__(self, name):

        if name == "value":

            return self.value

        for key, value in enumerate(self._fields_):

            if value == name:

                return key

    def __getitem__(self, lookupkey):

        for key, value in enumerate(self._fields_):

            if key == lookupkey:

                return value

        raise IndexError("Value %d not in range of possible enum values for %s!" % (lookupkey, self.__class__.__name__))

    def __str__(self):

        return self[self.value]

    def __repr__(self):

        return self.__str__()

    def __int__(self):

        return self.value

    def __eq__(self, other):

        if type(other) == str:

            try: return getattr(self, other) == self.value

            except AttributeError: return False

        else:

            return self.value == other

    def __lt__(self, other):

        if type(other) == str:

            try: return self.value < getattr(self, other)

            except AttributeError: return False

        else:

            return self.value < other

    def __gt__(self, other):

        if type(other) == str:

            try: return  self.value > getattr(self, other)

            except AttributeError: return False

        else:

            return self.value > other

    def __le__(self, other):

        if self.value == other or self.value < other:

            return True

        return False

    def __ge__(self, other):

        if self.value == other or self.value > other:

            return True

        return False

    def __ne__(self, other):

        if type(other) == str:

            try: return getattr(self, other) != self.value

            except AttributeError: return True

        else:

            return self.value != other

class ExtendedCStruct(LittleEndianStructure):

    """

        This is a subclass of the LittleEndianStructure.

        It implements functions to easily convert

        structures to/from strings and Bunches.

    """

    def __init__(self, data = None, base = 0, address = 0):

        LittleEndianStructure.__init__(self)

        if data != None:

            self._data_ = data

            self._base_ = base

            self._address_ = address

            if address < base or address + sizeof(self) > base + len(data):

                raise Exception("Range 0x%08X+0x%X out of bounds [0x%08X:0x%08X]" % (address, sizeof(self), base, base + len(data)))

            memmove(addressof(self), data[address - base : address - base + sizeof(self)], sizeof(self))

    def __int__(self):

        return self._address_

    def _from_bunch(self, bunch):

        for field, _ in self._fields_:

            if field in bunch:

                setattr(self, field, getattr(bunch, field))

    def _to_bunch(self):

        bunch = Bunch()

        for field, _ in self._fields_:

            setattr(bunch, field, getattr(self, field))

        return bunch

    def _from_string(self, string):

        memmove(addressof(self), string, sizeof(self))

    def _to_string(self):

        return string_at(addressof(self), sizeof(self))

def getthread(address, threads):

    """

        Returns the thread at <address> from the list of threads <threads>.

        Returns an empty thread if not found

    """

    for thread in threads:

        if address == thread.addr:

            return thread

    from libemcoredata import scheduler_thread

    thread = scheduler_thread()._to_bunch()

    thread.name = "[Invalid Thread 0x%08X]" % address

    return thread

def string_from_image(data, base, ptr, maxlen):

  if ptr == 0: name = "<NULL>"

  try:

    string = ""

    end = data.find(b"\0", ptr - base, ptr - base + maxlen)

    if end < 0: return "<BAD_STRING>"

    else:

      d = data[ptr - base : end]

      for i in range(len(d)):

        byte = ord(d[i : i + 1])

        if byte < 0x20: return "<BAD_STRING>"

        else: string = string + chr(byte)

  except: return "<BAD_PTR>"

  return string

def gethwname(id):

    try:

        from libemcoredata import hwtypes

        hwtype = hwtypes[id][1]

    except KeyError:

        hwtype = "UNKNOWN (ID = 0x%X)" % id

    return hwtype

def gethwid(shortname):

    from libemcoredata import hwtypes

    for hwid in hwtypes:

        if hwtypes[hwid][0] == shortname:

            return hwid

    return False

def trimdoc(docstring):

    """

        Trims whitespace from docstrings

    """

    if not docstring:

        return ''

    # Convert tabs to spaces (following the normal Python rules)

    # and split into a list of lines:

    lines = docstring.expandtabs().splitlines()

    # Determine minimum indentation (first line doesn't count):

    try: maxsize = sys.maxint

    except AttributeError: maxsize = sys.maxsize

    indent = maxsize

    for line in lines[1:]:

        stripped = line.lstrip()

        if stripped:

            indent = min(indent, len(line) - len(stripped))

    # Remove indentation (first line is special):

    trimmed = [lines[0].strip()]

    if indent < maxsize:

        for line in lines[1:]:

            trimmed.append(line[indent:].rstrip())

    # Strip off trailing and leading blank lines:

    while trimmed and not trimmed[-1]:

        trimmed.pop()

    while trimmed and not trimmed[0]:

        trimmed.pop(0)

    # Return a single string:

    return '\n'.join(trimmed)

def getfuncdoc(funcdict):

    """

        Extracts important information from a dict of functions like the

        docstring and arguments and returns them in a human readable format

    """

    import inspect

    import re

    functions = Bunch()

    for function in funcdict:

        function = funcdict[function].func

        docinfo = Bunch()

        name = function.__name__

        argspec = inspect.getargspec(function)

        args = argspec[0]

        docinfo['varargs'] = False

        if argspec[1]:

            docinfo['varargs'] = True

        kwargvalues = argspec[3]

        kwargs = []

        if args:

            if kwargvalues:

                argnum = len(args) - len(kwargvalues)

                kwargnum = len(kwargvalues)

                kwargs = list(zip(args[argnum:], kwargvalues))

            else:

                argnum = len(args)

        else:

            argnum = 0

        docinfo['args'] = args[1:argnum]

        docinfo['kwargs'] = kwargs

        if function.__doc__:

            # strip unneccessary whitespace

            docinfo['documentation'] = trimdoc(function.__doc__)

        else:

            docinfo['documentation'] = None

        functions[name] = docinfo

    return functions

def gendoc(funcdict, indentwidth = 4, logtarget = "string"):

    logger = Logger()

    doc = getfuncdoc(funcdict)

    ret = ""

    for function in sorted(doc.items()):

        function = function[0]

        ret += logger.log("def " + function + "(", target = logtarget)

        counter = 0

        if doc[function]['args']:

            for arg in doc[function]['args']:

                if counter > 0:

                    sys.stdout.write(", ")

                counter += 1

                ret += logger.log(arg, target = logtarget)

        if doc[function]['kwargs']:

            for kwarg, kwargvalue in doc[function]['kwargs'].items():

                if counter > 0:

                    sys.stdout.write(", ")

                counter += 1

                ret += logger.log(kwarg + "=" + str(kwargvalue), target = logtarget)

        if doc[function]['varargs']:

            ret += logger.log("*argv", target = logtarget)

        ret += logger.log("):\n", target = logtarget)

        if doc[function]['documentation']:

            ret += logger.log("\"\"\"\n", indent = indentwidth, target = logtarget)

            ret += logger.log(trimdoc(doc[function]['documentation']) + "\n", indent = 2 * indentwidth, target = logtarget)

            ret += logger.log("\"\"\"\n", indent = indentwidth, target = logtarget)

        ret += logger.log("\n", target = logtarget)

    return ret

def to_bool(something):

    """

        Converts quite everything into bool.

    """

    if type(something).__name__ in ("bool", "NoneType"):

        return something

    elif type(something).__name__ in ("int", "long"):

        return something != 0

    elif type(something) == str:

        if something.lower() in ['true', '1', 't', 'y', 'yes']:

            return True

        elif something.lower() in ['false', '0', 'f', 'n', 'no']:

            return False

    raise ArgumentTypeError("bool", "'%s' (%s)" % (something, type(something).__name__))

def to_int(something):

    """

        Converts quite everything to a hexadecimal represented integer.

        This works for default arguments too, because it returns

        None when it found that it got a NoneType object.

    """

    if type(something).__name__ in ("int", "long", "NoneType"):

        return something

    elif type(something) == str:

        try:

            if something[:2] == "0x": # Hexadecimal notation

                return int(something[2:], 16)

            elif something[:2] == "0b": # Binary notation

                return int(something[2:], 2)

            else: # Decimal notation

                return int(something, 10)

        except ValueError:

            raise ArgumentTypeError("integer", "'%s' (%s)" % (something, type(something).__name__))

    else:

        raise ArgumentTypeError("integer", "'%s' (%s)" % (something, type(something).__name__))

def majorver():

    """

        Returns the major version of python

    """

    import sys

    return sys.hexversion // 0x1000000