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
|
// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "discovery/mdns/mdns_receiver.h"
#include <memory>
#include <utility>
#include <vector>
#include "discovery/common/config.h"
#include "discovery/mdns/mdns_records.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "platform/api/time.h"
#include "platform/test/fake_udp_socket.h"
namespace openscreen {
namespace discovery {
using testing::_;
using testing::Return;
class MockMdnsReceiverDelegate : public MdnsReceiver::ResponseClient {
public:
MOCK_METHOD(void, OnMessageReceived, (const MdnsMessage&));
};
TEST(MdnsReceiverTest, ReceiveQuery) {
// clang-format off
const std::vector<uint8_t> kQueryBytes = {
0x00, 0x01, // ID = 1
0x00, 0x00, // FLAGS = None
0x00, 0x01, // Question count
0x00, 0x00, // Answer count
0x00, 0x00, // Authority count
0x00, 0x00, // Additional count
// Question
0x07, 't', 'e', 's', 't', 'i', 'n', 'g',
0x05, 'l', 'o', 'c', 'a', 'l',
0x00,
0x00, 0x01, // TYPE = A (1)
0x00, 0x01, // CLASS = IN (1)
};
// clang-format on
Config config;
FakeUdpSocket socket;
MockMdnsReceiverDelegate delegate;
MdnsReceiver receiver(config);
receiver.SetQueryCallback(
[&delegate](const MdnsMessage& message, const IPEndpoint& endpoint) {
delegate.OnMessageReceived(message);
});
receiver.Start();
MdnsQuestion question(DomainName{"testing", "local"}, DnsType::kA,
DnsClass::kIN, ResponseType::kMulticast);
MdnsMessage message(1, MessageType::Query);
message.AddQuestion(question);
UdpPacket packet(kQueryBytes.data(), kQueryBytes.data() + kQueryBytes.size());
packet.set_source(
IPEndpoint{.address = IPAddress(192, 168, 1, 1), .port = 31337});
packet.set_destination(
IPEndpoint{.address = IPAddress(kDefaultMulticastGroupIPv4),
.port = kDefaultMulticastPort});
// Imitate a call to OnRead from NetworkRunner by calling it manually here
EXPECT_CALL(delegate, OnMessageReceived(message)).Times(1);
receiver.OnRead(&socket, std::move(packet));
receiver.Stop();
}
TEST(MdnsReceiverTest, ReceiveResponse) {
// clang-format off
const std::vector<uint8_t> kResponseBytes = {
0x00, 0x01, // ID = 1
0x84, 0x00, // FLAGS = AA | RESPONSE
0x00, 0x00, // Question count
0x00, 0x01, // Answer count
0x00, 0x00, // Authority count
0x00, 0x00, // Additional count
// Answer
0x07, 't', 'e', 's', 't', 'i', 'n', 'g',
0x05, 'l', 'o', 'c', 'a', 'l',
0x00,
0x00, 0x01, // TYPE = A (1)
0x00, 0x01, // CLASS = IN (1)
0x00, 0x00, 0x00, 0x78, // TTL = 120 seconds
0x00, 0x04, // RDLENGTH = 4 bytes
0xac, 0x00, 0x00, 0x01, // 172.0.0.1
};
constexpr uint16_t kIPv6AddressHextets[] = {
0xfe80, 0x0000, 0x0000, 0x0000,
0x0202, 0xb3ff, 0xfe1e, 0x8329,
};
// clang-format on
Config config;
FakeUdpSocket socket;
MockMdnsReceiverDelegate delegate;
MdnsReceiver receiver(config);
receiver.AddResponseCallback(&delegate);
receiver.Start();
MdnsRecord record(DomainName{"testing", "local"}, DnsType::kA, DnsClass::kIN,
RecordType::kShared, std::chrono::seconds(120),
ARecordRdata(IPAddress{172, 0, 0, 1}));
MdnsMessage message(1, MessageType::Response);
message.AddAnswer(record);
UdpPacket packet(kResponseBytes.size());
packet.assign(kResponseBytes.data(),
kResponseBytes.data() + kResponseBytes.size());
packet.set_source(
IPEndpoint{.address = IPAddress(kIPv6AddressHextets), .port = 31337});
packet.set_destination(
IPEndpoint{.address = IPAddress(kDefaultMulticastGroupIPv6),
.port = kDefaultMulticastPort});
// Imitate a call to OnRead from NetworkRunner by calling it manually here
EXPECT_CALL(delegate, OnMessageReceived(message)).Times(1);
receiver.OnRead(&socket, std::move(packet));
receiver.Stop();
receiver.RemoveResponseCallback(&delegate);
}
} // namespace discovery
} // namespace openscreen
|
