src/lib/platform/windows/os-socket.h

   1 #pragma once
   2 /*
   3  * This file is part of the libCEC(R) library.
   4  *
   5  * libCEC(R) is Copyright (C) 2011-2012 Pulse-Eight Limited.  All rights reserved.
   6  * libCEC(R) is an original work, containing original code.
   7  *
   8  * libCEC(R) is a trademark of Pulse-Eight Limited.
   9  *
  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.
  14  *
  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.
  19  *
  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.
  23  *
  24  *
  25  * Alternatively, you can license this library under a commercial license,
  26  * please contact Pulse-Eight Licensing for more information.
  27  *
  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/
  32  */
  33
  34 #include "../os.h"
  35 #include "../util/timeutils.h"
  36
  37 #pragma comment(lib, "Ws2_32.lib")
  38 #include <ws2spi.h>
  39 #include <ws2ipdef.h>
  40 #include <ws2tcpip.h>
  41
  42 #define SHUT_RDWR SD_BOTH
  43
  44 #ifndef ETIMEDOUT
  45 #define ETIMEDOUT 138
  46 #endif
  47
  48 namespace PLATFORM
  49 {
  50   #ifndef MSG_WAITALL
  51   #define MSG_WAITALL 0x8
  52   #endif
  53
  54   inline int GetSocketError(void)
  55   {
  56     int error = WSAGetLastError();
  57     switch(error)
  58     {
  59       case WSAEINPROGRESS: return EINPROGRESS;
  60       case WSAECONNRESET : return ECONNRESET;
  61       case WSAETIMEDOUT  : return ETIMEDOUT;
  62       case WSAEWOULDBLOCK: return EAGAIN;
  63       default            : return error;
  64     }
  65   }
  66
  67   // Serial port
  68   //@{
  69   inline void SerialSocketClose(serial_socket_t socket)
  70   {
  71     if (socket != INVALID_HANDLE_VALUE)
  72       CloseHandle(socket);
  73   }
  74
  75   inline ssize_t SerialSocketWrite(serial_socket_t socket, int *iError, void* data, size_t len)
  76   {
  77     if (len != (DWORD)len)
  78     {
  79       *iError = EINVAL;
  80       return -1;
  81     }
  82
  83     DWORD iBytesWritten(0);
  84     if (socket != INVALID_HANDLE_VALUE)
  85     {
  86       if (!WriteFile(socket, data, (DWORD)len, &iBytesWritten, NULL))
  87       {
  88         *iError = GetLastError();
  89         return -1;
  90       }
  91       return (ssize_t)iBytesWritten;
  92     }
  93
  94     return -1;
  95   }
  96
  97   inline ssize_t SerialSocketRead(serial_socket_t socket, int *iError, void* data, size_t len, uint64_t iTimeoutMs /*= 0*/)
  98   {
  99     if (len != (DWORD)len)
 100     {
 101       *iError = EINVAL;
 102       return -1;
 103     }
 104
 105     DWORD iBytesRead(0);
 106     if (socket != INVALID_HANDLE_VALUE)
 107     {
 108       if(!ReadFile(socket, data, (DWORD)len, &iBytesRead, NULL) != 0)
 109       {
 110         *iError = GetLastError();
 111         return -1;
 112       }
 113       return (ssize_t)iBytesRead;
 114     }
 115     return -1;
 116   }
 117   //@}
 118
 119   // TCP
 120   //@{
 121   inline void TcpSocketSetBlocking(tcp_socket_t socket, bool bSetTo)
 122   {
 123     u_long iSetTo = bSetTo ? 0 : 1;
 124     ioctlsocket(socket, FIONBIO, &iSetTo);
 125   }
 126
 127   inline void TcpSocketClose(tcp_socket_t socket)
 128   {
 129     closesocket(socket);
 130   }
 131
 132   inline void TcpSocketShutdown(tcp_socket_t socket)
 133   {
 134     if (socket != INVALID_SOCKET &&
 135         socket != SOCKET_ERROR)
 136       shutdown(socket, SHUT_RDWR);
 137   }
 138
 139   inline ssize_t TcpSocketWrite(tcp_socket_t socket, int *iError, void* data, size_t len)
 140   {
 141     if (socket == INVALID_SOCKET ||
 142         socket == SOCKET_ERROR ||
 143         len != (int)len)
 144     {
 145       *iError = EINVAL;
 146       return -1;
 147     }
 148
 149     ssize_t iReturn = send(socket, (char*)data, (int)len, 0);
 150     if (iReturn < (ssize_t)len)
 151       *iError = errno;
 152     return iReturn;
 153   }
 154
 155   inline ssize_t TcpSocketRead(tcp_socket_t socket, int *iError, void* data, size_t len, uint64_t iTimeoutMs /*= 0*/)
 156   {
 157     int64_t iNow(0), iTarget(0);
 158     ssize_t iBytesRead(0);
 159     *iError = 0;
 160
 161     if (socket == INVALID_SOCKET ||
 162         socket == SOCKET_ERROR ||
 163         len != (int)len)
 164     {
 165       *iError = EINVAL;
 166       return -1;
 167     }
 168
 169     if (iTimeoutMs > 0)
 170     {
 171       iNow    = GetTimeMs();
 172       iTarget = iNow + (int64_t) iTimeoutMs;
 173     }
 174
 175     fd_set fd_read;
 176     struct timeval tv;
 177     while (iBytesRead >= 0 && iBytesRead < (ssize_t)len && (iTimeoutMs == 0 || iTarget > iNow))
 178     {
 179       if (iTimeoutMs > 0)
 180       {
 181         tv.tv_sec  =        (long)(iTimeoutMs / 1000);
 182         tv.tv_usec = 1000 * (long)(iTimeoutMs % 1000);
 183
 184         FD_ZERO(&fd_read);
 185         FD_SET(socket, &fd_read);
 186
 187         if (select((int)socket + 1, &fd_read, NULL, NULL, &tv) == 0)
 188           return ETIMEDOUT;
 189         TcpSocketSetBlocking(socket, false);
 190       }
 191
 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();
 196       if (iReadResult < 0)
 197       {
 198         if (errno == EAGAIN && iTimeoutMs > 0)
 199           continue;
 200         *iError = errno;
 201         return -1;
 202       }
 203       else if (iReadResult == 0 || (iReadResult != (ssize_t)len && iTimeoutMs == 0))
 204       {
 205         *iError = ECONNRESET;
 206         return -1;
 207       }
 208
 209       iBytesRead += iReadResult;
 210
 211       if (iTimeoutMs > 0)
 212       {
 213         TcpSocketSetBlocking(socket, true);
 214         iNow = GetTimeMs();
 215       }
 216     }
 217     return 0;
 218   }
 219
 220   inline bool TcpResolveAddress(const char *strHost, uint16_t iPort, int *iError, struct addrinfo **info)
 221   {
 222     struct   addrinfo hints;
 223     char     service[33];
 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);
 229
 230     *iError = getaddrinfo(strHost, service, &hints, info);
 231     return !(*iError);
 232   }
 233
 234   inline int TcpGetSocketError(tcp_socket_t socket)
 235   {
 236     int iReturn(0);
 237     socklen_t optLen = sizeof(tcp_socket_t);
 238     getsockopt(socket, SOL_SOCKET, SO_ERROR, (char *)&iReturn, &optLen);
 239     return iReturn;
 240   }
 241
 242   inline bool TcpSetNoDelay(tcp_socket_t socket)
 243   {
 244     int iSetTo(1);
 245     setsockopt(socket, IPPROTO_TCP, TCP_NODELAY, (const char*)&iSetTo, sizeof(iSetTo));
 246     return true;
 247   }
 248
 249   inline bool TcpConnectSocket(tcp_socket_t socket, struct addrinfo* addr, int *iError, uint64_t iTimeout = 0)
 250   {
 251     TcpSocketSetBlocking(socket, false);
 252
 253     *iError = 0;
 254     int iConnectResult = connect(socket, addr->ai_addr, (int)addr->ai_addrlen);
 255     if (iConnectResult == -1)
 256     {
 257       if (GetSocketError() == EINPROGRESS ||
 258           GetSocketError() == EAGAIN)
 259       {
 260         fd_set fd_write, fd_except;
 261         struct timeval tv;
 262         tv.tv_sec  =        (long)(iTimeout / 1000);
 263         tv.tv_usec = 1000 * (long)(iTimeout % 1000);
 264
 265         FD_ZERO(&fd_write);
 266         FD_ZERO(&fd_except);
 267         FD_SET(socket, &fd_write);
 268         FD_SET(socket, &fd_except);
 269
 270         int iPollResult = select(sizeof(socket)*8, NULL, &fd_write, &fd_except, &tv);
 271         if (iPollResult == 0)
 272           *iError = ETIMEDOUT;
 273         else if (iPollResult == -1)
 274           *iError = GetSocketError();
 275         else
 276         {
 277           socklen_t errlen = sizeof(int);
 278           getsockopt(socket, SOL_SOCKET, SO_ERROR, (char *)iError, &errlen);
 279         }
 280       }
 281       else
 282       {
 283         *iError = errno;
 284       }
 285     }
 286
 287     TcpSocketSetBlocking(socket, true);
 288
 289     return *iError == 0;
 290   }
 291 }