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