Hooking a function with a custom calling convention

import memory

from memory import DataType
from memory.hooks import PreHook

from _memory import Register
from _memory import CallingConvention


class x86MsCCSPlayerDeafen(CallingConvention):
    def get_registers(self):
        return [Register.ECX, Register.XMM1]

    def get_pop_size(self):
        return 0

    def get_argument_ptr(self, index, registers):
        if index == 0:
            return registers.ecx.address
        elif index == 1:
            return registers.xmm1.address

        raise IndexError

    def argument_ptr_changed(self, index, registers):
        pass

    def get_return_ptr(self, registers):
        pass

    def return_ptr_changed(self, registers, return_ptr):
        pass


server = memory.find_binary('server')

deafen = server[b'\x55\x8B\xEC\x83\xEC\x28\x56\x57\x8B\xF9\xF3\x0F\x11\x4D\xFC'].make_function(
    x86MsCCSPlayerDeafen,
    (DataType.POINTER, DataType.FLOAT),
    DataType.VOID
)

@PreHook(deafen)
def my_hook(args):
    print(tuple(args))
    return 0