1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
#include <QtNetwork>
#include "client.h"
#include "connection.h"
#include "peermanager.h"
Client::Client()
{
peerManager = new PeerManager(this);
peerManager->setServerPort(server.serverPort());
peerManager->startBroadcasting();
connect(peerManager, &PeerManager::newConnection,
this, &Client::newConnection);
connect(&server, &Server::newConnection,
this, &Client::newConnection);
}
void Client::sendMessage(const QString &message)
{
if (message.isEmpty())
return;
for (Connection *connection : std::as_const(peers))
connection->sendMessage(message);
}
QString Client::nickName() const
{
return peerManager->userName() + '@' + QHostInfo::localHostName()
+ ':' + QString::number(server.serverPort());
}
bool Client::hasConnection(const QHostAddress &senderIp, int senderPort) const
{
if (senderPort == -1)
return peers.contains(senderIp);
if (!peers.contains(senderIp))
return false;
const QList<Connection *> connections = peers.values(senderIp);
for (const Connection *connection : connections) {
if (connection->peerPort() == senderPort)
return true;
}
return false;
}
void Client::newConnection(Connection *connection)
{
connection->setGreetingMessage(peerManager->userName());
connect(connection, &Connection::errorOccurred, this, &Client::connectionError);
connect(connection, &Connection::disconnected, this, &Client::disconnected);
connect(connection, &Connection::readyForUse, this, &Client::readyForUse);
}
void Client::readyForUse()
{
Connection *connection = qobject_cast<Connection *>(sender());
if (!connection || hasConnection(connection->peerAddress(),
connection->peerPort()))
return;
connect(connection, &Connection::newMessage,
this, &Client::newMessage);
peers.insert(connection->peerAddress(), connection);
QString nick = connection->name();
if (!nick.isEmpty())
emit newParticipant(nick);
}
void Client::disconnected()
{
if (Connection *connection = qobject_cast<Connection *>(sender()))
removeConnection(connection);
}
void Client::connectionError(QAbstractSocket::SocketError /* socketError */)
{
if (Connection *connection = qobject_cast<Connection *>(sender()))
removeConnection(connection);
}
void Client::removeConnection(Connection *connection)
{
if (peers.contains(connection->peerAddress())) {
peers.remove(connection->peerAddress());
QString nick = connection->name();
if (!nick.isEmpty())
emit participantLeft(nick);
}
connection->deleteLater();
}
|
