use custom malloc function to alloc memory for converting go string to c string

Author: learnforpractice

pysrc/__init__.py

@@ -1,7 +1,11 @@
 from . import mixin_api
 from . import mixin_bot_api
+from . import _mixin
+
 __VERSION__ = '0.2.4'
 
+_mixin.init()
+
 default_api = mixin_api.MixinApi()
 
 def get_mixin_api():

src/CMakeLists.txt

@@ -8,7 +8,7 @@ add_library(_mixin MODULE
 if (WIN32)
     add_compile_definitions(MS_WIN64)
     SET(BUILD_TARGET mixin.lib)
-    SET(BUILD_LIB_CMD go build -o mixin.dll -buildmode=c-shared && cmake -E copy mixin.dll ${CMAKE_CURRENT_SOURCE_DIR}/../pysrc/mixin.dll && gendef mixin.dll && lib /def:mixin.def /machine:x64 /out:mixin.lib && cmake -E copy libmixin.bk.h libmixin.h)
+    SET(BUILD_LIB_CMD go build -o mixin.dll -buildmode=c-shared && cmake -E copy mixin.dll ${CMAKE_CURRENT_SOURCE_DIR}/../pysrc/mixin.dll && gendef mixin.dll && lib /def:mixin.def /machine:x64 /out:mixin.lib)
 else()
     SET(BUILD_TARGET libmixin.a)
     SET(BUILD_LIB_CMD go build -o ${BUILD_TARGET} -buildmode=c-archive)

src/_mixin.pyx

@@ -19,8 +19,9 @@ cdef extern from "<Python.h>":
     object PyBytes_FromStringAndSize(const char* str, int size)
     int _PyLong_AsByteArray(PyLongObject* v, unsigned char* bytes, size_t n, int little_endian, int is_signed)
 
-cdef extern from "libmixin.h" nogil:
-    void Init();
+cdef extern from "wrapper.h" nogil:
+    ctypedef char *(*fn_malloc)(uint64_t size)
+    void Init(fn_malloc fn)
     char* GetMixinVersion()
     int MixinMain(char* args)
     char* CreateAddress(char* _params);
@@ -63,8 +64,11 @@ def main(_args):
     with nogil:
         MixinMain(args)
 
+cdef char *user_malloc(uint64_t size):
+    return <char *>malloc(size)
+
 def init():
-    Init()
+    Init(<fn_malloc>user_malloc)
 
 def get_mixin_version():
     cdef char* ret