diff options
Diffstat (limited to 'portato')
| -rw-r--r-- | portato/mq.pxd | 3 | ||||
| -rw-r--r-- | portato/mq.pyx | 12 |
2 files changed, 8 insertions, 7 deletions
diff --git a/portato/mq.pxd b/portato/mq.pxd index ab5cfbe..be7631b 100644 --- a/portato/mq.pxd +++ b/portato/mq.pxd @@ -10,7 +10,8 @@ # # Written by René 'Necoro' Neumann <necoro@necoro.net> -from stdlib cimport * +from python_string cimport * +from python_mem cimport * cdef extern from "errno.h": int errno diff --git a/portato/mq.pyx b/portato/mq.pyx index a5d0745..e9ce464 100644 --- a/portato/mq.pyx +++ b/portato/mq.pyx @@ -84,7 +84,7 @@ cdef class MessageQueue (object): if size >= MAX_MESSAGE_SIZE: raise ValueError("Message must be smaller than %d", MAX_MESSAGE_SIZE) - msg = <msg_data*>malloc(sizeof(msg_data) + size) + msg = <msg_data*>PyMem_Malloc(sizeof(msg_data) + size) if msg is NULL: raise MemoryError("Out of memory") @@ -93,7 +93,7 @@ cdef class MessageQueue (object): msg.mtype = type with nogil: - ret = msgsnd(self.msgid, &msg, size, 0) + ret = msgsnd(self.msgid, msg, size, 0) try: if ret == -1: @@ -106,14 +106,14 @@ cdef class MessageQueue (object): else: raise OSError(errno, strerror(errno)) finally: - free(msg) + PyMem_Free(msg) def receive (self): cdef msg_data * msg cdef int ret cdef object retTuple - msg = <msg_data*>malloc(sizeof(msg_data) + MAX_MESSAGE_SIZE) + msg = <msg_data*>PyMem_Malloc(sizeof(msg_data) + MAX_MESSAGE_SIZE) if msg is NULL: raise MemoryError("Out of memory") @@ -134,9 +134,9 @@ cdef class MessageQueue (object): else: raise OSError(errno, strerror(errno)) - retTuple = (msg.mtext, msg.mtype) + retTuple = (PyString_FromStringAndSize(msg.mtext, ret), msg.mtype) finally: - free(msg) + PyMem_Free(msg) return retTuple |