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.
414 lines
10 KiB
414 lines
10 KiB
/*
|
|
* Copyright (C) 2003 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 <tqsocketnotifier.h>
|
|
#include <tqmutex.h>
|
|
|
|
#include "tdesocketaddress.h"
|
|
#include "kresolver.h"
|
|
#include "tdesocketbase.h"
|
|
#include "tdesocketdevice.h"
|
|
#include "kstreamsocket.h"
|
|
#include "kbufferedsocket.h"
|
|
#include "kserversocket.h"
|
|
|
|
using namespace KNetwork;
|
|
|
|
class KNetwork::TDEServerSocketPrivate
|
|
{
|
|
public:
|
|
KResolver resolver;
|
|
KResolverResults resolverResults;
|
|
|
|
enum { None, LookupDone, Bound, Listening } state;
|
|
int backlog;
|
|
int timeout;
|
|
|
|
bool bindWhenFound : 1, listenWhenBound : 1, useTDEBufferedSocket : 1;
|
|
|
|
TDEServerSocketPrivate()
|
|
: state(None), timeout(0), bindWhenFound(false), listenWhenBound(false),
|
|
useTDEBufferedSocket(true)
|
|
{
|
|
resolver.setFlags(KResolver::Passive);
|
|
resolver.setFamily(KResolver::KnownFamily);
|
|
}
|
|
};
|
|
|
|
TDEServerSocket::TDEServerSocket(TQObject* parent, const char *name)
|
|
: TQObject(parent, name), d(new TDEServerSocketPrivate)
|
|
{
|
|
TQObject::connect(&d->resolver, TQ_SIGNAL(finished(KResolverResults)),
|
|
this, TQ_SLOT(lookupFinishedSlot()));
|
|
}
|
|
|
|
TDEServerSocket::TDEServerSocket(const TQString& service, TQObject* parent, const char *name)
|
|
: TQObject(parent, name), d(new TDEServerSocketPrivate)
|
|
{
|
|
TQObject::connect(&d->resolver, TQ_SIGNAL(finished(KResolverResults)),
|
|
this, TQ_SLOT(lookupFinishedSlot()));
|
|
d->resolver.setServiceName(service);
|
|
}
|
|
|
|
TDEServerSocket::TDEServerSocket(const TQString& node, const TQString& service,
|
|
TQObject* parent, const char* name)
|
|
: TQObject(parent, name), d(new TDEServerSocketPrivate)
|
|
{
|
|
TQObject::connect(&d->resolver, TQ_SIGNAL(finished(KResolverResults)),
|
|
this, TQ_SLOT(lookupFinishedSlot()));
|
|
setAddress(node, service);
|
|
}
|
|
|
|
TDEServerSocket::~TDEServerSocket()
|
|
{
|
|
close();
|
|
delete d;
|
|
}
|
|
|
|
bool TDEServerSocket::setSocketOptions(int opts)
|
|
{
|
|
TQMutexLocker locker(mutex());
|
|
TDESocketBase::setSocketOptions(opts); // call parent
|
|
bool result = socketDevice()->setSocketOptions(opts); // and set the implementation
|
|
copyError();
|
|
return result;
|
|
}
|
|
|
|
KResolver& TDEServerSocket::resolver() const
|
|
{
|
|
return d->resolver;
|
|
}
|
|
|
|
const KResolverResults& TDEServerSocket::resolverResults() const
|
|
{
|
|
return d->resolverResults;
|
|
}
|
|
|
|
void TDEServerSocket::setResolutionEnabled(bool enable)
|
|
{
|
|
if (enable)
|
|
d->resolver.setFlags(d->resolver.flags() & ~KResolver::NoResolve);
|
|
else
|
|
d->resolver.setFlags(d->resolver.flags() | KResolver::NoResolve);
|
|
}
|
|
|
|
void TDEServerSocket::setFamily(int families)
|
|
{
|
|
d->resolver.setFamily(families);
|
|
}
|
|
|
|
void TDEServerSocket::setAddress(const TQString& service)
|
|
{
|
|
d->resolver.setNodeName(TQString::null);
|
|
d->resolver.setServiceName(service);
|
|
d->resolverResults.empty();
|
|
if (d->state <= TDEServerSocketPrivate::LookupDone)
|
|
d->state = TDEServerSocketPrivate::None;
|
|
}
|
|
|
|
void TDEServerSocket::setAddress(const TQString& node, const TQString& service)
|
|
{
|
|
d->resolver.setNodeName(node);
|
|
d->resolver.setServiceName(service);
|
|
d->resolverResults.empty();
|
|
if (d->state <= TDEServerSocketPrivate::LookupDone)
|
|
d->state = TDEServerSocketPrivate::None;
|
|
}
|
|
|
|
void TDEServerSocket::setTimeout(int msec)
|
|
{
|
|
d->timeout = msec;
|
|
}
|
|
|
|
bool TDEServerSocket::lookup()
|
|
{
|
|
setError(NoError);
|
|
if (d->resolver.isRunning() && !blocking())
|
|
return true; // already doing lookup
|
|
|
|
if (d->state >= TDEServerSocketPrivate::LookupDone)
|
|
return true; // results are already available
|
|
|
|
// make sure we have at least one parameter for lookup
|
|
if (d->resolver.serviceName().isNull() &&
|
|
!d->resolver.nodeName().isNull())
|
|
d->resolver.setServiceName(TQString::fromLatin1(""));
|
|
|
|
// don't restart the lookups if they had succeeded and
|
|
// the input values weren't changed
|
|
|
|
// reset results
|
|
d->resolverResults = KResolverResults();
|
|
|
|
if (d->resolver.status() <= 0)
|
|
// if it's already running, there's no harm in calling again
|
|
d->resolver.start(); // signal may emit
|
|
|
|
if (blocking())
|
|
{
|
|
// we're in blocking mode operation
|
|
// wait for the results
|
|
|
|
d->resolver.wait(); // signal may be emitted again
|
|
// lookupFinishedSlot has been called
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool TDEServerSocket::bind(const KResolverEntry& address)
|
|
{
|
|
if (socketDevice()->bind(address))
|
|
{
|
|
setError(NoError);
|
|
|
|
d->state = TDEServerSocketPrivate::Bound;
|
|
emit bound(address);
|
|
return true;
|
|
}
|
|
copyError();
|
|
return false;
|
|
}
|
|
|
|
bool TDEServerSocket::bind(const TQString& node, const TQString& service)
|
|
{
|
|
setAddress(node, service);
|
|
return bind();
|
|
}
|
|
|
|
bool TDEServerSocket::bind(const TQString& service)
|
|
{
|
|
setAddress(service);
|
|
return bind();
|
|
}
|
|
|
|
bool TDEServerSocket::bind()
|
|
{
|
|
if (d->state >= TDEServerSocketPrivate::Bound)
|
|
return true;
|
|
|
|
if (d->state < TDEServerSocketPrivate::LookupDone)
|
|
{
|
|
if (!blocking())
|
|
{
|
|
d->bindWhenFound = true;
|
|
bool ok = lookup(); // will call doBind
|
|
if (d->state >= TDEServerSocketPrivate::Bound)
|
|
d->bindWhenFound = false;
|
|
return ok;
|
|
}
|
|
|
|
// not blocking
|
|
if (!lookup())
|
|
return false;
|
|
}
|
|
|
|
return doBind();
|
|
}
|
|
|
|
bool TDEServerSocket::listen(int backlog)
|
|
{
|
|
// WARNING
|
|
// this function has to be reentrant
|
|
// due to the mechanisms used for binding, this function might
|
|
// end up calling itself
|
|
|
|
if (d->state == TDEServerSocketPrivate::Listening)
|
|
return true; // already listening
|
|
|
|
d->backlog = backlog;
|
|
|
|
if (d->state < TDEServerSocketPrivate::Bound)
|
|
{
|
|
// we must bind
|
|
// note that we can end up calling ourselves here
|
|
d->listenWhenBound = true;
|
|
if (!bind())
|
|
{
|
|
d->listenWhenBound = false;
|
|
return false;
|
|
}
|
|
|
|
if (d->state < TDEServerSocketPrivate::Bound)
|
|
// asynchronous lookup in progress...
|
|
// we can't be blocking here anyways
|
|
return true;
|
|
|
|
d->listenWhenBound = false;
|
|
}
|
|
|
|
if (d->state < TDEServerSocketPrivate::Listening)
|
|
return doListen();
|
|
|
|
return true;
|
|
}
|
|
|
|
void TDEServerSocket::close()
|
|
{
|
|
socketDevice()->close();
|
|
if (d->resolver.isRunning())
|
|
d->resolver.cancel(false);
|
|
d->state = TDEServerSocketPrivate::None;
|
|
emit closed();
|
|
}
|
|
|
|
void TDEServerSocket::setAcceptBuffered(bool enable)
|
|
{
|
|
d->useTDEBufferedSocket = enable;
|
|
}
|
|
|
|
KActiveSocketBase* TDEServerSocket::accept()
|
|
{
|
|
if (d->state < TDEServerSocketPrivate::Listening)
|
|
{
|
|
if (!blocking())
|
|
{
|
|
listen();
|
|
setError(WouldBlock);
|
|
return NULL;
|
|
}
|
|
else if (!listen())
|
|
// error happened during listen
|
|
return 0L;
|
|
}
|
|
|
|
// check to see if we're doing a timeout
|
|
if (blocking() && d->timeout > 0)
|
|
{
|
|
bool timedout;
|
|
if (!socketDevice()->poll(d->timeout, &timedout))
|
|
{
|
|
copyError();
|
|
return NULL;
|
|
}
|
|
|
|
if (timedout)
|
|
return 0L;
|
|
}
|
|
|
|
// we're listening here
|
|
TDESocketDevice* accepted = socketDevice()->accept();
|
|
if (!accepted)
|
|
{
|
|
// error happened during accept
|
|
copyError();
|
|
return NULL;
|
|
}
|
|
|
|
KStreamSocket* streamsocket;
|
|
if (d->useTDEBufferedSocket)
|
|
streamsocket = new TDEBufferedSocket();
|
|
else
|
|
streamsocket = new KStreamSocket();
|
|
streamsocket->setSocketDevice(accepted);
|
|
|
|
// FIXME!
|
|
// when KStreamSocket can find out the state of the socket passed through
|
|
// setSocketDevice, this will probably be unnecessary:
|
|
streamsocket->setState(KStreamSocket::Connected);
|
|
streamsocket->setFlags(IO_Sequential | IO_Raw | IO_ReadWrite | IO_Open | IO_Async);
|
|
|
|
return streamsocket;
|
|
}
|
|
|
|
TDESocketAddress TDEServerSocket::localAddress() const
|
|
{
|
|
return socketDevice()->localAddress();
|
|
}
|
|
|
|
TDESocketAddress TDEServerSocket::externalAddress() const
|
|
{
|
|
return socketDevice()->externalAddress();
|
|
}
|
|
|
|
void TDEServerSocket::lookupFinishedSlot()
|
|
{
|
|
if (d->resolver.isRunning() || d->state > TDEServerSocketPrivate::LookupDone)
|
|
return;
|
|
|
|
if (d->resolver.status() < 0)
|
|
{
|
|
setError(LookupFailure);
|
|
emit gotError(LookupFailure);
|
|
d->bindWhenFound = d->listenWhenBound = false;
|
|
d->state = TDEServerSocketPrivate::None;
|
|
return;
|
|
}
|
|
|
|
// lookup succeeded
|
|
d->resolverResults = d->resolver.results();
|
|
d->state = TDEServerSocketPrivate::LookupDone;
|
|
emit hostFound();
|
|
|
|
if (d->bindWhenFound)
|
|
doBind();
|
|
}
|
|
|
|
void TDEServerSocket::copyError()
|
|
{
|
|
setError(socketDevice()->error());
|
|
}
|
|
|
|
bool TDEServerSocket::doBind()
|
|
{
|
|
d->bindWhenFound = false;
|
|
// loop through the results and bind to the first that works
|
|
|
|
KResolverResults::ConstIterator it = d->resolverResults.begin();
|
|
for ( ; it != d->resolverResults.end(); ++it)
|
|
if (bind(*it))
|
|
{
|
|
if (d->listenWhenBound)
|
|
return doListen();
|
|
return true;
|
|
}
|
|
else
|
|
socketDevice()->close(); // didn't work, try again
|
|
|
|
// failed to bind
|
|
emit gotError(error());
|
|
return false;
|
|
}
|
|
|
|
bool TDEServerSocket::doListen()
|
|
{
|
|
if (!socketDevice()->listen(d->backlog))
|
|
{
|
|
copyError();
|
|
emit gotError(error());
|
|
return false; // failed to listen
|
|
}
|
|
|
|
// set up ready accept signal
|
|
TQObject::connect(socketDevice()->readNotifier(), TQ_SIGNAL(activated(int)),
|
|
this, TQ_SIGNAL(readyAccept()));
|
|
d->state = TDEServerSocketPrivate::Listening;
|
|
return true;
|
|
}
|
|
|
|
|
|
#include "kserversocket.moc"
|