blob: cf06f8cb22814cc7f746c98077f39cee40426491 (
plain)
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
|
// Copyright (c) the JPEG XL 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.
#ifndef LIB_JXL_SIZE_CONSTRAINTS_H_
#define LIB_JXL_SIZE_CONSTRAINTS_H_
#include <cstdint>
#include <type_traits>
#include "lib/jxl/base/status.h"
namespace jxl {
struct SizeConstraints {
// Upper limit on pixel dimensions/area, enforced by VerifyDimensions
// (called from decoders). Fuzzers set smaller values to limit memory use.
uint32_t dec_max_xsize = 0xFFFFFFFFu;
uint32_t dec_max_ysize = 0xFFFFFFFFu;
uint64_t dec_max_pixels = 0xFFFFFFFFu; // Might be up to ~0ull
};
template <typename T,
class = typename std::enable_if<std::is_unsigned<T>::value>::type>
Status VerifyDimensions(const SizeConstraints* constraints, T xs, T ys) {
if (!constraints) return true;
if (xs == 0 || ys == 0) return JXL_FAILURE("Empty image.");
if (xs > constraints->dec_max_xsize) return JXL_FAILURE("Image too wide.");
if (ys > constraints->dec_max_ysize) return JXL_FAILURE("Image too tall.");
const uint64_t num_pixels = static_cast<uint64_t>(xs) * ys;
if (num_pixels > constraints->dec_max_pixels) {
return JXL_FAILURE("Image too big.");
}
return true;
}
} // namespace jxl
#endif // LIB_JXL_SIZE_CONSTRAINTS_H_
|
