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
|
/*
RailControl - Model Railway Control Software
Copyright (c) 2017-2025 by Teddy / Dominik Mahrer - www.railcontrol.org
RailControl 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 3, or (at your option) any
later version.
RailControl 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 RailControl; see the file LICENCE. If not see
<http://www.gnu.org/licenses/>.
*/
#pragma once
#include <string>
#include <thread>
#include <vector>
#include <sys/socket.h>
#include "Logger/Logger.h"
#include "Utils/Network.h"
namespace Network
{
class UdpClient
{
public:
UdpClient() = delete;
UdpClient(const UdpClient&) = delete;
UdpClient& operator=(const UdpClient&) = delete;
inline UdpClient(const int serverSocket,
const struct sockaddr_storage* clientAddress,
const time_t timeout = 60)
: run(true),
lastUsed(time(nullptr)),
timeout(timeout),
serverSocket(serverSocket),
clientAddress(*clientAddress)
{
}
virtual ~UdpClient()
{
}
inline bool IsClient(const struct sockaddr_storage* address)
{
return Utils::Network::CompareAddresses(&clientAddress, address);
}
inline void Update()
{
lastUsed = time(nullptr);
}
inline bool Terminated()
{
return !run;
}
inline bool TimedOut()
{
return lastUsed + timeout < time(nullptr);
}
virtual void Work(const unsigned char* buffer, const size_t size) = 0;
protected:
inline void Send(const unsigned char* buffer, const size_t size)
{
sendto(serverSocket, buffer, size, 0, reinterpret_cast<const struct sockaddr*>(&clientAddress), sizeof(struct sockaddr_storage));
}
inline void Terminate()
{
run = false;
}
inline std::string AddressAsString()
{
return Utils::Network::AddressToString(&clientAddress);
}
private:
volatile bool run;
time_t lastUsed;
time_t timeout;
const int serverSocket;
const struct sockaddr_storage clientAddress;
};
}
|
