KDECore
kdatagramsocket.cpp
Go to the documentation of this file.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 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"