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
|
/*
CharmTCPCommandServer.cpp
This file is part of Charm, a task-based time tracking application.
Copyright (C) 2015-2018 Klarälvdalens Datakonsult AB, a KDAB Group company, info@kdab.com
Author: Guillermo A. Amaral <gamaral@kdab.com>
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "CharmTCPCommandServer.h"
#include <QTcpServer>
#include <QTcpSocket>
#include <QTimerEvent>
#include <QUdpSocket>
#include "CharmCommandSession.h"
#include "CharmCMake.h"
#ifndef CHARM_CI_TCPSERVER
#error Build system error: CHARM_CI_TCPSERVER should be defined
#endif
static const quint16 sCharmDefaultPort(5323);
static const int sCharmDiscoveryBroadcastRate(5000);
CharmTCPCommandServer::CharmTCPCommandServer(QObject *parent)
: CharmCommandServer(parent)
, m_address(QHostAddress::Any)
, m_port(sCharmDefaultPort)
, m_server(new QTcpServer(this))
, m_discovery(new QUdpSocket(this))
, m_discoveryTimer(0)
{
}
CharmTCPCommandServer::~CharmTCPCommandServer()
{
}
const QHostAddress &CharmTCPCommandServer::address() const
{
return m_address;
}
void CharmTCPCommandServer::setAddress(const QHostAddress &address)
{
m_address = address;
}
quint16 CharmTCPCommandServer::port() const
{
return m_port;
}
void CharmTCPCommandServer::setPort(quint16 port)
{
m_port = port;
}
bool CharmTCPCommandServer::listen()
{
if (!m_server->listen(m_address, m_port)) {
qWarning("Failed to bind to %s:%d", qPrintable(m_address.toString()), m_port);
return false;
}
connect(m_server, SIGNAL(newConnection()), SLOT(onNewConnection()));
m_discoveryTimer = startTimer(sCharmDiscoveryBroadcastRate);
return true;
}
void CharmTCPCommandServer::close()
{
if (m_discoveryTimer)
killTimer(m_discoveryTimer), m_discoveryTimer = 0;
m_server->close();
}
void CharmTCPCommandServer::timerEvent(QTimerEvent *event)
{
if (event->timerId() != m_discoveryTimer)
return;
static const char sBroadcastIdentifier[] = "LUCKY"; // CHARMS
m_discovery->writeDatagram(sBroadcastIdentifier, sizeof(sBroadcastIdentifier),
QHostAddress::Broadcast, m_port);
}
void CharmTCPCommandServer::onNewConnection()
{
while (m_server->hasPendingConnections()) {
QTcpSocket *conn = m_server->nextPendingConnection();
Q_ASSERT(conn);
qDebug("New TCP connection, creating command session...");
spawnSession(conn);
}
}
|
