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 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
|
// Copyright (C) 2015 Alex Trotsenko <alex1973tr@gmail.com>
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
#include <QtWidgets>
#include <QtNetwork>
#include <QtAlgorithms>
#include "server.h"
#include "movieprovider.h"
#include "timeprovider.h"
#include "chatprovider.h"
#include "../shared/sctpchannels.h"
Server::Server(QWidget *parent)
: QDialog(parent)
, providers(SctpChannels::NumberOfChannels)
{
setWindowTitle(tr("Multi-stream Server"));
sctpServer = new QSctpServer(this);
sctpServer->setMaximumChannelCount(NumberOfChannels);
statusLabel = new QLabel;
QPushButton *quitButton = new QPushButton(tr("Quit"));
providers[SctpChannels::Movie] = new MovieProvider(this);
providers[SctpChannels::Time] = new TimeProvider(this);
providers[SctpChannels::Chat] = new ChatProvider(this);
connect(sctpServer, &QSctpServer::newConnection, this, &Server::newConnection);
connect(quitButton, &QPushButton::clicked, this, &Server::accept);
connect(providers[SctpChannels::Movie], &Provider::writeDatagram, this, &Server::writeDatagram);
connect(providers[SctpChannels::Time], &Provider::writeDatagram, this, &Server::writeDatagram);
connect(providers[SctpChannels::Chat], &Provider::writeDatagram, this, &Server::writeDatagram);
QVBoxLayout *mainLayout = new QVBoxLayout;
mainLayout->addWidget(statusLabel);
mainLayout->addWidget(quitButton);
setLayout(mainLayout);
}
Server::~Server()
{
qDeleteAll(connections.begin(), connections.end());
}
int Server::exec()
{
if (!sctpServer->listen()) {
QMessageBox::critical(this, windowTitle(),
tr("Unable to start the server: %1.")
.arg(sctpServer->errorString()));
return QDialog::Rejected;
}
QString ipAddress;
QList<QHostAddress> ipAddressesList = QNetworkInterface::allAddresses();
// use the first non-localhost IPv4 address
for (int i = 0; i < ipAddressesList.size(); ++i) {
if (ipAddressesList.at(i) != QHostAddress::LocalHost &&
ipAddressesList.at(i).toIPv4Address()) {
ipAddress = ipAddressesList.at(i).toString();
break;
}
}
// if we did not find one, use IPv4 localhost
if (ipAddress.isEmpty())
ipAddress = QHostAddress(QHostAddress::LocalHost).toString();
statusLabel->setText(tr("The server is running on\n\nIP: %1\nport: %2\n\n"
"Run the Multi-stream Client example now.")
.arg(ipAddress).arg(sctpServer->serverPort()));
return QDialog::exec();
}
void Server::newConnection()
{
QSctpSocket *connection = sctpServer->nextPendingDatagramConnection();
connections.append(connection);
connect(connection, &QSctpSocket::channelReadyRead, this, &Server::readDatagram);
connect(connection, &QSctpSocket::disconnected, this, &Server::clientDisconnected);
for (Provider *provider : providers)
provider->newConnection(*connection);
}
void Server::clientDisconnected()
{
QSctpSocket *connection = static_cast<QSctpSocket *>(sender());
connections.removeOne(connection);
connection->disconnect();
for (Provider *provider : providers)
provider->clientDisconnected(*connection);
connection->deleteLater();
}
void Server::readDatagram(int channel)
{
QSctpSocket *connection = static_cast<QSctpSocket *>(sender());
connection->setCurrentReadChannel(channel);
providers[channel]->readDatagram(*connection, connection->readDatagram().data());
}
void Server::writeDatagram(QSctpSocket *to, const QByteArray &ba)
{
int channel = providers.indexOf(static_cast<Provider *>(sender()));
if (to) {
to->setCurrentWriteChannel(channel);
to->writeDatagram(ba);
return;
}
for (QSctpSocket *connection : connections) {
connection->setCurrentWriteChannel(channel);
connection->writeDatagram(ba);
}
}
|