| 1 | /**************************************************************************** |
| 2 | ** |
| 3 | ** Copyright (C) 2016 The Qt Company Ltd. |
| 4 | ** Contact: https://www.qt.io/licensing/ |
| 5 | ** |
| 6 | ** This file is part of the examples of the Qt Toolkit. |
| 7 | ** |
| 8 | ** $QT_BEGIN_LICENSE:BSD$ |
| 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 The Qt Company. For licensing terms |
| 14 | ** and conditions see https://www.qt.io/terms-conditions. For further |
| 15 | ** information use the contact form at https://www.qt.io/contact-us. |
| 16 | ** |
| 17 | ** BSD License Usage |
| 18 | ** Alternatively, you may use this file under the terms of the BSD license |
| 19 | ** as follows: |
| 20 | ** |
| 21 | ** "Redistribution and use in source and binary forms, with or without |
| 22 | ** modification, are permitted provided that the following conditions are |
| 23 | ** met: |
| 24 | ** * Redistributions of source code must retain the above copyright |
| 25 | ** notice, this list of conditions and the following disclaimer. |
| 26 | ** * Redistributions in binary form must reproduce the above copyright |
| 27 | ** notice, this list of conditions and the following disclaimer in |
| 28 | ** the documentation and/or other materials provided with the |
| 29 | ** distribution. |
| 30 | ** * Neither the name of The Qt Company Ltd nor the names of its |
| 31 | ** contributors may be used to endorse or promote products derived |
| 32 | ** from this software without specific prior written permission. |
| 33 | ** |
| 34 | ** |
| 35 | ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 36 | ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 37 | ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 38 | ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 39 | ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 40 | ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 41 | ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 42 | ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 43 | ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 44 | ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 45 | ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." |
| 46 | ** |
| 47 | ** $QT_END_LICENSE$ |
| 48 | ** |
| 49 | ****************************************************************************/ |
| 50 | |
| 51 | #ifndef CONNECTION_H |
| 52 | #define CONNECTION_H |
| 53 | |
| 54 | #include <QCborStreamReader> |
| 55 | #include <QCborStreamWriter> |
| 56 | #include <QElapsedTimer> |
| 57 | #include <QHostAddress> |
| 58 | #include <QString> |
| 59 | #include <QTcpSocket> |
| 60 | #include <QTimer> |
| 61 | |
| 62 | static const int MaxBufferSize = 1024000; |
| 63 | |
| 64 | class Connection : public QTcpSocket |
| 65 | { |
| 66 | Q_OBJECT |
| 67 | |
| 68 | public: |
| 69 | enum ConnectionState { |
| 70 | WaitingForGreeting, |
| 71 | ReadingGreeting, |
| 72 | ReadyForUse |
| 73 | }; |
| 74 | enum DataType { |
| 75 | PlainText, |
| 76 | Ping, |
| 77 | Pong, |
| 78 | Greeting, |
| 79 | Undefined |
| 80 | }; |
| 81 | |
| 82 | Connection(QObject *parent = nullptr); |
| 83 | Connection(qintptr socketDescriptor, QObject *parent = nullptr); |
| 84 | ~Connection(); |
| 85 | |
| 86 | QString name() const; |
| 87 | void setGreetingMessage(const QString &message); |
| 88 | bool sendMessage(const QString &message); |
| 89 | |
| 90 | signals: |
| 91 | void readyForUse(); |
| 92 | void newMessage(const QString &from, const QString &message); |
| 93 | |
| 94 | protected: |
| 95 | void timerEvent(QTimerEvent *timerEvent) override; |
| 96 | |
| 97 | private slots: |
| 98 | void processReadyRead(); |
| 99 | void sendPing(); |
| 100 | void sendGreetingMessage(); |
| 101 | |
| 102 | private: |
| 103 | bool hasEnoughData(); |
| 104 | void processGreeting(); |
| 105 | void processData(); |
| 106 | |
| 107 | QCborStreamReader reader; |
| 108 | QCborStreamWriter writer; |
| 109 | QString greetingMessage; |
| 110 | QString username; |
| 111 | QTimer pingTimer; |
| 112 | QElapsedTimer pongTime; |
| 113 | QString buffer; |
| 114 | ConnectionState state; |
| 115 | DataType currentDataType; |
| 116 | int transferTimerId; |
| 117 | bool isGreetingMessageSent; |
| 118 | }; |
| 119 | |
| 120 | #endif |
| 121 | |