kdatagramsocket.cpp00001
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 Q_LONG KDatagramSocket::writeBlock(const char *data, Q_ULONG len, const KSocketAddress& to)
00174 {
00175 if (to.family() != AF_UNSPEC)
00176 {
00177
00178 if (!socketDevice()->isOpen())
00179
00180 socketDevice()->create(to.family(), SOCK_DGRAM, 0);
00181 }
00182 return KClientSocketBase::writeBlock(data, len, to);
00183 }
00184
00185 void KDatagramSocket::lookupFinishedLocal()
00186 {
00187
00188
00189
00190 if (!doBind())
00191 return;
00192
00193 if (peerResults().count() > 0)
00194 {
00195 setState(Connecting);
00196 emit stateChanged(Connecting);
00197
00198 lookupFinishedPeer();
00199 }
00200 }
00201
00202 void KDatagramSocket::lookupFinishedPeer()
00203 {
00204
00205
00206
00207
00208 if (state() != Connecting)
00209 return;
00210
00211 if (peerResults().count() == 0)
00212 {
00213 setState(Unconnected);
00214 emit stateChanged(Unconnected);
00215 return;
00216 }
00217
00218 KResolverResults::ConstIterator it = peerResults().begin();
00219 for ( ; it != peerResults().end(); ++it)
00220 if (connect(*it))
00221 {
00222
00223
00224 setState(Connected);
00225
00226
00227 emit stateChanged(Connected);
00228 emit connected(*it);
00229 return;
00230 }
00231
00232
00233 copyError();
00234 setState(Unconnected);
00235 emit stateChanged(Unconnected);
00236 emit gotError(error());
00237 }
00238
00239 bool KDatagramSocket::doBind()
00240 {
00241 if (localResults().count() == 0)
00242 return true;
00243 if (state() >= Bound)
00244 return true;
00245
00246 KResolverResults::ConstIterator it = localResults().begin();
00247 for ( ; it != localResults().end(); ++it)
00248 if (bind(*it))
00249 {
00250
00251 setupSignals();
00252 return true;
00253 }
00254
00255
00256
00257 copyError();
00258 emit gotError(error());
00259 return false;
00260 }
00261
00262 void KDatagramSocket::setupSignals()
00263 {
00264 QSocketNotifier *n = socketDevice()->readNotifier();
00265 if (n)
00266 {
00267 n->setEnabled(emitsReadyRead());
00268 QObject::connect(n, SIGNAL(activated(int)), this, SLOT(slotReadActivity()));
00269 }
00270 else
00271 return;
00272
00273 n = socketDevice()->writeNotifier();
00274 if (n)
00275 {
00276 n->setEnabled(emitsReadyWrite());
00277 QObject::connect(n, SIGNAL(activated(int)), this, SLOT(slotWriteActivity()));
00278 }
00279 else
00280 return;
00281 }
00282
00283 #include "kdatagramsocket.moc"
|