src/lib/platform/posix/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 TcpGetSocketError(socket_t socket)
  40   {
  41     int iReturn(0);
  42     socklen_t optLen = sizeof(socket_t);
  43     getsockopt(socket, SOL_SOCKET, SO_ERROR, (void *)&iReturn, &optLen);
  44     return iReturn;
  45   }
  46
  47   inline bool TcpSetNoDelay(socket_t socket)
  48   {
  49     int iSetTo(1);
  50     setsockopt(socket, SOL_TCP, TCP_NODELAY, &iSetTo, sizeof(iSetTo));
  51     return true;
  52   }
  53
  54   inline bool TcpConnectSocket(socket_t socket, struct addrinfo* addr, int *iError, uint64_t iTimeout = 0)
  55   {
  56     bool bConnected = (connect(socket, addr->ai_addr, addr->ai_addrlen) == 0);
  57     if (!bConnected && errno == EINPROGRESS)
  58     {
  59       struct pollfd pfd;
  60       pfd.fd = socket;
  61       pfd.events = POLLOUT;
  62       pfd.revents = 0;
  63
  64       int iPollResult = poll(&pfd, 1, iTimeout);
  65       if (iPollResult == 0)
  66         *iError = ETIMEDOUT;
  67       else if (iPollResult == -1)
  68         *iError = errno;
  69       else
  70         *iError = TcpGetSocketError(socket);
  71     }
  72
  73     return bConnected;
  74   }
  75
  76   inline bool TcpResolveAddress(const char *strHost, uint16_t iPort, int *iError, struct addrinfo *info)
  77   {
  78     struct   addrinfo hints;
  79     char     service[33];
  80     memset(&hints, 0, sizeof(hints));
  81     hints.ai_family   = AF_UNSPEC;
  82     hints.ai_socktype = SOCK_STREAM;
  83     hints.ai_protocol = IPPROTO_TCP;
  84     sprintf(service, "%d", iPort);
  85
  86     return ((*iError = getaddrinfo(strHost, service, &hints, &info)) == 0);
  87   }
  88 }