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#include <QtNetwork>
52
53#include "client.h"
54#include "connection.h"
55#include "peermanager.h"
56
57Client::Client()
58{
59 peerManager = new PeerManager(this);
60 peerManager->setServerPort(server.serverPort());
61 peerManager->startBroadcasting();
62
63 connect(peerManager, &PeerManager::newConnection,
64 this, &Client::newConnection);
65 connect(&server, &Server::newConnection,
66 this, &Client::newConnection);
67}
68
69void Client::sendMessage(const QString &message)
70{
71 if (message.isEmpty())
72 return;
73
74 for (Connection *connection : qAsConst(peers))
75 connection->sendMessage(message);
76}
77
78QString Client::nickName() const
79{
80 return peerManager->userName() + '@' + QHostInfo::localHostName()
81 + ':' + QString::number(server.serverPort());
82}
83
84bool Client::hasConnection(const QHostAddress &senderIp, int senderPort) const
85{
86 if (senderPort == -1)
87 return peers.contains(senderIp);
88
89 if (!peers.contains(senderIp))
90 return false;
91
92 const QList<Connection *> connections = peers.values(senderIp);
93 for (const Connection *connection : connections) {
94 if (connection->peerPort() == senderPort)
95 return true;
96 }
97
98 return false;
99}
100
101void Client::newConnection(Connection *connection)
102{
103 connection->setGreetingMessage(peerManager->userName());
104
105 connect(connection, &Connection::errorOccurred, this, &Client::connectionError);
106 connect(connection, &Connection::disconnected, this, &Client::disconnected);
107 connect(connection, &Connection::readyForUse, this, &Client::readyForUse);
108}
109
110void Client::readyForUse()
111{
112 Connection *connection = qobject_cast<Connection *>(sender());
113 if (!connection || hasConnection(connection->peerAddress(),
114 connection->peerPort()))
115 return;
116
117 connect(connection, &Connection::newMessage,
118 this, &Client::newMessage);
119
120 peers.insert(connection->peerAddress(), connection);
121 QString nick = connection->name();
122 if (!nick.isEmpty())
123 emit newParticipant(nick);
124}
125
126void Client::disconnected()
127{
128 if (Connection *connection = qobject_cast<Connection *>(sender()))
129 removeConnection(connection);
130}
131
132void Client::connectionError(QAbstractSocket::SocketError /* socketError */)
133{
134 if (Connection *connection = qobject_cast<Connection *>(sender()))
135 removeConnection(connection);
136}
137
138void Client::removeConnection(Connection *connection)
139{
140 if (peers.contains(connection->peerAddress())) {
141 peers.remove(connection->peerAddress());
142 QString nick = connection->name();
143 if (!nick.isEmpty())
144 emit participantLeft(nick);
145 }
146 connection->deleteLater();
147}
148