1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
def recv(long s, object bufflist, object obj, unsigned long flags = 0):
cdef int rc, res
cdef myOVERLAPPED *ov
cdef WSABUF *ws_buf
cdef unsigned long bytes
cdef PyObject **buffers
cdef Py_ssize_t i, size, buffcount
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, &size)
ws_buf[i].len = <DWORD>size
ov = makeOV()
if obj is not None:
ov.obj = <PyObject *>obj
rc = WSARecv(s, ws_buf, <DWORD>buffcount, &bytes, &flags, <OVERLAPPED *>ov, NULL)
if rc == SOCKET_ERROR:
rc = WSAGetLastError()
if rc != ERROR_IO_PENDING:
PyMem_Free(ov)
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
cdef Py_ssize_t size
PyObject_AsWriteBuffer(buff, <void **>&ws_buf.buf, &size)
ws_buf.len = <DWORD>size
PyObject_AsWriteBuffer(addr_buff, <void **>&c_addr_buff, &size)
c_addr_buff_len = <int>size
PyObject_AsWriteBuffer(addr_len_buff, <void **>&c_addr_len_buff, &size)
c_addr_len_buff_len = <int>size
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:
PyMem_Free(ov)
return rc, 0
Py_XINCREF(obj)
return rc, bytes
|