import gdb

def _convert_args(raw_args):

        for raw_arg in raw_args:

            if isinstance(raw_arg, gdb.Value):

                arg = raw_arg

            else:

                arg = gdb.parse_and_eval(raw_arg)

            arg_t = arg.type.strip_typedefs()

            if arg_t.code == gdb.TYPE_CODE_INT:

                yield int(arg)

            elif arg_t.code == gdb.TYPE_CODE_PTR:

                yield int(str(arg), 16)

            else: # let's try passing it directly

                yield arg

# some magic to register commands more easily

class AutoCommand(gdb.Command):

    def __init__(self, command_name, method, *args, **kw):

        self._command_name = command_name

        self._method = method

        super(AutoCommand, self).__init__(command_name, *args, **kw)

    def invoke(self, arg_string, from_tty):

        raw_args = gdb.string_to_argv(arg_string)

        args = _convert_args(raw_args)

        return self._method(*args)

class AutoFunction(gdb.Function):

    def __init__(self, function_name, method):

        self._function_name = function_name

        self._method = method

        super(AutoFunction, self).__init__(function_name)

    def invoke(self, *raw_args):

        args = _convert_args(raw_args)

        return self._method(*args)