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
|
// Copyright (c) 2016 The WebM project authors. All Rights Reserved.
//
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file in the root of the source
// tree. An additional intellectual property rights grant can be found
// in the file PATENTS. All contributing project authors may
// be found in the AUTHORS file in the root of the source tree.
#include "src/parser_utils.h"
#include <cstdint>
#include <limits>
#include <memory>
#include "gtest/gtest.h"
#include "test_utils/limited_reader.h"
#include "webm/buffer_reader.h"
#include "webm/reader.h"
#include "webm/status.h"
using webm::BufferReader;
using webm::LimitedReader;
using webm::Reader;
using webm::Status;
namespace {
class ParserUtilsTest : public testing::Test {};
TEST_F(ParserUtilsTest, ReadByte) {
LimitedReader reader(
std::unique_ptr<Reader>(new BufferReader({0x12, 0x34, 0x56, 0x78})));
Status status;
std::uint8_t byte;
status = ReadByte(&reader, &byte);
EXPECT_EQ(Status::kOkCompleted, status.code);
EXPECT_EQ(0x12, byte);
status = ReadByte(&reader, &byte);
EXPECT_EQ(Status::kOkCompleted, status.code);
EXPECT_EQ(0x34, byte);
status = ReadByte(&reader, &byte);
EXPECT_EQ(Status::kOkCompleted, status.code);
EXPECT_EQ(0x56, byte);
reader.set_total_read_limit(0);
status = ReadByte(&reader, &byte);
EXPECT_EQ(Status::kWouldBlock, status.code);
EXPECT_EQ(0x56, byte);
reader.set_total_read_limit(std::numeric_limits<std::size_t>::max());
status = ReadByte(&reader, &byte);
EXPECT_EQ(Status::kOkCompleted, status.code);
EXPECT_EQ(0x78, byte);
status = ReadByte(&reader, &byte);
EXPECT_EQ(Status::kEndOfFile, status.code);
EXPECT_EQ(0x78, byte);
}
TEST_F(ParserUtilsTest, AccumulateIntegerBytes) {
LimitedReader reader(
std::unique_ptr<Reader>(new BufferReader({0x12, 0x34, 0x56, 0x78})));
std::uint32_t integer = 0;
Status status;
std::uint64_t num_bytes_actually_read;
reader.set_total_read_limit(1);
status =
AccumulateIntegerBytes(4, &reader, &integer, &num_bytes_actually_read);
EXPECT_EQ(Status::kWouldBlock, status.code);
EXPECT_EQ(static_cast<std::uint64_t>(1), num_bytes_actually_read);
EXPECT_EQ(static_cast<std::uint32_t>(0x12), integer);
reader.set_total_read_limit(std::numeric_limits<std::size_t>::max());
status =
AccumulateIntegerBytes(3, &reader, &integer, &num_bytes_actually_read);
EXPECT_EQ(Status::kOkCompleted, status.code);
EXPECT_EQ(static_cast<std::uint64_t>(3), num_bytes_actually_read);
EXPECT_EQ(static_cast<std::uint32_t>(0x12345678), integer);
// Make sure calling with num_bytes_remaining == 0 is a no-op.
status =
AccumulateIntegerBytes(0, &reader, &integer, &num_bytes_actually_read);
EXPECT_EQ(Status::kOkCompleted, status.code);
EXPECT_EQ(static_cast<std::uint64_t>(0), num_bytes_actually_read);
EXPECT_EQ(static_cast<std::uint32_t>(0x12345678), integer);
status =
AccumulateIntegerBytes(4, &reader, &integer, &num_bytes_actually_read);
EXPECT_EQ(Status::kEndOfFile, status.code);
EXPECT_EQ(static_cast<std::uint64_t>(0), num_bytes_actually_read);
EXPECT_EQ(static_cast<std::uint32_t>(0x12345678), integer);
}
} // namespace
|
