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

   1 #pragma once
   2 /*
   3  * This file is part of the libCEC(R) library.
   4  *
   5  * libCEC(R) is Copyright (C) 2011 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 "../sockets/socket.h"
  36
  37 namespace PLATFORM
  38 {
  39   inline int TcpResolveAddress(const CStdString &strHostname, uint16_t iPort, struct addrinfo *address)
  40   {
  41      struct addrinfo hints;
  42      char   service[33];
  43
  44      memset(&hints, 0, sizeof(hints));
  45      hints.ai_family   = AF_UNSPEC;
  46      hints.ai_socktype = SOCK_STREAM;
  47      hints.ai_protocol = IPPROTO_TCP;
  48      sprintf(service, "%d", iPort);
  49
  50      return getaddrinfo(strHostname.c_str(), service, &hints, &address);
  51    }
  52
  53   inline int TcpGetSocketError()
  54   {
  55     int error = WSAGetLastError();
  56     switch(error)
  57     {
  58       case WSAEINPROGRESS: return EINPROGRESS;
  59       case WSAECONNRESET : return ECONNRESET;
  60       case WSAETIMEDOUT  : return ETIMEDOUT;
  61       case WSAEWOULDBLOCK: return EAGAIN;
  62       default            : return error;
  63     }
  64   }
  65
  66   inline int TcpConnectSocket(struct addrinfo* addr, socket_t socket, int *iError, uint64_t iTimeout)
  67   {
  68     int nRes = 0;
  69     socklen_t errlen = sizeof(int);
  70
  71     SocketSetBlocking(socket, false);
  72
  73     /* connect to the other side */
  74     nRes = connect(socket, addr->ai_addr, addr->ai_addrlen);
  75
  76     /* poll until a connection is established */
  77     if (nRes == -1)
  78     {
  79       if (TcpGetSocketError() == EINPROGRESS || TcpGetSocketError() == EAGAIN)
  80       {
  81         fd_set fd_write, fd_except;
  82         struct timeval tv;
  83         tv.tv_sec  = (long)(iTimeout / 1000);
  84         tv.tv_usec = (long)(1000 * (iTimeout % 1000));
  85
  86         FD_ZERO(&fd_write);
  87         FD_ZERO(&fd_except);
  88         FD_SET(socket, &fd_write);
  89         FD_SET(socket, &fd_except);
  90
  91         nRes = select(sizeof(socket)*8, NULL, &fd_write, &fd_except, &tv);
  92         if (nRes == 0)
  93         {
  94           *iError = ETIMEDOUT;
  95           return SOCKET_ERROR;
  96         }
  97         else if (nRes == -1)
  98         {
  99           *iError = TcpGetSocketError();
 100           return SOCKET_ERROR;
 101         }
 102
 103         /* check for errors */
 104         getsockopt(socket, SOL_SOCKET, SO_ERROR, (char *)iError, &errlen);
 105       }
 106       else
 107       {
 108         *iError = TcpGetSocketError();
 109       }
 110     }
 111
 112     SocketSetBlocking(socket, true);
 113
 114     return 0;
 115   }
 116
 117   inline bool TcpSetNoDelay(socket_t socket)
 118   {
 119     int nVal = 1;
 120     setsockopt(socket, IPPROTO_TCP, TCP_NODELAY, (const char*)&nVal, sizeof(nVal));
 121     return true;
 122   }
 123 }