1 | /****************************************************************************
|
---|
2 | **
|
---|
3 | ** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
|
---|
4 | ** Contact: Qt Software Information ([email protected])
|
---|
5 | **
|
---|
6 | ** This file is part of the QtNetwork module of the Qt Toolkit.
|
---|
7 | **
|
---|
8 | ** $QT_BEGIN_LICENSE:LGPL$
|
---|
9 | ** Commercial Usage
|
---|
10 | ** Licensees holding valid Qt Commercial licenses may use this file in
|
---|
11 | ** accordance with the Qt Commercial License Agreement provided with the
|
---|
12 | ** Software or, alternatively, in accordance with the terms contained in
|
---|
13 | ** a written agreement between you and Nokia.
|
---|
14 | **
|
---|
15 | ** GNU Lesser General Public License Usage
|
---|
16 | ** Alternatively, this file may be used under the terms of the GNU Lesser
|
---|
17 | ** General Public License version 2.1 as published by the Free Software
|
---|
18 | ** Foundation and appearing in the file LICENSE.LGPL included in the
|
---|
19 | ** packaging of this file. Please review the following information to
|
---|
20 | ** ensure the GNU Lesser General Public License version 2.1 requirements
|
---|
21 | ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
---|
22 | **
|
---|
23 | ** In addition, as a special exception, Nokia gives you certain
|
---|
24 | ** additional rights. These rights are described in the Nokia Qt LGPL
|
---|
25 | ** Exception version 1.0, included in the file LGPL_EXCEPTION.txt in this
|
---|
26 | ** package.
|
---|
27 | **
|
---|
28 | ** GNU General Public License Usage
|
---|
29 | ** Alternatively, this file may be used under the terms of the GNU
|
---|
30 | ** General Public License version 3.0 as published by the Free Software
|
---|
31 | ** Foundation and appearing in the file LICENSE.GPL included in the
|
---|
32 | ** packaging of this file. Please review the following information to
|
---|
33 | ** ensure the GNU General Public License version 3.0 requirements will be
|
---|
34 | ** met: http://www.gnu.org/copyleft/gpl.html.
|
---|
35 | **
|
---|
36 | ** If you are unsure which license is appropriate for your use, please
|
---|
37 | ** contact the sales department at [email protected].
|
---|
38 | ** $QT_END_LICENSE$
|
---|
39 | **
|
---|
40 | ****************************************************************************/
|
---|
41 |
|
---|
42 | #include <winsock2.h>
|
---|
43 |
|
---|
44 | #include "qnativesocketengine_p.h"
|
---|
45 |
|
---|
46 | #include <qabstracteventdispatcher.h>
|
---|
47 | #include <qsocketnotifier.h>
|
---|
48 | #include <qdebug.h>
|
---|
49 | #include <qdatetime.h>
|
---|
50 |
|
---|
51 | //#define QNATIVESOCKETENGINE_DEBUG
|
---|
52 | #if defined(QNATIVESOCKETENGINE_DEBUG)
|
---|
53 | # include <qstring.h>
|
---|
54 | # include <qbytearray.h>
|
---|
55 | #endif
|
---|
56 |
|
---|
57 | QT_BEGIN_NAMESPACE
|
---|
58 |
|
---|
59 | #if defined(QNATIVESOCKETENGINE_DEBUG)
|
---|
60 |
|
---|
61 | void verboseWSErrorDebug(int r)
|
---|
62 | {
|
---|
63 | switch (r) {
|
---|
64 | case WSANOTINITIALISED : qDebug("WSA error : WSANOTINITIALISED"); break;
|
---|
65 | case WSAEINTR: qDebug("WSA error : WSAEINTR"); break;
|
---|
66 | case WSAEBADF: qDebug("WSA error : WSAEBADF"); break;
|
---|
67 | case WSAEACCES: qDebug("WSA error : WSAEACCES"); break;
|
---|
68 | case WSAEFAULT: qDebug("WSA error : WSAEFAULT"); break;
|
---|
69 | case WSAEINVAL: qDebug("WSA error : WSAEINVAL"); break;
|
---|
70 | case WSAEMFILE: qDebug("WSA error : WSAEMFILE"); break;
|
---|
71 | case WSAEWOULDBLOCK: qDebug("WSA error : WSAEWOULDBLOCK"); break;
|
---|
72 | case WSAEINPROGRESS: qDebug("WSA error : WSAEINPROGRESS"); break;
|
---|
73 | case WSAEALREADY: qDebug("WSA error : WSAEALREADY"); break;
|
---|
74 | case WSAENOTSOCK: qDebug("WSA error : WSAENOTSOCK"); break;
|
---|
75 | case WSAEDESTADDRREQ: qDebug("WSA error : WSAEDESTADDRREQ"); break;
|
---|
76 | case WSAEMSGSIZE: qDebug("WSA error : WSAEMSGSIZE"); break;
|
---|
77 | case WSAEPROTOTYPE: qDebug("WSA error : WSAEPROTOTYPE"); break;
|
---|
78 | case WSAENOPROTOOPT: qDebug("WSA error : WSAENOPROTOOPT"); break;
|
---|
79 | case WSAEPROTONOSUPPORT: qDebug("WSA error : WSAEPROTONOSUPPORT"); break;
|
---|
80 | case WSAESOCKTNOSUPPORT: qDebug("WSA error : WSAESOCKTNOSUPPORT"); break;
|
---|
81 | case WSAEOPNOTSUPP: qDebug("WSA error : WSAEOPNOTSUPP"); break;
|
---|
82 | case WSAEPFNOSUPPORT: qDebug("WSA error : WSAEPFNOSUPPORT"); break;
|
---|
83 | case WSAEAFNOSUPPORT: qDebug("WSA error : WSAEAFNOSUPPORT"); break;
|
---|
84 | case WSAEADDRINUSE: qDebug("WSA error : WSAEADDRINUSE"); break;
|
---|
85 | case WSAEADDRNOTAVAIL: qDebug("WSA error : WSAEADDRNOTAVAIL"); break;
|
---|
86 | case WSAENETDOWN: qDebug("WSA error : WSAENETDOWN"); break;
|
---|
87 | case WSAENETUNREACH: qDebug("WSA error : WSAENETUNREACH"); break;
|
---|
88 | case WSAENETRESET: qDebug("WSA error : WSAENETRESET"); break;
|
---|
89 | case WSAECONNABORTED: qDebug("WSA error : WSAECONNABORTED"); break;
|
---|
90 | case WSAECONNRESET: qDebug("WSA error : WSAECONNRESET"); break;
|
---|
91 | case WSAENOBUFS: qDebug("WSA error : WSAENOBUFS"); break;
|
---|
92 | case WSAEISCONN: qDebug("WSA error : WSAEISCONN"); break;
|
---|
93 | case WSAENOTCONN: qDebug("WSA error : WSAENOTCONN"); break;
|
---|
94 | case WSAESHUTDOWN: qDebug("WSA error : WSAESHUTDOWN"); break;
|
---|
95 | case WSAETOOMANYREFS: qDebug("WSA error : WSAETOOMANYREFS"); break;
|
---|
96 | case WSAETIMEDOUT: qDebug("WSA error : WSAETIMEDOUT"); break;
|
---|
97 | case WSAECONNREFUSED: qDebug("WSA error : WSAECONNREFUSED"); break;
|
---|
98 | case WSAELOOP: qDebug("WSA error : WSAELOOP"); break;
|
---|
99 | case WSAENAMETOOLONG: qDebug("WSA error : WSAENAMETOOLONG"); break;
|
---|
100 | case WSAEHOSTDOWN: qDebug("WSA error : WSAEHOSTDOWN"); break;
|
---|
101 | case WSAEHOSTUNREACH: qDebug("WSA error : WSAEHOSTUNREACH"); break;
|
---|
102 | case WSAENOTEMPTY: qDebug("WSA error : WSAENOTEMPTY"); break;
|
---|
103 | case WSAEPROCLIM: qDebug("WSA error : WSAEPROCLIM"); break;
|
---|
104 | case WSAEUSERS: qDebug("WSA error : WSAEUSERS"); break;
|
---|
105 | case WSAEDQUOT: qDebug("WSA error : WSAEDQUOT"); break;
|
---|
106 | case WSAESTALE: qDebug("WSA error : WSAESTALE"); break;
|
---|
107 | case WSAEREMOTE: qDebug("WSA error : WSAEREMOTE"); break;
|
---|
108 | case WSAEDISCON: qDebug("WSA error : WSAEDISCON"); break;
|
---|
109 | default: qDebug("WSA error : Unknown"); break;
|
---|
110 | }
|
---|
111 | qErrnoWarning(r, "more details");
|
---|
112 | }
|
---|
113 |
|
---|
114 | /*
|
---|
115 | Returns a human readable representation of the first \a len
|
---|
116 | characters in \a data.
|
---|
117 | */
|
---|
118 | static QByteArray qt_prettyDebug(const char *data, int len, int maxLength)
|
---|
119 | {
|
---|
120 | if (!data) return "(null)";
|
---|
121 | QByteArray out;
|
---|
122 | for (int i = 0; i < len; ++i) {
|
---|
123 | char c = data[i];
|
---|
124 | if (isprint(int(uchar(c)))) {
|
---|
125 | out += c;
|
---|
126 | } else switch (c) {
|
---|
127 | case '\n': out += "\\n"; break;
|
---|
128 | case '\r': out += "\\r"; break;
|
---|
129 | case '\t': out += "\\t"; break;
|
---|
130 | default:
|
---|
131 | QString tmp;
|
---|
132 | tmp.sprintf("\\%o", c);
|
---|
133 | out += tmp.toLatin1().constData();
|
---|
134 | }
|
---|
135 | }
|
---|
136 |
|
---|
137 | if (len < maxLength)
|
---|
138 | out += "...";
|
---|
139 |
|
---|
140 | return out;
|
---|
141 | }
|
---|
142 |
|
---|
143 |
|
---|
144 | #define WS_ERROR_DEBUG(x) verboseWSErrorDebug(x);
|
---|
145 |
|
---|
146 | #else
|
---|
147 |
|
---|
148 | #define WS_ERROR_DEBUG(x) Q_UNUSED(x)
|
---|
149 |
|
---|
150 | #endif
|
---|
151 |
|
---|
152 | #if !defined (QT_NO_IPV6)
|
---|
153 |
|
---|
154 | // Use our own defines and structs which we know are correct
|
---|
155 | # define QT_SS_MAXSIZE 128
|
---|
156 | # define QT_SS_ALIGNSIZE (sizeof(__int64))
|
---|
157 | # define QT_SS_PAD1SIZE (QT_SS_ALIGNSIZE - sizeof (short))
|
---|
158 | # define QT_SS_PAD2SIZE (QT_SS_MAXSIZE - (sizeof (short) + QT_SS_PAD1SIZE + QT_SS_ALIGNSIZE))
|
---|
159 | struct qt_sockaddr_storage {
|
---|
160 | short ss_family;
|
---|
161 | char __ss_pad1[QT_SS_PAD1SIZE];
|
---|
162 | __int64 __ss_align;
|
---|
163 | char __ss_pad2[QT_SS_PAD2SIZE];
|
---|
164 | };
|
---|
165 |
|
---|
166 | // sockaddr_in6 size changed between old and new SDK
|
---|
167 | // Only the new version is the correct one, so always
|
---|
168 | // use this structure.
|
---|
169 | struct qt_in6_addr {
|
---|
170 | u_char qt_s6_addr[16];
|
---|
171 | };
|
---|
172 | typedef struct {
|
---|
173 | short sin6_family; /* AF_INET6 */
|
---|
174 | u_short sin6_port; /* Transport level port number */
|
---|
175 | u_long sin6_flowinfo; /* IPv6 flow information */
|
---|
176 | struct qt_in6_addr sin6_addr; /* IPv6 address */
|
---|
177 | u_long sin6_scope_id; /* set of interfaces for a scope */
|
---|
178 | } qt_sockaddr_in6;
|
---|
179 |
|
---|
180 | #else
|
---|
181 |
|
---|
182 | typedef void * qt_sockaddr_in6 ;
|
---|
183 |
|
---|
184 |
|
---|
185 | #endif
|
---|
186 |
|
---|
187 | #ifndef AF_INET6
|
---|
188 | #define AF_INET6 23 /* Internetwork Version 6 */
|
---|
189 | #endif
|
---|
190 |
|
---|
191 | #ifndef SO_EXCLUSIVEADDRUSE
|
---|
192 | #define SO_EXCLUSIVEADDRUSE ((int)(~SO_REUSEADDR)) /* disallow local address reuse */
|
---|
193 | #endif
|
---|
194 |
|
---|
195 | //###
|
---|
196 | #define QT_SOCKLEN_T int
|
---|
197 | #define QT_SOCKOPTLEN_T int
|
---|
198 |
|
---|
199 |
|
---|
200 | /*
|
---|
201 | Extracts the port and address from a sockaddr, and stores them in
|
---|
202 | \a port and \a addr if they are non-null.
|
---|
203 | */
|
---|
204 | static inline void qt_socket_getPortAndAddress(SOCKET socketDescriptor, struct sockaddr *sa, quint16 *port, QHostAddress *address)
|
---|
205 | {
|
---|
206 | #if !defined (QT_NO_IPV6)
|
---|
207 | if (sa->sa_family == AF_INET6) {
|
---|
208 | qt_sockaddr_in6 *sa6 = (qt_sockaddr_in6 *)sa;
|
---|
209 | Q_IPV6ADDR tmp;
|
---|
210 | for (int i = 0; i < 16; ++i)
|
---|
211 | tmp.c[i] = sa6->sin6_addr.qt_s6_addr[i];
|
---|
212 | QHostAddress a;
|
---|
213 | a.setAddress(tmp);
|
---|
214 | if (address)
|
---|
215 | *address = a;
|
---|
216 | if (port)
|
---|
217 | WSANtohs(socketDescriptor, sa6->sin6_port, port);
|
---|
218 | } else
|
---|
219 | #endif
|
---|
220 | if (sa->sa_family == AF_INET) {
|
---|
221 | struct sockaddr_in *sa4 = (struct sockaddr_in *)sa;
|
---|
222 | unsigned long addr;
|
---|
223 | WSANtohl(socketDescriptor, sa4->sin_addr.s_addr, &addr);
|
---|
224 | QHostAddress a;
|
---|
225 | a.setAddress(addr);
|
---|
226 | if (address)
|
---|
227 | *address = a;
|
---|
228 | if (port)
|
---|
229 | WSANtohs(socketDescriptor, sa4->sin_port, port);
|
---|
230 | }
|
---|
231 | }
|
---|
232 |
|
---|
233 |
|
---|
234 | /*! \internal
|
---|
235 |
|
---|
236 | Sets the port and address to a sockaddr. Requires that sa point to the IPv6 struct if the address is IPv6.
|
---|
237 | */
|
---|
238 | static inline void qt_socket_setPortAndAddress(SOCKET socketDescriptor, sockaddr_in * sockAddrIPv4, qt_sockaddr_in6 * sockAddrIPv6,
|
---|
239 | quint16 port, const QHostAddress & address, sockaddr ** sockAddrPtr, QT_SOCKLEN_T *sockAddrSize)
|
---|
240 | {
|
---|
241 | #if !defined(QT_NO_IPV6)
|
---|
242 | if (address.protocol() == QAbstractSocket::IPv6Protocol) {
|
---|
243 | memset(sockAddrIPv6, 0, sizeof(qt_sockaddr_in6));
|
---|
244 | sockAddrIPv6->sin6_family = AF_INET6;
|
---|
245 | WSAHtons(socketDescriptor, port, &(sockAddrIPv6->sin6_port));
|
---|
246 | Q_IPV6ADDR tmp = address.toIPv6Address();
|
---|
247 | memcpy(&(sockAddrIPv6->sin6_addr.qt_s6_addr), &tmp, sizeof(tmp));
|
---|
248 | *sockAddrSize = sizeof(qt_sockaddr_in6);
|
---|
249 | *sockAddrPtr = (struct sockaddr *) sockAddrIPv6;
|
---|
250 | } else
|
---|
251 | #endif
|
---|
252 | if (address.protocol() == QAbstractSocket::IPv4Protocol
|
---|
253 | || address.protocol() == QAbstractSocket::UnknownNetworkLayerProtocol) {
|
---|
254 | memset(sockAddrIPv4, 0, sizeof(sockaddr_in));
|
---|
255 | sockAddrIPv4->sin_family = AF_INET;
|
---|
256 | WSAHtons(socketDescriptor, port, &(sockAddrIPv4->sin_port));
|
---|
257 | WSAHtonl(socketDescriptor, address.toIPv4Address(), &(sockAddrIPv4->sin_addr.s_addr));
|
---|
258 | *sockAddrSize = sizeof(sockaddr_in);
|
---|
259 | *sockAddrPtr = (struct sockaddr *) sockAddrIPv4;
|
---|
260 | } else {
|
---|
261 | // unreachable
|
---|
262 | }
|
---|
263 | }
|
---|
264 |
|
---|
265 | /*! \internal
|
---|
266 |
|
---|
267 | */
|
---|
268 | static inline QAbstractSocket::SocketType qt_socket_getType(int socketDescriptor)
|
---|
269 | {
|
---|
270 | int value = 0;
|
---|
271 | QT_SOCKLEN_T valueSize = sizeof(value);
|
---|
272 | if (::getsockopt(socketDescriptor, SOL_SOCKET, SO_TYPE, (char *) &value, &valueSize) != 0) {
|
---|
273 | WS_ERROR_DEBUG(WSAGetLastError());
|
---|
274 | } else {
|
---|
275 | if (value == SOCK_STREAM)
|
---|
276 | return QAbstractSocket::TcpSocket;
|
---|
277 | else if (value == SOCK_DGRAM)
|
---|
278 | return QAbstractSocket::UdpSocket;
|
---|
279 | }
|
---|
280 | return QAbstractSocket::UnknownSocketType;
|
---|
281 | }
|
---|
282 |
|
---|
283 | /*! \internal
|
---|
284 |
|
---|
285 | */
|
---|
286 | static inline int qt_socket_getMaxMsgSize(int socketDescriptor)
|
---|
287 | {
|
---|
288 | int value = 0;
|
---|
289 | QT_SOCKLEN_T valueSize = sizeof(value);
|
---|
290 | if (::getsockopt(socketDescriptor, SOL_SOCKET, SO_MAX_MSG_SIZE, (char *) &value, &valueSize) != 0) {
|
---|
291 | WS_ERROR_DEBUG(WSAGetLastError());
|
---|
292 | }
|
---|
293 | return value;
|
---|
294 | }
|
---|
295 |
|
---|
296 | QWindowsSockInit::QWindowsSockInit()
|
---|
297 | : version(0)
|
---|
298 | {
|
---|
299 | //### should we try for 2.2 on all platforms ??
|
---|
300 | WSAData wsadata;
|
---|
301 |
|
---|
302 | // IPv6 requires Winsock v2.0 or better.
|
---|
303 | if (WSAStartup(MAKEWORD(2,0), &wsadata) != 0) {
|
---|
304 | qWarning("QTcpSocketAPI: WinSock v2.0 initialization failed.");
|
---|
305 | } else {
|
---|
306 | version = 0x20;
|
---|
307 | }
|
---|
308 | }
|
---|
309 |
|
---|
310 | QWindowsSockInit::~QWindowsSockInit()
|
---|
311 | {
|
---|
312 | WSACleanup();
|
---|
313 | }
|
---|
314 |
|
---|
315 | // MS Transport Provider IOCTL to control
|
---|
316 | // reporting PORT_UNREACHABLE messages
|
---|
317 | // on UDP sockets via recv/WSARecv/etc.
|
---|
318 | // Path TRUE in input buffer to enable (default if supported),
|
---|
319 | // FALSE to disable.
|
---|
320 | #ifndef SIO_UDP_CONNRESET
|
---|
321 | # ifndef IOC_VENDOR
|
---|
322 | # define IOC_VENDOR 0x18000000
|
---|
323 | # endif
|
---|
324 | # ifndef _WSAIOW
|
---|
325 | # define _WSAIOW(x,y) (IOC_IN|(x)|(y))
|
---|
326 | # endif
|
---|
327 | # define SIO_UDP_CONNRESET _WSAIOW(IOC_VENDOR,12)
|
---|
328 | #endif
|
---|
329 |
|
---|
330 | bool QNativeSocketEnginePrivate::createNewSocket(QAbstractSocket::SocketType socketType, QAbstractSocket::NetworkLayerProtocol socketProtocol)
|
---|
331 | {
|
---|
332 |
|
---|
333 | //### no ip6 support on winsocket 1.1 but we will try not to use this !!!!!!!!!!!!1
|
---|
334 | /*
|
---|
335 | if (winsockVersion < 0x20 && socketProtocol == QAbstractSocket::IPv6Protocol) {
|
---|
336 | //### no ip6 support
|
---|
337 | return -1;
|
---|
338 | }
|
---|
339 | */
|
---|
340 |
|
---|
341 | int protocol = (socketProtocol == QAbstractSocket::IPv6Protocol) ? AF_INET6 : AF_INET;
|
---|
342 | int type = (socketType == QAbstractSocket::UdpSocket) ? SOCK_DGRAM : SOCK_STREAM;
|
---|
343 | // MSDN KB179942 states that on winnt 4 WSA_FLAG_OVERLAPPED is needed if socket is to be non blocking
|
---|
344 | // and recomends alwasy doing it for cross windows version comapablity.
|
---|
345 | SOCKET socket = ::WSASocket(protocol, type, 0, NULL, 0, WSA_FLAG_OVERLAPPED);
|
---|
346 |
|
---|
347 | if (socket == INVALID_SOCKET) {
|
---|
348 | int err = WSAGetLastError();
|
---|
349 | WS_ERROR_DEBUG(err);
|
---|
350 | switch (err) {
|
---|
351 | case WSANOTINITIALISED:
|
---|
352 | //###
|
---|
353 | break;
|
---|
354 | case WSAEAFNOSUPPORT:
|
---|
355 | case WSAESOCKTNOSUPPORT:
|
---|
356 | case WSAEPROTOTYPE:
|
---|
357 | case WSAEINVAL:
|
---|
358 | setError(QAbstractSocket::UnsupportedSocketOperationError, ProtocolUnsupportedErrorString);
|
---|
359 | break;
|
---|
360 | case WSAEMFILE:
|
---|
361 | case WSAENOBUFS:
|
---|
362 | setError(QAbstractSocket::SocketResourceError, ResourceErrorString);
|
---|
363 | break;
|
---|
364 | default:
|
---|
365 | break;
|
---|
366 | }
|
---|
367 |
|
---|
368 | return false;
|
---|
369 | }
|
---|
370 |
|
---|
371 | #if !defined(Q_OS_WINCE)
|
---|
372 | // enable new behavior using
|
---|
373 | // SIO_UDP_CONNRESET
|
---|
374 | DWORD dwBytesReturned = 0;
|
---|
375 | int bNewBehavior = 1;
|
---|
376 | if (::WSAIoctl(socket, SIO_UDP_CONNRESET, &bNewBehavior, sizeof(bNewBehavior),
|
---|
377 | NULL, 0, &dwBytesReturned, NULL, NULL) == SOCKET_ERROR) {
|
---|
378 | // not to worry isBogusUdpReadNotification() should handle this otherwise
|
---|
379 | int err = WSAGetLastError();
|
---|
380 | WS_ERROR_DEBUG(err);
|
---|
381 | }
|
---|
382 | #endif
|
---|
383 |
|
---|
384 | socketDescriptor = socket;
|
---|
385 | return true;
|
---|
386 |
|
---|
387 | }
|
---|
388 |
|
---|
389 | /*! \internal
|
---|
390 |
|
---|
391 | Returns the value of the socket option \a opt.
|
---|
392 | */
|
---|
393 | int QNativeSocketEnginePrivate::option(QNativeSocketEngine::SocketOption opt) const
|
---|
394 | {
|
---|
395 | Q_Q(const QNativeSocketEngine);
|
---|
396 | if (!q->isValid())
|
---|
397 | return -1;
|
---|
398 |
|
---|
399 | int n = -1;
|
---|
400 | switch (opt) {
|
---|
401 | case QNativeSocketEngine::ReceiveBufferSocketOption:
|
---|
402 | n = SO_RCVBUF;
|
---|
403 | break;
|
---|
404 | case QNativeSocketEngine::SendBufferSocketOption:
|
---|
405 | n = SO_SNDBUF;
|
---|
406 | break;
|
---|
407 | case QNativeSocketEngine::BroadcastSocketOption:
|
---|
408 | n = SO_BROADCAST;
|
---|
409 | break;
|
---|
410 | case QNativeSocketEngine::NonBlockingSocketOption: {
|
---|
411 | unsigned long buf = 0;
|
---|
412 | if (WSAIoctl(socketDescriptor, FIONBIO, 0,0, &buf, sizeof(buf), 0,0,0) == 0)
|
---|
413 | return buf;
|
---|
414 | else
|
---|
415 | return -1;
|
---|
416 | break;
|
---|
417 | }
|
---|
418 | case QNativeSocketEngine::AddressReusable:
|
---|
419 | n = SO_REUSEADDR;
|
---|
420 | break;
|
---|
421 | case QNativeSocketEngine::BindExclusively:
|
---|
422 | n = SO_EXCLUSIVEADDRUSE;
|
---|
423 | break;
|
---|
424 | case QNativeSocketEngine::ReceiveOutOfBandData:
|
---|
425 | n = SO_OOBINLINE;
|
---|
426 | break;
|
---|
427 | }
|
---|
428 |
|
---|
429 | int v = -1;
|
---|
430 | QT_SOCKOPTLEN_T len = sizeof(v);
|
---|
431 | if (getsockopt(socketDescriptor, SOL_SOCKET, n, (char *) &v, &len) != -1)
|
---|
432 | return v;
|
---|
433 | return -1;
|
---|
434 | }
|
---|
435 |
|
---|
436 |
|
---|
437 | /*! \internal
|
---|
438 | Sets the socket option \a opt to \a v.
|
---|
439 | */
|
---|
440 | bool QNativeSocketEnginePrivate::setOption(QNativeSocketEngine::SocketOption opt, int v)
|
---|
441 | {
|
---|
442 | Q_Q(const QNativeSocketEngine);
|
---|
443 | if (!q->isValid())
|
---|
444 | return false;
|
---|
445 |
|
---|
446 | int n = 0;
|
---|
447 | switch (opt) {
|
---|
448 | case QNativeSocketEngine::ReceiveBufferSocketOption:
|
---|
449 | n = SO_RCVBUF;
|
---|
450 | break;
|
---|
451 | case QNativeSocketEngine::SendBufferSocketOption:
|
---|
452 | n = SO_SNDBUF;
|
---|
453 | break;
|
---|
454 | case QNativeSocketEngine::BroadcastSocketOption:
|
---|
455 | n = SO_BROADCAST;
|
---|
456 | break;
|
---|
457 | case QNativeSocketEngine::NonBlockingSocketOption:
|
---|
458 | {
|
---|
459 | unsigned long buf = v;
|
---|
460 | unsigned long outBuf;
|
---|
461 | DWORD sizeWritten = 0;
|
---|
462 | if (::WSAIoctl(socketDescriptor, FIONBIO, &buf, sizeof(unsigned long), &outBuf, sizeof(unsigned long), &sizeWritten, 0,0) == SOCKET_ERROR) {
|
---|
463 | WS_ERROR_DEBUG(WSAGetLastError());
|
---|
464 | return false;
|
---|
465 | }
|
---|
466 | return true;
|
---|
467 | break;
|
---|
468 | }
|
---|
469 | case QNativeSocketEngine::AddressReusable:
|
---|
470 | n = SO_REUSEADDR;
|
---|
471 | break;
|
---|
472 | case QNativeSocketEngine::BindExclusively:
|
---|
473 | n = SO_EXCLUSIVEADDRUSE;
|
---|
474 | break;
|
---|
475 | case QNativeSocketEngine::ReceiveOutOfBandData:
|
---|
476 | n = SO_OOBINLINE;
|
---|
477 | break;
|
---|
478 | }
|
---|
479 |
|
---|
480 | if (::setsockopt(socketDescriptor, SOL_SOCKET, n, (char*)&v, sizeof(v)) != 0) {
|
---|
481 | WS_ERROR_DEBUG(WSAGetLastError());
|
---|
482 | return false;
|
---|
483 | }
|
---|
484 | return true;
|
---|
485 | }
|
---|
486 |
|
---|
487 | /*!
|
---|
488 | Fetches information about both ends of the connection: whatever is
|
---|
489 | available.
|
---|
490 | */
|
---|
491 | bool QNativeSocketEnginePrivate::fetchConnectionParameters()
|
---|
492 | {
|
---|
493 | localPort = 0;
|
---|
494 | localAddress.clear();
|
---|
495 | peerPort = 0;
|
---|
496 | peerAddress.clear();
|
---|
497 |
|
---|
498 | if (socketDescriptor == -1)
|
---|
499 | return false;
|
---|
500 |
|
---|
501 | #if !defined (QT_NO_IPV6)
|
---|
502 | struct qt_sockaddr_storage sa;
|
---|
503 | #else
|
---|
504 | struct sockaddr_in sa;
|
---|
505 | #endif
|
---|
506 | struct sockaddr *pSa = (struct sockaddr *) &sa;
|
---|
507 |
|
---|
508 | QT_SOCKLEN_T sz = sizeof(sa);
|
---|
509 |
|
---|
510 | memset(&sa, 0, sizeof(sa));
|
---|
511 | if (::getsockname(socketDescriptor, pSa, &sz) == 0) {
|
---|
512 | qt_socket_getPortAndAddress(socketDescriptor, pSa, &localPort, &localAddress);
|
---|
513 | // Determine protocol family
|
---|
514 | switch (pSa->sa_family) {
|
---|
515 | case AF_INET:
|
---|
516 | socketProtocol = QAbstractSocket::IPv4Protocol;
|
---|
517 | break;
|
---|
518 | #if !defined (QT_NO_IPV6)
|
---|
519 | case AF_INET6:
|
---|
520 | socketProtocol = QAbstractSocket::IPv6Protocol;
|
---|
521 | break;
|
---|
522 | #endif
|
---|
523 | default:
|
---|
524 | socketProtocol = QAbstractSocket::UnknownNetworkLayerProtocol;
|
---|
525 | break;
|
---|
526 | }
|
---|
527 | } else {
|
---|
528 | int err = WSAGetLastError();
|
---|
529 | WS_ERROR_DEBUG(err);
|
---|
530 | if (err == WSAENOTSOCK) {
|
---|
531 | setError(QAbstractSocket::UnsupportedSocketOperationError,
|
---|
532 | InvalidSocketErrorString);
|
---|
533 | return false;
|
---|
534 | }
|
---|
535 | }
|
---|
536 |
|
---|
537 | memset(&sa, 0, sizeof(sa));
|
---|
538 | if (::getpeername(socketDescriptor, pSa, &sz) == 0) {
|
---|
539 | qt_socket_getPortAndAddress(socketDescriptor, pSa, &peerPort, &peerAddress);
|
---|
540 | } else {
|
---|
541 | WS_ERROR_DEBUG(WSAGetLastError());
|
---|
542 | }
|
---|
543 |
|
---|
544 | socketType = qt_socket_getType(socketDescriptor);
|
---|
545 |
|
---|
546 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
547 | QString socketProtocolStr = "UnknownProtocol";
|
---|
548 | if (socketProtocol == QAbstractSocket::IPv4Protocol) socketProtocolStr = "IPv4Protocol";
|
---|
549 | else if (socketProtocol == QAbstractSocket::IPv6Protocol) socketProtocolStr = "IPv6Protocol";
|
---|
550 |
|
---|
551 | QString socketTypeStr = "UnknownSocketType";
|
---|
552 | if (socketType == QAbstractSocket::TcpSocket) socketTypeStr = "TcpSocket";
|
---|
553 | else if (socketType == QAbstractSocket::UdpSocket) socketTypeStr = "UdpSocket";
|
---|
554 |
|
---|
555 | qDebug("QNativeSocketEnginePrivate::fetchConnectionParameters() localAddress == %s, localPort = %i, peerAddress == %s, peerPort = %i, socketProtocol == %s, socketType == %s", localAddress.toString().toLatin1().constData(), localPort, peerAddress.toString().toLatin1().constData(), peerPort, socketProtocolStr.toLatin1().constData(), socketTypeStr.toLatin1().constData());
|
---|
556 | #endif
|
---|
557 |
|
---|
558 | return true;
|
---|
559 | }
|
---|
560 |
|
---|
561 |
|
---|
562 | bool QNativeSocketEnginePrivate::nativeConnect(const QHostAddress &address, quint16 port)
|
---|
563 | {
|
---|
564 |
|
---|
565 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
566 | qDebug("QNativeSocketEnginePrivate::nativeConnect() to %s :: %i", address.toString().toLatin1().constData(), port);
|
---|
567 | #endif
|
---|
568 |
|
---|
569 | struct sockaddr_in sockAddrIPv4;
|
---|
570 | qt_sockaddr_in6 sockAddrIPv6;
|
---|
571 | struct sockaddr *sockAddrPtr;
|
---|
572 | QT_SOCKLEN_T sockAddrSize;
|
---|
573 |
|
---|
574 | qt_socket_setPortAndAddress(socketDescriptor, &sockAddrIPv4, &sockAddrIPv6, port, address, &sockAddrPtr, &sockAddrSize);
|
---|
575 |
|
---|
576 | forever {
|
---|
577 | int connectResult = ::WSAConnect(socketDescriptor, sockAddrPtr, sockAddrSize, 0,0,0,0);
|
---|
578 | if (connectResult == SOCKET_ERROR) {
|
---|
579 | int err = WSAGetLastError();
|
---|
580 | WS_ERROR_DEBUG(err);
|
---|
581 |
|
---|
582 | switch (err) {
|
---|
583 | case WSANOTINITIALISED:
|
---|
584 | //###
|
---|
585 | break;
|
---|
586 | case WSAEISCONN:
|
---|
587 | socketState = QAbstractSocket::ConnectedState;
|
---|
588 | break;
|
---|
589 | case WSAEWOULDBLOCK: {
|
---|
590 | // If WSAConnect returns WSAEWOULDBLOCK on the second
|
---|
591 | // connection attempt, we have to check SO_ERROR's
|
---|
592 | // value to detect ECONNREFUSED. If we don't get
|
---|
593 | // ECONNREFUSED, we'll have to treat it as an
|
---|
594 | // unfinished operation.
|
---|
595 | int value = 0;
|
---|
596 | QT_SOCKLEN_T valueSize = sizeof(value);
|
---|
597 | if (::getsockopt(socketDescriptor, SOL_SOCKET, SO_ERROR, (char *) &value, &valueSize) == 0) {
|
---|
598 | if (value == WSAECONNREFUSED) {
|
---|
599 | setError(QAbstractSocket::ConnectionRefusedError, ConnectionRefusedErrorString);
|
---|
600 | socketState = QAbstractSocket::UnconnectedState;
|
---|
601 | break;
|
---|
602 | }
|
---|
603 | if (value == WSAETIMEDOUT) {
|
---|
604 | setError(QAbstractSocket::NetworkError, ConnectionTimeOutErrorString);
|
---|
605 | socketState = QAbstractSocket::UnconnectedState;
|
---|
606 | break;
|
---|
607 | }
|
---|
608 | }
|
---|
609 | // fall through
|
---|
610 | }
|
---|
611 | case WSAEINPROGRESS:
|
---|
612 | setError(QAbstractSocket::UnfinishedSocketOperationError, InvalidSocketErrorString);
|
---|
613 | socketState = QAbstractSocket::ConnectingState;
|
---|
614 | break;
|
---|
615 | case WSAEADDRINUSE:
|
---|
616 | setError(QAbstractSocket::NetworkError, AddressInuseErrorString);
|
---|
617 | break;
|
---|
618 | case WSAECONNREFUSED:
|
---|
619 | setError(QAbstractSocket::ConnectionRefusedError, ConnectionRefusedErrorString);
|
---|
620 | socketState = QAbstractSocket::UnconnectedState;
|
---|
621 | break;
|
---|
622 | case WSAETIMEDOUT:
|
---|
623 | setError(QAbstractSocket::NetworkError, ConnectionTimeOutErrorString);
|
---|
624 | break;
|
---|
625 | case WSAEACCES:
|
---|
626 | setError(QAbstractSocket::SocketAccessError, AccessErrorString);
|
---|
627 | socketState = QAbstractSocket::UnconnectedState;
|
---|
628 | break;
|
---|
629 | case WSAEHOSTUNREACH:
|
---|
630 | setError(QAbstractSocket::NetworkError, HostUnreachableErrorString);
|
---|
631 | socketState = QAbstractSocket::UnconnectedState;
|
---|
632 | break;
|
---|
633 | case WSAENETUNREACH:
|
---|
634 | setError(QAbstractSocket::NetworkError, NetworkUnreachableErrorString);
|
---|
635 | socketState = QAbstractSocket::UnconnectedState;
|
---|
636 | break;
|
---|
637 | case WSAEINVAL:
|
---|
638 | case WSAEALREADY:
|
---|
639 | setError(QAbstractSocket::UnfinishedSocketOperationError, InvalidSocketErrorString);
|
---|
640 | break;
|
---|
641 | default:
|
---|
642 | break;
|
---|
643 | }
|
---|
644 | if (socketState != QAbstractSocket::ConnectedState) {
|
---|
645 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
646 | qDebug("QNativeSocketEnginePrivate::nativeConnect(%s, %i) == false (%s)",
|
---|
647 | address.toString().toLatin1().constData(), port,
|
---|
648 | socketState == QAbstractSocket::ConnectingState
|
---|
649 | ? "Connection in progress" : socketErrorString.toLatin1().constData());
|
---|
650 | #endif
|
---|
651 | return false;
|
---|
652 | }
|
---|
653 | }
|
---|
654 | break;
|
---|
655 | }
|
---|
656 |
|
---|
657 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
658 | qDebug("QNativeSocketEnginePrivate::nativeConnect(%s, %i) == true",
|
---|
659 | address.toString().toLatin1().constData(), port);
|
---|
660 | #endif
|
---|
661 |
|
---|
662 | socketState = QAbstractSocket::ConnectedState;
|
---|
663 | return true;
|
---|
664 | }
|
---|
665 |
|
---|
666 |
|
---|
667 | bool QNativeSocketEnginePrivate::nativeBind(const QHostAddress &address, quint16 port)
|
---|
668 | {
|
---|
669 | struct sockaddr_in sockAddrIPv4;
|
---|
670 | qt_sockaddr_in6 sockAddrIPv6;
|
---|
671 | struct sockaddr *sockAddrPtr;
|
---|
672 | QT_SOCKLEN_T sockAddrSize;
|
---|
673 |
|
---|
674 | qt_socket_setPortAndAddress(socketDescriptor, &sockAddrIPv4, &sockAddrIPv6, port, address, &sockAddrPtr, &sockAddrSize);
|
---|
675 |
|
---|
676 |
|
---|
677 | int bindResult = ::bind(socketDescriptor, sockAddrPtr, sockAddrSize);
|
---|
678 | if (bindResult == SOCKET_ERROR) {
|
---|
679 | int err = WSAGetLastError();
|
---|
680 | WS_ERROR_DEBUG(err);
|
---|
681 | switch (err) {
|
---|
682 | case WSANOTINITIALISED:
|
---|
683 | //###
|
---|
684 | break;
|
---|
685 | case WSAEADDRINUSE:
|
---|
686 | case WSAEINVAL:
|
---|
687 | setError(QAbstractSocket::AddressInUseError, AddressInuseErrorString);
|
---|
688 | break;
|
---|
689 | case WSAEACCES:
|
---|
690 | setError(QAbstractSocket::SocketAccessError, AddressProtectedErrorString);
|
---|
691 | break;
|
---|
692 | case WSAEADDRNOTAVAIL:
|
---|
693 | setError(QAbstractSocket::SocketAddressNotAvailableError, AddressNotAvailableErrorString);
|
---|
694 | break;
|
---|
695 | default:
|
---|
696 | break;
|
---|
697 | }
|
---|
698 |
|
---|
699 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
700 | qDebug("QNativeSocketEnginePrivate::nativeBind(%s, %i) == false (%s)",
|
---|
701 | address.toString().toLatin1().constData(), port, socketErrorString.toLatin1().constData());
|
---|
702 | #endif
|
---|
703 |
|
---|
704 | return false;
|
---|
705 | }
|
---|
706 |
|
---|
707 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
708 | qDebug("QNativeSocketEnginePrivate::nativeBind(%s, %i) == true",
|
---|
709 | address.toString().toLatin1().constData(), port);
|
---|
710 | #endif
|
---|
711 | socketState = QAbstractSocket::BoundState;
|
---|
712 | return true;
|
---|
713 | }
|
---|
714 |
|
---|
715 |
|
---|
716 | bool QNativeSocketEnginePrivate::nativeListen(int backlog)
|
---|
717 | {
|
---|
718 | if (::listen(socketDescriptor, backlog) == SOCKET_ERROR) {
|
---|
719 | int err = WSAGetLastError();
|
---|
720 | WS_ERROR_DEBUG(err);
|
---|
721 | switch (err) {
|
---|
722 | case WSANOTINITIALISED:
|
---|
723 | //###
|
---|
724 | break;
|
---|
725 | case WSAEADDRINUSE:
|
---|
726 | setError(QAbstractSocket::AddressInUseError,
|
---|
727 | PortInuseErrorString);
|
---|
728 | break;
|
---|
729 | default:
|
---|
730 | break;
|
---|
731 | }
|
---|
732 |
|
---|
733 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
734 | qDebug("QNativeSocketEnginePrivate::nativeListen(%i) == false (%s)",
|
---|
735 | backlog, socketErrorString.toLatin1().constData());
|
---|
736 | #endif
|
---|
737 | return false;
|
---|
738 | }
|
---|
739 |
|
---|
740 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
741 | qDebug("QNativeSocketEnginePrivate::nativeListen(%i) == true", backlog);
|
---|
742 | #endif
|
---|
743 |
|
---|
744 | socketState = QAbstractSocket::ListeningState;
|
---|
745 | return true;
|
---|
746 | }
|
---|
747 |
|
---|
748 | int QNativeSocketEnginePrivate::nativeAccept()
|
---|
749 | {
|
---|
750 | int acceptedDescriptor = WSAAccept(socketDescriptor, 0,0,0,0);
|
---|
751 | if (acceptedDescriptor != -1 && QAbstractEventDispatcher::instance()) {
|
---|
752 | // Becuase of WSAAsyncSelect() WSAAccept returns a non blocking socket
|
---|
753 | // with the same attributes as the listening socket including the current
|
---|
754 | // WSAAsyncSelect(). To be able to change the socket to blocking mode the
|
---|
755 | // WSAAsyncSelect() call must be cancled.
|
---|
756 | QSocketNotifier n(acceptedDescriptor, QSocketNotifier::Read);
|
---|
757 | n.setEnabled(true);
|
---|
758 | n.setEnabled(false);
|
---|
759 | }
|
---|
760 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
761 | qDebug("QNativeSocketEnginePrivate::nativeAccept() == %i", acceptedDescriptor);
|
---|
762 | #endif
|
---|
763 | return acceptedDescriptor;
|
---|
764 | }
|
---|
765 |
|
---|
766 |
|
---|
767 | qint64 QNativeSocketEnginePrivate::nativeBytesAvailable() const
|
---|
768 | {
|
---|
769 | unsigned long nbytes = 0;
|
---|
770 | unsigned long dummy = 0;
|
---|
771 | DWORD sizeWritten = 0;
|
---|
772 | if (::WSAIoctl(socketDescriptor, FIONREAD, &dummy, sizeof(dummy), &nbytes, sizeof(nbytes), &sizeWritten, 0,0) == SOCKET_ERROR) {
|
---|
773 | WS_ERROR_DEBUG(WSAGetLastError());
|
---|
774 | return -1;
|
---|
775 | }
|
---|
776 |
|
---|
777 | // ioctlsocket sometimes reports 1 byte available for datagrams
|
---|
778 | // while the following recvfrom returns -1 and claims connection
|
---|
779 | // was reset (udp is connectionless). so we peek one byte to
|
---|
780 | // catch this case and return 0 bytes available if recvfrom
|
---|
781 | // fails.
|
---|
782 | if (nbytes == 1 && socketType == QAbstractSocket::UdpSocket) {
|
---|
783 | char c;
|
---|
784 | WSABUF buf;
|
---|
785 | buf.buf = &c;
|
---|
786 | buf.len = sizeof(c);
|
---|
787 | DWORD flags = MSG_PEEK;
|
---|
788 | if (::WSARecvFrom(socketDescriptor, &buf, 1, 0, &flags, 0,0,0,0) == SOCKET_ERROR)
|
---|
789 | return 0;
|
---|
790 | }
|
---|
791 | return nbytes;
|
---|
792 | }
|
---|
793 |
|
---|
794 |
|
---|
795 | bool QNativeSocketEnginePrivate::nativeHasPendingDatagrams() const
|
---|
796 | {
|
---|
797 | #if !defined(Q_OS_WINCE)
|
---|
798 | // Create a sockaddr struct and reset its port number.
|
---|
799 | #if !defined(QT_NO_IPV6)
|
---|
800 | qt_sockaddr_in6 storage;
|
---|
801 | qt_sockaddr_in6 *storagePtrIPv6 = reinterpret_cast<qt_sockaddr_in6 *>(&storage);
|
---|
802 | storagePtrIPv6->sin6_port = 0;
|
---|
803 | #else
|
---|
804 | struct sockaddr storage;
|
---|
805 | #endif
|
---|
806 | sockaddr *storagePtr = reinterpret_cast<sockaddr *>(&storage);
|
---|
807 | storagePtr->sa_family = 0;
|
---|
808 |
|
---|
809 | sockaddr_in *storagePtrIPv4 = reinterpret_cast<sockaddr_in *>(&storage);
|
---|
810 | storagePtrIPv4->sin_port = 0;
|
---|
811 | QT_SOCKLEN_T storageSize = sizeof(storage);
|
---|
812 |
|
---|
813 |
|
---|
814 | bool result = false;
|
---|
815 |
|
---|
816 | // Peek 0 bytes into the next message. The size of the message may
|
---|
817 | // well be 0, so we check if there was a sender.
|
---|
818 | char c;
|
---|
819 | WSABUF buf;
|
---|
820 | buf.buf = &c;
|
---|
821 | buf.len = sizeof(c);
|
---|
822 | DWORD available = 0;
|
---|
823 | DWORD flags = MSG_PEEK;
|
---|
824 | int ret = ::WSARecvFrom(socketDescriptor, &buf, 1, &available, &flags, storagePtr, &storageSize,0,0);
|
---|
825 | int err = WSAGetLastError();
|
---|
826 | if (ret == SOCKET_ERROR && err != WSAEMSGSIZE) {
|
---|
827 | WS_ERROR_DEBUG(err);
|
---|
828 | if (err == WSAECONNRESET) {
|
---|
829 | // Discard error message to prevent QAbstractSocket from
|
---|
830 | // getting this message repeatedly after reenabling the
|
---|
831 | // notifiers.
|
---|
832 | flags = 0;
|
---|
833 | ::WSARecvFrom(socketDescriptor, &buf, 1, &available, &flags,
|
---|
834 | storagePtr, &storageSize, 0, 0);
|
---|
835 | }
|
---|
836 | } else {
|
---|
837 | // If there's no error, or if our buffer was too small, there must be
|
---|
838 | // a pending datagram.
|
---|
839 | result = true;
|
---|
840 | }
|
---|
841 |
|
---|
842 | #else // Q_OS_WINCE
|
---|
843 | bool result = false;
|
---|
844 | fd_set readS;
|
---|
845 | FD_ZERO(&readS);
|
---|
846 | FD_SET(socketDescriptor, &readS);
|
---|
847 | timeval timeout;
|
---|
848 | timeout.tv_sec = 0;
|
---|
849 | timeout.tv_usec = 5000;
|
---|
850 | int available = ::select(1, &readS, 0, 0, &timeout);
|
---|
851 | result = available > 0 ? true : false;
|
---|
852 | #endif
|
---|
853 |
|
---|
854 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
855 | qDebug("QNativeSocketEnginePrivate::nativeHasPendingDatagrams() == %s",
|
---|
856 | result ? "true" : "false");
|
---|
857 | #endif
|
---|
858 | return result;
|
---|
859 | }
|
---|
860 |
|
---|
861 |
|
---|
862 | qint64 QNativeSocketEnginePrivate::nativePendingDatagramSize() const
|
---|
863 | {
|
---|
864 | qint64 ret = -1;
|
---|
865 | #if !defined(Q_OS_WINCE)
|
---|
866 | int recvResult = 0;
|
---|
867 | DWORD flags;
|
---|
868 | DWORD bufferCount = 5;
|
---|
869 | WSABUF * buf = 0;
|
---|
870 | for (;;) {
|
---|
871 | // the data written to udpMessagePeekBuffer is discarded, so
|
---|
872 | // this function is still reentrant although it might not look
|
---|
873 | // so.
|
---|
874 | static char udpMessagePeekBuffer[8192];
|
---|
875 |
|
---|
876 | buf = new WSABUF[bufferCount];
|
---|
877 | for (DWORD i=0; i<bufferCount; i++) {
|
---|
878 | buf[i].buf = udpMessagePeekBuffer;
|
---|
879 | buf[i].len = sizeof(udpMessagePeekBuffer);
|
---|
880 | }
|
---|
881 | flags = MSG_PEEK;
|
---|
882 | DWORD bytesRead = 0;
|
---|
883 | recvResult = ::WSARecv(socketDescriptor, buf, bufferCount, &bytesRead, &flags, 0,0);
|
---|
884 | int err = WSAGetLastError();
|
---|
885 | if (recvResult != SOCKET_ERROR) {
|
---|
886 | ret = qint64(bytesRead);
|
---|
887 | break;
|
---|
888 | } else if (recvResult == SOCKET_ERROR && err == WSAEMSGSIZE) {
|
---|
889 | bufferCount += 5;
|
---|
890 | delete[] buf;
|
---|
891 | } else if (recvResult == SOCKET_ERROR) {
|
---|
892 | WS_ERROR_DEBUG(err);
|
---|
893 | ret = -1;
|
---|
894 | break;
|
---|
895 | }
|
---|
896 | }
|
---|
897 |
|
---|
898 | if (buf)
|
---|
899 | delete[] buf;
|
---|
900 |
|
---|
901 | #else // Q_OS_WINCE
|
---|
902 | DWORD size = -1;
|
---|
903 | DWORD bytesReturned;
|
---|
904 | int ioResult = WSAIoctl(socketDescriptor, FIONREAD, 0,0, &size, sizeof(size), &bytesReturned, 0, 0);
|
---|
905 | if (ioResult == SOCKET_ERROR) {
|
---|
906 | int err = WSAGetLastError();
|
---|
907 | WS_ERROR_DEBUG(err);
|
---|
908 | } else {
|
---|
909 | ret = qint64(size);
|
---|
910 | }
|
---|
911 | #endif
|
---|
912 |
|
---|
913 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
914 | qDebug("QNativeSocketEnginePrivate::nativePendingDatagramSize() == %li", ret);
|
---|
915 | #endif
|
---|
916 |
|
---|
917 | return ret;
|
---|
918 | }
|
---|
919 |
|
---|
920 |
|
---|
921 | qint64 QNativeSocketEnginePrivate::nativeReceiveDatagram(char *data, qint64 maxLength,
|
---|
922 | QHostAddress *address, quint16 *port)
|
---|
923 | {
|
---|
924 | qint64 ret = 0;
|
---|
925 |
|
---|
926 | #if !defined(QT_NO_IPV6)
|
---|
927 | qt_sockaddr_storage aa;
|
---|
928 | #else
|
---|
929 | struct sockaddr_in aa;
|
---|
930 | #endif
|
---|
931 | memset(&aa, 0, sizeof(aa));
|
---|
932 | QT_SOCKLEN_T sz;
|
---|
933 | sz = sizeof(aa);
|
---|
934 | WSABUF buf;
|
---|
935 | buf.buf = data;
|
---|
936 | buf.len = maxLength;
|
---|
937 | #if !defined(Q_OS_WINCE)
|
---|
938 | buf.buf = data;
|
---|
939 | buf.len = maxLength;
|
---|
940 | #else
|
---|
941 | char tmpChar;
|
---|
942 | buf.buf = data ? data : &tmpChar;
|
---|
943 | buf.len = maxLength;
|
---|
944 | #endif
|
---|
945 |
|
---|
946 | DWORD flags = 0;
|
---|
947 | DWORD bytesRead = 0;
|
---|
948 | int wsaRet = ::WSARecvFrom(socketDescriptor, &buf, 1, &bytesRead, &flags, (struct sockaddr *) &aa, &sz,0,0);
|
---|
949 | if (wsaRet == SOCKET_ERROR) {
|
---|
950 | int err = WSAGetLastError();
|
---|
951 | if (err == WSAEMSGSIZE) {
|
---|
952 | // it is ok the buffer was to small if bytesRead is larger than
|
---|
953 | // maxLength (win 9x) then assume bytes read is really maxLenth
|
---|
954 | ret = qint64(bytesRead) > maxLength ? maxLength : qint64(bytesRead);
|
---|
955 | } else {
|
---|
956 | WS_ERROR_DEBUG(err);
|
---|
957 | setError(QAbstractSocket::NetworkError, ReceiveDatagramErrorString);
|
---|
958 | ret = -1;
|
---|
959 | }
|
---|
960 | } else {
|
---|
961 | ret = qint64(bytesRead);
|
---|
962 | }
|
---|
963 |
|
---|
964 | qt_socket_getPortAndAddress(socketDescriptor, (struct sockaddr *) &aa, port, address);
|
---|
965 |
|
---|
966 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
967 | qDebug("QNativeSocketEnginePrivate::nativeReceiveDatagram(%p \"%s\", %li, %s, %i) == %li",
|
---|
968 | data, qt_prettyDebug(data, qMin<qint64>(ret, 16), ret).data(), maxLength,
|
---|
969 | address ? address->toString().toLatin1().constData() : "(nil)",
|
---|
970 | port ? *port : 0, ret);
|
---|
971 | #endif
|
---|
972 |
|
---|
973 | return ret;
|
---|
974 | }
|
---|
975 |
|
---|
976 |
|
---|
977 | qint64 QNativeSocketEnginePrivate::nativeSendDatagram(const char *data, qint64 len,
|
---|
978 | const QHostAddress &address, quint16 port)
|
---|
979 | {
|
---|
980 | qint64 ret = -1;
|
---|
981 | struct sockaddr_in sockAddrIPv4;
|
---|
982 | qt_sockaddr_in6 sockAddrIPv6;
|
---|
983 | struct sockaddr *sockAddrPtr;
|
---|
984 | QT_SOCKLEN_T sockAddrSize;
|
---|
985 |
|
---|
986 | qt_socket_setPortAndAddress(socketDescriptor, &sockAddrIPv4, &sockAddrIPv6, port, address, &sockAddrPtr, &sockAddrSize);
|
---|
987 |
|
---|
988 | if (QSysInfo::WindowsVersion & QSysInfo::WV_DOS_based && len > qint64(qt_socket_getMaxMsgSize(socketDescriptor))) {
|
---|
989 | // WSAEMSGSIZE is not reliable enough (win 9x) so we check max size our self.
|
---|
990 | setError(QAbstractSocket::DatagramTooLargeError, DatagramTooLargeErrorString);
|
---|
991 | } else {
|
---|
992 | WSABUF buf;
|
---|
993 | #if !defined(Q_OS_WINCE)
|
---|
994 | buf.buf = len ? (char*)data : 0;
|
---|
995 | #else
|
---|
996 | char tmp;
|
---|
997 | buf.buf = len ? (char*)data : &tmp;
|
---|
998 | #endif
|
---|
999 | buf.len = len;
|
---|
1000 | DWORD flags = 0;
|
---|
1001 | DWORD bytesSent = 0;
|
---|
1002 | if (::WSASendTo(socketDescriptor, &buf, 1, &bytesSent, flags, sockAddrPtr, sockAddrSize, 0,0) == SOCKET_ERROR) {
|
---|
1003 | int err = WSAGetLastError();
|
---|
1004 | WS_ERROR_DEBUG(err);
|
---|
1005 | switch (err) {
|
---|
1006 | case WSAEMSGSIZE:
|
---|
1007 | setError(QAbstractSocket::DatagramTooLargeError, DatagramTooLargeErrorString);
|
---|
1008 | break;
|
---|
1009 | default:
|
---|
1010 | setError(QAbstractSocket::NetworkError, SendDatagramErrorString);
|
---|
1011 | break;
|
---|
1012 | }
|
---|
1013 | ret = -1;
|
---|
1014 | } else {
|
---|
1015 | ret = qint64(bytesSent);
|
---|
1016 | }
|
---|
1017 | }
|
---|
1018 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
1019 | qDebug("QNativeSocketEnginePrivate::nativeSendDatagram(%p \"%s\", %li, \"%s\", %i) == %li", data,
|
---|
1020 | qt_prettyDebug(data, qMin<qint64>(len, 16), len).data(), 0, address.toString().toLatin1().constData(),
|
---|
1021 | port, ret);
|
---|
1022 | #endif
|
---|
1023 |
|
---|
1024 | return ret;
|
---|
1025 | }
|
---|
1026 |
|
---|
1027 |
|
---|
1028 | qint64 QNativeSocketEnginePrivate::nativeWrite(const char *data, qint64 len)
|
---|
1029 | {
|
---|
1030 | Q_Q(QNativeSocketEngine);
|
---|
1031 | qint64 ret = 0;
|
---|
1032 | // don't send more than 49152 per call to WSASendTo to avoid getting a WSAENOBUFS
|
---|
1033 | for (;;) {
|
---|
1034 | qint64 bytesToSend = qMin<qint64>(49152, len - ret);
|
---|
1035 | WSABUF buf;
|
---|
1036 | buf.buf = (char*)data + ret;
|
---|
1037 | buf.len = bytesToSend;
|
---|
1038 | DWORD flags = 0;
|
---|
1039 | DWORD bytesWritten = 0;
|
---|
1040 |
|
---|
1041 | int socketRet = ::WSASend(socketDescriptor, &buf, 1, &bytesWritten, flags, 0,0);
|
---|
1042 |
|
---|
1043 | ret += qint64(bytesWritten);
|
---|
1044 |
|
---|
1045 | if (socketRet != SOCKET_ERROR) {
|
---|
1046 | if (ret == len)
|
---|
1047 | break;
|
---|
1048 | else
|
---|
1049 | continue;
|
---|
1050 | } else if (WSAGetLastError() == WSAEWOULDBLOCK) {
|
---|
1051 | break;
|
---|
1052 | } else {
|
---|
1053 | int err = WSAGetLastError();
|
---|
1054 | WS_ERROR_DEBUG(err);
|
---|
1055 | switch (err) {
|
---|
1056 | case WSAECONNRESET:
|
---|
1057 | case WSAECONNABORTED:
|
---|
1058 | ret = -1;
|
---|
1059 | setError(QAbstractSocket::NetworkError, WriteErrorString);
|
---|
1060 | q->close();
|
---|
1061 | break;
|
---|
1062 | default:
|
---|
1063 | break;
|
---|
1064 | }
|
---|
1065 | break;
|
---|
1066 | }
|
---|
1067 | }
|
---|
1068 |
|
---|
1069 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
1070 | qDebug("QNativeSocketEnginePrivate::nativeWrite(%p \"%s\", %li) == %li",
|
---|
1071 | data, qt_prettyDebug(data, qMin((int)ret, 16), (int)ret).data(), (int)len, (int)ret);
|
---|
1072 | #endif
|
---|
1073 |
|
---|
1074 | return ret;
|
---|
1075 | }
|
---|
1076 |
|
---|
1077 | qint64 QNativeSocketEnginePrivate::nativeRead(char *data, qint64 maxLength)
|
---|
1078 | {
|
---|
1079 | qint64 ret = -1;
|
---|
1080 | WSABUF buf;
|
---|
1081 | buf.buf = data;
|
---|
1082 | buf.len = maxLength;
|
---|
1083 | DWORD flags = 0;
|
---|
1084 | DWORD bytesRead = 0;
|
---|
1085 | #if defined(Q_OS_WINCE)
|
---|
1086 | WSASetLastError(0);
|
---|
1087 | #endif
|
---|
1088 | if (::WSARecv(socketDescriptor, &buf, 1, &bytesRead, &flags, 0,0) == SOCKET_ERROR) {
|
---|
1089 | int err = WSAGetLastError();
|
---|
1090 | WS_ERROR_DEBUG(err);
|
---|
1091 | switch (err) {
|
---|
1092 | case WSAEWOULDBLOCK:
|
---|
1093 | ret = -2;
|
---|
1094 | break;
|
---|
1095 | case WSAEBADF:
|
---|
1096 | case WSAEINVAL:
|
---|
1097 | setError(QAbstractSocket::NetworkError, ReadErrorString);
|
---|
1098 | break;
|
---|
1099 | case WSAECONNRESET:
|
---|
1100 | case WSAECONNABORTED:
|
---|
1101 | // for tcp sockets this will be handled in QNativeSocketEngine::read
|
---|
1102 | ret = 0;
|
---|
1103 | break;
|
---|
1104 | default:
|
---|
1105 | break;
|
---|
1106 | }
|
---|
1107 | } else {
|
---|
1108 | if (WSAGetLastError() == WSAEWOULDBLOCK)
|
---|
1109 | ret = -2;
|
---|
1110 | else
|
---|
1111 | ret = qint64(bytesRead);
|
---|
1112 | }
|
---|
1113 |
|
---|
1114 | #if defined (QNATIVESOCKETENGINE_DEBUG)
|
---|
1115 | if (ret != -2) {
|
---|
1116 | qDebug("QNativeSocketEnginePrivate::nativeRead(%p \"%s\", %l) == %li",
|
---|
1117 | data, qt_prettyDebug(data, qMin((int)bytesRead, 16), (int)bytesRead).data(), (int)maxLength, (int)ret);
|
---|
1118 | } else {
|
---|
1119 | qDebug("QNativeSocketEnginePrivate::nativeRead(%p, %l) == -2 (WOULD BLOCK)",
|
---|
1120 | data, int(maxLength));
|
---|
1121 | }
|
---|
1122 | #endif
|
---|
1123 |
|
---|
1124 | return ret;
|
---|
1125 | }
|
---|
1126 |
|
---|
1127 | int QNativeSocketEnginePrivate::nativeSelect(int timeout, bool selectForRead) const
|
---|
1128 | {
|
---|
1129 | bool readEnabled = selectForRead && readNotifier && readNotifier->isEnabled();
|
---|
1130 | if (readEnabled)
|
---|
1131 | readNotifier->setEnabled(false);
|
---|
1132 |
|
---|
1133 | fd_set fds;
|
---|
1134 |
|
---|
1135 | int ret = 0;
|
---|
1136 |
|
---|
1137 | memset(&fds, 0, sizeof(fd_set));
|
---|
1138 | fds.fd_count = 1;
|
---|
1139 | fds.fd_array[0] = socketDescriptor;
|
---|
1140 |
|
---|
1141 | struct timeval tv;
|
---|
1142 | tv.tv_sec = timeout / 1000;
|
---|
1143 | tv.tv_usec = (timeout % 1000) * 1000;
|
---|
1144 |
|
---|
1145 | if (selectForRead)
|
---|
1146 | ret = select(0, &fds, 0, 0, timeout < 0 ? 0 : &tv);
|
---|
1147 | else
|
---|
1148 | ret = select(0, 0, &fds, 0, timeout < 0 ? 0 : &tv);
|
---|
1149 |
|
---|
1150 | if (readEnabled)
|
---|
1151 | readNotifier->setEnabled(true);
|
---|
1152 |
|
---|
1153 | return ret;
|
---|
1154 | }
|
---|
1155 |
|
---|
1156 | int QNativeSocketEnginePrivate::nativeSelect(int timeout,
|
---|
1157 | bool checkRead, bool checkWrite,
|
---|
1158 | bool *selectForRead, bool *selectForWrite) const
|
---|
1159 | {
|
---|
1160 | bool readEnabled = checkRead && readNotifier && readNotifier->isEnabled();
|
---|
1161 | if (readEnabled)
|
---|
1162 | readNotifier->setEnabled(false);
|
---|
1163 |
|
---|
1164 | fd_set fdread;
|
---|
1165 | fd_set fdwrite;
|
---|
1166 |
|
---|
1167 | int ret = 0;
|
---|
1168 |
|
---|
1169 | memset(&fdread, 0, sizeof(fd_set));
|
---|
1170 | if (checkRead) {
|
---|
1171 | fdread.fd_count = 1;
|
---|
1172 | fdread.fd_array[0] = socketDescriptor;
|
---|
1173 | }
|
---|
1174 | memset(&fdwrite, 0, sizeof(fd_set));
|
---|
1175 | if (checkWrite) {
|
---|
1176 | fdwrite.fd_count = 1;
|
---|
1177 | fdwrite.fd_array[0] = socketDescriptor;
|
---|
1178 | }
|
---|
1179 |
|
---|
1180 | struct timeval tv;
|
---|
1181 | tv.tv_sec = timeout / 1000;
|
---|
1182 | tv.tv_usec = (timeout % 1000) * 1000;
|
---|
1183 |
|
---|
1184 | #if !defined(Q_OS_WINCE)
|
---|
1185 | ret = select(socketDescriptor + 1, &fdread, &fdwrite, 0, timeout < 0 ? 0 : &tv);
|
---|
1186 | #else
|
---|
1187 | ret = select(1, &fdread, &fdwrite, 0, timeout < 0 ? 0 : &tv);
|
---|
1188 | #endif
|
---|
1189 | if (readEnabled)
|
---|
1190 | readNotifier->setEnabled(true);
|
---|
1191 |
|
---|
1192 | if (ret <= 0)
|
---|
1193 | return ret;
|
---|
1194 |
|
---|
1195 | *selectForRead = FD_ISSET(socketDescriptor, &fdread);
|
---|
1196 | *selectForWrite = FD_ISSET(socketDescriptor, &fdwrite);
|
---|
1197 |
|
---|
1198 | return ret;
|
---|
1199 | }
|
---|
1200 |
|
---|
1201 | void QNativeSocketEnginePrivate::nativeClose()
|
---|
1202 | {
|
---|
1203 | #if defined (QTCPSOCKETENGINE_DEBUG)
|
---|
1204 | qDebug("QNativeSocketEnginePrivate::nativeClose()");
|
---|
1205 | #endif
|
---|
1206 | linger l = {1, 0};
|
---|
1207 | ::setsockopt(socketDescriptor, SOL_SOCKET, SO_DONTLINGER, (char*)&l, sizeof(l));
|
---|
1208 | ::closesocket(socketDescriptor);
|
---|
1209 | }
|
---|
1210 |
|
---|
1211 | QT_END_NAMESPACE
|
---|