/src/qpdf/libqpdf/SF_FlateLzwDecode.cc
Line | Count | Source |
1 | | #include <qpdf/SF_FlateLzwDecode.hh> |
2 | | |
3 | | #include <qpdf/Pl_Flate.hh> |
4 | | #include <qpdf/Pl_LZWDecoder.hh> |
5 | | #include <qpdf/Pl_PNGFilter.hh> |
6 | | #include <qpdf/Pl_TIFFPredictor.hh> |
7 | | #include <qpdf/QIntC.hh> |
8 | | #include <qpdf/QPDFObjectHandle_private.hh> |
9 | | #include <qpdf/QTC.hh> |
10 | | |
11 | | bool |
12 | | SF_FlateLzwDecode::setDecodeParms(QPDFObjectHandle decode_parms) |
13 | 15.2k | { |
14 | 15.2k | if (decode_parms.null()) { |
15 | 11.1k | return true; |
16 | 11.1k | } |
17 | | |
18 | 4.14k | auto memory_limit = Pl_Flate::memory_limit(); |
19 | | |
20 | 4.14k | std::set<std::string> keys = decode_parms.getKeys(); |
21 | 9.67k | for (auto const& key: keys) { |
22 | 9.67k | QPDFObjectHandle value = decode_parms.getKey(key); |
23 | 9.67k | if (key == "/Predictor") { |
24 | 2.54k | if (value.isInteger()) { |
25 | 2.53k | predictor = value.getIntValueAsInt(); |
26 | 2.53k | if (!(predictor == 1 || predictor == 2 || (predictor >= 10 && predictor <= 15))) { |
27 | 63 | return false; |
28 | 63 | } |
29 | 2.53k | } else { |
30 | 12 | return false; |
31 | 12 | } |
32 | 7.12k | } else if (key == "/Columns" || key == "/Colors" || key == "/BitsPerComponent") { |
33 | 2.52k | if (value.isInteger()) { |
34 | 2.51k | int val = value.getIntValueAsInt(); |
35 | 2.51k | if (memory_limit && static_cast<unsigned int>(val) > memory_limit) { |
36 | 17 | QPDFLogger::defaultLogger()->warn( |
37 | 17 | "SF_FlateLzwDecode parameter exceeds PL_Flate memory limit\n"); |
38 | 17 | return false; |
39 | 17 | } |
40 | 2.49k | if (key == "/Columns") { |
41 | 1.43k | columns = val; |
42 | 1.43k | } else if (key == "/Colors") { |
43 | 687 | colors = val; |
44 | 687 | } else if (key == "/BitsPerComponent") { |
45 | 376 | bits_per_component = val; |
46 | 376 | } |
47 | 2.49k | } else { |
48 | 10 | return false; |
49 | 10 | } |
50 | 4.60k | } else if (lzw && (key == "/EarlyChange")) { |
51 | 175 | if (value.isInteger()) { |
52 | 169 | int earlychange = value.getIntValueAsInt(); |
53 | 169 | early_code_change = (earlychange == 1); |
54 | 169 | if (!(earlychange == 0 || earlychange == 1)) { |
55 | 36 | return false; |
56 | 36 | } |
57 | 169 | } else { |
58 | 6 | return false; |
59 | 6 | } |
60 | 175 | } |
61 | 9.67k | } |
62 | | |
63 | 4.00k | if (predictor > 1 && columns == 0) { |
64 | 3 | return false; |
65 | 3 | } |
66 | | |
67 | 3.99k | return true; |
68 | 4.00k | } |
69 | | |
70 | | Pipeline* |
71 | | SF_FlateLzwDecode::getDecodePipeline(Pipeline* next) |
72 | 15.1k | { |
73 | 15.1k | std::unique_ptr<Pipeline> pipeline; |
74 | 15.1k | if (predictor >= 10 && predictor <= 15) { |
75 | 1.54k | QTC::TC("qpdf", "SF_FlateLzwDecode PNG filter"); |
76 | 1.54k | pipeline = std::make_unique<Pl_PNGFilter>( |
77 | 1.54k | "png decode", |
78 | 1.54k | next, |
79 | 1.54k | Pl_PNGFilter::a_decode, |
80 | 1.54k | QIntC::to_uint(columns), |
81 | 1.54k | QIntC::to_uint(colors), |
82 | 1.54k | QIntC::to_uint(bits_per_component)); |
83 | 1.54k | next = pipeline.get(); |
84 | 1.54k | pipelines.push_back(std::move(pipeline)); |
85 | 13.5k | } else if (predictor == 2) { |
86 | 831 | QTC::TC("qpdf", "SF_FlateLzwDecode TIFF predictor"); |
87 | 831 | pipeline = std::make_unique<Pl_TIFFPredictor>( |
88 | 831 | "tiff decode", |
89 | 831 | next, |
90 | 831 | Pl_TIFFPredictor::a_decode, |
91 | 831 | QIntC::to_uint(columns), |
92 | 831 | QIntC::to_uint(colors), |
93 | 831 | QIntC::to_uint(bits_per_component)); |
94 | 831 | next = pipeline.get(); |
95 | 831 | pipelines.push_back(std::move(pipeline)); |
96 | 831 | } |
97 | | |
98 | 15.1k | if (lzw) { |
99 | 3.28k | pipeline = std::make_unique<Pl_LZWDecoder>("lzw decode", next, early_code_change); |
100 | 11.8k | } else { |
101 | 11.8k | pipeline = std::make_unique<Pl_Flate>("stream inflate", next, Pl_Flate::a_inflate); |
102 | 11.8k | } |
103 | 15.1k | next = pipeline.get(); |
104 | 15.1k | pipelines.push_back(std::move(pipeline)); |
105 | 15.1k | return next; |
106 | 15.1k | } |