3 * This file is part of the libCEC(R) library.
5 * libCEC(R) is Copyright (C) 2011-2012 Pulse-Eight Limited. All rights reserved.
6 * libCEC(R) is an original work, containing original code.
8 * libCEC(R) is a trademark of Pulse-Eight Limited.
10 * This program is dual-licensed; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
25 * Alternatively, you can license this library under a commercial license,
26 * please contact Pulse-Eight Licensing for more information.
28 * For more information contact:
29 * Pulse-Eight Licensing <license@pulse-eight.com>
30 * http://www.pulse-eight.com/
31 * http://www.pulse-eight.net/
35 #include "../util/timeutils.h"
37 #pragma comment(lib, "Ws2_32.lib")
42 #define SHUT_RDWR SD_BOTH
51 #define MSG_WAITALL 0x8
54 inline int GetSocketError(void)
56 int error
= WSAGetLastError();
59 case WSAEINPROGRESS
: return EINPROGRESS
;
60 case WSAECONNRESET
: return ECONNRESET
;
61 case WSAETIMEDOUT
: return ETIMEDOUT
;
62 case WSAEWOULDBLOCK
: return EAGAIN
;
63 default : return error
;
69 inline void SerialSocketClose(serial_socket_t socket
)
71 if (socket
!= INVALID_HANDLE_VALUE
)
75 inline ssize_t
SerialSocketWrite(serial_socket_t socket
, int *iError
, void* data
, size_t len
)
77 if (len
!= (DWORD
)len
)
83 DWORD
iBytesWritten(0);
84 if (socket
!= INVALID_HANDLE_VALUE
)
86 if (!WriteFile(socket
, data
, (DWORD
)len
, &iBytesWritten
, NULL
))
88 *iError
= GetLastError();
91 return (ssize_t
)iBytesWritten
;
97 inline ssize_t
SerialSocketRead(serial_socket_t socket
, int *iError
, void* data
, size_t len
, uint64_t iTimeoutMs
/*= 0*/)
99 if (len
!= (DWORD
)len
)
106 if (socket
!= INVALID_HANDLE_VALUE
)
108 if(!ReadFile(socket
, data
, (DWORD
)len
, &iBytesRead
, NULL
) != 0)
110 *iError
= GetLastError();
113 return (ssize_t
)iBytesRead
;
121 inline void TcpSocketSetBlocking(tcp_socket_t socket
, bool bSetTo
)
123 u_long iSetTo
= bSetTo
? 0 : 1;
124 ioctlsocket(socket
, FIONBIO
, &iSetTo
);
127 inline void TcpSocketClose(tcp_socket_t socket
)
132 inline void TcpSocketShutdown(tcp_socket_t socket
)
134 if (socket
!= INVALID_SOCKET
&&
135 socket
!= SOCKET_ERROR
)
136 shutdown(socket
, SHUT_RDWR
);
139 inline ssize_t
TcpSocketWrite(tcp_socket_t socket
, int *iError
, void* data
, size_t len
)
141 if (socket
== INVALID_SOCKET
||
142 socket
== SOCKET_ERROR
||
149 ssize_t iReturn
= send(socket
, (char*)data
, (int)len
, 0);
150 if (iReturn
< (ssize_t
)len
)
155 inline ssize_t
TcpSocketRead(tcp_socket_t socket
, int *iError
, void* data
, size_t len
, uint64_t iTimeoutMs
/*= 0*/)
157 int64_t iNow(0), iTarget(0);
158 ssize_t
iBytesRead(0);
161 if (socket
== INVALID_SOCKET
||
162 socket
== SOCKET_ERROR
||
172 iTarget
= iNow
+ (int64_t) iTimeoutMs
;
177 while (iBytesRead
>= 0 && iBytesRead
< (ssize_t
)len
&& (iTimeoutMs
== 0 || iTarget
> iNow
))
181 tv
.tv_sec
= (long)(iTimeoutMs
/ 1000);
182 tv
.tv_usec
= 1000 * (long)(iTimeoutMs
% 1000);
185 FD_SET(socket
, &fd_read
);
187 if (select((int)socket
+ 1, &fd_read
, NULL
, NULL
, &tv
) == 0)
189 TcpSocketSetBlocking(socket
, false);
192 ssize_t iReadResult
= (iTimeoutMs
> 0) ?
193 recv(socket
, (char*)data
+ iBytesRead
, (int)(len
- iBytesRead
), MSG_WAITALL
) :
194 recv(socket
, (char*)data
, (int)len
, MSG_WAITALL
);
195 *iError
= GetSocketError();
198 if (errno
== EAGAIN
&& iTimeoutMs
> 0)
203 else if (iReadResult
== 0 || (iReadResult
!= (ssize_t
)len
&& iTimeoutMs
== 0))
205 *iError
= ECONNRESET
;
209 iBytesRead
+= iReadResult
;
213 TcpSocketSetBlocking(socket
, true);
220 inline bool TcpResolveAddress(const char *strHost
, uint16_t iPort
, int *iError
, struct addrinfo
**info
)
222 struct addrinfo hints
;
224 memset(&hints
, 0, sizeof(hints
));
225 hints
.ai_family
= AF_UNSPEC
;
226 hints
.ai_socktype
= SOCK_STREAM
;
227 hints
.ai_protocol
= IPPROTO_TCP
;
228 sprintf(service
, "%d", iPort
);
230 *iError
= getaddrinfo(strHost
, service
, &hints
, info
);
234 inline int TcpGetSocketError(tcp_socket_t socket
)
237 socklen_t optLen
= sizeof(tcp_socket_t
);
238 getsockopt(socket
, SOL_SOCKET
, SO_ERROR
, (char *)&iReturn
, &optLen
);
242 inline bool TcpSetNoDelay(tcp_socket_t socket
)
245 setsockopt(socket
, IPPROTO_TCP
, TCP_NODELAY
, (const char*)&iSetTo
, sizeof(iSetTo
));
249 inline bool TcpConnectSocket(tcp_socket_t socket
, struct addrinfo
* addr
, int *iError
, uint64_t iTimeout
= 0)
251 TcpSocketSetBlocking(socket
, false);
254 int iConnectResult
= connect(socket
, addr
->ai_addr
, (int)addr
->ai_addrlen
);
255 if (iConnectResult
== -1)
257 if (GetSocketError() == EINPROGRESS
||
258 GetSocketError() == EAGAIN
)
260 fd_set fd_write
, fd_except
;
262 tv
.tv_sec
= (long)(iTimeout
/ 1000);
263 tv
.tv_usec
= 1000 * (long)(iTimeout
% 1000);
267 FD_SET(socket
, &fd_write
);
268 FD_SET(socket
, &fd_except
);
270 int iPollResult
= select(sizeof(socket
)*8, NULL
, &fd_write
, &fd_except
, &tv
);
271 if (iPollResult
== 0)
273 else if (iPollResult
== -1)
274 *iError
= GetSocketError();
277 socklen_t errlen
= sizeof(int);
278 getsockopt(socket
, SOL_SOCKET
, SO_ERROR
, (char *)iError
, &errlen
);
287 TcpSocketSetBlocking(socket
, true);