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
|
/*
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 <cstring>
#include <string>
#include "Utils/Network.h"
namespace Network
{
class TcpConnection
{
public:
TcpConnection() = delete;
TcpConnection& operator=(const TcpConnection&) = delete;
inline TcpConnection(int socket,
const struct sockaddr_storage* address = nullptr)
: connectionSocket(socket),
connected(socket != 0)
{
if (address)
{
this->address = *address;
}
else
{
memset(&(this->address), 0, sizeof(struct sockaddr_storage));
}
}
inline TcpConnection(const TcpConnection& other)
: connectionSocket(other.connectionSocket),
connected(other.connected),
address(other.address)
{
}
inline ~TcpConnection()
{
Terminate();
}
void Terminate() const;
int Send(const unsigned char* buffer, const size_t bufferLength, const int flags = 0) const;
inline int Send(const char* buffer, const size_t bufferLength, const int flags = 0) const
{
return Send(reinterpret_cast<const unsigned char*>(buffer), bufferLength, flags);
}
inline int Send(const std::string& string, const int flags = 0) const
{
return Send(string.c_str(), string.size(), flags);
}
int Receive(unsigned char* buffer, const size_t bufferLength, const int flags = 0) const;
inline int Receive(char* buffer, const size_t bufferLength, const int flags = 0) const
{
return Receive(reinterpret_cast<unsigned char*>(buffer), bufferLength, flags);
}
bool Receive(std::string& data, const size_t maxData = 1024, const int flags = 0);
int ReceiveExact(unsigned char* buffer, const size_t bufferLength, const int flags = 0) const;
inline bool IsConnected() const
{
return connected;
}
inline std::string AddressAsString()
{
return Utils::Network::AddressToString(&address);
}
private:
mutable int connectionSocket;
mutable volatile bool connected;
struct sockaddr_storage address;
};
}
|
