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
|
// Copyright 2011 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include "remoting/protocol/message_decoder.h"
#include <stdint.h>
#include <list>
#include <memory>
#include <string>
#include "base/strings/string_number_conversions.h"
#include "remoting/proto/event.pb.h"
#include "remoting/proto/internal.pb.h"
#include "remoting/protocol/message_serialization.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace remoting::protocol {
static const unsigned int kTestKey = 142;
static void AppendMessage(const EventMessage& msg, std::string* buffer) {
// Contains one encoded message.
scoped_refptr<net::IOBufferWithSize> encoded_msg;
encoded_msg = SerializeAndFrameMessage(msg);
buffer->append(encoded_msg->data(), encoded_msg->size());
}
// Construct and prepare data in the |output_stream|.
static void PrepareData(uint8_t** buffer, int* size) {
// Contains all encoded messages.
std::string encoded_data;
// Then append 10 update sequences to the data.
for (int i = 0; i < 10; ++i) {
EventMessage msg;
msg.set_timestamp(i);
msg.mutable_key_event()->set_usb_keycode(kTestKey + i);
msg.mutable_key_event()->set_pressed((i % 2) != 0);
AppendMessage(msg, &encoded_data);
}
*size = encoded_data.length();
*buffer = new uint8_t[*size];
memcpy(*buffer, encoded_data.c_str(), *size);
}
void SimulateReadSequence(const int read_sequence[], int sequence_size) {
// Prepare encoded data for testing.
int size;
uint8_t* test_data;
PrepareData(&test_data, &size);
std::unique_ptr<uint8_t[]> memory_deleter(test_data);
// Then simulate using MessageDecoder to decode variable
// size of encoded data.
// The first thing to do is to generate a variable size of data. This is done
// by iterating the following array for read sizes.
MessageDecoder decoder;
// Then feed the protocol decoder using the above generated data and the
// read pattern.
std::list<std::unique_ptr<EventMessage>> message_list;
for (int pos = 0; pos < size;) {
SCOPED_TRACE("Input position: " + base::NumberToString(pos));
// First generate the amount to feed the decoder.
int read = std::min(size - pos, read_sequence[pos % sequence_size]);
// And then prepare an IOBuffer for feeding it.
auto buffer = base::MakeRefCounted<net::IOBufferWithSize>(read);
memcpy(buffer->data(), test_data + pos, read);
decoder.AddData(buffer, read);
while (true) {
std::unique_ptr<CompoundBuffer> message(decoder.GetNextMessage());
if (!message.get()) {
break;
}
std::unique_ptr<EventMessage> event = std::make_unique<EventMessage>();
CompoundBufferInputStream stream(message.get());
ASSERT_TRUE(event->ParseFromZeroCopyStream(&stream));
message_list.push_back(std::move(event));
}
pos += read;
}
// Then verify the decoded messages.
EXPECT_EQ(10u, message_list.size());
unsigned int index = 0;
for (const auto& message : message_list) {
SCOPED_TRACE("Message " + base::NumberToString(index));
// Partial update stream.
EXPECT_TRUE(message->has_key_event());
// TODO(sergeyu): Don't use index here. Instead store the expected values
// in an array.
EXPECT_EQ(kTestKey + index, message->key_event().usb_keycode());
EXPECT_EQ((index % 2) != 0, message->key_event().pressed());
++index;
}
}
TEST(MessageDecoderTest, SmallReads) {
const int kReads[] = {1, 2, 3, 1};
SimulateReadSequence(kReads, std::size(kReads));
}
TEST(MessageDecoderTest, LargeReads) {
const int kReads[] = {50, 50, 5};
SimulateReadSequence(kReads, std::size(kReads));
}
TEST(MessageDecoderTest, EmptyReads) {
const int kReads[] = {4, 0, 50, 0};
SimulateReadSequence(kReads, std::size(kReads));
}
} // namespace remoting::protocol
|
