try:
    import pywraps
    pywraps_there = True
except:
    pywraps_there = False
import _idaapi
import random
import operator
import datetime

if pywraps_there:
    _idaapi.appcall = pywraps.appcall
    from py_idaapi import *
else:
    import idaapi
    from idaapi import *

# ----------------------------------------------------------------------------------------------------------------------------------------------
#<pycode(py_idd)>
import types

# -----------------------------------------------------------------------
class Appcall_array__(object):
    """This class is used with Appcall.array() method"""
    def __init__(self, tp):
        self.__type = tp

    def pack(self, L):
        """Packs a list or tuple into a byref buffer"""
        t = type(L)
        if not (t == types.ListType or t == types.TupleType):
            raise ValueError, "Either a list or a tuple must be passed"
        self.__size = len(L)
        if self.__size == 1:
            self.__typedobj = Appcall__.typedobj(self.__type + ";")
        else:
            self.__typedobj = Appcall__.typedobj("%s x[%d];" % (self.__type, self.__size))
        # Now store the object in a string buffer
        ok, buf = self.__typedobj.store(L)
        if ok:
            return Appcall__.byref(buf)
        else:
            return None

    def try_to_convert_to_list(self, obj):
        """Is this object a list? We check for the existance of attribute zero and attribute self.size-1"""
        if not (hasattr(obj, "0") and hasattr(obj, str(self.__size-1))):
            return obj
        # at this point, we are sure we have an "idc list"
        # let us convert to a Python list
        return [getattr(obj, str(x)) for x in xrange(0, self.__size)]

    def unpack(self, buf, as_list=True):
        """Unpacks an array back into a list or an object"""
        # take the value from the special ref object
        if isinstance(buf, PyIdc_cvt_refclass__):
            buf = buf.value

        # we can only unpack from strings
        if type(buf) != types.StringType:
            raise ValueError, "Cannot unpack this type!"
        # now unpack
        ok, obj = self.__typedobj.retrieve(buf)
        if not ok:
            raise ValueError, "Failed while unpacking!"
        if not as_list:
            return obj
        return self.try_to_convert_to_list(obj)


# -----------------------------------------------------------------------
# Wrapper class for the appcall()
class Appcall_callable__(object):
    """
    Helper class to issue appcalls using a natural syntax:
      appcall.FunctionNameInTheDatabase(arguments, ....)
    or
      appcall["Function@8"](arguments, ...)
    or
      f8 = appcall["Function@8"]
      f8(arg1, arg2, ...)
    or
      o = appcall.obj()
      i = byref(5)
      appcall.funcname(arg1, i, "hello", o)
    """
    def __init__(self, ea, tp = None, fld = None):
        """Initializes an appcall with a given function ea"""
        self.__ea     = ea
        self.__type   = tp
        self.__fields = fld
        self.__options = None # Appcall options
        self.__timeout = None # Appcall timeout

    def __get_timeout(self):
        return self.__timeout

    def __set_timeout(self, v):
        self.__timeout = v

    timeout = property(__get_timeout, __set_timeout)
    """An Appcall instance can change its timeout value with this attribute"""

    def __get_options(self):
        return self.__options if self.__options != None else Appcall__.get_appcall_options()

    def __set_options(self, v):
        if self.timeout:
            # If timeout value is set, then put the timeout flag and encode the timeout value
            v |= Appcall__.APPCALL_TIMEOUT | (self.timeout << 16)
        else:
            # Timeout is not set, then clear the timeout flag
            v &= ~Appcall__.APPCALL_TIMEOUT

        self.__options = v

    options = property(__get_options, __set_options)
    """Sets the Appcall options locally to this Appcall instance"""

    def __call__(self, *args):
        """Make object callable. We redirect execution to idaapi.appcall()"""
        if self.ea is None:
            raise ValueError, "Object not callable!"

        # convert arguments to a list
        arg_list = list(args)

        # Save appcall options and set new global options
        old_opt = Appcall__.get_appcall_options()
        Appcall__.set_appcall_options(self.options)

        # Do the Appcall (use the wrapped version)
        e_obj = None
        try:
            r = _idaapi.appcall(
               self.ea,
               _idaapi.get_current_thread(),
               self.type,
               self.fields,
               arg_list)
        except Exception as e:
            e_obj = e

        # Restore appcall options
        Appcall__.set_appcall_options(old_opt)

        # Return or re-raise exception
        if e_obj:
            raise Exception, e_obj

        return r

    def __get_ea(self):
        return self.__ea

    def __set_ea(self, val):
        self.__ea = val

    ea = property(__get_ea, __set_ea)
    """Returns or sets the EA associated with this object"""

    def __get_size(self):
        if self.__type == None:
            return -1
        r = _idaapi.calc_type_size(_idaapi.cvar.idati, self.__type)
        if not r:
            return -1
        return r

    size = property(__get_size)
    """Returns the size of the type"""

    def __get_type(self):
        return self.__type

    type = property(__get_type)
    """Returns the typestring"""

    def __get_fields(self):
        return self.__fields

    fields = property(__get_fields)
    """Returns the field names"""


    def retrieve(self, src=None, flags=0):
        """
        Unpacks a typed object from the database if an ea is given or from a string if a string was passed
        @param src: the address of the object or a string
        @return: Returns a tuple of boolean and object or error number (Bool, Error | Object).
        """

        # Nothing passed? Take the address and unpack from the database
        if src is None:
            src = self.ea

        if type(src) == types.StringType:
            return _idaapi.unpack_object_from_bv(_idaapi.cvar.idati, self.type, self.fields, src, flags)
        else:
            return _idaapi.unpack_object_from_idb(_idaapi.cvar.idati, self.type, self.fields, src, flags)

    def store(self, obj, dest_ea=None, base_ea=0, flags=0):
        """
        Packs an object into a given ea if provided or into a string if no address was passed.
        @param obj: The object to pack
        @param dest_ea: If packing to idb this will be the store location
        @param base_ea: If packing to a buffer, this will be the base that will be used to relocate the pointers

        @return:
            - If packing to a string then a Tuple(Boolean, packed_string or error code)
            - If packing to the database then a return code is returned (0 is success)
        """

        # no ea passed? thus pack to a string
        if dest_ea is None:
            return _idaapi.pack_object_to_bv(obj,
                                             _idaapi.cvar.idati,
                                             self.type,
                                             self.fields,
                                             base_ea,
                                             flags)
        else:
            return _idaapi.pack_object_to_idb(obj,
                                              _idaapi.cvar.idati,
                                              self.type,
                                              self.fields,
                                              dest_ea,
                                              flags)

# -----------------------------------------------------------------------
class Appcall_consts__(object):
    """Helper class used by Appcall.Consts attribute
    It is used to retrieve constants via attribute access"""
    def __init__(self, default=0):
        self.__default = default

    def __getattr__(self, attr):
        return Appcall__.valueof(attr, self.__default)

# -----------------------------------------------------------------------
class Appcall__(object):
    APPCALL_MANUAL = 0x1
    """
    Only set up the appcall, do not run it.
    you should call CleanupAppcall() when finished
    """

    APPCALL_DEBEV  = 0x2
    """
    Return debug event information
    If this bit is set, exceptions during appcall
    will generate idc exceptions with full
    information about the exception
    """

    APPCALL_TIMEOUT = 0x4
    """
    Appcall with timeout
    The timeout value in milliseconds is specified
    in the high 2 bytes of the 'options' argument:
    If timed out, errbuf will contain "timeout".
    """

    def __init__(self):
        self.__consts = Appcall_consts__()
    def __get_consts(self):
        return self.__consts
    Consts = property(__get_consts)
    """Use Appcall.Consts.CONST_NAME to access constants"""

    @staticmethod
    def __name_or_ea(name_or_ea):
        """
        Function that accepts a name or an ea and checks if the address is enabled.
        If a name is passed then idaapi.get_name_ea() is applied to retrieve the name
        @return:
            - Returns the resolved EA or
            - Raises an exception if the address is not enabled
        """

        # a string? try to resolve it
        if type(name_or_ea) == types.StringType:
            ea = _idaapi.get_name_ea(_idaapi.BADADDR, name_or_ea)
        else:
            ea = name_or_ea
        # could not resolve name or invalid address?
        if ea == _idaapi.BADADDR or not _idaapi.isEnabled(ea):
            raise ValueError, "Undefined function " + name_or_ea
        return ea

    @staticmethod
    def proto(name_or_ea, prototype, flags = None):
        """
        Allows you to instantiate an appcall (callable object) with the desired prototype
        @param name_or_ea: The name of the function (will be resolved with LocByName())
        @param prototype:
        @return:
            - On failure it raises an exception if the prototype could not be parsed
              or the address is not resolvable
            - Returns a callbable Appcall instance with the given prototypes and flags
        """

        # resolve and raise exception on error
        ea = Appcall__.__name_or_ea(name_or_ea)
        # parse the type
        if flags is None:
            flags = 1 | 2 | 4 # PT_SIL | PT_NDC | PT_TYP

        result = _idaapi.idc_parse_decl(_idaapi.cvar.idati, prototype, flags)
        if result is None:
            raise ValueError, "Could not parse type: " + prototype

        # Return the callable method with type info
        return Appcall_callable__(ea, result[1], result[2])

    def __getattr__(self, name_or_ea):
        """Allows you to call functions as if they were member functions (by returning a callable object)"""
        # resolve and raise exception on error
        ea = self.__name_or_ea(name_or_ea)
        if ea == _idaapi.BADADDR:
            raise ValueError, "Undefined function " + name
        # Return the callable method
        return Appcall_callable__(ea)

    def __getitem__(self, idx):
        """
        Use self[func_name] syntax if the function name contains invalid characters for an attribute name
        See __getattr___
        """
        return self.__getattr__(idx)

    @staticmethod
    def valueof(name, default=0):
        """
        Returns the numeric value of a given name string.
        If the name could not be resolved then the default value will be returned
        """
        t, v = _idaapi.get_name_value(_idaapi.BADADDR, name)
        if t == 0: # NT_NONE
          v = default
        return v

    @staticmethod
    def int64(v):
        """Whenever a 64bit number is needed use this method to construct an object"""
        return PyIdc_cvt_int64__(v)

    @staticmethod
    def byref(val):
        """
        Method to create references to immutable objects
        Currently we support references to int/strings
        Objects need not be passed by reference (this will be done automatically)
        """
        return PyIdc_cvt_refclass__(val)

    @staticmethod
    def buffer(str = None, size = 0, fill="\x00"):
        """
        Creates a string buffer. The returned value (r) will be a byref object.
        Use r.value to get the contents and r.size to get the buffer's size
        """
        if str is None:
            str = ""
        left = size - len(str)
        if left > 0:
            str = str + (fill * left)
        r = Appcall__.byref(str)
        r.size = size
        return r

    @staticmethod
    def obj(**kwds):
        """Returns an empty object or objects with attributes as passed via its keywords arguments"""
        return object_t(**kwds)

    @staticmethod
    def cstr(val):
        return as_cstr(val)

    @staticmethod
    def unicode(s):
        return as_unicode(s)

    @staticmethod
    def array(type_name):
        """Defines an array type. Later you need to pack() / unpack()"""
        return Appcall_array__(type_name)

    @staticmethod
    def typedobj(typestr, ea=None):
        """
        Parses a type string and returns an appcall object.
        One can then use retrieve() member method
        @param ea: Optional parameter that later can be used to retrieve the type
        @return: Appcall object or raises ValueError exception
        """
        # parse the type
        result = _idaapi.idc_parse_decl(_idaapi.cvar.idati, typestr, 1 | 2 | 4) # PT_SIL | PT_NDC | PT_TYP
        if result is None:
            raise ValueError, "Could not parse type: " + typestr
        # Return the callable method with type info
        return Appcall_callable__(ea, result[1], result[2])

    @staticmethod
    def set_appcall_options(opt):
        """Method to change the Appcall options globally (not per Appcall)"""
        old_opt = Appcall__.get_appcall_options()
        _idaapi.cvar.inf.appcall_options = opt
        return old_opt

    @staticmethod
    def get_appcall_options():
        """Return the global Appcall options"""
        return _idaapi.cvar.inf.appcall_options

    @staticmethod
    def cleanup_appcall(tid = 0):
        """Equivalent to IDC's CleanupAppcall()"""
        return _idaapi.cleanup_appcall(tid)

Appcall = Appcall__()
#</pycode(py_idd)>

# ----------------------------------------------------------------------------------------------------------------------------------------------

#<pycode(appcalltest)>
a = Appcall # Take a shortcut to Appcall
c = Appcall.Consts # take shortcut to constants

# -----------------------------------------------------------------------
# - Adds missing types
# - ...
def init():
    # add neeeded types
    if a.valueof("ERROR_FILE_NOT_FOUND") == 0:
        add_type("MACRO_ERROR")

    setattr(c, "ERROR_FILE_NOT_FOUND", 2)

    if a.valueof("MEM_COMMIT") == 0:
        add_type("MACRO_PAGE")

    # open log file
    global test_log
    try:
        test_log = file("python_test.log", "a")
    except:
        test_log = None

# -----------------------------------------------------------------------
def deinit():
    global test_log
    test_log.close()
    test_log = None
    return 1

# -----------------------------------------------------------------------
def logprint(s):
    if not test_log:
        print s
    else:
        test_log.write(s + "\n")
    return 1

# -----------------------------------------------------------------------
def add_type(t):
        print "adding type:", t
        idaapi.import_type(idaapi.cvar.idati, 0, t)

# -----------------------------------------------------------------------
def test_init():
    hmod = getmodulehandlew(a.unicode("kernel32.dll"))
    print "k32hmod=%x" % hmod
    if hmod == 0:
        return -1
    p = getprocaddr(hmod, "VirtualAlloc")
    if p == 0:
        return -2
    print "VirtualAlloc->%x" % p
    virtualalloc.ea = p

    p = getprocaddr(hmod, "VirtualFree")
    if p == 0:
        return -3
    print "VirtualFree->%x" % p
    virtualfree.ea = p

    m = virtualalloc(0, c.MEM_COMMIT, 0x1000, c.PAGE_EXECUTE_READWRITE)
    idc.RefreshDebuggerMemory()
    print "%x: allocated memory\n" % m
    global WRITE_AREA
    WRITE_AREA = m

    return 1

# -----------------------------------------------------------------------
def test_deinit():
    virtualfree(WRITE_AREA, 0, c.MEM_FREE)

# -----------------------------------------------------------------------
# Tests changedir/setdir/buffer creation (two methods) and cstr()
def test_enum_files():
    # create a buffer
    savedpath = a.byref("\x00" * 260)
    # get current directory
    n = getcurdir(250, savedpath)
    out = []
    out.append("curdir=%s" % savedpath.value[0:n])

    # get windir
    windir = a.buffer(size=260) # create a buffer using helper function
    n = getwindir(windir, windir.size)
    if n == 0:
        return -1 # could not get current directory

    windir = windir.value[:n]
    out.append("windir=%s" % windir)

    # change to windows folder
    setcurdir(windir)

    # initiate find
    fd = a.obj()
    h = findfirst("*.exe", fd)
    if h == -1:
        return -2 # no files found!

    found = -6
    while True:
        fn = a.cstr(fd.cFileName)
        if "regedit" in fn:
            found = 1
        out.append("fn=%s<" % fn)
        fd = a.obj() # reset the FD object
        ok = findnext(h, fd)
        if not ok:
            break
    #
    findclose(h)

    # restore cur dir
    setcurdir(savedpath.value)

    # verify
    t = a.buffer(size=260)
    n = getcurdir(t.size, t)
    if t.cstr() != savedpath.cstr():
        return -4 # could not restore cur dir

    out.append("curdir=%s<" % t.cstr())
#    print "all done!"
#    for l in out:
#        print l

    return found
# -----------------------------------------------------------------------
def test_gpa():
    h = loadlib("user32.dll")
    if h == 0:
        print "failed to load library!"
        return -1
    p = getprocaddr(h, "FindWindowA")
    if p == 0:
        print "failed to gpa!"
        return -2
    findwin = a.proto(p, "int FindWindow(LPCTSTR lpClassName, LPCTSTR lpWindowName);")
    hwnd = findwin("TIdaWindow", 0)
    freelib(h)
    print "%x: ok!->hwnd=%x" % (p, hwnd)

    return 1

# -----------------------------------------------------------------------
def test_loaddll():
    h = loadlib("twain_32.dll")
    if h == 0:
        print "failed to load twain32 library!"
        return -1
    return 1

# -----------------------------------------------------------------------
# Packs a simple structure (into the database) and unpacks it back using the idaapi methods
def test_pck_idb_raw():
    name, tp, flds = idc.ParseType("struct { int a, b; char x[5];};", 0)
    o = a.obj(a=15, b=17,x="hi")
    idaapi.pack_object_to_idb(o, idaapi.cvar.idati, tp, flds, WRITE_AREA, 0)

    ok, obj = idaapi.unpack_object_from_idb(idaapi.cvar.idati, tp, flds, WRITE_AREA, 0)
    if obj.a != 15 and obj.b != 17 and obj != "hi":
        return -2
    return 1

# -----------------------------------------------------------------------
# Packs a simple structure (into a string) and unpacks it back using the idaapi methods
def test_pck_bv_raw():
    name, tp, flds = idc.ParseType("struct { int a; char x[5]; int b;};", 0)
    va,vb,vx = 15,17,"hi"
    o = a.obj(a=va, b=vb,x=vx)
    ok, s = idaapi.pack_object_to_bv(o, idaapi.cvar.idati, tp, flds, WRITE_AREA, 0)
    if not ok:
        return -1
    ok, obj = idaapi.unpack_object_from_idb(idaapi.cvar.idati, tp, flds, WRITE_AREA, 0)
    if obj.a != va and obj.b != vb and obj.x != vx:
        return -2
    return 1

# -----------------------------------------------------------------------
# 1. Unpacks the DOS header at 0x400000 and verify the fields
# 2. Unpacks a string and see if it is unpacked correctly
def test_unpack_raw():
    name, tp, flds = idc.ParseType("IMAGE_DOS_HEADER;", 0)
    ok, obj = idaapi.unpack_object_from_idb(idaapi.cvar.idati, tp, flds, 0x400000, 0)
    if obj.e_magic != 23117 and obj.e_cblp != 144:
        return -1

    name, tp, flds = idc.ParseType("struct abc_t { int a, b;};", 0)
    ok, obj = idaapi.unpack_object_from_bv(idaapi.cvar.idati, tp, flds, "\x01\x00\x00\x00\x02\x00\x00\x00", 0)
    if obj.a != 1 and obj.b != 2:
        return -2

    return 1

# -----------------------------------------------------------------------
# Packs/Unpacks a structure to the database using appcall facilities
def test_pck_idb():
    print "%x: ..." % WRITE_AREA
    tp = a.typedobj("struct { int a, b; char x[5];};")
    o = a.obj(a=16, b=17,x="zzzhi")
    if tp.store(o, WRITE_AREA) != 0:
        return -1
    idc.RefreshDebuggerMemory()

    ok, r = tp.retrieve(WRITE_AREA)
    if not ok:
        return -2
    if r.a != o.a and r.b != o.b and r.x != o.x:
        return -3

    return 1

# -----------------------------------------------------------------------
# Packs/Unpacks a structure to/from a string
def test_pck_bv():
    tp = a.typedobj("struct { int a, b; char x[5];};")
    o = a.obj(a=16, b=17,x="zzzhi")
    ok, packed = tp.store(o)
    if not ok:
        return -1
    print "packed->", repr(packed)
    ok, r = tp.retrieve(packed)
    if not ok:
        return -2

    if r.a != o.a and r.b != o.b and r.x != o.x:
        return -3

    return 1

# -----------------------------------------------------------------------
def test_local_types():
  (type, fields) = GetLocalTinfo(1)
  if not type:
    return -1
  decl = GetLocalType(1, PRTYPE_MULTI)
  if decl != "enum\n"\
           + "{\n"\
           + "  FEATURE_OBJECT_CACHING = 0x0,\n"\
           + "  FEATURE_ZONE_ELEVATION = 0x1,\n"\
           + "  FEATURE_MIME_HANDLING = 0x2,\n"\
           + "  FEATURE_MIME_SNIFFING = 0x3,\n"\
           + "  FEATURE_WINDOW_RESTRICTIONS = 0x4,\n"\
           + "  FEATURE_WEBOC_POPUPMANAGEMENT = 0x5,\n"\
           + "  FEATURE_BEHAVIORS = 0x6,\n"\
           + "  FEATURE_DISABLE_MK_PROTOCOL = 0x7,\n"\
           + "  FEATURE_LOCALMACHINE_LOCKDOWN = 0x8,\n"\
           + "  FEATURE_SECURITYBAND = 0x9,\n"\
           + "  FEATURE_RESTRICT_ACTIVEXINSTALL = 0xA,\n"\
           + "  FEATURE_VALIDATE_NAVIGATE_URL = 0xB,\n"\
           + "  FEATURE_RESTRICT_FILEDOWNLOAD = 0xC,\n"\
           + "  FEATURE_ADDON_MANAGEMENT = 0xD,\n"\
           + "  FEATURE_PROTOCOL_LOCKDOWN = 0xE,\n"\
           + "  FEATURE_HTTP_USERNAME_PASSWORD_DISABLE = 0xF,\n"\
           + "  FEATURE_SAFE_BINDTOOBJECT = 0x10,\n"\
           + "  FEATURE_UNC_SAVEDFILECHECK = 0x11,\n"\
           + "  FEATURE_GET_URL_DOM_FILEPATH_UNENCODED = 0x12,\n"\
           + "  FEATURE_TABBED_BROWSING = 0x13,\n"\
           + "  FEATURE_SSLUX = 0x14,\n"\
           + "  FEATURE_DISABLE_NAVIGATION_SOUNDS = 0x15,\n"\
           + "  FEATURE_DISABLE_LEGACY_COMPRESSION = 0x16,\n"\
           + "  FEATURE_FORCE_ADDR_AND_STATUS = 0x17,\n"\
           + "  FEATURE_XMLHTTP = 0x18,\n"\
           + "  FEATURE_DISABLE_TELNET_PROTOCOL = 0x19,\n"\
           + "  FEATURE_FEEDS = 0x1A,\n"\
           + "  FEATURE_BLOCK_INPUT_PROMPTS = 0x1B,\n"\
           + "  FEATURE_ENTRY_COUNT = 0x1C,\n"\
           + "} _tagINTERNETFEATURELIST\n":
      print "decl = " + decl
      return -2
  return 1
# various tests
def test1(stage):
    # call a method that takes a string buffer and appends a dot to its end
    if stage == st_ref2:
        buf = a.buffer("test", 100)
        vals = [378, 424, 470]
        for i in xrange(0, 2+1):
            n = a._ref2(buf)
            if buf.value[4+i] != '.':
                return -st_ref2
            if vals[i] != n:
                return -stage
    # call a method that takes an integer reference
    elif stage == st_ref1:
        v = 5
        i = a.byref(v)
        a._ref1(i)
        if v + 1 != i.value:
            return -stage
    # call a method that takes an array of integers
    elif stage == st_ref3:
        # create an array type
        arr = a.array("int")
        # create a list
        L = [x for x in xrange(1, 10)]
        # pack the list
        p_list = arr.pack(L)
        # appcall to compute the total
        c_total = a._ref3(p_list, len(L))
        # internally compute the total
        total = reduce(operator.add, L)
        if total != c_total:
            return -stage
    # subst()
    elif stage == st_subst:
        v = a._subst(5, 1)
        if v != 4:
            return -stage
        v = subst(5, 1) # subst() / pascal
        if v != -4:
            return -stage*2
    elif stage == st_make2:
        x = a._str2_make(5)
        s = a.cstr(x.next.str)
#        print "len=%d;<%s>" % (len(s), s)
        if s != "This is string 2":
            return -stage
        n = a._str2_print(x)
        if n != 5:
            return -st_make2*2
    elif stage == st_make1:
        x = a._str1_make(6)
        if x.val != 1 and x.next.val != 2:
            return -st_make1
        n = a._str1_print(x)
        if n != 6:
            return -stage
    # 64bit test
    elif stage == st_make3:
        global gr
        try:
            x = a._str3_make(6)
            gr = x
        except Exception as e:
            print "Exception: ", str(e)
            return -stage
        print "x.val32=", x.val32
        if (x.val32 != 1) and (x.val64 != (1 << 32)):
            return -stage * 2
        va = 0
        vb = 0
        r = x
        i = 0
        while x != 0:
            i += 1
            print "1"
            va += x.val32
            print "2"
            vb += x.val64.value
            print "3"
            x = x.next
            print "i=", i, "a=", va, "b=", vb
        if va != 21 and vb != 90194313216:
            return -stage*3
    elif stage == st_asm:
        n = asm1(5, 1)
        if n != 4:
            return -stage
        n = asm2(5, 1, 1)
        if n != 7:
            return -stage*2
    elif stage == st_byvalref1:
        v1 = a.obj(val=5, next=0)
        v2 = a.obj(str="Hello", next=0)
        n = a._byvalref1(v1, 1, v2)
        if n != 78:
            return -stage
        # v1 is passed by ref, thus it will be changed
        if n + 1 != v1.val:
            return -stage * 2
    elif stage == st_altsum:
        # 1 + 2 - 3 + 5 = 5
        n = altsum(1, 2, 3, 5, 0)
        if n != 5:
          return -stage;
    elif stage == st_op64:
        # invalid opcode, should return -1
        r = a._op_two64(1, 2, 6).value
        print "r=", r
        if r != -1:
            return -stage
        r = a._op_two64(6, a.int64(2), 3).value
        if r != 12:
            return -stage * 2
        return 1
    elif stage == st_byval3:
        o = a.obj(val=6,next=a.obj(val=-1, next=0))
#        print "before: o.val=%d o.next.val=%d" % (o.val, o.next.val)
        n = a._byval3(o)
        if n != 5:
            return -stage
#        print "after: o.val=%d o.next.val=%d, n=%d" % (o.val, o.next.val, n)
    #---------
    elif stage == st_ex:
        def doit():
            try:
                # causes a DebugBreak()
                a._ex2(1)
            except Exception as e:
                if not "Software breakpoint" in e.message:
                    return -stage
            try:
                a._ex1(1, 2)
                return -st_ex * 2
            except Exception as e:
                if not "referenced memory" in e.message:
                    return -stage
            return 1
        old = a.set_appcall_options(0)
        r = doit()
        a.set_appcall_options(old)
        if r <= 0:
            return r
    #---------
    elif stage == st_ref4:
        i = a.int64(5)
        v = a.byref(i)
        if a._ref4(v) != 1:
            return -st_ref4
        # test (in case recycling failed)
        if v.value.value != 6:
            return -st_ref4 * 2
        # test recycling
        if i.value != 6:
            return -st_ref4 * 3
    # return success
    return 1

# -----------------------------------------------------------------------
def test2():
    fn = "pc_win32_appcall.pe"

    # Try to open an non-existing file
    h = a._file_open(fn + ".1")
    if h != 0:
        return -1

    n = getlasterror()
    print "gle=", n
    if n != c.ERROR_FILE_NOT_FOUND:
        return -2

    # Should succeed
    h = a._file_open(fn)
    if h == 0:
        return -3

    s = a.buffer("", 10, "!")
    n = a._file_read(h, s, 2)
#    print "read=%d; buf=%s" % (n, repr(s.value))
    if s.value[:2] != "MZ":
        return -4

    n = a._file_close(h)
    if n != h-1:
        return -5

    return 1

# -----------------------------------------------------------------------
# This test changes the appcall options and sees if the appcall
# generates an exception and if we get it properly
# An appcall can throw an exception:
# - ValueError: conversion from/to idc/py failed
# - OSError: an OSError when APPCALL_DEBEV in Appcall_Options. In that case check the exc.args[0] to get the debug event
# - Exception: in all other cases
def test_exec_throw():
    old = a.set_appcall_options(0)
    print "old_opt=", old, "new=0"
    try:
        # causes a divide by zero exception (will be reported as an Exception)
        print a._op_two64(2,0,4).value
        return -1
    except Exception as e:
        print "runtime error!"

    # now appcall exceptions will be reported as OSErr and other exceptions as Exception
    print "old_opt=", a.set_appcall_options(a.APPCALL_DEBEV), "new=2"
    try:
        # causes an error: Wrong number of arguments
        print a._op_two64(2).value
        return -2
    except OSError, e:
        return -3
    except Exception as e:
        print "Got other exception:", e # good

    try:
        # causes an OS error and "e" will contain the last debug_event
        # in our case exception, and the code is int_divide_zero = 0xC0000094
        print a._op_two64(2, 0, 4).value
        return -4
    except OSError, e:
        if idaapi.as_uint32(e.args[0].code) != 0xC0000094:
            return -5
    except Exception as e:
        print "Got other exception:", e
        return -6
    a.set_appcall_options(old)
    return 1
# -----------------------------------------------------------------------
# all the tests that take zero parameters
tests0 = (test_gpa, test_pck_idb_raw, test_pck_bv_raw,
          test_unpack_raw, test_pck_idb, test_pck_bv, test_local_types,
          test_enum_files, test2, test_exec_throw, test_loaddll)
test_log = None # test log file

# -----------------------------------------------------------------------
def test_all():
    if test_init() <= 0:
        print "test_init() failed!"
        return -1

    # tests 0
    for t in tests0:
        print "testing->", t
        r = t()
        if r <= 0:
            return r
    # test 1
    for i in xrange(1, st_last):
        print "test1 #", i
        r = test1(i)
        if r <= 0:
            return r

    logprint(datetime.date.today().strftime("Python was here: %Y-%m-%d @ %I:%M:%S%p"))

    test_deinit()

    return 1

# -----------------------------------------------------------------------
init()

# reference to an integer. use i.value to dereference
i = a.byref(5)

# object representing the str1_t type
o = a.obj(val=5,next=a.obj(val=-2, next=0))

# dictionary representing the str1_t type
# (dictionaries will be converted into IDC objects)
d = {'val':5, 'next': {'val':-2, 'next':0} }

# initialize some pointers
findfirst           = a["__imp__FindFirstFileA@8"]
findnext            = a["__imp__FindNextFileA@8"]
findclose           = a["__imp__FindClose@4"]
getlasterror        = a["__imp__GetLastError@0"]
setcurdir           = a["__imp__SetCurrentDirectoryA@4"]
beep                = a["__imp__Beep@8"]
getwindir           = a["__imp__GetWindowsDirectoryA@8"]
getprocaddr         = a.proto("__imp__GetProcAddress@8", "int (__stdcall *GetProcAddress)(int hModule, LPCSTR lpProcName);")
getcurdir           = a["__imp__GetCurrentDirectoryA@8"]
loadlib             = a.proto("__imp__LoadLibraryA@4", "int (__stdcall *LoadLibraryA)(const char *lpLibFileName);")
freelib             = a.proto("__imp__FreeLibrary@4", "int (__stdcall *FreeLibrary)(int hLibModule);")
setlasterror        = a.typedobj("void __stdcall SetLastError(int dwErrCode);")
getmodulehandlew    = a.proto("__imp__GetModuleHandleW@4", "int (__stdcall *GetModuleHandleW)(LPCWSTR lpModuleName);")
virtualalloc        = a.typedobj("int __stdcall VirtualAlloc(int lpAddress, SIZE_T dwSize, DWORD flAllocationType, DWORD flProtect);")
# typed objects can become calling if an EA was provided. Thus virtualfree.ea = some_address, then we can call virtualfree(., .., ...)
virtualfree         = a.typedobj("BOOL __stdcall VirtualFree(int lpAddress, SIZE_T dwSize, DWORD dwFreeType);")
asm1                = a.proto("_asm1", "int __usercall asm1<eax>(int a<esi>, int b<edi>);")
asm2                = a.proto("_asm2", "int __usercall asm2<eax>(int a, int b, int c<ecx>);")
asm3                = a.proto("_asm3", "int __usercall asm3<edx>(int f<eax>, int a, int b, int c<ecx>);")
asm4                = a.proto("_asm4", "unsigned __int16 __usercall asm4<al:ah>(unsigned __int8 a<bl>, unsigned __int8 b<cl>);")
asm5                = a.proto("_asm5", "unsigned int __usercall asm5<si:bx>(unsigned __int8 a<al>, unsigned __int8 b<cl>, unsigned __int8 c<dl>, int d<ebx>);")
altsum              = a.proto("_va_altsum", "int __cdecl va_altsum(int n1, ...);")
getcommandline      = a.proto("__imp__GetCommandLineA@0", "LPSTR (__stdcall *GetCommandLineA)();")

# make an appcall with a user defined prototype
subst               = a.proto("_subst", "int __pascal subst(int a, int b);")

# some test identifiers
st_ref1 = 1
st_ref2 = 2
st_ref3 = 3
st_subst = 4
st_make1 = 5
st_make2 = 6
st_asm = 7
st_ex = 8
st_byval3 = 9
st_byvalref1 = 10
st_altsum = 11
st_make3 = 12
st_op64 = 13
st_ref4 = 14
st_last = 15
# some area where we can write some bytes _safely_
WRITE_AREA   = 0x400020
gr = None
#</pycode(appcalltest)>

# initialize the test
#test_init()