|
 |
dd89bb |
/********************************************************/
|
|
|
dd89bb |
/* ntapi: Native API core library */
|
|
|
dde53a |
/* Copyright (C) 2013--2017 Z. Gilboa */
|
|
|
dd89bb |
/* Released under GPLv2 and GPLv3; see COPYING.NTAPI. */
|
|
|
dd89bb |
/********************************************************/
|
|
|
dd89bb |
|
|
|
dd89bb |
#include <psxtypes/psxtypes.h>
|
|
|
dd89bb |
#include <ntapi/nt_object.h>
|
|
|
dd89bb |
#include <ntapi/nt_file.h>
|
|
|
dd89bb |
#include <ntapi/nt_socket.h>
|
|
|
dd89bb |
#include <ntapi/ntapi.h>
|
|
|
dd89bb |
#include "ntapi_impl.h"
|
|
|
dd89bb |
|
|
|
f6c77a |
typedef struct _nt_afd_recv_udp_info {
|
|
|
f6c77a |
nt_afd_recv_info afd_recv;
|
|
|
f6c77a |
nt_sockaddr * addr;
|
|
|
f6c77a |
nt_sockaddr_reply * raddr;
|
|
|
f6c77a |
} nt_afd_recv_udp_info;
|
|
|
f6c77a |
|
|
|
dd89bb |
int32_t __cdecl __ntapi_sc_recv(
|
|
|
dd89bb |
__in nt_socket * hssocket,
|
|
|
dd89bb |
__in const void * buffer,
|
|
|
dd89bb |
__in size_t len,
|
|
|
dd89bb |
__out ssize_t * bytes_received __optional,
|
|
|
f6c77a |
__in nt_sockaddr_reply * raddr __optional,
|
|
|
dd89bb |
__in uintptr_t afdflags __optional,
|
|
|
dd89bb |
__in uintptr_t tdiflags __optional,
|
|
|
dd89bb |
__out nt_io_status_block * iosb __optional)
|
|
|
dd89bb |
{
|
|
|
dd89bb |
nt_afd_buffer afd_buffer;
|
|
|
f6c77a |
nt_afd_recv_udp_info afd_recv;
|
|
|
dd89bb |
nt_io_status_block siosb;
|
|
|
dd89bb |
|
|
|
dd89bb |
iosb = iosb ? iosb : &siosb;
|
|
|
dd89bb |
|
|
|
f6c77a |
/* conditional connect */
|
|
|
f6c77a |
if (raddr && (hssocket->protocol == NT_IPPROTO_UDP)) {
|
|
|
f6c77a |
afd_recv.raddr = raddr;
|
|
|
f6c77a |
raddr->addrlen = sizeof(raddr->addr);
|
|
|
f6c77a |
afd_recv.addr = &raddr->addr;
|
|
|
f6c77a |
} else
|
|
|
f6c77a |
raddr = 0;
|
|
|
f6c77a |
|
|
|
dd89bb |
/* tdiflags */
|
|
|
dd89bb |
if (tdiflags == 0)
|
|
|
dd89bb |
tdiflags = NT_TDI_RECEIVE_NORMAL;
|
|
|
dd89bb |
|
|
|
dd89bb |
/* afd_buffer */
|
|
|
dd89bb |
afd_buffer.length = len;
|
|
|
dd89bb |
afd_buffer.buffer = (char *)buffer;
|
|
|
dd89bb |
|
|
|
f6c77a |
/* afd_recv.afd_recv */
|
|
|
f6c77a |
afd_recv.afd_recv.afd_buffer_array = &afd_buffer;
|
|
|
f6c77a |
afd_recv.afd_recv.buffer_count = 1;
|
|
|
dd89bb |
|
|
|
f6c77a |
afd_recv.afd_recv.afd_flags = (uint32_t)afdflags;
|
|
|
f6c77a |
afd_recv.afd_recv.tdi_flags = (uint32_t)tdiflags;
|
|
|
dd89bb |
|
|
|
dd89bb |
hssocket->iostatus = __ntapi->zw_device_io_control_file(
|
|
|
dd89bb |
hssocket->hsocket,
|
|
|
dd89bb |
hssocket->hevent,
|
|
|
dd89bb |
0,
|
|
|
dd89bb |
0,
|
|
|
dd89bb |
iosb,
|
|
|
f6c77a |
raddr ? NT_AFD_IOCTL_UDP_RECV : NT_AFD_IOCTL_RECV,
|
|
|
dd89bb |
&afd_recv,
|
|
|
f6c77a |
raddr ? sizeof(afd_recv) : sizeof(afd_recv.afd_recv),
|
|
|
dd89bb |
0,
|
|
|
dd89bb |
0);
|
|
|
dd89bb |
|
|
|
dd89bb |
if (hssocket->iostatus && (hssocket->ntflags & __NT_FILE_SYNC_IO))
|
|
|
dd89bb |
__ntapi->sc_wait(hssocket,iosb,&hssocket->timeout);
|
|
|
dd89bb |
|
|
|
dd89bb |
if (!hssocket->iostatus && bytes_received)
|
|
|
dd89bb |
*bytes_received = iosb->info;
|
|
|
dd89bb |
|
|
|
dd89bb |
return hssocket->iostatus;
|
|
|
dd89bb |
}
|