examples/sslechoserver/sslechoserver.cpp

   1 /****************************************************************************
   2 **
   3 ** Copyright (C) 2014 Kurt Pattyn <pattyn.kurt@gmail.com>.
   4 ** Contact: http://www.qt-project.org/legal
   5 **
   6 ** This file is part of the QtWebSockets module of the Qt Toolkit.
   7 **
   8 ** $QT_BEGIN_LICENSE:LGPL$
   9 ** Commercial License Usage
  10 ** Licensees holding valid commercial Qt licenses may use this file in
  11 ** accordance with the commercial license agreement provided with the
  12 ** Software or, alternatively, in accordance with the terms contained in
  13 ** a written agreement between you and Digia.  For licensing terms and
  14 ** conditions see http://qt.digia.com/licensing.  For further information
  15 ** use the contact form at http://qt.digia.com/contact-us.
  16 **
  17 ** GNU Lesser General Public License Usage
  18 ** Alternatively, this file may be used under the terms of the GNU Lesser
  19 ** General Public License version 2.1 as published by the Free Software
  20 ** Foundation and appearing in the file LICENSE.LGPL included in the
  21 ** packaging of this file.  Please review the following information to
  22 ** ensure the GNU Lesser General Public License version 2.1 requirements
  23 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
  24 **
  25 ** In addition, as a special exception, Digia gives you certain additional
  26 ** rights.  These rights are described in the Digia Qt LGPL Exception
  27 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
  28 **
  29 ** GNU General Public License Usage
  30 ** Alternatively, this file may be used under the terms of the GNU
  31 ** General Public License version 3.0 as published by the Free Software
  32 ** Foundation and appearing in the file LICENSE.GPL included in the
  33 ** packaging of this file.  Please review the following information to
  34 ** ensure the GNU General Public License version 3.0 requirements will be
  35 ** met: http://www.gnu.org/copyleft/gpl.html.
  36 **
  37 **
  38 ** $QT_END_LICENSE$
  39 **
  40 ****************************************************************************/
  41 #include "sslechoserver.h"
  42 #include "QtWebSockets/QWebSocketServer"
  43 #include "QtWebSockets/QWebSocket"
  44 #include <QtCore/QDebug>
  45 #include <QtCore/QFile>
  46 #include <QtNetwork/QSslCertificate>
  47 #include <QtNetwork/QSslKey>
  48
  49 QT_USE_NAMESPACE
  50
  51 //! [constructor]
  52 SslEchoServer::SslEchoServer(quint16 port, QObject *parent) :
  53     QObject(parent),
  54     m_pWebSocketServer(Q_NULLPTR),
  55     m_clients()
  56 {
  57     m_pWebSocketServer = new QWebSocketServer(QStringLiteral("SSL Echo Server"),
  58                                               QWebSocketServer::SecureMode,
  59                                               this);
  60     QSslConfiguration sslConfiguration;
  61     QFile certFile(QStringLiteral("./localhost.cert"));
  62     QFile keyFile(QStringLiteral("./localhost.key"));
  63     certFile.open(QIODevice::ReadOnly);
  64     keyFile.open(QIODevice::ReadOnly);
  65     QSslCertificate certificate(&certFile, QSsl::Pem);
  66     QSslKey sslKey(&keyFile, QSsl::Rsa, QSsl::Pem);
  67     certFile.close();
  68     keyFile.close();
  69     sslConfiguration.setProtocol(QSsl::AnyProtocol);
  70     sslConfiguration.setPeerVerifyMode(QSslSocket::VerifyNone);
  71     sslConfiguration.setLocalCertificate(certificate);
  72     sslConfiguration.setPrivateKey(sslKey);
  73     sslConfiguration.setProtocol(QSsl::TlsV1SslV3);
  74     m_pWebSocketServer->setSslConfiguration(sslConfiguration);
  75
  76     if (m_pWebSocketServer->listen(QHostAddress::Any, port))
  77     {
  78         qDebug() << "SSL Echo Server listening on port" << port;
  79         connect(m_pWebSocketServer, &QWebSocketServer::newConnection,
  80                 this, &SslEchoServer::onNewConnection);
  81         connect(m_pWebSocketServer, &QWebSocketServer::sslErrors,
  82                 this, &SslEchoServer::onSslErrors);
  83     }
  84 }
  85 //! [constructor]
  86
  87 SslEchoServer::~SslEchoServer()
  88 {
  89     m_pWebSocketServer->close();
  90     qDeleteAll(m_clients.begin(), m_clients.end());
  91 }
  92
  93 //! [onNewConnection]
  94 void SslEchoServer::onNewConnection()
  95 {
  96     QWebSocket *pSocket = m_pWebSocketServer->nextPendingConnection();
  97
  98     qDebug() << "Client connected:" << pSocket->peerName() << pSocket->origin();
  99
 100     connect(pSocket, &QWebSocket::textMessageReceived, this, &SslEchoServer::processTextMessage);
 101     connect(pSocket, &QWebSocket::binaryMessageReceived,
 102             this, &SslEchoServer::processBinaryMessage);
 103     connect(pSocket, &QWebSocket::disconnected, this, &SslEchoServer::socketDisconnected);
 104     //connect(pSocket, &QWebSocket::pong, this, &SslEchoServer::processPong);
 105
 106     m_clients << pSocket;
 107 }
 108 //! [onNewConnection]
 109
 110 //! [processTextMessage]
 111 void SslEchoServer::processTextMessage(QString message)
 112 {
 113     QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
 114     if (pClient)
 115     {
 116         pClient->sendTextMessage(message);
 117     }
 118 }
 119 //! [processTextMessage]
 120
 121 //! [processBinaryMessage]
 122 void SslEchoServer::processBinaryMessage(QByteArray message)
 123 {
 124     QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
 125     if (pClient)
 126     {
 127         pClient->sendBinaryMessage(message);
 128     }
 129 }
 130 //! [processBinaryMessage]
 131
 132 //! [socketDisconnected]
 133 void SslEchoServer::socketDisconnected()
 134 {
 135     qDebug() << "Client disconnected";
 136     QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
 137     if (pClient)
 138     {
 139         m_clients.removeAll(pClient);
 140         pClient->deleteLater();
 141     }
 142 }
 143
 144 void SslEchoServer::onSslErrors(const QList<QSslError> &)
 145 {
 146     qDebug() << "Ssl errors occurred";
 147 }
 148 //! [socketDisconnected]