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
79
80
81
82
83
|
//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
// This file (along with its corresponding .cpp) defines a very thin platform abstraction layer for the use of
// networking sockets. Thankfully the underlying APIs on Windows and Linux are very similar so not much conversion
// is needed (typically just forwarding the parameters to a differently named function).
// Some of the APIs are in fact completely identical and so no forwarding function is needed.
#if !defined(ARMNN_DISABLE_SOCKETS)
#pragma once
#if defined(__unix__) || defined(__APPLE__)
#include <poll.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <sys/un.h>
#elif defined(_MSC_VER)
#include <WindowsWrapper.hpp>
#include <winsock2.h>
#include <afunix.h>
#elif defined(__MINGW32__)
#include <WindowsWrapper.hpp>
#include <winsock2.h>
#endif
namespace arm
{
namespace pipe
{
#if defined(__unix__)
using Socket = int;
using PollFd = pollfd;
#elif defined(__APPLE__)
using Socket = int;
using PollFd = pollfd;
#define SOCK_CLOEXEC 0
#elif defined(_MSC_VER)
using Socket = SOCKET;
using PollFd = WSAPOLLFD;
using nfds_t = int;
using socklen_t = int;
#define SOCK_CLOEXEC 0
#elif defined(__MINGW32__)
using Socket = SOCKET;
using PollFd = WSAPOLLFD;
using nfds_t = int;
using socklen_t = int;
#define SOCK_CLOEXEC 0
#endif
/// Performs any required one-time setup.
bool Initialize();
int Close(Socket s);
bool SetNonBlocking(Socket s);
long Write(Socket s, const void* buf, size_t len);
long Read(Socket s, void* buf, size_t len);
int Ioctl(Socket s, unsigned long int cmd, void* arg);
int Poll(PollFd* fds, nfds_t numFds, int timeout);
Socket Accept(Socket s, sockaddr* addr, socklen_t* addrlen, int flags);
} // namespace arm
} // namespace pipe
#endif
|
