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