/src/zopfli_deflate_fuzzer.cc
Line | Count | Source |
1 | | // Copyright 2019 Google LLC |
2 | | // |
3 | | // Licensed under the Apache License, Version 2.0 (the "License"); |
4 | | // you may not use this file except in compliance with the License. |
5 | | // You may obtain a copy of the License at |
6 | | // |
7 | | // http://www.apache.org/licenses/LICENSE-2.0 |
8 | | // |
9 | | // Unless required by applicable law or agreed to in writing, software |
10 | | // distributed under the License is distributed on an "AS IS" BASIS, |
11 | | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | | // See the License for the specific language governing permissions and |
13 | | // limitations under the License. |
14 | | #include <fuzzer/FuzzedDataProvider.h> |
15 | | |
16 | | #include <string> |
17 | | |
18 | | #include "deflate.h" |
19 | | #include "zopfli.h" |
20 | | |
21 | 2.97k | extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { |
22 | 2.97k | ZopfliOptions options; |
23 | 2.97k | ZopfliInitOptions(&options); |
24 | | |
25 | 2.97k | FuzzedDataProvider stream(data, size); |
26 | | |
27 | | // From documentation: valid values for btype are 0, 1, or 2. |
28 | 2.97k | const int btype = stream.PickValueInArray({0, 1, 2}); |
29 | | // The final parameter is an int but it is used as a bool. |
30 | 2.97k | const int is_final = stream.ConsumeIntegralInRange(0, 1); |
31 | 2.97k | const std::string input = stream.ConsumeRemainingBytesAsString(); |
32 | | |
33 | 2.97k | unsigned char* out = nullptr; |
34 | 2.97k | size_t outsize = 0; |
35 | 2.97k | unsigned char bp = 0; // Apparently must be zero. |
36 | 2.97k | ZopfliDeflate(&options, btype, is_final, |
37 | 2.97k | reinterpret_cast<const unsigned char*>(input.data()), |
38 | 2.97k | input.size(), &bp, &out, &outsize); |
39 | | |
40 | 2.97k | if (out != nullptr) { |
41 | 2.97k | free(out); |
42 | 2.97k | } |
43 | | |
44 | 2.97k | return 0; |
45 | 2.97k | } |