/src/qpdf/fuzz/runlength_fuzzer.cc
Line | Count | Source |
1 | | #include <qpdf/Pl_Discard.hh> |
2 | | #include <qpdf/Pl_RunLength.hh> |
3 | | #include <iostream> |
4 | | #include <stdexcept> |
5 | | |
6 | | class FuzzHelper |
7 | | { |
8 | | public: |
9 | | FuzzHelper(unsigned char const* data, size_t size); |
10 | | void run(); |
11 | | |
12 | | private: |
13 | | void doChecks(); |
14 | | |
15 | | unsigned char const* data; |
16 | | size_t size; |
17 | | }; |
18 | | |
19 | | FuzzHelper::FuzzHelper(unsigned char const* data, size_t size) : |
20 | 225 | data(data), |
21 | 225 | size(size) |
22 | 225 | { |
23 | 225 | } |
24 | | |
25 | | void |
26 | | FuzzHelper::doChecks() |
27 | 225 | { |
28 | 225 | Pl_RunLength::setMemoryLimit(1'000'000); |
29 | 225 | Pl_Discard discard; |
30 | 225 | Pl_RunLength p("decode", &discard, Pl_RunLength::a_decode); |
31 | 225 | p.write(const_cast<unsigned char*>(data), size); |
32 | 225 | p.finish(); |
33 | 225 | } |
34 | | |
35 | | void |
36 | | FuzzHelper::run() |
37 | 225 | { |
38 | 225 | try { |
39 | 225 | doChecks(); |
40 | 225 | } catch (std::runtime_error const& e) { |
41 | 75 | std::cerr << "runtime_error: " << e.what() << '\n'; |
42 | 75 | } |
43 | 225 | } |
44 | | |
45 | | extern "C" int |
46 | | LLVMFuzzerTestOneInput(unsigned char const* data, size_t size) |
47 | 225 | { |
48 | 225 | FuzzHelper f(data, size); |
49 | 225 | f.run(); |
50 | 225 | return 0; |
51 | 225 | } |