syscall_handlers.py

import logging

from unicorn import *
from unicorn.arm_const import *

from androidemu.cpu.interrupt_handler import InterruptHandler
from androidemu.cpu.syscall_handler import SyscallHandler
from androidemu.utils import memory_helpers

logger = logging.getLogger(__name__)


class SyscallHandlers:

    """
    :type interrupt_handler InterruptHandler
    """
    def __init__(self, interrupt_handler):
        self._handlers = dict()
        interrupt_handler.set_handler(2, self._handle_syscall)

    def set_handler(self, idx, name, arg_count, callback):
        self._handlers[idx] = SyscallHandler(idx, name, arg_count, callback)

    def _handle_syscall(self, mu):
        idx = mu.reg_read(UC_ARM_REG_R7)
        args = [mu.reg_read(reg_idx) for reg_idx in range(UC_ARM_REG_R0, UC_ARM_REG_R6 + 1)]

        if idx in self._handlers:
            handler = self._handlers[idx]
            args = args[:handler.arg_count]
            args_formatted = ", ".join(["%08x" % arg for arg in args])
            logger.debug("Executing syscall %s(%s)" % (handler.name, args_formatted))

            try:
                result = handler.callback(mu, *args)
            except:
                logger.error("An error occured during in %x syscall hander, stopping emulation" % idx)
                mu.emu_stop()
                raise

            if result is not None:
                mu.reg_write(UC_ARM_REG_R0, result)
        else:
            logger.error("Unhandled syscall 0x%x at 0x%x, stopping emulation" % (idx, mu.reg_read(UC_ARM_REG_PC)))
            mu.emu_stop()