mirror of
https://github.com/pocoproject/poco.git
synced 2025-10-26 18:42:41 +01:00
initial import
This commit is contained in:
Guenter Obiltschnig
845
Net/src/SocketImpl.cpp
Normal file
845
Net/src/SocketImpl.cpp
Normal file
@@ -0,0 +1,845 @@
|
||||
//
|
||||
// SocketImpl.cpp
|
||||
//
|
||||
// $Id: //poco/1.1.0/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 "Net/SocketImpl.h"
|
||||
#include "Net/NetException.h"
|
||||
#include "Net/StreamSocketImpl.h"
|
||||
#include "Foundation/NumberFormatter.h"
|
||||
#include <string.h>
|
||||
|
||||
|
||||
using Foundation::IOException;
|
||||
using Foundation::TimeoutException;
|
||||
using Foundation::InvalidArgumentException;
|
||||
using Foundation::NumberFormatter;
|
||||
using Foundation::Timespan;
|
||||
|
||||
|
||||
Net_BEGIN
|
||||
|
||||
|
||||
SocketImpl::SocketImpl():
|
||||
_sockfd(POCO_INVALID_SOCKET)
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
SocketImpl::SocketImpl(poco_socket_t sockfd):
|
||||
_sockfd(sockfd)
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
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());
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::connect(const SocketAddress& address, const Foundation::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 Foundation::TimeoutException("connect timed out", address.toString());
|
||||
int err = socketError();
|
||||
if (err != 0) error(err);
|
||||
}
|
||||
}
|
||||
catch (Foundation::Exception&)
|
||||
{
|
||||
setBlocking(true);
|
||||
throw;
|
||||
}
|
||||
setBlocking(true);
|
||||
}
|
||||
|
||||
|
||||
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());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
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());
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::listen(int backlog)
|
||||
{
|
||||
poco_assert (_sockfd != POCO_INVALID_SOCKET);
|
||||
|
||||
int rc = ::listen(_sockfd, backlog);
|
||||
if (rc != 0) error();
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::close()
|
||||
{
|
||||
if (_sockfd != POCO_INVALID_SOCKET)
|
||||
{
|
||||
poco_closesocket(_sockfd);
|
||||
_sockfd = POCO_INVALID_SOCKET;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::shutdownReceive()
|
||||
{
|
||||
poco_assert (_sockfd != POCO_INVALID_SOCKET);
|
||||
|
||||
int rc = ::shutdown(_sockfd, 0);
|
||||
if (rc != 0) error();
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::shutdownSend()
|
||||
{
|
||||
poco_assert (_sockfd != POCO_INVALID_SOCKET);
|
||||
|
||||
int rc = ::shutdown(_sockfd, 1);
|
||||
if (rc != 0) error();
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::shutdown()
|
||||
{
|
||||
poco_assert (_sockfd != POCO_INVALID_SOCKET);
|
||||
|
||||
int rc = ::shutdown(_sockfd, 2);
|
||||
if (rc != 0) error();
|
||||
}
|
||||
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::sendUrgent(unsigned char data)
|
||||
{
|
||||
int rc = ::send(_sockfd, reinterpret_cast<const char*>(&data), sizeof(data), MSG_OOB);
|
||||
if (rc < 0) error();
|
||||
}
|
||||
|
||||
|
||||
int SocketImpl::available()
|
||||
{
|
||||
int result;
|
||||
ioctl(FIONREAD, result);
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
bool SocketImpl::poll(const Foundation::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);
|
||||
}
|
||||
struct timeval tv;
|
||||
tv.tv_sec = (long) timeout.totalSeconds();
|
||||
tv.tv_usec = (long) timeout.useconds();
|
||||
int rc = ::select(int(_sockfd) + 1, &fdRead, &fdWrite, &fdExcept, &tv);
|
||||
if (rc < 0) error();
|
||||
return rc > 0;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setSendBufferSize(int size)
|
||||
{
|
||||
setOption(SOL_SOCKET, SO_SNDBUF, size);
|
||||
}
|
||||
|
||||
|
||||
int SocketImpl::getSendBufferSize()
|
||||
{
|
||||
int result;
|
||||
getOption(SOL_SOCKET, SO_SNDBUF, result);
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setReceiveBufferSize(int size)
|
||||
{
|
||||
setOption(SOL_SOCKET, SO_RCVBUF, size);
|
||||
}
|
||||
|
||||
|
||||
int SocketImpl::getReceiveBufferSize()
|
||||
{
|
||||
int result;
|
||||
getOption(SOL_SOCKET, SO_RCVBUF, result);
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setSendTimeout(const Foundation::Timespan& timeout)
|
||||
{
|
||||
#if defined(_WIN32)
|
||||
int value = (int) timeout.totalMilliseconds();
|
||||
setOption(SOL_SOCKET, SO_SNDTIMEO, value);
|
||||
#else
|
||||
setOption(SOL_SOCKET, SO_SNDTIMEO, timeout);
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
Foundation::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;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setReceiveTimeout(const Foundation::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
|
||||
}
|
||||
|
||||
|
||||
Foundation::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;
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setOption(int level, int option, int value)
|
||||
{
|
||||
setRawOption(level, option, &value, sizeof(value));
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setOption(int level, int option, unsigned value)
|
||||
{
|
||||
setRawOption(level, option, &value, sizeof(value));
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setOption(int level, int option, unsigned char value)
|
||||
{
|
||||
setRawOption(level, option, &value, sizeof(value));
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setOption(int level, int option, const IPAddress& value)
|
||||
{
|
||||
setRawOption(level, option, value.addr(), value.length());
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setOption(int level, int option, const Foundation::Timespan& value)
|
||||
{
|
||||
struct timeval tv;
|
||||
tv.tv_sec = (long) value.totalSeconds();
|
||||
tv.tv_usec = (long) value.useconds();
|
||||
|
||||
setRawOption(level, option, &tv, sizeof(tv));
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::getOption(int level, int option, int& value)
|
||||
{
|
||||
poco_socklen_t len = sizeof(value);
|
||||
getRawOption(level, option, &value, len);
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::getOption(int level, int option, unsigned& value)
|
||||
{
|
||||
poco_socklen_t len = sizeof(value);
|
||||
getRawOption(level, option, &value, len);
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::getOption(int level, int option, unsigned char& value)
|
||||
{
|
||||
poco_socklen_t len = sizeof(value);
|
||||
getRawOption(level, option, &value, len);
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::getOption(int level, int option, Foundation::Timespan& value)
|
||||
{
|
||||
struct timeval tv;
|
||||
poco_socklen_t len = sizeof(tv);
|
||||
getRawOption(level, option, &tv, len);
|
||||
value.assign(tv.tv_sec, tv.tv_usec);
|
||||
}
|
||||
|
||||
|
||||
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);
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
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));
|
||||
}
|
||||
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setNoDelay(bool flag)
|
||||
{
|
||||
int value = flag ? 1 : 0;
|
||||
setOption(IPPROTO_TCP, TCP_NODELAY, value);
|
||||
}
|
||||
|
||||
|
||||
bool SocketImpl::getNoDelay()
|
||||
{
|
||||
int value;
|
||||
getOption(IPPROTO_TCP, TCP_NODELAY, value);
|
||||
return value != 0;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setKeepAlive(bool flag)
|
||||
{
|
||||
int value = flag ? 1 : 0;
|
||||
setOption(SOL_SOCKET, SO_KEEPALIVE, value);
|
||||
}
|
||||
|
||||
|
||||
bool SocketImpl::getKeepAlive()
|
||||
{
|
||||
int value;
|
||||
getOption(SOL_SOCKET, SO_KEEPALIVE, value);
|
||||
return value != 0;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setReuseAddress(bool flag)
|
||||
{
|
||||
int value = flag ? 1 : 0;
|
||||
setOption(SOL_SOCKET, SO_REUSEADDR, value);
|
||||
}
|
||||
|
||||
|
||||
bool SocketImpl::getReuseAddress()
|
||||
{
|
||||
int value;
|
||||
getOption(SOL_SOCKET, SO_REUSEADDR, value);
|
||||
return value != 0;
|
||||
}
|
||||
|
||||
|
||||
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
|
||||
}
|
||||
|
||||
bool SocketImpl::getReusePort()
|
||||
{
|
||||
#ifdef SO_REUSEPORT
|
||||
int value;
|
||||
getOption(SOL_SOCKET, SO_REUSEPORT, value);
|
||||
return value != 0;
|
||||
#else
|
||||
return false;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setOOBInline(bool flag)
|
||||
{
|
||||
int value = flag ? 1 : 0;
|
||||
setOption(SOL_SOCKET, SO_OOBINLINE, value);
|
||||
}
|
||||
|
||||
|
||||
bool SocketImpl::getOOBInline()
|
||||
{
|
||||
int value;
|
||||
getOption(SOL_SOCKET, SO_OOBINLINE, value);
|
||||
return value != 0;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setBroadcast(bool flag)
|
||||
{
|
||||
int value = flag ? 1 : 0;
|
||||
setOption(SOL_SOCKET, SO_BROADCAST, value);
|
||||
}
|
||||
|
||||
|
||||
bool SocketImpl::getBroadcast()
|
||||
{
|
||||
int value;
|
||||
getOption(SOL_SOCKET, SO_BROADCAST, value);
|
||||
return value != 0;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setBlocking(bool flag)
|
||||
{
|
||||
int arg = flag ? 0 : 1;
|
||||
ioctl(FIONBIO, arg);
|
||||
}
|
||||
|
||||
|
||||
int SocketImpl::socketError()
|
||||
{
|
||||
int result;
|
||||
getOption(SOL_SOCKET, SO_ERROR, result);
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::init(int af)
|
||||
{
|
||||
initSocket(af, SOCK_STREAM);
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
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();
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::setSockfd(poco_socket_t aSocket)
|
||||
{
|
||||
poco_assert(sockfd() == POCO_INVALID_SOCKET);
|
||||
_sockfd = aSocket;
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::error()
|
||||
{
|
||||
std::string empty;
|
||||
error(lastError(), empty);
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::error(const std::string& arg)
|
||||
{
|
||||
error(lastError(), arg);
|
||||
}
|
||||
|
||||
|
||||
void SocketImpl::error(int code)
|
||||
{
|
||||
std::string arg;
|
||||
error(code, arg);
|
||||
}
|
||||
|
||||
|
||||
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);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Net_END
|
||||
Reference in New Issue
Block a user