00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
#include <config.h>
00026
00027
#include <sys/types.h>
00028
#include <sys/socket.h>
00029
00030
#include "ksocketaddress.h"
00031
#include "kresolver.h"
00032
#include "ksocketdevice.h"
00033
#include "kdatagramsocket.h"
00034
00035
using namespace KNetwork;
00036
00037
00038
00039
00040
00041
00042
00043
00044 KDatagramSocket::KDatagramSocket(
QObject* parent,
const char *name)
00045 :
KClientSocketBase(parent, name), d(0L)
00046 {
00047
peerResolver().
setFamily(KResolver::KnownFamily);
00048
localResolver().
setFamily(KResolver::KnownFamily);
00049
00050
peerResolver().
setSocketType(SOCK_DGRAM);
00051
localResolver().
setSocketType(SOCK_DGRAM);
00052
00053
localResolver().
setFlags(KResolver::Passive);
00054
00055
00056
00057 QObject::connect(&
peerResolver(), SIGNAL(finished(
KResolverResults)),
00058
this, SLOT(lookupFinishedPeer()));
00059 QObject::connect(
this, SIGNAL(
hostFound()),
this, SLOT(lookupFinishedLocal()));
00060 }
00061
00062 KDatagramSocket::~KDatagramSocket()
00063 {
00064
00065
00066
00067 }
00068
00069 bool KDatagramSocket::bind(
const QString& node,
const QString& service)
00070 {
00071
if (
state() >= Bound)
00072
return false;
00073
00074
if (
localResolver().
isRunning())
00075
localResolver().
cancel(
false);
00076
00077
00078
localResolver().
setAddress(node, service);
00079
00080
if (!
lookup())
00081
return false;
00082
00083
00084
00085
00086
if (
state() > HostLookup)
00087
return doBind();
00088
00089
return true;
00090 }
00091
00092 bool KDatagramSocket::connect(
const QString& node,
const QString& service)
00093 {
00094
if (
state() >= Connected)
00095
return true;
00096
00097
if (
peerResolver().
nodeName() != node ||
00098
peerResolver().
serviceName() != service)
00099
peerResolver().
setAddress(node, service);
00100
00101
00102
00103
00104 SocketState s =
state();
00105 setState(s == Connecting ? HostLookup : Idle);
00106
bool ok =
lookup();
00107
if (!ok)
00108 {
00109 setState(s);
00110
return false;
00111 }
00112
00113
00114
00115
if (
state() == HostLookup)
00116 {
00117
00118 setState(Connecting);
00119 emit stateChanged(Connecting);
00120
return true;
00121 }
00122
00123
00124
if (
state() != Connected)
00125 {
00126 setState(Connecting);
00127 emit stateChanged(Connecting);
00128 lookupFinishedPeer();
00129 }
00130
00131
return state() == Connected;
00132 }
00133
00134 KDatagramPacket KDatagramSocket::receive()
00135 {
00136 Q_LONG size =
bytesAvailable();
00137
if (size == 0)
00138 {
00139
00140
00141
if (
blocking())
00142
socketDevice()->
waitForMore(-1);
00143
else
00144 {
00145
00146 setError(IO_ReadError, WouldBlock);
00147 emit gotError(WouldBlock);
00148
return KDatagramPacket();
00149 }
00150
00151
00152 size =
bytesAvailable();
00153 }
00154
00155
QByteArray data(size);
00156
KSocketAddress address;
00157
00158
00159 size = readBlock(data.data(), size, address);
00160
if (size < 0)
00161
00162
return KDatagramPacket();
00163
00164 data.resize(size);
00165
return KDatagramPacket(data, address);
00166 }
00167
00168 Q_LONG KDatagramSocket::send(
const KDatagramPacket& packet)
00169 {
00170
return writeBlock(packet.
data(), packet.
size(), packet.
address());
00171 }
00172
00173
void KDatagramSocket::lookupFinishedLocal()
00174 {
00175
00176
00177
00178
if (!doBind())
00179
return;
00180
00181
if (
peerResults().count() > 0)
00182 {
00183 setState(Connecting);
00184 emit stateChanged(Connecting);
00185
00186 lookupFinishedPeer();
00187 }
00188 }
00189
00190
void KDatagramSocket::lookupFinishedPeer()
00191 {
00192
00193
00194
00195
00196
if (
state() != Connecting)
00197
return;
00198
00199
if (
peerResults().count() == 0)
00200 {
00201
setState(Unconnected);
00202 emit
stateChanged(Unconnected);
00203
return;
00204 }
00205
00206 KResolverResults::ConstIterator it =
peerResults().begin();
00207
for ( ; it !=
peerResults().end(); ++it)
00208
if (
connect(*it))
00209 {
00210
00211
00212
setState(Connected);
00213
00214
00215 emit
stateChanged(Connected);
00216 emit
connected(*it);
00217
return;
00218 }
00219
00220
00221
copyError();
00222
setState(Unconnected);
00223 emit
stateChanged(Unconnected);
00224 emit
gotError(
error());
00225 }
00226
00227
bool KDatagramSocket::doBind()
00228 {
00229
if (
localResults().count() == 0)
00230
return true;
00231
if (
state() >= Bound)
00232
return true;
00233
00234 KResolverResults::ConstIterator it =
localResults().begin();
00235
for ( ; it !=
localResults().end(); ++it)
00236
if (
bind(*it))
00237 {
00238
00239 setupSignals();
00240
return true;
00241 }
00242
00243
00244
00245
copyError();
00246 emit gotError(
error());
00247
return false;
00248 }
00249
00250
void KDatagramSocket::setupSignals()
00251 {
00252
QSocketNotifier *n =
socketDevice()->
readNotifier();
00253
if (n)
00254 {
00255 n->setEnabled(
emitsReadyRead());
00256 QObject::connect(n, SIGNAL(activated(
int)),
this, SLOT(
slotReadActivity()));
00257 }
00258
else
00259
return;
00260
00261 n =
socketDevice()->
writeNotifier();
00262
if (n)
00263 {
00264 n->setEnabled(
emitsReadyWrite());
00265 QObject::connect(n, SIGNAL(activated(
int)),
this, SLOT(
slotWriteActivity()));
00266 }
00267
else
00268
return;
00269 }
00270
00271
#include "kdatagramsocket.moc"