dfhack/dfhack-python/process.py

from ctypes import *
from dftypes import libdfhack
from util import check_pointer_cache

libdfhack.Process_readQuad.argtypes = [ c_void_p, c_uint, POINTER(c_ulong) ]
libdfhack.Process_writeQuad.argtypes = [ c_void_p, c_uint, c_ulong ]

libdfhack.Process_readDWord.argtypes = [ c_void_p, c_uint, POINTER(c_uint) ]
libdfhack.Process_writeDWord.argtypes = [ c_void_p, c_uint, c_uint ]

libdfhack.Process_readWord.argtypes = [ c_void_p, c_uint, POINTER(c_ushort) ]
libdfhack.Process_writeWord.argtypes = [ c_void_p, c_uint, c_ushort ]

libdfhack.Process_readByte.argtypes = [ c_void_p, c_uint, POINTER(c_ubyte) ]
libdfhack.Process_writeByte.argtypes = [ c_void_p, c_uint, c_ubyte ]

libdfhack.Process_readFloat.argtypes = [ c_void_p, c_uint, POINTER(c_float) ]

libdfhack.Process_read.argtypes = [ c_void_p, c_uint, c_uint ]
libdfhack.Process_read.restype = c_void_p

libdfhack.Process_write.argtypes = [ c_void_p, c_uint, c_uint, POINTER(c_ubyte) ]

libdfhack.Process_getThreadIDs.restype = c_void_p

class Process(object):
    def __init__(self, ptr):
        self._p_ptr = ptr
    
    def attach(self):
        return libdfhack.Process_attach(self._p_ptr) > 0
    
    def detach(self):
        return libdfhack.Process_detach(self._p_ptr) > 0
    
    def suspend(self):
        return libdfhack.Process_suspend(self._p_ptr) > 0
    
    def async_suspend(self):
        return libdfhack.Process_asyncSuspend(self._p_ptr) > 0
    
    def resume(self):
        return libdfhack.Process_resume(self._p_ptr) > 0
    
    def force_resume(self):
        return libdfhack.Process_forceresume(self._p_ptr) > 0
    
    def read_quad(self, address):
        q = c_ulong(0)
        
        if libdfhack.Process_readQuad(self._p_ptr, address, byref(q)) > 0:
            return long(q.value)
        else:
            return -1
    
    def write_quad(self, address, value):
        return libdfhack.Process_writeQuad(self._p_ptr, address, value) > 0
    
    def read_dword(self, address):
        d = c_uint(0)
        
        if libdfhack.Process_readDWord(self._p_ptr, address, byref(d)) > 0:
            return int(d.value)
        else:
            return -1
    
    def write_dword(self, address, value):
        return libdfhack.Process_writeDWord(self._p_ptr, address, value) > 0
    
    def read_word(self, address):
        s = c_ushort(0)
        
        if libdfhack.Process_readWord(self._p_ptr, address, byref(s)) > 0:
            return int(s.value)
        else:
            return -1
    
    def write_word(self, address, value):
        return libdfhack.Process_writeWord(self._p_ptr, address, value) > 0
    
    def read_byte(self, address):
        b = c_ubyte(0)
        
        if libdfhack.Process_readByte(self._p_ptr, address, byref(b)) > 0:
            return int(b.value)
        else:
            return -1
    
    def write_byte(self, address, value):
        return libdfhack.Process_writeByte(self._p_ptr, address, value) > 0
    
    def read_float(self, address):
        f = c_float(0)
        
        if libdfhack.Process_readFloat(self._p_ptr, address, byref(f)) > 0:
            return float(f.value)
        else:
            return -1
    
    def read(self, address, length):
        return check_pointer_cache(libdfhack.Process_read(self._p_ptr, address, length))
    
    def write(self, address, length, buffer):
        libdfhack.Process_write(self._p_ptr, address, length, byref(buffer))
    
    def get_thread_ids(self):
        return check_pointer_cache(libdfhack.Process_getThreadIDs(self._p_ptr))
    
    def set_and_wait(self, state):
        return libdfhack.Process_SetAndWait(self._p_ptr, state) > 0
    
    @property
    def is_suspended(self):
        return libdfhack.Process_isSuspended(self._p_ptr) > 0
    
    @property
    def is_attached(self):
        return libdfhack.Process_isAttached(self._p_ptr) > 0
    
    @property
    def is_identified(self):
        return libdfhack.Process_isIdentified(self._p_ptr) > 0
    
    @property
    def is_snapshot(self):
        return libdfhack.Process_isSnapshot(self._p_ptr) > 0
    
    @property
    def pid(self):
        p = c_int(0)
        
        if libdfhack.Process_getPID(self._p_ptr, byref(p)) > 0:
            return int(p.value)
        else:
            return -1