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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
/*
Copyright (C) 2018 Red Hat, Inc.
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, see <http://www.gnu.org/licenses/>.
*/
/* Small compatibility layer for sockets, mostly to make easier portability
* for Windows but without loosing performances under Unix, the most supported
* system */
#ifndef RED_SYS_SOCKET_H_
#define RED_SYS_SOCKET_H_
#ifndef _WIN32
# include <sys/socket.h>
#define socket_read(sock, buf, len) read(sock, buf, len)
#define socket_write(sock, buf, len) write(sock, buf, len)
#define socket_writev(sock, iov, n) writev(sock, iov, n)
#define socket_close(sock) close(sock)
#else
# include <winsock2.h>
# include <windows.h>
# include <spice/macros.h>
SPICE_BEGIN_DECLS
typedef int socklen_t;
// this definition is ABI compatible with WSABUF
struct iovec {
u_long iov_len;
void FAR *iov_base;
};
void socket_win32_set_errno(void);
static inline ssize_t socket_read(int sock, void *buf, size_t count)
{
ssize_t res = recv(sock, (char *) buf, count, 0);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
static inline ssize_t socket_write(int sock, const void *buf, size_t count)
{
ssize_t res = send(sock, (const char *) buf, count, 0);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
static inline ssize_t socket_writev(int sock, const struct iovec *iov, int n_iov)
{
DWORD sent;
int res = WSASend(sock, (LPWSABUF) iov, n_iov, &sent, 0, NULL, NULL);
if (res) {
socket_win32_set_errno();
return -1;
}
return sent;
}
#define socket_close(sock) closesocket(sock)
#define SHUT_RDWR SD_BOTH
static inline int
socket_getsockopt(int sock, int lvl, int type, void *value, socklen_t *len)
{
int res = getsockopt(sock, lvl, type, (char *) value, len);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
#undef getsockopt
#define getsockopt socket_getsockopt
static inline int
socket_setsockopt(int sock, int lvl, int type, const void *value, socklen_t len)
{
int res = setsockopt(sock, lvl, type, (const char*) value, len);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
#undef setsockopt
#define setsockopt socket_setsockopt
static inline int
socket_listen(int sock, int backlog)
{
int res = listen(sock, backlog);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
#undef listen
#define listen socket_listen
static inline int
socket_bind(int sock, const struct sockaddr *addr, int addrlen)
{
int res = bind(sock, addr, addrlen);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
#undef bind
#define bind socket_bind
static inline int
socket_accept(int sock, struct sockaddr *addr, int *addrlen)
{
int res = accept(sock, addr, addrlen);
if (res < 0) {
socket_win32_set_errno();
}
return res;
}
#undef accept
#define accept socket_accept
int socket_newpair(int type, int protocol, int sv[2]);
#define socketpair(family, type, protocol, sv) socket_newpair(type, protocol, sv)
SPICE_END_DECLS
#endif
#if defined(SO_NOSIGPIPE) && defined(__APPLE__)
#define MSG_NOSIGNAL 0
#endif
#endif // RED_SYS_SOCKET_H_
|
