OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	03/31/2022 06:22:38 AM	rwxr-xr-x
📄 acceptex.pxi	1.24 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 connectex.pxi	1.09 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 iocpsupport.c	388.6 KB	09/08/2017 10:38:35 AM	rwxr-xr-x
📄 iocpsupport.pyx	12.92 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 winsock_pointers.c	1.59 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 winsock_pointers.h	1.26 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 wsarecv.pxi	2.17 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 wsasend.pxi	624 bytes	09/08/2017 10:38:36 AM	rw-r--r--

Editing: wsarecv.pxi

# 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(obj, bufflist)

rc = WSARecv(s, ws_buf, <DWORD>buffcount, &bytes, &flags, <OVERLAPPED *>ov, NULL)

if rc == SOCKET_ERROR:
            rc = WSAGetLastError()
            if rc != ERROR_IO_PENDING:
                unmakeOV(ov)
                return rc, 0

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(obj, (buff, addr_buff, addr_len_buff))

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:
            unmakeOV(ov)
            return rc, 0

return rc, bytes