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