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
|
//===-- PortMapTest.cpp ---------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "llvm/Testing/Support/Error.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "Plugins/Process/gdb-remote/GDBRemoteCommunicationServerPlatform.h"
using namespace lldb_private::process_gdb_remote;
TEST(PortMapTest, Constructors) {
// Default construct to empty map
GDBRemoteCommunicationServerPlatform::PortMap p1;
ASSERT_TRUE(p1.empty());
// Empty means no restrictions, return 0 and and bind to get a port
llvm::Expected<uint16_t> available_port = p1.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::HasValue(0));
// Adding any port makes it not empty
p1.AllowPort(1);
ASSERT_FALSE(p1.empty());
// So we will return the added port this time
available_port = p1.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::HasValue(1));
// Construct from a range of ports
GDBRemoteCommunicationServerPlatform::PortMap p2(1, 4);
ASSERT_FALSE(p2.empty());
// Use up all the ports
for (uint16_t expected = 1; expected < 4; ++expected) {
available_port = p2.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::HasValue(expected));
p2.AssociatePortWithProcess(*available_port, 1);
}
// Now we fail since we're not an empty port map but all ports are used
available_port = p2.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::Failed());
}
TEST(PortMapTest, FreePort) {
GDBRemoteCommunicationServerPlatform::PortMap p(1, 4);
// Use up all the ports
for (uint16_t port = 1; port < 4; ++port) {
p.AssociatePortWithProcess(port, 1);
}
llvm::Expected<uint16_t> available_port = p.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::Failed());
// Can't free a port that isn't in the map
ASSERT_FALSE(p.FreePort(0));
ASSERT_FALSE(p.FreePort(4));
// After freeing a port it becomes available
ASSERT_TRUE(p.FreePort(2));
available_port = p.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::HasValue(2));
}
TEST(PortMapTest, FreePortForProcess) {
GDBRemoteCommunicationServerPlatform::PortMap p;
p.AllowPort(1);
p.AllowPort(2);
ASSERT_TRUE(p.AssociatePortWithProcess(1, 11));
ASSERT_TRUE(p.AssociatePortWithProcess(2, 22));
// All ports have been used
llvm::Expected<uint16_t> available_port = p.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::Failed());
// Can't free a port for a process that doesn't have any
ASSERT_FALSE(p.FreePortForProcess(33));
// You can move a used port to a new pid
ASSERT_TRUE(p.AssociatePortWithProcess(1, 99));
ASSERT_TRUE(p.FreePortForProcess(22));
available_port = p.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::Succeeded());
ASSERT_EQ(2, *available_port);
// proces 22 no longer has a port
ASSERT_FALSE(p.FreePortForProcess(22));
}
TEST(PortMapTest, AllowPort) {
GDBRemoteCommunicationServerPlatform::PortMap p;
// Allow port 1 and tie it to process 11
p.AllowPort(1);
ASSERT_TRUE(p.AssociatePortWithProcess(1, 11));
// Allowing it a second time shouldn't change existing mapping
p.AllowPort(1);
llvm::Expected<uint16_t> available_port = p.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::Failed());
// A new port is marked as free when allowed
p.AllowPort(2);
available_port = p.GetNextAvailablePort();
ASSERT_THAT_EXPECTED(available_port, llvm::HasValue(2));
// 11 should still be tied to port 1
ASSERT_TRUE(p.FreePortForProcess(11));
}
|
