source: trunk/examples/network/network-chat/connection.cpp@ 5

Last change on this file since 5 was 2, checked in by Dmitry A. Kuminov, 16 years ago

Initially imported qt-all-opensource-src-4.5.1 from Trolltech.
File size: 7.4 KB
Line 
1/****************************************************************************
2**
3** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
4** Contact: Qt Software Information ([email protected])
5**
6** This file is part of the examples of the Qt Toolkit.
7**
8** $QT_BEGIN_LICENSE:LGPL$
9** Commercial Usage
10** Licensees holding valid Qt Commercial licenses may use this file in
11** accordance with the Qt Commercial License Agreement provided with the
12** Software or, alternatively, in accordance with the terms contained in
13** a written agreement between you and Nokia.
14**
15** GNU Lesser General Public License Usage
16** Alternatively, this file may be used under the terms of the GNU Lesser
17** General Public License version 2.1 as published by the Free Software
18** Foundation and appearing in the file LICENSE.LGPL included in the
19** packaging of this file. Please review the following information to
20** ensure the GNU Lesser General Public License version 2.1 requirements
21** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
22**
23** In addition, as a special exception, Nokia gives you certain
24** additional rights. These rights are described in the Nokia Qt LGPL
25** Exception version 1.0, included in the file LGPL_EXCEPTION.txt in this
26** package.
27**
28** GNU General Public License Usage
29** Alternatively, this file may be used under the terms of the GNU
30** General Public License version 3.0 as published by the Free Software
31** Foundation and appearing in the file LICENSE.GPL included in the
32** packaging of this file. Please review the following information to
33** ensure the GNU General Public License version 3.0 requirements will be
34** met: http://www.gnu.org/copyleft/gpl.html.
35**
36** If you are unsure which license is appropriate for your use, please
37** contact the sales department at [email protected].
38** $QT_END_LICENSE$
39**
40****************************************************************************/
41
42#include "connection.h"
43
44#include <QtNetwork>
45
46static const int TransferTimeout = 30 * 1000;
47static const int PongTimeout = 60 * 1000;
48static const int PingInterval = 5 * 1000;
49static const char SeparatorToken = ' ';
50
51Connection::Connection(QObject *parent)
52 : QTcpSocket(parent)
53{
54 greetingMessage = tr("undefined");
55 username = tr("unknown");
56 state = WaitingForGreeting;
57 currentDataType = Undefined;
58 numBytesForCurrentDataType = -1;
59 transferTimerId = 0;
60 isGreetingMessageSent = false;
61 pingTimer.setInterval(PingInterval);
62
63 QObject::connect(this, SIGNAL(readyRead()), this, SLOT(processReadyRead()));
64 QObject::connect(this, SIGNAL(disconnected()), &pingTimer, SLOT(stop()));
65 QObject::connect(&pingTimer, SIGNAL(timeout()), this, SLOT(sendPing()));
66 QObject::connect(this, SIGNAL(connected()),
67 this, SLOT(sendGreetingMessage()));
68}
69
70QString Connection::name() const
71{
72 return username;
73}
74
75void Connection::setGreetingMessage(const QString &message)
76{
77 greetingMessage = message;
78}
79
80bool Connection::sendMessage(const QString &message)
81{
82 if (message.isEmpty())
83 return false;
84
85 QByteArray msg = message.toUtf8();
86 QByteArray data = "MESSAGE " + QByteArray::number(msg.size()) + " " + msg;
87 return write(data) == data.size();
88}
89
90void Connection::timerEvent(QTimerEvent *timerEvent)
91{
92 if (timerEvent->timerId() == transferTimerId) {
93 abort();
94 killTimer(transferTimerId);
95 transferTimerId = 0;
96 }
97}
98
99void Connection::processReadyRead()
100{
101 if (state == WaitingForGreeting) {
102 if (!readProtocolHeader())
103 return;
104 if (currentDataType != Greeting) {
105 abort();
106 return;
107 }
108 state = ReadingGreeting;
109 }
110
111 if (state == ReadingGreeting) {
112 if (!hasEnoughData())
113 return;
114
115 buffer = read(numBytesForCurrentDataType);
116 if (buffer.size() != numBytesForCurrentDataType) {
117 abort();
118 return;
119 }
120
121 username = QString(buffer) + "@" + peerAddress().toString() + ":"
122 + QString::number(peerPort());
123 currentDataType = Undefined;
124 numBytesForCurrentDataType = 0;
125 buffer.clear();
126
127 if (!isValid()) {
128 abort();
129 return;
130 }
131
132 if (!isGreetingMessageSent)
133 sendGreetingMessage();
134
135 pingTimer.start();
136 pongTime.start();
137 state = ReadyForUse;
138 emit readyForUse();
139 }
140
141 do {