You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdelibs/kdecore/network/ksocketdevice.cpp

892 lines
20 KiB

/* -*- C++ -*-
* Copyright (C) 2003,2005 Thiago Macieira <thiago.macieira@kdemail.net>
*
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* 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 AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#include <config.h>
#include <tqmap.h>
#ifdef USE_SOLARIS
# include <sys/filio.h>
#endif
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <unistd.h>
#ifdef HAVE_POLL
# include <sys/poll.h>
#else
# ifdef HAVE_SYS_SELECT
# include <sys/select.h>
# endif
#endif
// Include syssocket before our local includes
#include "syssocket.h"
#include <tqmutex.h>
#include <tqsocketnotifier.h>
#include "kresolver.h"
#include "ksocketaddress.h"
#include "ksocketbase.h"
#include "ksocketdevice.h"
#include "ksockssocketdevice.h"
using namespace KNetwork;
class KNetwork::KSocketDevicePrivate
{
public:
mutable TQSocketNotifier *input, *output, *exception;
KSocketAddress local, peer;
int af;
inline KSocketDevicePrivate()
{
input = output = exception = 0L;
af = 0;
}
};
KSocketDevice::KSocketDevice(const KSocketBase* parent)
: m_sockfd(-1), d(new KSocketDevicePrivate)
{
setSocketDevice(this);
if (parent)
setSocketOptions(parent->socketOptions());
}
KSocketDevice::KSocketDevice(int fd)
: m_sockfd(fd), d(new KSocketDevicePrivate)
{
setState(IO_Open);
setFlags(IO_Sequential | IO_Raw | IO_ReadWrite);
setSocketDevice(this);
d->af = localAddress().family();
}
KSocketDevice::KSocketDevice(bool, const KSocketBase* parent)
: m_sockfd(-1), d(new KSocketDevicePrivate)
{
// do not set parent
if (parent)
setSocketOptions(parent->socketOptions());
}
KSocketDevice::~KSocketDevice()
{
close(); // deletes the notifiers
unsetSocketDevice(); // prevent double deletion
delete d;
}
bool KSocketDevice::setSocketOptions(int opts)
{
// must call parent
TQMutexLocker locker(mutex());
KSocketBase::setSocketOptions(opts);
if (m_sockfd == -1)
return true; // flags are stored
{
int fdflags = fcntl(m_sockfd, F_GETFL, 0);
if (fdflags == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false; // error
}
if (opts & Blocking)
fdflags &= ~O_NONBLOCK;
else
fdflags |= O_NONBLOCK;
if (fcntl(m_sockfd, F_SETFL, fdflags) == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false; // error
}
}
{
int on = opts & AddressReuseable ? 1 : 0;
if (setsockopt(m_sockfd, SOL_SOCKET, SO_REUSEADDR, (char*)&on, sizeof(on)) == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false; // error
}
}
#if defined(IPV6_V6ONLY) && defined(AF_INET6)
if (d->af == AF_INET6)
{
// don't try this on non-IPv6 sockets, or we'll get an error
int on = opts & IPv6Only ? 1 : 0;
if (setsockopt(m_sockfd, IPPROTO_IPV6, IPV6_V6ONLY, (char*)&on, sizeof(on)) == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false; // error
}
}
#endif
{
int on = opts & Broadcast ? 1 : 0;
if (setsockopt(m_sockfd, SOL_SOCKET, SO_BROADCAST, (char*)&on, sizeof(on)) == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false; // error
}
}
return true; // all went well
}
bool KSocketDevice::open(int)
{
resetError();
return false;
}
void KSocketDevice::close()
{
resetError();
if (m_sockfd != -1)
{
delete d->input;
delete d->output;
delete d->exception;
d->input = d->output = d->exception = 0L;
d->local.setFamily(AF_UNSPEC);
d->peer.setFamily(AF_UNSPEC);
::close(m_sockfd);
}
setState(0);
m_sockfd = -1;
}
bool KSocketDevice::create(int family, int type, int protocol)
{
resetError();
if (m_sockfd != -1)
{
// it's already created!
setError(IO_SocketCreateError, AlreadyCreated);
return false;
}
// no socket yet; we have to create it
m_sockfd = kde_socket(family, type, protocol);
if (m_sockfd == -1)
{
setError(IO_SocketCreateError, NotSupported);
return false;
}
d->af = family;
setSocketOptions(socketOptions());
setState(IO_Open);
return true; // successfully created
}
bool KSocketDevice::create(const KResolverEntry& address)
{
return create(address.family(), address.socketType(), address.protocol());
}
bool KSocketDevice::bind(const KResolverEntry& address)
{
resetError();
if (m_sockfd == -1 && !create(address))
return false; // failed creating
// we have a socket, so try and bind
if (kde_bind(m_sockfd, address.address(), address.length()) == -1)
{
if (errno == EADDRINUSE)
setError(IO_BindError, AddressInUse);
else if (errno == EINVAL)
setError(IO_BindError, AlreadyBound);
else
// assume the address is the cause
setError(IO_BindError, NotSupported);
return false;
}
return true;
}
bool KSocketDevice::listen(int backlog)
{
if (m_sockfd != -1)
{
if (kde_listen(m_sockfd, backlog) == -1)
{
setError(IO_ListenError, NotSupported);
return false;
}
resetError();
setFlags(IO_Sequential | IO_Raw | IO_ReadWrite);
return true;
}
// we don't have a socket
// can't listen
setError(IO_ListenError, NotCreated);
return false;
}
bool KSocketDevice::connect(const KResolverEntry& address)
{
resetError();
if (m_sockfd == -1 && !create(address))
return false; // failed creating!
if (kde_connect(m_sockfd, address.address(), address.length()) == -1)
{
if (errno == EISCONN)
return true; // we're already connected
else if (errno == EALREADY || errno == EINPROGRESS)
{
setError(IO_ConnectError, InProgress);
return true;
}
else if (errno == ECONNREFUSED)
setError(IO_ConnectError, ConnectionRefused);
else if (errno == ENETDOWN || errno == ENETUNREACH ||
errno == ENETRESET || errno == ECONNABORTED ||
errno == ECONNRESET || errno == EHOSTDOWN ||
errno == EHOSTUNREACH)
setError(IO_ConnectError, NetFailure);
else
setError(IO_ConnectError, NotSupported);
return false;
}
setFlags(IO_Sequential | IO_Raw | IO_ReadWrite);
return true; // all is well
}
KSocketDevice* KSocketDevice::accept()
{
if (m_sockfd == -1)
{
// can't accept without a socket
setError(IO_AcceptError, NotCreated);
return 0L;
}
struct sockaddr sa;
socklen_t len = sizeof(sa);
int newfd = kde_accept(m_sockfd, &sa, &len);
if (newfd == -1)
{
if (errno == EAGAIN || errno == EWOULDBLOCK)
setError(IO_AcceptError, WouldBlock);
else
setError(IO_AcceptError, UnknownError);
return NULL;
}
return new KSocketDevice(newfd);
}
bool KSocketDevice::disconnect()
{
resetError();
if (m_sockfd == -1)
return false; // can't create
KSocketAddress address;
address.setFamily(AF_UNSPEC);
if (kde_connect(m_sockfd, address.address(), address.length()) == -1)
{
if (errno == EALREADY || errno == EINPROGRESS)
{
setError(IO_ConnectError, InProgress);
return false;
}
else if (errno == ECONNREFUSED)
setError(IO_ConnectError, ConnectionRefused);
else if (errno == ENETDOWN || errno == ENETUNREACH ||
errno == ENETRESET || errno == ECONNABORTED ||
errno == ECONNRESET || errno == EHOSTDOWN ||
errno == EHOSTUNREACH)
setError(IO_ConnectError, NetFailure);
else
setError(IO_ConnectError, NotSupported);
return false;
}
setFlags(IO_Sequential | IO_Raw | IO_ReadWrite);
setState(IO_Open);
return true; // all is well
}
#ifdef USE_QT3
Q_LONG KSocketDevice::bytesAvailable() const
#endif
#ifdef USE_QT4
qint64 KSocketDevice::bytesAvailable() const
#endif
{
if (m_sockfd == -1)
return -1; // there's nothing to read in a closed socket
int nchars;
if (ioctl(m_sockfd, FIONREAD, &nchars) == -1)
return -1; // error!
return nchars;
}
Q_LONG KSocketDevice::waitForMore(int msecs, bool *timeout)
{
if (m_sockfd == -1)
return -1; // there won't ever be anything to read...
bool input;
if (!poll(&input, 0, 0, msecs, timeout))
return -1; // failed polling
return bytesAvailable();
}
static int do_read_common(int sockfd, char *data, Q_ULONG maxlen, KSocketAddress* from, ssize_t &retval, bool peek = false)
{
socklen_t len;
if (from)
{
from->setLength(len = 128); // arbitrary length
retval = ::recvfrom(sockfd, data, maxlen, peek ? MSG_PEEK : 0, from->address(), &len);
}
else
retval = ::recvfrom(sockfd, data, maxlen, peek ? MSG_PEEK : 0, NULL, NULL);
if (retval == -1)
{
if (errno == EAGAIN || errno == EWOULDBLOCK)
return KSocketDevice::WouldBlock;
else
return KSocketDevice::UnknownError;
}
if (retval == 0)
return KSocketDevice::RemotelyDisconnected;
if (from)
from->setLength(len);
return 0;
}
Q_LONG KSocketDevice::readBlock(char *data, Q_ULONG maxlen)
{
resetError();
if (m_sockfd == -1)
return -1;
if (maxlen == 0 || data == 0L)
return 0; // can't read
ssize_t retval;
int err = do_read_common(m_sockfd, data, maxlen, 0L, retval);
if (err)
{
setError(IO_ReadError, static_cast<SocketError>(err));
return -1;
}
return retval;
}
Q_LONG KSocketDevice::readBlock(char *data, Q_ULONG maxlen, KSocketAddress &from)
{
resetError();
if (m_sockfd == -1)
return -1; // nothing to do here
if (data == 0L || maxlen == 0)
return 0; // user doesn't want to read
ssize_t retval;
int err = do_read_common(m_sockfd, data, maxlen, &from, retval);
if (err)
{
setError(IO_ReadError, static_cast<SocketError>(err));
return -1;
}
return retval;
}
Q_LONG KSocketDevice::peekBlock(char *data, Q_ULONG maxlen)
{
resetError();
if (m_sockfd == -1)
return -1;
if (maxlen == 0 || data == 0L)
return 0; // can't read
ssize_t retval;
int err = do_read_common(m_sockfd, data, maxlen, 0L, retval, true);
if (err)
{
setError(IO_ReadError, static_cast<SocketError>(err));
return -1;
}
return retval;
}
Q_LONG KSocketDevice::peekBlock(char *data, Q_ULONG maxlen, KSocketAddress& from)
{
resetError();
if (m_sockfd == -1)
return -1; // nothing to do here
if (data == 0L || maxlen == 0)
return 0; // user doesn't want to read
ssize_t retval;
int err = do_read_common(m_sockfd, data, maxlen, &from, retval, true);
if (err)
{
setError(IO_ReadError, static_cast<SocketError>(err));
return -1;
}
return retval;
}
Q_LONG KSocketDevice::writeBlock(const char *data, Q_ULONG len)
{
return writeBlock(data, len, KSocketAddress());
}
Q_LONG KSocketDevice::writeBlock(const char *data, Q_ULONG len, const KSocketAddress& to)
{
resetError();
if (m_sockfd == -1)
return -1; // can't write to unopen socket
if (data == 0L || len == 0)
return 0; // nothing to be written
ssize_t retval = ::sendto(m_sockfd, data, len, 0, to.address(), to.length());
if (retval == -1)
{
if (errno == EAGAIN || errno == EWOULDBLOCK)
setError(IO_WriteError, WouldBlock);
else
setError(IO_WriteError, UnknownError);
return -1; // nothing written
}
else if (retval == 0)
setError(IO_WriteError, RemotelyDisconnected);
return retval;
}
KSocketAddress KSocketDevice::localAddress() const
{
if (m_sockfd == -1)
return KSocketAddress(); // not open, empty value
if (d->local.family() != AF_UNSPEC)
return d->local;
socklen_t len;
KSocketAddress localAddress;
localAddress.setLength(len = 32); // arbitrary value
if (kde_getsockname(m_sockfd, localAddress.address(), &len) == -1)
// error!
return d->local = KSocketAddress();
#ifdef HAVE_STRUCT_SOCKADDR_SA_LEN
len = localAddress.address()->sa_len;
#endif
if (len <= localAddress.length())
{
// it has fit already
localAddress.setLength(len);
return d->local = localAddress;
}
// no, the socket address is actually larger than we had anticipated
// call again
localAddress.setLength(len);
if (kde_getsockname(m_sockfd, localAddress.address(), &len) == -1)
// error!
return d->local = KSocketAddress();
return d->local = localAddress;
}
KSocketAddress KSocketDevice::peerAddress() const
{
if (m_sockfd == -1)
return KSocketAddress(); // not open, empty value
if (d->peer.family() != AF_UNSPEC)
return d->peer;
socklen_t len;
KSocketAddress peerAddress;
peerAddress.setLength(len = 32); // arbitrary value
if (kde_getpeername(m_sockfd, peerAddress.address(), &len) == -1)
// error!
return d->peer = KSocketAddress();
#ifdef HAVE_STRUCT_SOCKADDR_SA_LEN
len = peerAddress.address()->sa_len;
#endif
if (len <= peerAddress.length())
{
// it has fit already
peerAddress.setLength(len);
return d->peer = peerAddress;
}
// no, the socket address is actually larger than we had anticipated
// call again
peerAddress.setLength(len);
if (kde_getpeername(m_sockfd, peerAddress.address(), &len) == -1)
// error!
return d->peer = KSocketAddress();
return d->peer = peerAddress;
}
KSocketAddress KSocketDevice::externalAddress() const
{
// for normal sockets, the externally visible address is the same
// as the local address
return localAddress();
}
TQSocketNotifier* KSocketDevice::readNotifier() const
{
if (d->input)
return d->input;
TQMutexLocker locker(mutex());
if (d->input)
return d->input;
if (m_sockfd == -1)
{
// socket doesn't exist; can't create notifier
return 0L;
}
return d->input = createNotifier(TQSocketNotifier::Read);
}
TQSocketNotifier* KSocketDevice::writeNotifier() const
{
if (d->output)
return d->output;
TQMutexLocker locker(mutex());
if (d->output)
return d->output;
if (m_sockfd == -1)
{
// socket doesn't exist; can't create notifier
return 0L;
}
return d->output = createNotifier(TQSocketNotifier::Write);
}
TQSocketNotifier* KSocketDevice::exceptionNotifier() const
{
if (d->exception)
return d->exception;
TQMutexLocker locker(mutex());
if (d->exception)
return d->exception;
if (m_sockfd == -1)
{
// socket doesn't exist; can't create notifier
return 0L;
}
return d->exception = createNotifier(TQSocketNotifier::Exception);
}
bool KSocketDevice::poll(bool *input, bool *output, bool *exception,
int timeout, bool* timedout)
{
if (m_sockfd == -1)
{
setError(IO_UnspecifiedError, NotCreated);
return false;
}
resetError();
#ifdef HAVE_POLL
struct pollfd fds;
fds.fd = m_sockfd;
fds.events = 0;
if (input)
{
fds.events |= POLLIN;
*input = false;
}
if (output)
{
fds.events |= POLLOUT;
*output = false;
}
if (exception)
{
fds.events |= POLLPRI;
*exception = false;
}
int retval = ::poll(&fds, 1, timeout);
if (retval == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false;
}
if (retval == 0)
{
// timeout
if (timedout)
*timedout = true;
return true;
}
if (input && fds.revents & POLLIN)
*input = true;
if (output && fds.revents & POLLOUT)
*output = true;
if (exception && fds.revents & POLLPRI)
*exception = true;
if (timedout)
*timedout = false;
return true;
#else
/*
* We don't have poll(2). We'll have to make do with select(2).
*/
fd_set readfds, writefds, exceptfds;
fd_set *preadfds = 0L, *pwritefds = 0L, *pexceptfds = 0L;
if (input)
{
preadfds = &readfds;
FD_ZERO(preadfds);
FD_SET(m_sockfd, preadfds);
*input = false;
}
if (output)
{
pwritefds = &writefds;
FD_ZERO(pwritefds);
FD_SET(m_sockfd, pwritefds);
*output = false;
}
if (exception)
{
pexceptfds = &exceptfds;
FD_ZERO(pexceptfds);
FD_SET(m_sockfd, pexceptfds);
*exception = false;
}
int retval;
if (timeout < 0)
retval = select(m_sockfd + 1, preadfds, pwritefds, pexceptfds, 0L);
else
{
// convert the milliseconds to timeval
struct timeval tv;
tv.tv_sec = timeout / 1000;
tv.tv_usec = timeout % 1000 * 1000;
retval = select(m_sockfd + 1, preadfds, pwritefds, pexceptfds, &tv);
}
if (retval == -1)
{
setError(IO_UnspecifiedError, UnknownError);
return false;
}
if (retval == 0)
{
// timeout
if (timedout)
*timedout = true;
return true;
}
if (input && FD_ISSET(m_sockfd, preadfds))
*input = true;
if (output && FD_ISSET(m_sockfd, pwritefds))
*output = true;
if (exception && FD_ISSET(m_sockfd, pexceptfds))
*exception = true;
return true;
#endif
}
bool KSocketDevice::poll(int timeout, bool *timedout)
{
bool input, output, exception;
return poll(&input, &output, &exception, timeout, timedout);
}
TQSocketNotifier* KSocketDevice::createNotifier(TQSocketNotifier::Type type) const
{
if (m_sockfd == -1)
return 0L;
return new TQSocketNotifier(m_sockfd, type);
}
namespace
{
// simple class to avoid pointer stuff
template<class T> class ptr
{
typedef T type;
type* obj;
public:
ptr() : obj(0)
{ }
ptr(const ptr<T>& other) : obj(other.obj)
{ }
ptr(type* _obj) : obj(_obj)
{ }
~ptr()
{ }
ptr<T>& operator=(const ptr<T>& other)
{ obj = other.obj; return *this; }
ptr<T>& operator=(T* _obj)
{ obj = _obj; return *this; }
type* operator->() const { return obj; }
operator T*() const { return obj; }
bool isNull() const
{ return obj == 0; }
};
}
static KSocketDeviceFactoryBase* defaultImplFactory;
static TQMutex defaultImplFactoryMutex;
typedef TQMap<int, KSocketDeviceFactoryBase* > factoryMap;
static factoryMap factories;
KSocketDevice* KSocketDevice::createDefault(KSocketBase* parent)
{
KSocketDevice* device = dynamic_cast<KSocketDevice*>(parent);
if (device != 0L)
return device;
KSocksSocketDevice::initSocks();
if (defaultImplFactory)
return defaultImplFactory->create(parent);
// the really default
return new KSocketDevice(parent);
}
KSocketDevice* KSocketDevice::createDefault(KSocketBase* parent, int capabilities)
{
KSocketDevice* device = dynamic_cast<KSocketDevice*>(parent);
if (device != 0L)
return device;
TQMutexLocker locker(&defaultImplFactoryMutex);
factoryMap::ConstIterator it = factories.constBegin();
for ( ; it != factories.constEnd(); ++it)
if ((it.key() & capabilities) == capabilities)
// found a match
return it.data()->create(parent);
return 0L; // no default
}
KSocketDeviceFactoryBase*
KSocketDevice::setDefaultImpl(KSocketDeviceFactoryBase* factory)
{
TQMutexLocker locker(&defaultImplFactoryMutex);
KSocketDeviceFactoryBase* old = defaultImplFactory;
defaultImplFactory = factory;
return old;
}
void KSocketDevice::addNewImpl(KSocketDeviceFactoryBase* factory, int capabilities)
{
TQMutexLocker locker(&defaultImplFactoryMutex);
if (factories.contains(capabilities))
delete factories[capabilities];
factories.insert(capabilities, factory);
}