304 lines
12 KiB
C++
304 lines
12 KiB
C++
/****************************************************************************
|
|
**
|
|
** Copyright (C) 2015 The Qt Company Ltd.
|
|
** Contact: http://www.qt.io/licensing/
|
|
**
|
|
** This file is part of the QtNetwork module of the Qt Toolkit.
|
|
**
|
|
** $QT_BEGIN_LICENSE:LGPL21$
|
|
** Commercial License Usage
|
|
** Licensees holding valid commercial Qt licenses may use this file in
|
|
** accordance with the commercial license agreement provided with the
|
|
** Software or, alternatively, in accordance with the terms contained in
|
|
** a written agreement between you and The Qt Company. For licensing terms
|
|
** and conditions see http://www.qt.io/terms-conditions. For further
|
|
** information use the contact form at http://www.qt.io/contact-us.
|
|
**
|
|
** GNU Lesser General Public License Usage
|
|
** Alternatively, this file may be used under the terms of the GNU Lesser
|
|
** General Public License version 2.1 or version 3 as published by the Free
|
|
** Software Foundation and appearing in the file LICENSE.LGPLv21 and
|
|
** LICENSE.LGPLv3 included in the packaging of this file. Please review the
|
|
** following information to ensure the GNU Lesser General Public License
|
|
** requirements will be met: https://www.gnu.org/licenses/lgpl.html and
|
|
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
|
|
**
|
|
** As a special exception, The Qt Company gives you certain additional
|
|
** rights. These rights are described in The Qt Company LGPL Exception
|
|
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
|
|
**
|
|
** $QT_END_LICENSE$
|
|
**
|
|
****************************************************************************/
|
|
|
|
#ifndef QNATIVESOCKETENGINE_P_H
|
|
#define QNATIVESOCKETENGINE_P_H
|
|
|
|
//
|
|
// W A R N I N G
|
|
// -------------
|
|
//
|
|
// This file is not part of the Qt API. It exists for the convenience
|
|
// of the QLibrary class. This header file may change from
|
|
// version to version without notice, or even be removed.
|
|
//
|
|
// We mean it.
|
|
//
|
|
#include "QtNetwork/qhostaddress.h"
|
|
#include "QtNetwork/qnetworkinterface.h"
|
|
#include "private/qabstractsocketengine_p.h"
|
|
#ifndef Q_OS_WIN
|
|
# include "qplatformdefs.h"
|
|
# include <netinet/in.h>
|
|
#else
|
|
# include <winsock2.h>
|
|
# include <ws2tcpip.h>
|
|
# include <mswsock.h>
|
|
#endif
|
|
|
|
QT_BEGIN_NAMESPACE
|
|
|
|
#ifdef Q_OS_WIN
|
|
#define QT_SOCKLEN_T int
|
|
#define QT_SOCKOPTLEN_T int
|
|
|
|
// The following definitions are copied from the MinGW header mswsock.h which
|
|
// was placed in the public domain. The WSASendMsg and WSARecvMsg functions
|
|
// were introduced with Windows Vista, so some Win32 headers are lacking them.
|
|
// There are no known versions of Windows CE or Embedded that contain them.
|
|
#ifndef Q_OS_WINCE
|
|
# ifndef WSAID_WSARECVMSG
|
|
typedef INT (WINAPI *LPFN_WSARECVMSG)(SOCKET s, LPWSAMSG lpMsg,
|
|
LPDWORD lpdwNumberOfBytesRecvd,
|
|
LPWSAOVERLAPPED lpOverlapped,
|
|
LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine);
|
|
# define WSAID_WSARECVMSG {0xf689d7c8,0x6f1f,0x436b,{0x8a,0x53,0xe5,0x4f,0xe3,0x51,0xc3,0x22}}
|
|
# endif
|
|
# ifndef WSAID_WSASENDMSG
|
|
typedef struct {
|
|
LPWSAMSG lpMsg;
|
|
DWORD dwFlags;
|
|
LPDWORD lpNumberOfBytesSent;
|
|
LPWSAOVERLAPPED lpOverlapped;
|
|
LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine;
|
|
} WSASENDMSG, *LPWSASENDMSG;
|
|
|
|
typedef INT (WSAAPI *LPFN_WSASENDMSG)(SOCKET s, LPWSAMSG lpMsg, DWORD dwFlags,
|
|
LPDWORD lpNumberOfBytesSent,
|
|
LPWSAOVERLAPPED lpOverlapped,
|
|
LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine);
|
|
|
|
# define WSAID_WSASENDMSG {0xa441e712,0x754f,0x43ca,{0x84,0xa7,0x0d,0xee,0x44,0xcf,0x60,0x6d}}
|
|
# endif
|
|
#endif
|
|
#endif
|
|
|
|
union qt_sockaddr {
|
|
sockaddr a;
|
|
sockaddr_in a4;
|
|
sockaddr_in6 a6;
|
|
};
|
|
|
|
class QNativeSocketEnginePrivate;
|
|
#ifndef QT_NO_NETWORKINTERFACE
|
|
class QNetworkInterface;
|
|
#endif
|
|
|
|
class Q_AUTOTEST_EXPORT QNativeSocketEngine : public QAbstractSocketEngine
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
QNativeSocketEngine(QObject *parent = 0);
|
|
~QNativeSocketEngine();
|
|
|
|
bool initialize(QAbstractSocket::SocketType type, QAbstractSocket::NetworkLayerProtocol protocol = QAbstractSocket::IPv4Protocol) Q_DECL_OVERRIDE;
|
|
bool initialize(qintptr socketDescriptor, QAbstractSocket::SocketState socketState = QAbstractSocket::ConnectedState) Q_DECL_OVERRIDE;
|
|
|
|
qintptr socketDescriptor() const Q_DECL_OVERRIDE;
|
|
|
|
bool isValid() const Q_DECL_OVERRIDE;
|
|
|
|
bool connectToHost(const QHostAddress &address, quint16 port) Q_DECL_OVERRIDE;
|
|
bool connectToHostByName(const QString &name, quint16 port) Q_DECL_OVERRIDE;
|
|
bool bind(const QHostAddress &address, quint16 port) Q_DECL_OVERRIDE;
|
|
bool listen() Q_DECL_OVERRIDE;
|
|
int accept() Q_DECL_OVERRIDE;
|
|
void close() Q_DECL_OVERRIDE;
|
|
|
|
qint64 bytesAvailable() const Q_DECL_OVERRIDE;
|
|
|
|
qint64 read(char *data, qint64 maxlen) Q_DECL_OVERRIDE;
|
|
qint64 write(const char *data, qint64 len) Q_DECL_OVERRIDE;
|
|
|
|
#ifndef QT_NO_UDPSOCKET
|
|
#ifndef QT_NO_NETWORKINTERFACE
|
|
bool joinMulticastGroup(const QHostAddress &groupAddress,
|
|
const QNetworkInterface &iface) Q_DECL_OVERRIDE;
|
|
bool leaveMulticastGroup(const QHostAddress &groupAddress,
|
|
const QNetworkInterface &iface) Q_DECL_OVERRIDE;
|
|
QNetworkInterface multicastInterface() const Q_DECL_OVERRIDE;
|
|
bool setMulticastInterface(const QNetworkInterface &iface) Q_DECL_OVERRIDE;
|
|
#endif
|
|
|
|
qint64 readDatagram(char *data, qint64 maxlen, QIpPacketHeader * = 0,
|
|
PacketHeaderOptions = WantNone) Q_DECL_OVERRIDE;
|
|
qint64 writeDatagram(const char *data, qint64 len, const QIpPacketHeader &) Q_DECL_OVERRIDE;
|
|
bool hasPendingDatagrams() const Q_DECL_OVERRIDE;
|
|
qint64 pendingDatagramSize() const Q_DECL_OVERRIDE;
|
|
#endif // QT_NO_UDPSOCKET
|
|
|
|
qint64 bytesToWrite() const Q_DECL_OVERRIDE;
|
|
|
|
qint64 receiveBufferSize() const;
|
|
void setReceiveBufferSize(qint64 bufferSize);
|
|
|
|
qint64 sendBufferSize() const;
|
|
void setSendBufferSize(qint64 bufferSize);
|
|
|
|
int option(SocketOption option) const Q_DECL_OVERRIDE;
|
|
bool setOption(SocketOption option, int value) Q_DECL_OVERRIDE;
|
|
|
|
bool waitForRead(int msecs = 30000, bool *timedOut = 0) Q_DECL_OVERRIDE;
|
|
bool waitForWrite(int msecs = 30000, bool *timedOut = 0) Q_DECL_OVERRIDE;
|
|
bool waitForReadOrWrite(bool *readyToRead, bool *readyToWrite,
|
|
bool checkRead, bool checkWrite,
|
|
int msecs = 30000, bool *timedOut = 0) Q_DECL_OVERRIDE;
|
|
|
|
bool isReadNotificationEnabled() const Q_DECL_OVERRIDE;
|
|
void setReadNotificationEnabled(bool enable) Q_DECL_OVERRIDE;
|
|
bool isWriteNotificationEnabled() const Q_DECL_OVERRIDE;
|
|
void setWriteNotificationEnabled(bool enable) Q_DECL_OVERRIDE;
|
|
bool isExceptionNotificationEnabled() const Q_DECL_OVERRIDE;
|
|
void setExceptionNotificationEnabled(bool enable) Q_DECL_OVERRIDE;
|
|
|
|
public Q_SLOTS:
|
|
// non-virtual override;
|
|
void connectionNotification();
|
|
|
|
private:
|
|
Q_DECLARE_PRIVATE(QNativeSocketEngine)
|
|
Q_DISABLE_COPY(QNativeSocketEngine)
|
|
};
|
|
|
|
class QSocketNotifier;
|
|
|
|
class QNativeSocketEnginePrivate : public QAbstractSocketEnginePrivate
|
|
{
|
|
Q_DECLARE_PUBLIC(QNativeSocketEngine)
|
|
public:
|
|
QNativeSocketEnginePrivate();
|
|
~QNativeSocketEnginePrivate();
|
|
|
|
qintptr socketDescriptor;
|
|
|
|
QSocketNotifier *readNotifier, *writeNotifier, *exceptNotifier;
|
|
|
|
#if defined(Q_OS_WIN) && !defined(Q_OS_WINCE)
|
|
LPFN_WSASENDMSG sendmsg;
|
|
LPFN_WSARECVMSG recvmsg;
|
|
# endif
|
|
enum ErrorString {
|
|
NonBlockingInitFailedErrorString,
|
|
BroadcastingInitFailedErrorString,
|
|
NoIpV6ErrorString,
|
|
RemoteHostClosedErrorString,
|
|
TimeOutErrorString,
|
|
ResourceErrorString,
|
|
OperationUnsupportedErrorString,
|
|
ProtocolUnsupportedErrorString,
|
|
InvalidSocketErrorString,
|
|
HostUnreachableErrorString,
|
|
NetworkUnreachableErrorString,
|
|
AccessErrorString,
|
|
ConnectionTimeOutErrorString,
|
|
ConnectionRefusedErrorString,
|
|
AddressInuseErrorString,
|
|
AddressNotAvailableErrorString,
|
|
AddressProtectedErrorString,
|
|
DatagramTooLargeErrorString,
|
|
SendDatagramErrorString,
|
|
ReceiveDatagramErrorString,
|
|
WriteErrorString,
|
|
ReadErrorString,
|
|
PortInuseErrorString,
|
|
NotSocketErrorString,
|
|
InvalidProxyTypeString,
|
|
TemporaryErrorString,
|
|
|
|
UnknownSocketErrorString = -1
|
|
};
|
|
|
|
void setError(QAbstractSocket::SocketError error, ErrorString errorString) const;
|
|
QHostAddress adjustAddressProtocol(const QHostAddress &address) const;
|
|
|
|
// native functions
|
|
int option(QNativeSocketEngine::SocketOption option) const;
|
|
bool setOption(QNativeSocketEngine::SocketOption option, int value);
|
|
|
|
bool createNewSocket(QAbstractSocket::SocketType type, QAbstractSocket::NetworkLayerProtocol &protocol);
|
|
|
|
bool nativeConnect(const QHostAddress &address, quint16 port);
|
|
bool nativeBind(const QHostAddress &address, quint16 port);
|
|
bool nativeListen(int backlog);
|
|
int nativeAccept();
|
|
#ifndef QT_NO_NETWORKINTERFACE
|
|
bool nativeJoinMulticastGroup(const QHostAddress &groupAddress,
|
|
const QNetworkInterface &iface);
|
|
bool nativeLeaveMulticastGroup(const QHostAddress &groupAddress,
|
|
const QNetworkInterface &iface);
|
|
QNetworkInterface nativeMulticastInterface() const;
|
|
bool nativeSetMulticastInterface(const QNetworkInterface &iface);
|
|
#endif
|
|
qint64 nativeBytesAvailable() const;
|
|
|
|
bool nativeHasPendingDatagrams() const;
|
|
qint64 nativePendingDatagramSize() const;
|
|
qint64 nativeReceiveDatagram(char *data, qint64 maxLength, QIpPacketHeader *header,
|
|
QAbstractSocketEngine::PacketHeaderOptions options);
|
|
qint64 nativeSendDatagram(const char *data, qint64 length, const QIpPacketHeader &header);
|
|
qint64 nativeRead(char *data, qint64 maxLength);
|
|
qint64 nativeWrite(const char *data, qint64 length);
|
|
int nativeSelect(int timeout, bool selectForRead) const;
|
|
int nativeSelect(int timeout, bool checkRead, bool checkWrite,
|
|
bool *selectForRead, bool *selectForWrite) const;
|
|
|
|
void nativeClose();
|
|
|
|
bool checkProxy(const QHostAddress &address);
|
|
bool fetchConnectionParameters();
|
|
|
|
static uint scopeIdFromString(const QString &scopeid)
|
|
{ return QNetworkInterface::interfaceIndexFromName(scopeid); }
|
|
|
|
/*! \internal
|
|
Sets \a address and \a port in the \a aa sockaddr structure and the size in \a sockAddrSize.
|
|
The address \a is converted to IPv6 if the current socket protocol is also IPv6.
|
|
*/
|
|
void setPortAndAddress(quint16 port, const QHostAddress &address, qt_sockaddr *aa, QT_SOCKLEN_T *sockAddrSize)
|
|
{
|
|
if (address.protocol() == QAbstractSocket::IPv6Protocol
|
|
|| address.protocol() == QAbstractSocket::AnyIPProtocol
|
|
|| socketProtocol == QAbstractSocket::IPv6Protocol
|
|
|| socketProtocol == QAbstractSocket::AnyIPProtocol) {
|
|
memset(&aa->a6, 0, sizeof(sockaddr_in6));
|
|
aa->a6.sin6_family = AF_INET6;
|
|
aa->a6.sin6_scope_id = scopeIdFromString(address.scopeId());
|
|
aa->a6.sin6_port = htons(port);
|
|
Q_IPV6ADDR tmp = address.toIPv6Address();
|
|
memcpy(&aa->a6.sin6_addr, &tmp, sizeof(tmp));
|
|
*sockAddrSize = sizeof(sockaddr_in6);
|
|
} else {
|
|
memset(&aa->a, 0, sizeof(sockaddr_in));
|
|
aa->a4.sin_family = AF_INET;
|
|
aa->a4.sin_port = htons(port);
|
|
aa->a4.sin_addr.s_addr = htonl(address.toIPv4Address());
|
|
*sockAddrSize = sizeof(sockaddr_in);
|
|
}
|
|
}
|
|
|
|
};
|
|
|
|
QT_END_NAMESPACE
|
|
|
|
#endif // QNATIVESOCKETENGINE_P_H
|