Your IP : 172.28.240.42
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
def recv(long s, object bufflist, object obj, unsigned long flags = 0):
cdef int rc, buffcount, i, res
cdef myOVERLAPPED *ov
cdef WSABUF *ws_buf
cdef unsigned long bytes
cdef PyObject **buffers
bufflist = PySequence_Fast(bufflist, 'second argument needs to be a list')
buffcount = PySequence_Fast_GET_SIZE(bufflist)
buffers = PySequence_Fast_ITEMS(bufflist)
ws_buf = <WSABUF *>PyMem_Malloc(buffcount*sizeof(WSABUF))
try:
for i from 0 <= i < buffcount:
PyObject_AsWriteBuffer(<object>buffers[i], <void **>&ws_buf[i].buf, <int *>&ws_buf[i].len)
ov = makeOV()
if obj is not None:
ov.obj = <PyObject *>obj
rc = WSARecv(s, ws_buf, buffcount, &bytes, &flags, <OVERLAPPED *>ov, NULL)
if rc == SOCKET_ERROR:
rc = WSAGetLastError()
if rc != ERROR_IO_PENDING:
return rc, 0
Py_XINCREF(obj)
return rc, bytes
finally:
PyMem_Free(ws_buf)
def recvfrom(long s, object buff, object addr_buff, object addr_len_buff, object obj, unsigned long flags = 0):
cdef int rc, c_addr_buff_len, c_addr_len_buff_len
cdef myOVERLAPPED *ov
cdef WSABUF ws_buf
cdef unsigned long bytes
cdef sockaddr *c_addr_buff
cdef int *c_addr_len_buff
PyObject_AsWriteBuffer(buff, <void **>&ws_buf.buf, <int *>&ws_buf.len)
PyObject_AsWriteBuffer(addr_buff, <void **>&c_addr_buff, &c_addr_buff_len)
PyObject_AsWriteBuffer(addr_len_buff, <void **>&c_addr_len_buff, &c_addr_len_buff_len)
if c_addr_len_buff_len != sizeof(int):
raise ValueError, 'length of address length buffer needs to be sizeof(int)'
c_addr_len_buff[0] = c_addr_buff_len
ov = makeOV()
if obj is not None:
ov.obj = <PyObject *>obj
rc = WSARecvFrom(s, &ws_buf, 1, &bytes, &flags, c_addr_buff, c_addr_len_buff, <OVERLAPPED *>ov, NULL)
if rc == SOCKET_ERROR:
rc = WSAGetLastError()
if rc != ERROR_IO_PENDING:
return rc, 0
Py_XINCREF(obj)
return rc, bytes