mirror of
https://bitbucket.org/smil3y/katie.git
synced 2025-02-27 20:32:57 +00:00
299 lines
9.8 KiB
C++
299 lines
9.8 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:LGPL$
|
|
** 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.
|
|
**
|
|
** 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.
|
|
**
|
|
** $QT_END_LICENSE$
|
|
**
|
|
****************************************************************************/
|
|
|
|
#ifndef QSOCKS5SOCKETENGINE_P_H
|
|
#define QSOCKS5SOCKETENGINE_P_H
|
|
|
|
//
|
|
// W A R N I N G
|
|
// -------------
|
|
//
|
|
// This file is not part of the Qt API. It exists purely as an
|
|
// implementation detail. This header file may change from version to
|
|
// version without notice, or even be removed.
|
|
//
|
|
// We mean it.
|
|
//
|
|
|
|
#include "qabstractsocketengine_p.h"
|
|
#include "qnetworkproxy.h"
|
|
|
|
QT_BEGIN_NAMESPACE
|
|
|
|
#ifndef QT_NO_SOCKS5
|
|
|
|
class QSocks5SocketEnginePrivate;
|
|
|
|
class Q_AUTOTEST_EXPORT QSocks5SocketEngine : public QAbstractSocketEngine
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
QSocks5SocketEngine(QObject *parent = 0);
|
|
~QSocks5SocketEngine();
|
|
|
|
bool initialize(QAbstractSocket::SocketType type, QAbstractSocket::NetworkLayerProtocol protocol = QAbstractSocket::IPv4Protocol);
|
|
bool initialize(int socketDescriptor, QAbstractSocket::SocketState socketState = QAbstractSocket::ConnectedState);
|
|
|
|
void setProxy(const QNetworkProxy &networkProxy);
|
|
|
|
int socketDescriptor() const;
|
|
|
|
bool isValid() const;
|
|
|
|
bool connectInternal();
|
|
bool connectToHost(const QHostAddress &address, quint16 port);
|
|
bool connectToHostByName(const QString &name, quint16 port);
|
|
bool bind(const QHostAddress &address, quint16 port);
|
|
bool listen();
|
|
int accept();
|
|
void close();
|
|
|
|
qint64 bytesAvailable() const;
|
|
|
|
qint64 read(char *data, qint64 maxlen);
|
|
qint64 write(const char *data, qint64 len);
|
|
|
|
#ifndef QT_NO_UDPSOCKET
|
|
#ifndef QT_NO_NETWORKINTERFACE
|
|
bool joinMulticastGroup(const QHostAddress &groupAddress,
|
|
const QNetworkInterface &interface);
|
|
bool leaveMulticastGroup(const QHostAddress &groupAddress,
|
|
const QNetworkInterface &interface);
|
|
QNetworkInterface multicastInterface() const;
|
|
bool setMulticastInterface(const QNetworkInterface &iface);
|
|
#endif // QT_NO_NETWORKINTERFACE
|
|
|
|
qint64 readDatagram(char *data, qint64 maxlen, QHostAddress *addr = 0,
|
|
quint16 *port = 0);
|
|
qint64 writeDatagram(const char *data, qint64 len, const QHostAddress &addr,
|
|
quint16 port);
|
|
bool hasPendingDatagrams() const;
|
|
qint64 pendingDatagramSize() const;
|
|
#endif // QT_NO_UDPSOCKET
|
|
|
|
qint64 bytesToWrite() const;
|
|
|
|
int option(SocketOption option) const;
|
|
bool setOption(SocketOption option, int value);
|
|
|
|
bool waitForRead(int msecs = 30000, bool *timedOut = 0);
|
|
bool waitForWrite(int msecs = 30000, bool *timedOut = 0);
|
|
bool waitForReadOrWrite(bool *readyToRead, bool *readyToWrite,
|
|
bool checkRead, bool checkWrite,
|
|
int msecs = 30000, bool *timedOut = 0);
|
|
|
|
bool isReadNotificationEnabled() const;
|
|
void setReadNotificationEnabled(bool enable);
|
|
bool isWriteNotificationEnabled() const;
|
|
void setWriteNotificationEnabled(bool enable);
|
|
bool isExceptionNotificationEnabled() const;
|
|
void setExceptionNotificationEnabled(bool enable);
|
|
|
|
private:
|
|
Q_DECLARE_PRIVATE(QSocks5SocketEngine)
|
|
Q_DISABLE_COPY(QSocks5SocketEngine)
|
|
Q_PRIVATE_SLOT(d_func(), void _q_controlSocketConnected())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_controlSocketReadNotification())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_controlSocketError(QAbstractSocket::SocketError))
|
|
#ifndef QT_NO_UDPSOCKET
|
|
Q_PRIVATE_SLOT(d_func(), void _q_udpSocketReadNotification())
|
|
#endif
|
|
Q_PRIVATE_SLOT(d_func(), void _q_controlSocketBytesWritten())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_emitPendingReadNotification())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_emitPendingWriteNotification())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_emitPendingConnectionNotification())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_controlSocketDisconnected())
|
|
Q_PRIVATE_SLOT(d_func(), void _q_controlSocketStateChanged(QAbstractSocket::SocketState))
|
|
|
|
};
|
|
|
|
|
|
class QTcpSocket;
|
|
|
|
class QSocks5Authenticator
|
|
{
|
|
public:
|
|
QSocks5Authenticator();
|
|
virtual ~QSocks5Authenticator();
|
|
virtual char methodId();
|
|
virtual bool beginAuthenticate(QTcpSocket *socket, bool *completed);
|
|
virtual bool continueAuthenticate(QTcpSocket *socket, bool *completed);
|
|
|
|
virtual bool seal(const QByteArray buf, QByteArray *sealedBuf);
|
|
virtual bool unSeal(const QByteArray sealedBuf, QByteArray *buf);
|
|
virtual bool unSeal(QTcpSocket *sealedSocket, QByteArray *buf);
|
|
|
|
virtual QString errorString() { return QString(); }
|
|
};
|
|
|
|
class QSocks5PasswordAuthenticator : public QSocks5Authenticator
|
|
{
|
|
public:
|
|
QSocks5PasswordAuthenticator(const QString &userName, const QString &password);
|
|
char methodId();
|
|
bool beginAuthenticate(QTcpSocket *socket, bool *completed);
|
|
bool continueAuthenticate(QTcpSocket *socket, bool *completed);
|
|
|
|
QString errorString();
|
|
|
|
private:
|
|
QString userName;
|
|
QString password;
|
|
};
|
|
|
|
struct QSocks5Data;
|
|
struct QSocks5ConnectData;
|
|
struct QSocks5UdpAssociateData;
|
|
struct QSocks5BindData;
|
|
|
|
class QSocks5SocketEnginePrivate : public QAbstractSocketEnginePrivate
|
|
{
|
|
Q_DECLARE_PUBLIC(QSocks5SocketEngine)
|
|
public:
|
|
QSocks5SocketEnginePrivate();
|
|
~QSocks5SocketEnginePrivate();
|
|
|
|
enum Socks5State
|
|
{
|
|
Uninitialized = 0,
|
|
ConnectError,
|
|
AuthenticationMethodsSent,
|
|
Authenticating,
|
|
AuthenticatingError,
|
|
RequestMethodSent,
|
|
RequestError,
|
|
Connected,
|
|
UdpAssociateSuccess,
|
|
BindSuccess,
|
|
ControlSocketError,
|
|
SocksError,
|
|
HostNameLookupError
|
|
};
|
|
Socks5State socks5State;
|
|
|
|
enum Socks5Mode
|
|
{
|
|
NoMode,
|
|
ConnectMode,
|
|
BindMode,
|
|
UdpAssociateMode
|
|
};
|
|
Socks5Mode mode;
|
|
|
|
enum Socks5Error
|
|
{
|
|
SocksFailure = 0x01,
|
|
ConnectionNotAllowed = 0x02,
|
|
NetworkUnreachable = 0x03,
|
|
HostUnreachable = 0x04,
|
|
ConnectionRefused = 0x05,
|
|
TTLExpired = 0x06,
|
|
CommandNotSupported = 0x07,
|
|
AddressTypeNotSupported = 0x08,
|
|
LastKnownError = AddressTypeNotSupported,
|
|
UnknownError
|
|
};
|
|
|
|
void initialize(Socks5Mode socks5Mode);
|
|
|
|
void setErrorState(Socks5State state, const QString &extraMessage = QString());
|
|
void setErrorState(Socks5State state, Socks5Error socks5error);
|
|
|
|
void reauthenticate();
|
|
void parseAuthenticationMethodReply();
|
|
void parseAuthenticatingReply();
|
|
void sendRequestMethod();
|
|
void parseRequestMethodReply();
|
|
void parseNewConnection();
|
|
|
|
bool waitForConnected(int msecs, bool *timedOut);
|
|
|
|
void _q_controlSocketConnected();
|
|
void _q_controlSocketReadNotification();
|
|
void _q_controlSocketError(QAbstractSocket::SocketError);
|
|
#ifndef QT_NO_UDPSOCKET
|
|
void checkForDatagrams() const;
|
|
void _q_udpSocketReadNotification();
|
|
#endif
|
|
void _q_controlSocketBytesWritten();
|
|
void _q_controlSocketDisconnected();
|
|
void _q_controlSocketStateChanged(QAbstractSocket::SocketState);
|
|
|
|
QNetworkProxy proxyInfo;
|
|
|
|
bool readNotificationEnabled, writeNotificationEnabled, exceptNotificationEnabled;
|
|
|
|
int socketDescriptor;
|
|
|
|
QSocks5Data *data;
|
|
QSocks5ConnectData *connectData;
|
|
#ifndef QT_NO_UDPSOCKET
|
|
QSocks5UdpAssociateData *udpData;
|
|
#endif
|
|
QSocks5BindData *bindData;
|
|
QString peerName;
|
|
|
|
mutable bool readNotificationActivated;
|
|
mutable bool writeNotificationActivated;
|
|
|
|
bool readNotificationPending;
|
|
void _q_emitPendingReadNotification();
|
|
void emitReadNotification();
|
|
bool writeNotificationPending;
|
|
void _q_emitPendingWriteNotification();
|
|
void emitWriteNotification();
|
|
bool connectionNotificationPending;
|
|
void _q_emitPendingConnectionNotification();
|
|
void emitConnectionNotification();
|
|
};
|
|
|
|
class Q_AUTOTEST_EXPORT QSocks5SocketEngineHandler : public QSocketEngineHandler
|
|
{
|
|
public:
|
|
virtual QAbstractSocketEngine *createSocketEngine(QAbstractSocket::SocketType socketType,
|
|
const QNetworkProxy &, QObject *parent);
|
|
virtual QAbstractSocketEngine *createSocketEngine(int socketDescripter, QObject *parent);
|
|
};
|
|
|
|
|
|
QT_END_NAMESPACE
|
|
#endif // QT_NO_SOCKS5
|
|
#endif // QSOCKS5SOCKETENGINE_H
|