Logo Search packages:      
Sourcecode: poco version File versions  Download package

SocketImpl.cpp

//
// SocketImpl.cpp
//
// $Id: //poco/1.2/Net/src/SocketImpl.cpp#2 $
//
// Library: Net
// Package: Sockets
// Module:  SocketImpl
//
// Copyright (c) 2005-2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
// 
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//


#include "Poco/Net/SocketImpl.h"
#include "Poco/Net/NetException.h"
#include "Poco/Net/StreamSocketImpl.h"
#include "Poco/NumberFormatter.h"
#include "Poco/Timestamp.h"
#include <string.h>


using Poco::IOException;
using Poco::TimeoutException;
using Poco::InvalidArgumentException;
using Poco::NumberFormatter;
using Poco::Timespan;


namespace Poco {
namespace Net {


00056 SocketImpl::SocketImpl():
      _sockfd(POCO_INVALID_SOCKET)
{
}


00062 SocketImpl::SocketImpl(poco_socket_t sockfd):
      _sockfd(sockfd)
{
}


00068 SocketImpl::~SocketImpl()
{
      close();
}

      
SocketImpl* SocketImpl::acceptConnection(SocketAddress& clientAddr)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

      char buffer[SocketAddress::MAX_ADDRESS_LENGTH];
      struct sockaddr* pSA = reinterpret_cast<struct sockaddr*>(buffer);
      poco_socklen_t saLen = sizeof(buffer);
      poco_socket_t sd;
      do
      {
            sd = ::accept(_sockfd, pSA, &saLen);
      }
      while (sd == POCO_INVALID_SOCKET && lastError() == POCO_EINTR);
      if (sd != POCO_INVALID_SOCKET)
      {
            clientAddr = SocketAddress(pSA, saLen);
            return new StreamSocketImpl(sd);
      }
      error(); // will throw
      return 0;
}


00097 void SocketImpl::connect(const SocketAddress& address)
{
      if (_sockfd == POCO_INVALID_SOCKET)
      {
            init(address.af());
      }
      int rc;
      do
      {
            rc = ::connect(_sockfd, address.addr(), address.length());
      }
      while (rc != 0 && lastError() == POCO_EINTR);
      if (rc != 0) error(address.toString());
}


00113 void SocketImpl::connect(const SocketAddress& address, const Poco::Timespan& timeout)
{
      poco_assert (_sockfd == POCO_INVALID_SOCKET);
      
      init(address.af());
      setBlocking(false);
      try
      {
            int rc = ::connect(_sockfd, address.addr(), address.length());
            if (rc != 0)
            {
                  if (lastError() != POCO_EINPROGRESS && lastError() != POCO_EWOULDBLOCK)
                        error(address.toString());
                  if (!poll(timeout, SELECT_READ | SELECT_WRITE))
                        throw Poco::TimeoutException("connect timed out", address.toString());
                  int err = socketError();
                  if (err != 0) error(err);
            }
      }
      catch (Poco::Exception&)
      {
            setBlocking(true);
            throw;
      }
      setBlocking(true);
}


00141 void SocketImpl::connectNB(const SocketAddress& address)
{
      if (_sockfd == POCO_INVALID_SOCKET)
      {
            init(address.af());
      }
      setBlocking(false);
      int rc = ::connect(_sockfd, address.addr(), address.length());
      if (rc != 0)
      {
            if (lastError() != POCO_EINPROGRESS && lastError() != POCO_EWOULDBLOCK)
                  error(address.toString());
      }
}


00157 void SocketImpl::bind(const SocketAddress& address, bool reuseAddress)
{
      if (_sockfd == POCO_INVALID_SOCKET)
      {
            init(address.af());
      }
      if (reuseAddress)
      {
            setReuseAddress(true);
            setReusePort(true);
      }
      int rc = ::bind(_sockfd, address.addr(), address.length());
      if (rc != 0) error(address.toString());
}

      
00173 void SocketImpl::listen(int backlog)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);
      
      int rc = ::listen(_sockfd, backlog);
      if (rc != 0) error();
}


00182 void SocketImpl::close()
{
      if (_sockfd != POCO_INVALID_SOCKET)
      {
            poco_closesocket(_sockfd);
            _sockfd = POCO_INVALID_SOCKET;
      }
}


00192 void SocketImpl::shutdownReceive()
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

      int rc = ::shutdown(_sockfd, 0);
      if (rc != 0) error();
}

      
00201 void SocketImpl::shutdownSend()
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

      int rc = ::shutdown(_sockfd, 1);
      if (rc != 0) error();
}

      
00210 void SocketImpl::shutdown()
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

      int rc = ::shutdown(_sockfd, 2);
      if (rc != 0) error();
}


00219 int SocketImpl::sendBytes(const void* buffer, int length, int flags)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

      int rc;
      do
      {
            rc = ::send(_sockfd, reinterpret_cast<const char*>(buffer), length, flags);
      }
      while (rc < 0 && lastError() == POCO_EINTR);
      if (rc < 0) error();
      return rc;
}


00234 int SocketImpl::receiveBytes(void* buffer, int length, int flags)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

#if defined(POCO_BROKEN_TIMEOUTS)
      if (_recvTimeout.totalMicroseconds() != 0)
      {
            if (!poll(_recvTimeout, SELECT_READ))
                  throw TimeoutException();
      }
#endif

      int rc;
      do
      {
            rc = ::recv(_sockfd, reinterpret_cast<char*>(buffer), length, flags);
      }
      while (rc < 0 && lastError() == POCO_EINTR);
      if (rc < 0) 
      {
            if (lastError() == POCO_EAGAIN || lastError() == POCO_ETIMEDOUT)
                  throw TimeoutException();
            else
                  error();
      }
      return rc;
}


00263 int SocketImpl::sendTo(const void* buffer, int length, const SocketAddress& address, int flags)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

      int rc;
      do
      {
            rc = ::sendto(_sockfd, reinterpret_cast<const char*>(buffer), length, flags, address.addr(), address.length());
      }
      while (rc < 0 && lastError() == POCO_EINTR);
      if (rc < 0) error();
      return rc;
}


00278 int SocketImpl::receiveFrom(void* buffer, int length, SocketAddress& address, int flags)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);

#if defined(POCO_BROKEN_TIMEOUTS)
      if (_recvTimeout.totalMicroseconds() != 0)
      {
            if (!poll(_recvTimeout, SELECT_READ))
                  throw TimeoutException();
      }
#endif
      
      char abuffer[SocketAddress::MAX_ADDRESS_LENGTH];
      struct sockaddr* pSA = reinterpret_cast<struct sockaddr*>(abuffer);
      poco_socklen_t saLen = sizeof(abuffer);
      int rc;
      do
      {
            rc = ::recvfrom(_sockfd, reinterpret_cast<char*>(buffer), length, flags, pSA, &saLen);
      }
      while (rc < 0 && lastError() == POCO_EINTR);
      if (rc >= 0)
      {
            address = SocketAddress(pSA, saLen);
      }
      else
      {
            if (lastError() == POCO_EAGAIN || lastError() == POCO_ETIMEDOUT)
                  throw TimeoutException();
            else
                  error();
      }
      return rc;
}


00314 void SocketImpl::sendUrgent(unsigned char data)
{
      int rc = ::send(_sockfd, reinterpret_cast<const char*>(&data), sizeof(data), MSG_OOB);
      if (rc < 0) error();
}


00321 int SocketImpl::available()
{
      int result;
      ioctl(FIONREAD, result);
      return result;
}


00329 bool SocketImpl::poll(const Poco::Timespan& timeout, int mode)
{
      fd_set fdRead;
      fd_set fdWrite;
      fd_set fdExcept;
      FD_ZERO(&fdRead);
      FD_ZERO(&fdWrite);
      FD_ZERO(&fdExcept);
      if (mode & SELECT_READ)
      {
            FD_SET(_sockfd, &fdRead);
      }
      if (mode & SELECT_WRITE)
      {
            FD_SET(_sockfd, &fdWrite);
      }
      if (mode & SELECT_ERROR)
      {
            FD_SET(_sockfd, &fdExcept);
      }
      Poco::Timespan remainingTime(timeout);
      int rc;
      do
      {
            struct timeval tv;
            tv.tv_sec  = (long) remainingTime.totalSeconds();
            tv.tv_usec = (long) remainingTime.useconds();
            Poco::Timestamp start;
            rc = ::select(int(_sockfd) + 1, &fdRead, &fdWrite, &fdExcept, &tv);
            if (rc < 0 && lastError() == POCO_EINTR)
            {
                  Poco::Timestamp end;
                  Poco::Timespan waited = end - start;
                  if (waited > remainingTime)
                        remainingTime -= waited;
                  else
                        remainingTime = 0;
            }
      }
      while (rc < 0 && lastError() == POCO_EINTR);
      if (rc < 0) error();
      return rc > 0; 
}

      
00374 void SocketImpl::setSendBufferSize(int size)
{
      setOption(SOL_SOCKET, SO_SNDBUF, size);
}

      
00380 int SocketImpl::getSendBufferSize()
{
      int result;
      getOption(SOL_SOCKET, SO_SNDBUF, result);
      return result;
}


00388 void SocketImpl::setReceiveBufferSize(int size)
{
      setOption(SOL_SOCKET, SO_RCVBUF, size);
}

      
00394 int SocketImpl::getReceiveBufferSize()
{
      int result;
      getOption(SOL_SOCKET, SO_RCVBUF, result);
      return result;
}


00402 void SocketImpl::setSendTimeout(const Poco::Timespan& timeout)
{
#if defined(_WIN32)
      int value = (int) timeout.totalMilliseconds();
      setOption(SOL_SOCKET, SO_SNDTIMEO, value);
#else
      setOption(SOL_SOCKET, SO_SNDTIMEO, timeout);
#endif
}


00413 Poco::Timespan SocketImpl::getSendTimeout()
{
      Timespan result;
#if defined(_WIN32)
      int value;
      getOption(SOL_SOCKET, SO_SNDTIMEO, value);
      result = Timespan::TimeDiff(value)*1000;
#else
      getOption(SOL_SOCKET, SO_SNDTIMEO, result);
#endif
      return result;
}


00427 void SocketImpl::setReceiveTimeout(const Poco::Timespan& timeout)
{
#if defined(_WIN32)
      int value = (int) timeout.totalMilliseconds();
      setOption(SOL_SOCKET, SO_RCVTIMEO, value);
#else
      setOption(SOL_SOCKET, SO_RCVTIMEO, timeout);
#endif
#if defined(POCO_BROKEN_TIMEOUTS)
      _recvTimeout = timeout;
#endif
}


00441 Poco::Timespan SocketImpl::getReceiveTimeout()
{
      Timespan result;
#if defined(_WIN32)
      int value;
      getOption(SOL_SOCKET, SO_RCVTIMEO, value);
      result = Timespan::TimeDiff(value)*1000;
#elif defined(POCO_BROKEN_TIMEOUTS)
      result = _recvTimeout;
#else
      getOption(SOL_SOCKET, SO_RCVTIMEO, result);
#endif
      return result;
}

      
00457 SocketAddress SocketImpl::address()
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);
      
      char buffer[SocketAddress::MAX_ADDRESS_LENGTH];
      struct sockaddr* pSA = reinterpret_cast<struct sockaddr*>(buffer);
      poco_socklen_t saLen = sizeof(buffer);
      int rc = ::getsockname(_sockfd, pSA, &saLen);
      if (rc == 0)
            return SocketAddress(pSA, saLen);
      else 
            error();
      return SocketAddress();
}

      
00473 SocketAddress SocketImpl::peerAddress()
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);
      
      char buffer[SocketAddress::MAX_ADDRESS_LENGTH];
      struct sockaddr* pSA = reinterpret_cast<struct sockaddr*>(buffer);
      poco_socklen_t saLen = sizeof(buffer);
      int rc = ::getpeername(_sockfd, pSA, &saLen);
      if (rc == 0)
            return SocketAddress(pSA, saLen);
      else 
            error();
      return SocketAddress();
}


00489 void SocketImpl::setOption(int level, int option, int value)
{
      setRawOption(level, option, &value, sizeof(value));
}


00495 void SocketImpl::setOption(int level, int option, unsigned value)
{
      setRawOption(level, option, &value, sizeof(value));
}


00501 void SocketImpl::setOption(int level, int option, unsigned char value)
{
      setRawOption(level, option, &value, sizeof(value));
}


00507 void SocketImpl::setOption(int level, int option, const IPAddress& value)
{
      setRawOption(level, option, value.addr(), value.length());
}


00513 void SocketImpl::setOption(int level, int option, const Poco::Timespan& value)
{
      struct timeval tv;
      tv.tv_sec  = (long) value.totalSeconds();
      tv.tv_usec = (long) value.useconds();
      
      setRawOption(level, option, &tv, sizeof(tv));
}

      
00523 void SocketImpl::setRawOption(int level, int option, const void* value, poco_socklen_t length)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);
      
      int rc = ::setsockopt(_sockfd, level, option, reinterpret_cast<const char*>(value), length);
      if (rc == -1) error();
}


00532 void SocketImpl::getOption(int level, int option, int& value)
{
      poco_socklen_t len = sizeof(value);
      getRawOption(level, option, &value, len);
}


00539 void SocketImpl::getOption(int level, int option, unsigned& value)
{
      poco_socklen_t len = sizeof(value);
      getRawOption(level, option, &value, len);
}


00546 void SocketImpl::getOption(int level, int option, unsigned char& value)
{
      poco_socklen_t len = sizeof(value);
      getRawOption(level, option, &value, len);
}


00553 void SocketImpl::getOption(int level, int option, Poco::Timespan& value)
{
      struct timeval tv;
      poco_socklen_t len = sizeof(tv);
      getRawOption(level, option, &tv, len);
      value.assign(tv.tv_sec, tv.tv_usec);
}


00562 void SocketImpl::getOption(int level, int option, IPAddress& value)
{
      char buffer[IPAddress::MAX_ADDRESS_LENGTH];
      poco_socklen_t len = sizeof(buffer);
      getRawOption(level, option, buffer, len);
      value = IPAddress(buffer, len);
}


00571 void SocketImpl::getRawOption(int level, int option, void* value, poco_socklen_t& length)
{
      poco_assert (_sockfd != POCO_INVALID_SOCKET);
      
      int rc = ::getsockopt(_sockfd, level, option, reinterpret_cast<char*>(value), &length);
      if (rc == -1) error();
}


00580 void SocketImpl::setLinger(bool on, int seconds)
{
      struct linger l;
      l.l_onoff  = on ? 1 : 0;
      l.l_linger = seconds;
      setRawOption(SOL_SOCKET, SO_LINGER, &l, sizeof(l));
}

      
00589 void SocketImpl::getLinger(bool& on, int& seconds)
{
      struct linger l;
      poco_socklen_t len = sizeof(l);
      getRawOption(SOL_SOCKET, SO_LINGER, &l, len);
      on      = l.l_onoff != 0;
      seconds = l.l_linger;
}


00599 void SocketImpl::setNoDelay(bool flag)
{
      int value = flag ? 1 : 0;
      setOption(IPPROTO_TCP, TCP_NODELAY, value);
}


00606 bool SocketImpl::getNoDelay()
{
      int value;
      getOption(IPPROTO_TCP, TCP_NODELAY, value);
      return value != 0;
}


00614 void SocketImpl::setKeepAlive(bool flag)
{
      int value = flag ? 1 : 0;
      setOption(SOL_SOCKET, SO_KEEPALIVE, value);
}


00621 bool SocketImpl::getKeepAlive()
{
      int value;
      getOption(SOL_SOCKET, SO_KEEPALIVE, value);
      return value != 0;
}


00629 void SocketImpl::setReuseAddress(bool flag)
{
      int value = flag ? 1 : 0;
      setOption(SOL_SOCKET, SO_REUSEADDR, value);
}


00636 bool SocketImpl::getReuseAddress()
{
      int value;
      getOption(SOL_SOCKET, SO_REUSEADDR, value);
      return value != 0;
}


00644 void SocketImpl::setReusePort(bool flag)
{
#ifdef SO_REUSEPORT
      try
      {
            int value = flag ? 1 : 0;
            setOption(SOL_SOCKET, SO_REUSEPORT, value);
      }
      catch (IOException&)
      {
            // ignore error, since not all implementations
            // support SO_REUSEPORT, even if the macro
            // is defined.
      }
#endif
}


00662 bool SocketImpl::getReusePort()
{
#ifdef SO_REUSEPORT
      int value;
      getOption(SOL_SOCKET, SO_REUSEPORT, value);
      return value != 0;
#else
      return false;
#endif
}


00674 void SocketImpl::setOOBInline(bool flag)
{
      int value = flag ? 1 : 0;
      setOption(SOL_SOCKET, SO_OOBINLINE, value);
}


00681 bool SocketImpl::getOOBInline()
{
      int value;
      getOption(SOL_SOCKET, SO_OOBINLINE, value);
      return value != 0;
}


00689 void SocketImpl::setBroadcast(bool flag)
{
      int value = flag ? 1 : 0;
      setOption(SOL_SOCKET, SO_BROADCAST, value);
}

      
00696 bool SocketImpl::getBroadcast()
{
      int value;
      getOption(SOL_SOCKET, SO_BROADCAST, value);
      return value != 0;
}


00704 void SocketImpl::setBlocking(bool flag)
{
      int arg = flag ? 0 : 1;
      ioctl(FIONBIO, arg);
}


00711 int SocketImpl::socketError()
{
      int result;
      getOption(SOL_SOCKET, SO_ERROR, result);
      return result;
}


00719 void SocketImpl::init(int af)
{
      initSocket(af, SOCK_STREAM);
}


00725 void SocketImpl::initSocket(int af, int type, int proto)
{
      poco_assert (_sockfd == POCO_INVALID_SOCKET);

      _sockfd = ::socket(af, type, proto);
      if (_sockfd == POCO_INVALID_SOCKET)
            error();
}


00735 void SocketImpl::ioctl(int request, int& arg)
{
#if defined(_WIN32)
      int rc = ioctlsocket(_sockfd, request, reinterpret_cast<u_long*>(&arg));
#else
      int rc = ::ioctl(_sockfd, request, &arg);
#endif
      if (rc != 0) error();
}


00746 void SocketImpl::ioctl(int request, void* arg)
{
#if defined(_WIN32)
      int rc = ioctlsocket(_sockfd, request, reinterpret_cast<u_long*>(arg));
#else
      int rc = ::ioctl(_sockfd, request, arg);
#endif
      if (rc != 0) error();
}


00757 void SocketImpl::setSockfd(poco_socket_t aSocket)
{
      poco_assert(sockfd() == POCO_INVALID_SOCKET);
      _sockfd = aSocket;
}


00764 void SocketImpl::error()
{
      std::string empty;
      error(lastError(), empty);
}


00771 void SocketImpl::error(const std::string& arg)
{
      error(lastError(), arg);
}


00777 void SocketImpl::error(int code)
{
      std::string arg;
      error(code, arg);
}


00784 void SocketImpl::error(int code, const std::string& arg)
{
      switch (code)
      {
      case POCO_ESYSNOTREADY:
            throw NetException("Net subsystem not ready");
      case POCO_ENOTINIT:
            throw NetException("Net subsystem not initialized");
      case POCO_EINTR:
            throw IOException("Interrupted");
      case POCO_EACCES:
            throw IOException("Permission denied");
      case POCO_EFAULT:
            throw IOException("Bad address");
      case POCO_EINVAL:
            throw InvalidArgumentException();
      case POCO_EMFILE:
            throw IOException("Too many open files");
      case POCO_EWOULDBLOCK:
            throw IOException("Operation would block");
      case POCO_EINPROGRESS:
            throw IOException("Operation now in progress");
      case POCO_EALREADY:
            throw IOException("Operation already in progress");
      case POCO_ENOTSOCK:
            throw IOException("Socket operation attempted on non-socket");
      case POCO_EDESTADDRREQ:
            throw NetException("Destination address required");
      case POCO_EMSGSIZE:
            throw NetException("Message too long");
      case POCO_EPROTOTYPE:
            throw NetException("Wrong protocol type");
      case POCO_ENOPROTOOPT:
            throw NetException("Protocol not available");
      case POCO_EPROTONOSUPPORT:
            throw NetException("Protocol not supported");
      case POCO_ESOCKTNOSUPPORT:
            throw NetException("Socket type not supported");
      case POCO_ENOTSUP:
            throw NetException("Operation not supported");
      case POCO_EPFNOSUPPORT:
            throw NetException("Protocol family not supported");
      case POCO_EAFNOSUPPORT:
            throw NetException("Address family not supported");
      case POCO_EADDRINUSE:
            throw NetException("Address already in use", arg);
      case POCO_EADDRNOTAVAIL:
            throw NetException("Cannot assign requested address", arg);
      case POCO_ENETDOWN:
            throw NetException("Network is down");
      case POCO_ENETUNREACH:
            throw NetException("Network is unreachable");
      case POCO_ENETRESET:
            throw NetException("Network dropped connection on reset");
      case POCO_ECONNABORTED:
            throw ConnectionAbortedException();
      case POCO_ECONNRESET:
            throw ConnectionResetException();
      case POCO_ENOBUFS:
            throw IOException("No buffer space available");
      case POCO_EISCONN:
            throw NetException("Socket is already connected");
      case POCO_ENOTCONN:
            throw NetException("Socket is not connected");
      case POCO_ESHUTDOWN:
            throw NetException("Cannot send after socket shutdown");
      case POCO_ETIMEDOUT:
            throw TimeoutException();
      case POCO_ECONNREFUSED:
            throw ConnectionRefusedException(arg);
      case POCO_EHOSTDOWN:
            throw NetException("Host is down");
      case POCO_EHOSTUNREACH:
            throw NetException("No route to host");
      default:
            throw IOException(NumberFormatter::format(code) + arg);
      }
}


} } // namespace Poco::Net

Generated by  Doxygen 1.6.0   Back to index