| /**************************************************************************** |
| ** |
| ** Copyright (C) 2011 Nokia Corporation and/or its subsidiary(-ies). |
| ** All rights reserved. |
| ** Contact: Nokia Corporation (qt-info@nokia.com) |
| ** |
| ** This file is part of the QtNetwork module of the Qt Toolkit. |
| ** |
| ** $QT_BEGIN_LICENSE:LGPL$ |
| ** GNU Lesser General Public License Usage |
| ** This file may be used under the terms of the GNU Lesser General Public |
| ** License version 2.1 as published by the Free Software Foundation and |
| ** appearing in the file LICENSE.LGPL included in the packaging of this |
| ** file. Please review the following information to ensure the GNU Lesser |
| ** General Public License version 2.1 requirements will be met: |
| ** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. |
| ** |
| ** In addition, as a special exception, Nokia gives you certain additional |
| ** rights. These rights are described in the Nokia Qt LGPL Exception |
| ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package. |
| ** |
| ** GNU General Public License Usage |
| ** Alternatively, this file may be used under the terms of the GNU General |
| ** Public License version 3.0 as published by the Free Software Foundation |
| ** and appearing in the file LICENSE.GPL included in the packaging of this |
| ** file. Please review the following information to ensure the GNU General |
| ** Public License version 3.0 requirements will be met: |
| ** http://www.gnu.org/copyleft/gpl.html. |
| ** |
| ** Other Usage |
| ** Alternatively, this file may be used in accordance with the terms and |
| ** conditions contained in a signed written agreement between you and Nokia. |
| ** |
| ** |
| ** |
| ** |
| ** |
| ** $QT_END_LICENSE$ |
| ** |
| ****************************************************************************/ |
| |
| //#define QNATIVESOCKETENGINE_DEBUG |
| #include "qnativesocketengine_p.h" |
| #include "private/qnet_unix_p.h" |
| #include "qiodevice.h" |
| #include "qhostaddress.h" |
| #include "qelapsedtimer.h" |
| #include "qvarlengtharray.h" |
| #include <time.h> |
| #include <errno.h> |
| #include <fcntl.h> |
| #ifndef QT_NO_IPV6IFNAME |
| #include <net/if.h> |
| #endif |
| #ifndef QT_NO_IPV6IFNAME |
| #include <net/if.h> |
| #endif |
| #ifdef QT_LINUXBASE |
| #include <arpa/inet.h> |
| #endif |
| |
| #if defined QNATIVESOCKETENGINE_DEBUG |
| #include <qstring.h> |
| #include <ctype.h> |
| #endif |
| |
| #ifdef Q_OS_SYMBIAN // ### TODO: Are these headers right? |
| #include <sys/socket.h> |
| #include <netinet/in.h> |
| #else |
| #include <netinet/tcp.h> |
| #endif |
| |
| QT_BEGIN_NAMESPACE |
| |
| #if defined QNATIVESOCKETENGINE_DEBUG |
| |
| /* |
| Returns a human readable representation of the first \a len |
| characters in \a data. |
| */ |
| static QByteArray qt_prettyDebug(const char *data, int len, int maxSize) |
| { |
| if (!data) return "(null)"; |
| QByteArray out; |
| for (int i = 0; i < len; ++i) { |
| char c = data[i]; |
| if (isprint(c)) { |
| out += c; |
| } else switch (c) { |
| case '\n': out += "\\n"; break; |
| case '\r': out += "\\r"; break; |
| case '\t': out += "\\t"; break; |
| default: |
| QString tmp; |
| tmp.sprintf("\\%o", c); |
| out += tmp.toLatin1(); |
| } |
| } |
| |
| if (len < maxSize) |
| out += "..."; |
| |
| return out; |
| } |
| #endif |
| |
| static void qt_ignore_sigpipe() |
| { |
| #ifndef Q_NO_POSIX_SIGNALS |
| // Set to ignore SIGPIPE once only. |
| static QBasicAtomicInt atom = Q_BASIC_ATOMIC_INITIALIZER(0); |
| if (atom.testAndSetRelaxed(0, 1)) { |
| struct sigaction noaction; |
| memset(&noaction, 0, sizeof(noaction)); |
| noaction.sa_handler = SIG_IGN; |
| ::sigaction(SIGPIPE, &noaction, 0); |
| } |
| #else |
| // Posix signals are not supported by the underlying platform |
| // so we don't need to ignore sigpipe signal explicitly |
| #endif |
| } |
| |
| /* |
| Extracts the port and address from a sockaddr, and stores them in |
| \a port and \a addr if they are non-null. |
| */ |
| static inline void qt_socket_getPortAndAddress(const qt_sockaddr *s, quint16 *port, QHostAddress *addr) |
| { |
| #if !defined(QT_NO_IPV6) |
| if (s->a.sa_family == AF_INET6) { |
| Q_IPV6ADDR tmp; |
| memcpy(&tmp, &s->a6.sin6_addr, sizeof(tmp)); |
| if (addr) { |
| QHostAddress tmpAddress; |
| tmpAddress.setAddress(tmp); |
| *addr = tmpAddress; |
| #ifndef QT_NO_IPV6IFNAME |
| char scopeid[IFNAMSIZ]; |
| if (::if_indextoname(s->a6.sin6_scope_id, scopeid)) { |
| addr->setScopeId(QLatin1String(scopeid)); |
| } else |
| #endif |
| addr->setScopeId(QString::number(s->a6.sin6_scope_id)); |
| } |
| if (port) |
| *port = ntohs(s->a6.sin6_port); |
| return; |
| } |
| #endif |
| if (port) |
| *port = ntohs(s->a4.sin_port); |
| if (addr) { |
| QHostAddress tmpAddress; |
| tmpAddress.setAddress(ntohl(s->a4.sin_addr.s_addr)); |
| *addr = tmpAddress; |
| } |
| } |
| |
| /*! \internal |
| |
| Creates and returns a new socket descriptor of type \a socketType |
| and \a socketProtocol. Returns -1 on failure. |
| */ |
| bool QNativeSocketEnginePrivate::createNewSocket(QAbstractSocket::SocketType socketType, |
| QAbstractSocket::NetworkLayerProtocol socketProtocol) |
| { |
| #ifndef QT_NO_IPV6 |
| int protocol = (socketProtocol == QAbstractSocket::IPv6Protocol) ? AF_INET6 : AF_INET; |
| #else |
| Q_UNUSED(socketProtocol); |
| int protocol = AF_INET; |
| #endif |
| int type = (socketType == QAbstractSocket::UdpSocket) ? SOCK_DGRAM : SOCK_STREAM; |
| #ifdef Q_OS_SYMBIAN |
| int socket = ::socket(protocol, type, 0); |
| #else |
| int socket = qt_safe_socket(protocol, type, 0); |
| #endif |
| |
| if (socket <= 0) { |
| switch (errno) { |
| case EPROTONOSUPPORT: |
| case EAFNOSUPPORT: |
| case EINVAL: |
| setError(QAbstractSocket::UnsupportedSocketOperationError, ProtocolUnsupportedErrorString); |
| break; |
| case ENFILE: |
| case EMFILE: |
| case ENOBUFS: |
| case ENOMEM: |
| setError(QAbstractSocket::SocketResourceError, ResourceErrorString); |
| break; |
| case EACCES: |
| setError(QAbstractSocket::SocketAccessError, AccessErrorString); |
| break; |
| default: |
| break; |
| } |
| |
| return false; |
| } |
| |
| socketDescriptor = socket; |
| return true; |
| } |
| |
| /* |
| Returns the value of the socket option \a opt. |
| */ |
| int QNativeSocketEnginePrivate::option(QNativeSocketEngine::SocketOption opt) const |
| { |
| Q_Q(const QNativeSocketEngine); |
| if (!q->isValid()) |
| return -1; |
| |
| int n = -1; |
| int level = SOL_SOCKET; // default |
| |
| switch (opt) { |
| case QNativeSocketEngine::ReceiveBufferSocketOption: |
| n = SO_RCVBUF; |
| break; |
| case QNativeSocketEngine::SendBufferSocketOption: |
| n = SO_SNDBUF; |
| break; |
| case QNativeSocketEngine::NonBlockingSocketOption: |
| break; |
| case QNativeSocketEngine::BroadcastSocketOption: |
| break; |
| case QNativeSocketEngine::AddressReusable: |
| n = SO_REUSEADDR; |
| break; |
| case QNativeSocketEngine::BindExclusively: |
| return true; |
| case QNativeSocketEngine::ReceiveOutOfBandData: |
| n = SO_OOBINLINE; |
| break; |
| case QNativeSocketEngine::LowDelayOption: |
| level = IPPROTO_TCP; |
| n = TCP_NODELAY; |
| break; |
| case QNativeSocketEngine::KeepAliveOption: |
| n = SO_KEEPALIVE; |
| break; |
| } |
| |
| int v = -1; |
| QT_SOCKOPTLEN_T len = sizeof(v); |
| if (::getsockopt(socketDescriptor, level, n, (char *) &v, &len) != -1) |
| return v; |
| |
| return -1; |
| } |
| |
| |
| /* |
| Sets the socket option \a opt to \a v. |
| */ |
| bool QNativeSocketEnginePrivate::setOption(QNativeSocketEngine::SocketOption opt, int v) |
| { |
| Q_Q(QNativeSocketEngine); |
| if (!q->isValid()) |
| return false; |
| |
| int n = 0; |
| int level = SOL_SOCKET; // default |
| |
| switch (opt) { |
| case QNativeSocketEngine::ReceiveBufferSocketOption: |
| n = SO_RCVBUF; |
| break; |
| case QNativeSocketEngine::SendBufferSocketOption: |
| n = SO_SNDBUF; |
| break; |
| case QNativeSocketEngine::BroadcastSocketOption: |
| n = SO_BROADCAST; |
| break; |
| case QNativeSocketEngine::NonBlockingSocketOption: { |
| // Make the socket nonblocking. |
| #if !defined(Q_OS_VXWORKS) |
| int flags = ::fcntl(socketDescriptor, F_GETFL, 0); |
| if (flags == -1) { |
| #ifdef QNATIVESOCKETENGINE_DEBUG |
| perror("QNativeSocketEnginePrivate::setOption(): fcntl(F_GETFL) failed"); |
| #endif |
| return false; |
| } |
| if (::fcntl(socketDescriptor, F_SETFL, flags | O_NONBLOCK) == -1) { |
| #ifdef QNATIVESOCKETENGINE_DEBUG |
| perror("QNativeSocketEnginePrivate::setOption(): fcntl(F_SETFL) failed"); |
| #endif |
| return false; |
| } |
| #else // Q_OS_VXWORKS |
| int onoff = 1; |
| #ifdef Q_OS_SYMBIAN |
| if (::ioctl(socketDescriptor, FIONBIO, &onoff) < 0) { |
| #else |
| if (qt_safe_ioctl(socketDescriptor, FIONBIO, &onoff) < 0) { |
| #endif |
| #ifdef QNATIVESOCKETENGINE_DEBUG |
| perror("QNativeSocketEnginePrivate::setOption(): ioctl(FIONBIO, 1) failed"); |
| #endif |
| return false; |
| } |
| #endif // Q_OS_VXWORKS |
| return true; |
| } |
| case QNativeSocketEngine::AddressReusable: |
| #if defined(SO_REUSEPORT) && !defined(Q_OS_SYMBIAN) |
| n = SO_REUSEPORT; |
| #else |
| n = SO_REUSEADDR; |
| #endif |
| break; |
| case QNativeSocketEngine::BindExclusively: |
| return true; |
| case QNativeSocketEngine::ReceiveOutOfBandData: |
| n = SO_OOBINLINE; |
| break; |
| case QNativeSocketEngine::LowDelayOption: |
| level = IPPROTO_TCP; |
| n = TCP_NODELAY; |
| break; |
| case QNativeSocketEngine::KeepAliveOption: |
| n = SO_KEEPALIVE; |
| break; |
| } |
| |
| return ::setsockopt(socketDescriptor, level, n, (char *) &v, sizeof(v)) == 0; |
| } |
| |
| bool QNativeSocketEnginePrivate::nativeConnect(const QHostAddress &addr, quint16 port) |
| { |
| #ifdef QNATIVESOCKETENGINE_DEBUG |
| qDebug("QNativeSocketEnginePrivate::nativeConnect() : %d ", socketDescriptor); |
| #endif |
| |
| struct sockaddr_in sockAddrIPv4; |
| struct sockaddr *sockAddrPtr = 0; |
| QT_SOCKLEN_T sockAddrSize = 0; |
| |
| #if !defined(QT_NO_IPV6) |
| struct sockaddr_in6 sockAddrIPv6; |
| |
| if (addr.protocol() == QAbstractSocket::IPv6Protocol) { |
| memset(&sockAddrIPv6, 0, sizeof(sockAddrIPv6)); |
| sockAddrIPv6.sin6_family = AF_INET6; |
| sockAddrIPv6.sin6_port = htons(port); |
| |
| QString scopeid = addr.scopeId(); |
| bool ok; |
| sockAddrIPv6.sin6_scope_id = scopeid.toInt(&ok); |
| #ifndef QT_NO_IPV6IFNAME |
| if (!ok) |
| sockAddrIPv6.sin6_scope_id = ::if_nametoindex(scopeid.toLatin1()); |
| #endif |
| Q_IPV6ADDR ip6 = addr.toIPv6Address(); |
| memcpy(&sockAddrIPv6.sin6_addr.s6_addr, &ip6, sizeof(ip6)); |
| |
| sockAddrSize = sizeof(sockAddrIPv6); |
| sockAddrPtr = (struct sockaddr *) &sockAddrIPv6; |
| } else |
| #if 0 |
| {} |
| #endif |
| #endif |
| if (addr.protocol() == QAbstractSocket::IPv4Protocol) { |
| memset(&sockAddrIPv4, 0, sizeof(sockAddrIPv4)); |
| sockAddrIPv4.sin_family = AF_INET; |
| sockAddrIPv4.sin_port = htons(port); |
| sockAddrIPv4.sin_addr.s_addr = htonl(addr.toIPv4Address()); |
| |
| sockAddrSize = sizeof(sockAddrIPv4); |
| sockAddrPtr = (struct sockaddr *) &sockAddrIPv4; |
| } else { |
| // unreachable |
| } |
| #ifdef Q_OS_SYMBIAN |
| int connectResult = ::connect(socketDescriptor, sockAddrPtr, sockAddrSize); |
| #else |
| int connectResult = qt_safe_connect(socketDescriptor, sockAddrPtr, sockAddrSize); |
| #endif |
| if (connectResult == -1) { |
| switch (errno) { |
| case EISCONN: |
| socketState = QAbstractSocket::ConnectedState; |
| break; |
| case ECONNREFUSED: |
| case EINVAL: |
| setError(QAbstractSocket::ConnectionRefusedError, ConnectionRefusedErrorString); |
| socketState = QAbstractSocket::UnconnectedState; |
| break; |
| case ETIMEDOUT: |
| setError(QAbstractSocket::NetworkError, ConnectionTimeOutErrorString); |
| break; |
| case EHOSTUNREACH: |
| setError(QAbstractSocket::NetworkError, HostUnreachableErrorString); |
| socketState = QAbstractSocket::UnconnectedState; |
| break; |
| case ENETUNREACH: |
| setError(QAbstractSocket::NetworkError, NetworkUnreachableErrorString); |
| socketState = QAbstractSocket::UnconnectedState; |
| break; |
| case EADDRINUSE: |
| setError(QAbstractSocket::NetworkError, AddressInuseErrorString); |
| break; |
| case EINPROGRESS: |
| case EALREADY: |
| setError(QAbstractSocket::UnfinishedSocketOperationError, InvalidSocketErrorString); |
| socketState = QAbstractSocket::ConnectingState; |
| break; |
| case EAGAIN: |
| setError(QAbstractSocket::UnfinishedSocketOperationError, InvalidSocketErrorString); |
| setError(QAbstractSocket::SocketResourceError, ResourceErrorString); |
| break; |
| case EACCES: |
| case EPERM: |
| setError(QAbstractSocket::SocketAccessError, AccessErrorString); |
| socketState = QAbstractSocket::UnconnectedState; |
| break; |
| case EAFNOSUPPORT: |
| case EBADF: |
| case EFAULT: |
| case ENOTSOCK: |
| #ifdef Q_OS_SYMBIAN |
| case EPIPE: |
| #endif |
| socketState = QAbstractSocket::UnconnectedState; |
| default: |
| break; |
| } |
| |
| if (socketState != QAbstractSocket::ConnectedState) { |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeConnect(%s, %i) == false (%s)", |
| addr.toString().toLatin1().constData(), port, |
| socketState == QAbstractSocket::ConnectingState |
| ? "Connection in progress" : socketErrorString.toLatin1().constData()); |
| #endif |
| return false; |
| } |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeConnect(%s, %i) == true", |
| addr.toString().toLatin1().constData(), port); |
| #endif |
| |
| socketState = QAbstractSocket::ConnectedState; |
| return true; |
| } |
| |
| bool QNativeSocketEnginePrivate::nativeBind(const QHostAddress &address, quint16 port) |
| { |
| struct sockaddr_in sockAddrIPv4; |
| struct sockaddr *sockAddrPtr = 0; |
| QT_SOCKLEN_T sockAddrSize = 0; |
| |
| #if !defined(QT_NO_IPV6) |
| struct sockaddr_in6 sockAddrIPv6; |
| |
| if (address.protocol() == QAbstractSocket::IPv6Protocol) { |
| memset(&sockAddrIPv6, 0, sizeof(sockAddrIPv6)); |
| sockAddrIPv6.sin6_family = AF_INET6; |
| sockAddrIPv6.sin6_port = htons(port); |
| #ifndef QT_NO_IPV6IFNAME |
| sockAddrIPv6.sin6_scope_id = ::if_nametoindex(address.scopeId().toLatin1().data()); |
| #else |
| sockAddrIPv6.sin6_scope_id = address.scopeId().toInt(); |
| #endif |
| Q_IPV6ADDR tmp = address.toIPv6Address(); |
| memcpy(&sockAddrIPv6.sin6_addr.s6_addr, &tmp, sizeof(tmp)); |
| sockAddrSize = sizeof(sockAddrIPv6); |
| sockAddrPtr = (struct sockaddr *) &sockAddrIPv6; |
| } else |
| #endif |
| if (address.protocol() == QAbstractSocket::IPv4Protocol) { |
| memset(&sockAddrIPv4, 0, sizeof(sockAddrIPv4)); |
| sockAddrIPv4.sin_family = AF_INET; |
| sockAddrIPv4.sin_port = htons(port); |
| sockAddrIPv4.sin_addr.s_addr = htonl(address.toIPv4Address()); |
| sockAddrSize = sizeof(sockAddrIPv4); |
| sockAddrPtr = (struct sockaddr *) &sockAddrIPv4; |
| } else { |
| // unreachable |
| } |
| |
| int bindResult = QT_SOCKET_BIND(socketDescriptor, sockAddrPtr, sockAddrSize); |
| |
| if (bindResult < 0) { |
| switch(errno) { |
| case EADDRINUSE: |
| setError(QAbstractSocket::AddressInUseError, AddressInuseErrorString); |
| break; |
| case EACCES: |
| setError(QAbstractSocket::SocketAccessError, AddressProtectedErrorString); |
| break; |
| case EINVAL: |
| setError(QAbstractSocket::UnsupportedSocketOperationError, OperationUnsupportedErrorString); |
| break; |
| case EADDRNOTAVAIL: |
| setError(QAbstractSocket::SocketAddressNotAvailableError, AddressNotAvailableErrorString); |
| break; |
| default: |
| break; |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeBind(%s, %i) == false (%s)", |
| address.toString().toLatin1().constData(), port, socketErrorString.toLatin1().constData()); |
| #endif |
| |
| return false; |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeBind(%s, %i) == true", |
| address.toString().toLatin1().constData(), port); |
| #endif |
| socketState = QAbstractSocket::BoundState; |
| return true; |
| } |
| |
| bool QNativeSocketEnginePrivate::nativeListen(int backlog) |
| { |
| #ifdef Q_OS_SYMBIAN |
| if (::listen(socketDescriptor, backlog) < 0) { |
| #else |
| if (qt_safe_listen(socketDescriptor, backlog) < 0) { |
| #endif |
| switch (errno) { |
| case EADDRINUSE: |
| setError(QAbstractSocket::AddressInUseError, |
| PortInuseErrorString); |
| break; |
| default: |
| break; |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeListen(%i) == false (%s)", |
| backlog, socketErrorString.toLatin1().constData()); |
| #endif |
| return false; |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeListen(%i) == true", backlog); |
| #endif |
| |
| socketState = QAbstractSocket::ListeningState; |
| return true; |
| } |
| |
| int QNativeSocketEnginePrivate::nativeAccept() |
| { |
| #ifdef Q_OS_SYMBIAN |
| int acceptedDescriptor = ::accept(socketDescriptor, 0, 0); |
| #else |
| int acceptedDescriptor = qt_safe_accept(socketDescriptor, 0, 0); |
| #endif |
| |
| return acceptedDescriptor; |
| } |
| |
| qint64 QNativeSocketEnginePrivate::nativeBytesAvailable() const |
| { |
| int nbytes = 0; |
| // gives shorter than true amounts on Unix domain sockets. |
| qint64 available = 0; |
| #ifdef Q_OS_SYMBIAN |
| if (::ioctl(socketDescriptor, FIONREAD, (char *) &nbytes) >= 0) |
| #else |
| if (qt_safe_ioctl(socketDescriptor, FIONREAD, (char *) &nbytes) >= 0) |
| #endif |
| available = (qint64) nbytes; |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeBytesAvailable() == %lli", available); |
| #endif |
| return available; |
| } |
| |
| bool QNativeSocketEnginePrivate::nativeHasPendingDatagrams() const |
| { |
| // Create a sockaddr struct and reset its port number. |
| qt_sockaddr storage; |
| QT_SOCKLEN_T storageSize = sizeof(storage); |
| memset(&storage, 0, storageSize); |
| |
| // Peek 0 bytes into the next message. The size of the message may |
| // well be 0, so we can't check recvfrom's return value. |
| ssize_t readBytes; |
| #ifdef Q_OS_SYMBIAN |
| char c; |
| readBytes = ::recvfrom(socketDescriptor, &c, 1, MSG_PEEK, &storage.a, &storageSize); |
| #else |
| do { |
| char c; |
| readBytes = ::recvfrom(socketDescriptor, &c, 1, MSG_PEEK, &storage.a, &storageSize); |
| } while (readBytes == -1 && errno == EINTR); |
| #endif |
| |
| // If there's no error, or if our buffer was too small, there must be a |
| // pending datagram. |
| bool result = (readBytes != -1) || errno == EMSGSIZE; |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeHasPendingDatagrams() == %s", |
| result ? "true" : "false"); |
| #endif |
| return result; |
| } |
| |
| #ifdef Q_OS_SYMBIAN |
| qint64 QNativeSocketEnginePrivate::nativePendingDatagramSize() const |
| { |
| size_t nbytes = 0; |
| ::ioctl(socketDescriptor, E32IONREAD, (char *) &nbytes); |
| return qint64(nbytes-28); |
| } |
| #else |
| qint64 QNativeSocketEnginePrivate::nativePendingDatagramSize() const |
| { |
| QVarLengthArray<char, 8192> udpMessagePeekBuffer(8192); |
| ssize_t recvResult = -1; |
| |
| for (;;) { |
| // the data written to udpMessagePeekBuffer is discarded, so |
| // this function is still reentrant although it might not look |
| // so. |
| recvResult = ::recv(socketDescriptor, udpMessagePeekBuffer.data(), |
| udpMessagePeekBuffer.size(), MSG_PEEK); |
| if (recvResult == -1 && errno == EINTR) |
| continue; |
| |
| if (recvResult != (ssize_t) udpMessagePeekBuffer.size()) |
| break; |
| |
| udpMessagePeekBuffer.resize(udpMessagePeekBuffer.size() * 2); |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativePendingDatagramSize() == %i", recvResult); |
| #endif |
| |
| return qint64(recvResult); |
| } |
| #endif |
| qint64 QNativeSocketEnginePrivate::nativeReceiveDatagram(char *data, qint64 maxSize, |
| QHostAddress *address, quint16 *port) |
| { |
| qt_sockaddr aa; |
| memset(&aa, 0, sizeof(aa)); |
| QT_SOCKLEN_T sz; |
| sz = sizeof(aa); |
| |
| ssize_t recvFromResult = 0; |
| #ifdef Q_OS_SYMBIAN |
| char c; |
| recvFromResult = ::recvfrom(socketDescriptor, maxSize ? data : &c, maxSize ? maxSize : 1, |
| 0, &aa.a, &sz); |
| #else |
| do { |
| char c; |
| recvFromResult = ::recvfrom(socketDescriptor, maxSize ? data : &c, maxSize ? maxSize : 1, |
| 0, &aa.a, &sz); |
| } while (recvFromResult == -1 && errno == EINTR); |
| #endif |
| |
| if (recvFromResult == -1) { |
| setError(QAbstractSocket::NetworkError, ReceiveDatagramErrorString); |
| } else if (port || address) { |
| qt_socket_getPortAndAddress(&aa, port, address); |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeReceiveDatagram(%p \"%s\", %lli, %s, %i) == %lli", |
| data, qt_prettyDebug(data, qMin(recvFromResult, ssize_t(16)), recvFromResult).data(), maxSize, |
| address ? address->toString().toLatin1().constData() : "(nil)", |
| port ? *port : 0, (qint64) recvFromResult); |
| #endif |
| |
| return qint64(maxSize ? recvFromResult : recvFromResult == -1 ? -1 : 0); |
| } |
| |
| qint64 QNativeSocketEnginePrivate::nativeSendDatagram(const char *data, qint64 len, |
| const QHostAddress &host, quint16 port) |
| { |
| struct sockaddr_in sockAddrIPv4; |
| struct sockaddr *sockAddrPtr = 0; |
| QT_SOCKLEN_T sockAddrSize = 0; |
| |
| #if !defined(QT_NO_IPV6) |
| struct sockaddr_in6 sockAddrIPv6; |
| if (host.protocol() == QAbstractSocket::IPv6Protocol) { |
| memset(&sockAddrIPv6, 0, sizeof(sockAddrIPv6)); |
| sockAddrIPv6.sin6_family = AF_INET6; |
| sockAddrIPv6.sin6_port = htons(port); |
| |
| Q_IPV6ADDR tmp = host.toIPv6Address(); |
| memcpy(&sockAddrIPv6.sin6_addr.s6_addr, &tmp, sizeof(tmp)); |
| sockAddrSize = sizeof(sockAddrIPv6); |
| sockAddrPtr = (struct sockaddr *)&sockAddrIPv6; |
| } else |
| #endif |
| if (host.protocol() == QAbstractSocket::IPv4Protocol) { |
| memset(&sockAddrIPv4, 0, sizeof(sockAddrIPv4)); |
| sockAddrIPv4.sin_family = AF_INET; |
| sockAddrIPv4.sin_port = htons(port); |
| sockAddrIPv4.sin_addr.s_addr = htonl(host.toIPv4Address()); |
| sockAddrSize = sizeof(sockAddrIPv4); |
| sockAddrPtr = (struct sockaddr *)&sockAddrIPv4; |
| } |
| |
| // ignore the SIGPIPE signal |
| qt_ignore_sigpipe(); |
| #ifdef Q_OS_SYMBIAN |
| ssize_t sentBytes = ::sendto(socketDescriptor, data, len, |
| 0, sockAddrPtr, sockAddrSize); |
| #else |
| ssize_t sentBytes = qt_safe_sendto(socketDescriptor, data, len, |
| 0, sockAddrPtr, sockAddrSize); |
| #endif |
| |
| if (sentBytes < 0) { |
| switch (errno) { |
| case EMSGSIZE: |
| setError(QAbstractSocket::DatagramTooLargeError, DatagramTooLargeErrorString); |
| break; |
| default: |
| setError(QAbstractSocket::NetworkError, SendDatagramErrorString); |
| } |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEngine::sendDatagram(%p \"%s\", %lli, \"%s\", %i) == %lli", data, |
| qt_prettyDebug(data, qMin<int>(len, 16), len).data(), len, host.toString().toLatin1().constData(), |
| port, (qint64) sentBytes); |
| #endif |
| |
| return qint64(sentBytes); |
| } |
| |
| bool QNativeSocketEnginePrivate::fetchConnectionParameters() |
| { |
| localPort = 0; |
| localAddress.clear(); |
| peerPort = 0; |
| peerAddress.clear(); |
| |
| if (socketDescriptor == -1) |
| return false; |
| |
| qt_sockaddr sa; |
| QT_SOCKLEN_T sockAddrSize = sizeof(sa); |
| |
| // Determine local address |
| memset(&sa, 0, sizeof(sa)); |
| if (::getsockname(socketDescriptor, &sa.a, &sockAddrSize) == 0) { |
| qt_socket_getPortAndAddress(&sa, &localPort, &localAddress); |
| |
| // Determine protocol family |
| switch (sa.a.sa_family) { |
| case AF_INET: |
| socketProtocol = QAbstractSocket::IPv4Protocol; |
| break; |
| #if !defined (QT_NO_IPV6) |
| case AF_INET6: |
| socketProtocol = QAbstractSocket::IPv6Protocol; |
| break; |
| #endif |
| default: |
| socketProtocol = QAbstractSocket::UnknownNetworkLayerProtocol; |
| break; |
| } |
| |
| } else if (errno == EBADF) { |
| setError(QAbstractSocket::UnsupportedSocketOperationError, InvalidSocketErrorString); |
| return false; |
| } |
| |
| // Determine the remote address |
| if (!::getpeername(socketDescriptor, &sa.a, &sockAddrSize)) |
| qt_socket_getPortAndAddress(&sa, &peerPort, &peerAddress); |
| |
| // Determine the socket type (UDP/TCP) |
| int value = 0; |
| QT_SOCKOPTLEN_T valueSize = sizeof(int); |
| if (::getsockopt(socketDescriptor, SOL_SOCKET, SO_TYPE, &value, &valueSize) == 0) { |
| if (value == SOCK_STREAM) |
| socketType = QAbstractSocket::TcpSocket; |
| else if (value == SOCK_DGRAM) |
| socketType = QAbstractSocket::UdpSocket; |
| else |
| socketType = QAbstractSocket::UnknownSocketType; |
| } |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| QString socketProtocolStr = "UnknownProtocol"; |
| if (socketProtocol == QAbstractSocket::IPv4Protocol) socketProtocolStr = "IPv4Protocol"; |
| else if (socketProtocol == QAbstractSocket::IPv6Protocol) socketProtocolStr = "IPv6Protocol"; |
| |
| QString socketTypeStr = "UnknownSocketType"; |
| if (socketType == QAbstractSocket::TcpSocket) socketTypeStr = "TcpSocket"; |
| else if (socketType == QAbstractSocket::UdpSocket) socketTypeStr = "UdpSocket"; |
| |
| qDebug("QNativeSocketEnginePrivate::fetchConnectionParameters() local == %s:%i," |
| " peer == %s:%i, socket == %s - %s", |
| localAddress.toString().toLatin1().constData(), localPort, |
| peerAddress.toString().toLatin1().constData(), peerPort,socketTypeStr.toLatin1().constData(), |
| socketProtocolStr.toLatin1().constData()); |
| #endif |
| return true; |
| } |
| |
| void QNativeSocketEnginePrivate::nativeClose() |
| { |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEngine::nativeClose()"); |
| #endif |
| |
| #ifdef Q_OS_SYMBIAN |
| ::close(socketDescriptor); |
| #else |
| qt_safe_close(socketDescriptor); |
| #endif |
| } |
| |
| qint64 QNativeSocketEnginePrivate::nativeWrite(const char *data, qint64 len) |
| { |
| Q_Q(QNativeSocketEngine); |
| |
| // ignore the SIGPIPE signal |
| qt_ignore_sigpipe(); |
| |
| ssize_t writtenBytes; |
| #ifdef Q_OS_SYMBIAN |
| // Symbian does not support signals natively and Open C returns EINTR when moving to offline |
| writtenBytes = ::write(socketDescriptor, data, len); |
| #else |
| writtenBytes = qt_safe_write(socketDescriptor, data, len); |
| #endif |
| |
| if (writtenBytes < 0) { |
| switch (errno) { |
| case EPIPE: |
| case ECONNRESET: |
| writtenBytes = -1; |
| setError(QAbstractSocket::RemoteHostClosedError, RemoteHostClosedErrorString); |
| q->close(); |
| break; |
| case EAGAIN: |
| writtenBytes = 0; |
| break; |
| case EMSGSIZE: |
| setError(QAbstractSocket::DatagramTooLargeError, DatagramTooLargeErrorString); |
| break; |
| default: |
| break; |
| } |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeWrite(%p \"%s\", %llu) == %i", |
| data, qt_prettyDebug(data, qMin((int) len, 16), |
| (int) len).data(), len, (int) writtenBytes); |
| #endif |
| |
| return qint64(writtenBytes); |
| } |
| /* |
| */ |
| qint64 QNativeSocketEnginePrivate::nativeRead(char *data, qint64 maxSize) |
| { |
| Q_Q(QNativeSocketEngine); |
| if (!q->isValid()) { |
| qWarning("QNativeSocketEngine::unbufferedRead: Invalid socket"); |
| return -1; |
| } |
| |
| ssize_t r = 0; |
| #ifdef Q_OS_SYMBIAN |
| r = ::read(socketDescriptor, data, maxSize); |
| #else |
| r = qt_safe_read(socketDescriptor, data, maxSize); |
| #endif |
| |
| if (r < 0) { |
| r = -1; |
| switch (errno) { |
| #if EWOULDBLOCK-0 && EWOULDBLOCK != EAGAIN |
| case EWOULDBLOCK: |
| #endif |
| case EAGAIN: |
| // No data was available for reading |
| r = -2; |
| break; |
| case EBADF: |
| case EINVAL: |
| case EIO: |
| //error string is now set in read(), not here in nativeRead() |
| break; |
| #ifdef Q_OS_SYMBIAN |
| case EPIPE: |
| #endif |
| case ECONNRESET: |
| #if defined(Q_OS_VXWORKS) |
| case ESHUTDOWN: |
| #endif |
| r = 0; |
| break; |
| default: |
| break; |
| } |
| } |
| |
| #if defined (QNATIVESOCKETENGINE_DEBUG) |
| qDebug("QNativeSocketEnginePrivate::nativeRead(%p \"%s\", %llu) == %i", |
| data, qt_prettyDebug(data, qMin(r, ssize_t(16)), r).data(), |
| maxSize, r); |
| #endif |
| |
| return qint64(r); |
| } |
| |
| int QNativeSocketEnginePrivate::nativeSelect(int timeout, bool selectForRead) const |
| { |
| fd_set fds; |
| FD_ZERO(&fds); |
| FD_SET(socketDescriptor, &fds); |
| |
| struct timeval tv; |
| tv.tv_sec = timeout / 1000; |
| tv.tv_usec = (timeout % 1000) * 1000; |
| |
| #ifdef Q_OS_SYMBIAN |
| fd_set fdexception; |
| FD_ZERO(&fdexception); |
| FD_SET(socketDescriptor, &fdexception); |
| #endif |
| |
| int retval; |
| if (selectForRead) |
| #ifdef Q_OS_SYMBIAN |
| retval = ::select(socketDescriptor + 1, &fds, 0, &fdexception, timeout < 0 ? 0 : &tv); |
| #else |
| retval = qt_safe_select(socketDescriptor + 1, &fds, 0, 0, timeout < 0 ? 0 : &tv); |
| #endif |
| else |
| #ifdef Q_OS_SYMBIAN |
| retval = ::select(socketDescriptor + 1, 0, &fds, &fdexception, timeout < 0 ? 0 : &tv); |
| #else |
| retval = qt_safe_select(socketDescriptor + 1, 0, &fds, 0, timeout < 0 ? 0 : &tv); |
| #endif |
| |
| |
| #ifdef Q_OS_SYMBIAN |
| bool selectForExec = false; |
| if(retval != 0) { |
| if(retval < 0) { |
| qWarning("nativeSelect(....) returned < 0 for socket %d", socketDescriptor); |
| } |
| selectForExec = FD_ISSET(socketDescriptor, &fdexception); |
| } |
| if(selectForExec) { |
| qWarning("nativeSelect (selectForRead %d, retVal %d, errno %d) Unexpected exception for fd %d", |
| selectForRead, retval, errno, socketDescriptor); |
| } |
| #endif |
| |
| return retval; |
| } |
| |
| int QNativeSocketEnginePrivate::nativeSelect(int timeout, bool checkRead, bool checkWrite, |
| bool *selectForRead, bool *selectForWrite) const |
| { |
| fd_set fdread; |
| FD_ZERO(&fdread); |
| if (checkRead) |
| FD_SET(socketDescriptor, &fdread); |
| |
| fd_set fdwrite; |
| FD_ZERO(&fdwrite); |
| if (checkWrite) |
| FD_SET(socketDescriptor, &fdwrite); |
| |
| #ifdef Q_OS_SYMBIAN |
| fd_set fdexception; |
| FD_ZERO(&fdexception); |
| FD_SET(socketDescriptor, &fdexception); |
| #endif |
| |
| struct timeval tv; |
| tv.tv_sec = timeout / 1000; |
| tv.tv_usec = (timeout % 1000) * 1000; |
| |
| int ret; |
| #ifndef Q_OS_SYMBIAN |
| ret = qt_safe_select(socketDescriptor + 1, &fdread, &fdwrite, 0, timeout < 0 ? 0 : &tv); |
| #else |
| QElapsedTimer timer; |
| timer.start(); |
| |
| do { |
| ret = ::select(socketDescriptor + 1, &fdread, &fdwrite, &fdexception, timeout < 0 ? 0 : &tv); |
| bool selectForExec = false; |
| if(ret != 0) { |
| if(ret < 0) { |
| qWarning("nativeSelect(....) returned < 0 for socket %d", socketDescriptor); |
| } |
| selectForExec = FD_ISSET(socketDescriptor, &fdexception); |
| } |
| if(selectForExec) { |
| qWarning("nativeSelect (checkRead %d, checkWrite %d, ret %d, errno %d): Unexpected expectfds ready in fd %d", |
| checkRead, checkWrite, ret, errno, socketDescriptor); |
| if (checkWrite){ |
| FD_CLR(socketDescriptor, &fdread); |
| FD_SET(socketDescriptor, &fdwrite); |
| } else if (checkRead) |
| FD_SET(socketDescriptor, &fdread); |
| |
| |
| if ((ret == -1) && ( errno == ECONNREFUSED || errno == EPIPE )) |
| ret = 1; |
| |
| } |
| |
| if (ret != -1 || errno != EINTR) { |
| break; |
| } |
| |
| if (timeout > 0) { |
| // recalculate the timeout |
| int t = timeout - timer.elapsed(); |
| if (t < 0) { |
| // oops, timeout turned negative? |
| ret = -1; |
| break; |
| } |
| |
| tv.tv_sec = t / 1000; |
| tv.tv_usec = (t % 1000) * 1000; |
| } |
| } while (true); |
| #endif |
| |
| if (ret <= 0) |
| return ret; |
| *selectForRead = FD_ISSET(socketDescriptor, &fdread); |
| *selectForWrite = FD_ISSET(socketDescriptor, &fdwrite); |
| |
| return ret; |
| } |
| |
| QT_END_NAMESPACE |