summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/websockets/qwebsocketserver.cpp14
-rw-r--r--src/websockets/qwebsocketserver.h3
-rw-r--r--src/websockets/qwebsocketserver_p.cpp26
-rw-r--r--src/websockets/qwebsocketserver_p.h3
4 files changed, 35 insertions, 11 deletions
diff --git a/src/websockets/qwebsocketserver.cpp b/src/websockets/qwebsocketserver.cpp
index f8ecdf2..e4cfe12 100644
--- a/src/websockets/qwebsocketserver.cpp
+++ b/src/websockets/qwebsocketserver.cpp
@@ -612,4 +612,18 @@ QList<QWebSocketProtocol::Version> QWebSocketServer::supportedVersions() const
return d->supportedVersions();
}
+/*!
+ Upgrades a tcp \a socket to websocket.
+
+ The QWebSocketServer object will take ownership of the socket
+ object and delete it when appropriate.
+
+ \since 5.9
+*/
+void QWebSocketServer::handleConnection(QTcpSocket *socket) const
+{
+ Q_D(const QWebSocketServer);
+ d->handleConnection(socket);
+}
+
QT_END_NAMESPACE
diff --git a/src/websockets/qwebsocketserver.h b/src/websockets/qwebsocketserver.h
index 511a55b..f846290 100644
--- a/src/websockets/qwebsocketserver.h
+++ b/src/websockets/qwebsocketserver.h
@@ -54,6 +54,7 @@
QT_BEGIN_NAMESPACE
+class QTcpSocket;
class QWebSocketServerPrivate;
class QWebSocket;
class QWebSocketCorsAuthenticator;
@@ -118,6 +119,8 @@ public:
QList<QWebSocketProtocol::Version> supportedVersions() const;
+ void handleConnection(QTcpSocket *socket) const;
+
Q_SIGNALS:
void acceptError(QAbstractSocket::SocketError socketError);
void serverError(QWebSocketProtocol::CloseCode closeCode);
diff --git a/src/websockets/qwebsocketserver_p.cpp b/src/websockets/qwebsocketserver_p.cpp
index d675056..b7972b0 100644
--- a/src/websockets/qwebsocketserver_p.cpp
+++ b/src/websockets/qwebsocketserver_p.cpp
@@ -384,17 +384,8 @@ void QWebSocketServerPrivate::setError(QWebSocketProtocol::CloseCode code, const
*/
void QWebSocketServerPrivate::onNewConnection()
{
- while (m_pTcpServer->hasPendingConnections()) {
- QTcpSocket *pTcpSocket = m_pTcpServer->nextPendingConnection();
- //use a queued connection because a QSslSocket
- //needs the event loop to process incoming data
- //if not queued, data is incomplete when handshakeReceived is called
- QObjectPrivate::connect(pTcpSocket, &QTcpSocket::readyRead,
- this, &QWebSocketServerPrivate::handshakeReceived,
- Qt::QueuedConnection);
- QObjectPrivate::connect(pTcpSocket, &QTcpSocket::disconnected,
- this, &QWebSocketServerPrivate::onSocketDisconnected);
- }
+ while (m_pTcpServer->hasPendingConnections())
+ handleConnection(m_pTcpServer->nextPendingConnection());
}
/*!
@@ -491,4 +482,17 @@ void QWebSocketServerPrivate::handshakeReceived()
}
}
+void QWebSocketServerPrivate::handleConnection(QTcpSocket *pTcpSocket) const
+{
+ if (Q_LIKELY(pTcpSocket)) {
+ // Use a queued connection because a QSslSocket needs the event loop to process incoming
+ // data. If not queued, data is incomplete when handshakeReceived is called.
+ QObjectPrivate::connect(pTcpSocket, &QTcpSocket::readyRead,
+ this, &QWebSocketServerPrivate::handshakeReceived,
+ Qt::QueuedConnection);
+ QObjectPrivate::connect(pTcpSocket, &QTcpSocket::disconnected,
+ this, &QWebSocketServerPrivate::onSocketDisconnected);
+ }
+}
+
QT_END_NAMESPACE
diff --git a/src/websockets/qwebsocketserver_p.h b/src/websockets/qwebsocketserver_p.h
index 6aabebd..0ca7495 100644
--- a/src/websockets/qwebsocketserver_p.h
+++ b/src/websockets/qwebsocketserver_p.h
@@ -65,6 +65,7 @@
QT_BEGIN_NAMESPACE
class QTcpServer;
+class QTcpSocket;
class QWebSocketServer;
class QWebSocketServerPrivate : public QObjectPrivate
@@ -120,6 +121,8 @@ public:
void setError(QWebSocketProtocol::CloseCode code, const QString &errorString);
+ void handleConnection(QTcpSocket *pTcpSocket) const;
+
QWebSocketServer * const q_ptr;
private: