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
125
126
127
128
129
130
131
132
133
134
135
|
/**
* (c) 2025 by Mega Limited, New Zealand
*
* This file is part of the MEGA SDK - Client Access Engine.
*
* Applications using the MEGA API must present a valid application key
* and comply with the the rules set forth in the Terms of Service.
*
* The MEGA SDK 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.
*
* @copyright Simplified (2-clause) BSD License.
*
* You should have received a copy of the license along with this
* program.
*/
#ifndef MEGA_UDP_SOCKET_TESTER_H
#define MEGA_UDP_SOCKET_TESTER_H
#include "mega/udp_socket.h"
#include <chrono>
#include <map>
#include <memory>
#include <string>
#include <vector>
namespace mega
{
/**
* @brief Mechanism for running a network connectivity test, by sending multiple messages on a
* single socket.
* It includes:
* - encapsulate the logic for building messages;
* - send all messages on the required socket;
* - receive and validate replies;
* - provide the results after all communication has finished.
*/
class UdpSocketTester
{
public:
UdpSocketTester(const std::string& ip, uint16_t port);
~UdpSocketTester();
struct TestSuite
{
TestSuite(uint16_t loops, uint16_t shorts, uint16_t longs, uint16_t dnss, uint64_t userId);
const uint16_t loopCount{};
const uint16_t shortMessageCount{};
const uint16_t longMessageCount{};
const uint16_t dnsMessageCount{};
const std::string& getShortMessage() const
{
return mShortMessage;
}
const std::string& getLongMessage() const
{
return mLongMessage;
}
const std::string& getDnsIPv4Message() const
{
return mDnsIPv4Message;
}
const std::string& getDnsIPv6Message() const
{
return mDnsIPv6Message;
}
enum class MessageType : char
{
SHORT = 'S',
LONG = 'L',
DNS = 'D',
};
uint16_t totalMessageCount() const
{
return static_cast<uint16_t>(loopCount *
(shortMessageCount + longMessageCount + dnsMessageCount));
}
private:
std::string mShortMessage;
std::string mLongMessage;
std::string mDnsIPv4Message;
std::string mDnsIPv6Message;
};
bool startSuite(const TestSuite& suite);
struct MessageResult
{
TestSuite::MessageType messageType;
int errorCode;
};
struct SocketResults
{
const uint16_t port;
std::vector<MessageResult> messageResults;
std::map<std::string, uint16_t> log; // {log message, counter}
};
// Return {port, {{messageType, error}, ...}, logged messages}
SocketResults getSocketResults(const std::chrono::high_resolution_clock::time_point& timeout);
private:
void sendMessage(TestSuite::MessageType type, const std::string& message);
static void sleepIfMultipleOf(uint16_t multiFactor, uint16_t factor);
void confirmFirst(TestSuite::MessageType type);
void log(std::string&& action, std::string&& error);
std::unique_ptr<UdpSocket> mSocket;
static constexpr int REPLY_NOT_RECEIVED = -1111;
SocketResults mTestResults;
bool mRunning{};
std::string mShortMessage;
std::string mLongMessage;
std::string mDnsMessage;
};
} // namespace mega
#endif // MEGA_UDP_SOCKET_TESTER_H
|
