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
|
// Copyright 2021 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
#include "pw_router/static_router.h"
#include "gtest/gtest.h"
#include "pw_router/egress_function.h"
namespace pw::router {
namespace {
struct BasicPacket {
static constexpr uint32_t kMagic = 0x8badf00d;
constexpr BasicPacket(uint32_t addr, uint64_t data)
: magic(kMagic), address(addr), payload(data) {}
ConstByteSpan data() const { return std::as_bytes(std::span(this, 1)); }
uint32_t magic;
uint32_t address;
uint64_t payload;
};
class BasicPacketParser : public PacketParser {
public:
constexpr BasicPacketParser() : packet_(nullptr) {}
bool Parse(pw::ConstByteSpan packet) final {
packet_ = reinterpret_cast<const BasicPacket*>(packet.data());
return packet_->magic == BasicPacket::kMagic;
}
std::optional<uint32_t> GetDestinationAddress() const final {
PW_DCHECK_NOTNULL(packet_);
return packet_->address;
}
private:
const BasicPacket* packet_;
};
EgressFunction GoodEgress(+[](ConstByteSpan) { return OkStatus(); });
EgressFunction BadEgress(+[](ConstByteSpan) {
return Status::ResourceExhausted();
});
TEST(StaticRouter, RoutePacket_RoutesToAnEgress) {
BasicPacketParser parser;
constexpr StaticRouter::Route routes[] = {{1, GoodEgress}, {2, BadEgress}};
StaticRouter router(parser, std::span(routes));
EXPECT_EQ(router.RoutePacket(BasicPacket(1, 0xdddd).data()), OkStatus());
EXPECT_EQ(router.RoutePacket(BasicPacket(2, 0xdddd).data()),
Status::Unavailable());
}
TEST(StaticRouter, RoutePacket_ReturnsParserError) {
BasicPacketParser parser;
constexpr StaticRouter::Route routes[] = {{1, GoodEgress}, {2, BadEgress}};
StaticRouter router(parser, std::span(routes));
BasicPacket bad_magic(1, 0xdddd);
bad_magic.magic = 0x1badda7a;
EXPECT_EQ(router.RoutePacket(bad_magic.data()), Status::DataLoss());
}
TEST(StaticRouter, RoutePacket_ReturnsNotFoundOnInvalidRoute) {
BasicPacketParser parser;
constexpr StaticRouter::Route routes[] = {{1, GoodEgress}, {2, BadEgress}};
StaticRouter router(parser, std::span(routes));
EXPECT_EQ(router.RoutePacket(BasicPacket(42, 0xdddd).data()),
Status::NotFound());
}
TEST(StaticRouter, RoutePacket_TracksNumberOfDrops) {
BasicPacketParser parser;
constexpr StaticRouter::Route routes[] = {{1, GoodEgress}, {2, BadEgress}};
StaticRouter router(parser, std::span(routes));
// Good
EXPECT_EQ(router.RoutePacket(BasicPacket(1, 0xdddd).data()), OkStatus());
// Egress error
EXPECT_EQ(router.RoutePacket(BasicPacket(2, 0xdddd).data()),
Status::Unavailable());
// Parser error
BasicPacket bad_magic(1, 0xdddd);
bad_magic.magic = 0x1badda7a;
EXPECT_EQ(router.RoutePacket(bad_magic.data()), Status::DataLoss());
// Good
EXPECT_EQ(router.RoutePacket(BasicPacket(1, 0xdddd).data()), OkStatus());
// Bad route
EXPECT_EQ(router.RoutePacket(BasicPacket(42, 0xdddd).data()),
Status::NotFound());
EXPECT_EQ(router.dropped_packets(), 3u);
}
} // namespace
} // namespace pw::router
|
