blob: 9e4770793c98ac57670cca485b7cbc0624f744be (
plain)
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
|
//
// Copyright © 2020 Arm Ltd. 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.
#pragma once
#if defined(__unix__)
#include <poll.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <sys/un.h>
#elif defined(_MSC_VER)
#include <winsock2.h>
#include <afunix.h>
#endif
namespace armnnUtils
{
namespace Sockets
{
#if defined(__unix__)
using Socket = int;
using PollFd = pollfd;
#elif defined(_MSC_VER)
using Socket = SOCKET;
using PollFd = WSAPOLLFD;
#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 cmd, void* arg);
int Poll(PollFd* fds, size_t numFds, int timeout);
Socket Accept(Socket s, sockaddr* addr, unsigned int* addrlen, int flags);
}
}
|
