forked from Haivision/srt
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_main.cpp
More file actions
231 lines (196 loc) · 5.61 KB
/
test_main.cpp
File metadata and controls
231 lines (196 loc) · 5.61 KB
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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
#include <string>
#include <iterator>
#include <vector>
#include <sstream>
#include <thread>
#include "gtest/gtest.h"
#include "test_env.h"
#include "srt.h"
#include "netinet_any.h"
#include "core.h"
#include "api.h"
using namespace std;
int main(int argc, char **argv)
{
string command_line_arg(argc == 2 ? argv[1] : "");
testing::InitGoogleTest(&argc, argv);
testing::AddGlobalTestEnvironment(new srt::TestEnv(argc, argv));
return RUN_ALL_TESTS();
}
namespace srt
{
TestEnv* TestEnv::me = 0;
void TestEnv::FillArgMap()
{
// The rule is:
// - first arguments go to an empty string key
// - if an argument has - in the beginning, name the key
// - key followed by args collected in a list
// - double dash prevents interpreting further args as option keys
string key;
bool expectkey = true;
argmap[""];
for (auto& a: args)
{
if (a.size() > 1)
{
if (expectkey && a[0] == '-')
{
if (a[1] == '-')
expectkey = false;
else if (a[1] == '/')
key = "";
else
{
key = a.substr(1);
argmap[key]; // Make sure it exists even empty
}
continue;
}
}
argmap[key].push_back(a);
}
return;
}
std::string TestEnv::OptionValue(const std::string& key)
{
std::ostringstream out;
auto it = argmap.find(key);
if (it != argmap.end() && !it->second.empty())
{
auto iv = it->second.begin();
out << (*iv);
while (++iv != it->second.end())
{
out << " " << (*iv);
}
}
return out.str();
}
// Specific functions
bool TestEnv::Allowed_IPv6()
{
if (TestEnv::me->OptionPresent("disable-ipv6"))
{
std::cout << "TEST: IPv6 testing disabled, FORCED PASS\n";
return false;
}
return true;
}
void TestInit::start(int& w_retstatus)
{
ASSERT_GE(w_retstatus = (int)srt_startup(), 0);
}
void TestInit::stop()
{
if (TestEnv::me->OptionPresent("check-close"))
{
std::string remaining_sockets = CUDT::uglobal().testSocketsClear();
// Give it one more chance to be deleted by GC
if (remaining_sockets.size())
{
std::cout << "TestInit::stop: waiting extra 1.5s to ensure closed sockets\n";
std::this_thread::sleep_for(std::chrono::milliseconds(1500));
remaining_sockets = CUDT::uglobal().testSocketsClear();
}
EXPECT_TRUE(remaining_sockets.empty()) << "STILL NOT CLOSED: " << remaining_sockets;
}
EXPECT_NE((int)srt_cleanup(), -1);
}
// This function finds some interesting options among command
// line arguments and does specific things.
void TestInit::HandlePerTestOptions()
{
// As a short example:
// use '-logdebug' option to turn on debug logging.
if (TestEnv::me->OptionPresent("logdebug"))
{
srt_setloglevel(LOG_DEBUG);
}
if (TestEnv::me->OptionPresent("lognote"))
{
srt_setloglevel(LOG_NOTICE);
}
}
// Copied from ../apps/apputil.cpp, can't really link this file here.
sockaddr_any CreateAddr(const std::string& name, unsigned short port, int pref_family)
{
using namespace std;
// Handle empty name.
// If family is specified, empty string resolves to ANY of that family.
// If not, it resolves to IPv4 ANY (to specify IPv6 any, use [::]).
if (name == "")
{
sockaddr_any result(pref_family == AF_INET6 ? pref_family : AF_INET);
result.hport(port);
return result;
}
bool first6 = pref_family != AF_INET;
int families[2] = {AF_INET6, AF_INET};
if (!first6)
{
families[0] = AF_INET;
families[1] = AF_INET6;
}
for (int i = 0; i < 2; ++i)
{
int family = families[i];
sockaddr_any result (family);
// Try to resolve the name by pton first
if (inet_pton(family, name.c_str(), result.get_addr()) == 1)
{
result.hport(port); // same addr location in ipv4 and ipv6
return result;
}
}
// If not, try to resolve by getaddrinfo
// This time, use the exact value of pref_family
sockaddr_any result;
addrinfo fo = {
0,
pref_family,
0, 0,
0, 0,
NULL, NULL
};
addrinfo* val = nullptr;
int erc = getaddrinfo(name.c_str(), nullptr, &fo, &val);
if (erc == 0)
{
result.set(val->ai_addr);
result.len = result.size();
result.hport(port); // same addr location in ipv4 and ipv6
}
freeaddrinfo(val);
return result;
}
UniqueSocket::~UniqueSocket()
{
// Could be closed explicitly
if (sock != -1)
close();
}
void UniqueSocket::close()
{
int close_result = srt_close(sock);
int close_error = srt_getlasterror(nullptr);
// XXX SRT_EINVSOCK is reported when the socket
// has been already wiped out, which may happen to a broken socket.
// This isn't exactly intended, although trying to close a nonexistent
// socket is not a problem, as long as it happens before the id value rollover
// (that is, when it's closed immediately after getting broken).
// This solution is still slick though and should be fixed.
//
// Restore this, when fixed
// EXPECT_NE(srt_close(sock), SRT_ERROR) << lab << " CREATED: "<< f << ":" << l;
if (close_result == SRT_ERROR)
{
EXPECT_NE(close_error, SRT_EINVSOCK) << lab << " CREATED: "<< f << ":" << l;
}
else
{
EXPECT_EQ(close_result, 0) << lab << " CREATED: "<< f << ":" << l;
}
sock = -1;
}
}