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
|
// Copyright 2017 PDFium 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 "core/fpdfapi/parser/cpdf_read_validator.h"
#include <algorithm>
#include "third_party/base/logging.h"
namespace {
constexpr FX_FILESIZE kAlignBlockValue = 512;
FX_FILESIZE AlignDown(FX_FILESIZE offset) {
return offset > 0 ? (offset - offset % kAlignBlockValue) : 0;
}
FX_FILESIZE AlignUp(FX_FILESIZE offset) {
FX_SAFE_FILESIZE safe_result = AlignDown(offset);
safe_result += kAlignBlockValue;
if (safe_result.IsValid())
return safe_result.ValueOrDie();
return offset;
}
} // namespace
CPDF_ReadValidator::Session::Session(CPDF_ReadValidator* validator)
: validator_(validator) {
ASSERT(validator_);
saved_read_error_ = validator_->read_error_;
saved_has_unavailable_data_ = validator_->has_unavailable_data_;
validator_->ResetErrors();
}
CPDF_ReadValidator::Session::~Session() {
validator_->read_error_ |= saved_read_error_;
validator_->has_unavailable_data_ |= saved_has_unavailable_data_;
}
CPDF_ReadValidator::CPDF_ReadValidator(
const CFX_RetainPtr<IFX_SeekableReadStream>& file_read,
CPDF_DataAvail::FileAvail* file_avail)
: file_read_(file_read),
file_avail_(file_avail),
read_error_(false),
has_unavailable_data_(false) {
ASSERT(file_read_);
}
CPDF_ReadValidator::~CPDF_ReadValidator() {}
void CPDF_ReadValidator::ResetErrors() {
read_error_ = false;
has_unavailable_data_ = false;
}
bool CPDF_ReadValidator::ReadBlock(void* buffer,
FX_FILESIZE offset,
size_t size) {
// correct values checks:
if (!pdfium::base::IsValueInRangeForNumericType<uint32_t>(size))
return false;
FX_SAFE_FILESIZE end_offset = offset;
end_offset += size;
if (!end_offset.IsValid() || end_offset.ValueOrDie() > GetSize())
return false;
if (!file_avail_ ||
file_avail_->IsDataAvail(offset, static_cast<uint32_t>(size))) {
if (file_read_->ReadBlock(buffer, offset, size))
return true;
read_error_ = true;
}
has_unavailable_data_ = true;
ScheduleDownload(offset, size);
return false;
}
FX_FILESIZE CPDF_ReadValidator::GetSize() {
return file_read_->GetSize();
}
void CPDF_ReadValidator::ScheduleDownload(FX_FILESIZE offset, size_t size) {
if (!hints_ || size == 0)
return;
const FX_FILESIZE start_segment_offset = AlignDown(offset);
FX_SAFE_FILESIZE end_segment_offset = offset;
end_segment_offset += size;
if (!end_segment_offset.IsValid()) {
NOTREACHED();
return;
}
end_segment_offset =
std::min(GetSize(), AlignUp(end_segment_offset.ValueOrDie()));
FX_SAFE_UINT32 segment_size = end_segment_offset;
segment_size -= start_segment_offset;
if (!segment_size.IsValid()) {
NOTREACHED();
return;
}
hints_->AddSegment(start_segment_offset, segment_size.ValueOrDie());
}
|
