examples/websockets/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.setPeerVerifyMode(QSslSocket::VerifyNone);
  70     sslConfiguration.setLocalCertificate(certificate);
  71     sslConfiguration.setPrivateKey(sslKey);
  72     sslConfiguration.setProtocol(QSsl::TlsV1SslV3);
  73     m_pWebSocketServer->setSslConfiguration(sslConfiguration);
  74
  75     if (m_pWebSocketServer->listen(QHostAddress::Any, port))
  76     {
  77         qDebug() << "SSL Echo Server listening on port" << port;
  78         connect(m_pWebSocketServer, &QWebSocketServer::newConnection,
  79                 this, &SslEchoServer::onNewConnection);
  80         connect(m_pWebSocketServer, &QWebSocketServer::sslErrors,
  81                 this, &SslEchoServer::onSslErrors);
  82     }
  83 }
  84 //! [constructor]
  85
  86 SslEchoServer::~SslEchoServer()
  87 {
  88     m_pWebSocketServer->close();
  89     qDeleteAll(m_clients.begin(), m_clients.end());
  90 }
  91
  92 //! [onNewConnection]
  93 void SslEchoServer::onNewConnection()
  94 {
  95     QWebSocket *pSocket = m_pWebSocketServer->nextPendingConnection();
  96
  97     qDebug() << "Client connected:" << pSocket->peerName() << pSocket->origin();
  98
  99     connect(pSocket, &QWebSocket::textMessageReceived, this, &SslEchoServer::processTextMessage);
 100     connect(pSocket, &QWebSocket::binaryMessageReceived,
 101             this, &SslEchoServer::processBinaryMessage);
 102     connect(pSocket, &QWebSocket::disconnected, this, &SslEchoServer::socketDisconnected);
 103     //connect(pSocket, &QWebSocket::pong, this, &SslEchoServer::processPong);
 104
 105     m_clients << pSocket;
 106 }
 107 //! [onNewConnection]
 108
 109 //! [processTextMessage]
 110 void SslEchoServer::processTextMessage(QString message)
 111 {
 112     QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
 113     if (pClient)
 114     {
 115         pClient->sendTextMessage(message);
 116     }
 117 }
 118 //! [processTextMessage]
 119
 120 //! [processBinaryMessage]
 121 void SslEchoServer::processBinaryMessage(QByteArray message)
 122 {
 123     QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
 124     if (pClient)
 125     {
 126         pClient->sendBinaryMessage(message);
 127     }
 128 }
 129 //! [processBinaryMessage]
 130
 131 //! [socketDisconnected]
 132 void SslEchoServer::socketDisconnected()
 133 {
 134     qDebug() << "Client disconnected";
 135     QWebSocket *pClient = qobject_cast<QWebSocket *>(sender());
 136     if (pClient)
 137     {
 138         m_clients.removeAll(pClient);
 139         pClient->deleteLater();
 140     }
 141 }
 142
 143 void SslEchoServer::onSslErrors(const QList<QSslError> &)
 144 {
 145     qDebug() << "Ssl errors occurred";
 146 }
 147 //! [socketDisconnected]