/src/zopfli_compress_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 | | |
15 | | #include <cstddef> |
16 | | #include <cstdint> |
17 | | #include <cstdlib> |
18 | | #include <string> |
19 | | |
20 | | #include <fuzzer/FuzzedDataProvider.h> |
21 | | |
22 | | #include "zopfli.h" |
23 | | |
24 | 2.66k | extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { |
25 | 2.66k | if (size > 8192) |
26 | 9 | return 0; |
27 | | |
28 | 2.65k | FuzzedDataProvider stream(data, size); |
29 | | |
30 | 2.65k | ZopfliOptions options; |
31 | 2.65k | ZopfliInitOptions(&options); |
32 | | |
33 | 2.65k | const ZopfliFormat format = stream.PickValueInArray( |
34 | 2.65k | {ZOPFLI_FORMAT_GZIP, ZOPFLI_FORMAT_ZLIB, ZOPFLI_FORMAT_DEFLATE}); |
35 | | |
36 | 2.65k | unsigned char* outbuf = nullptr; |
37 | 2.65k | size_t outsize = 0; |
38 | 2.65k | std::vector<unsigned char> input = |
39 | 2.65k | stream.ConsumeRemainingBytes<unsigned char>(); |
40 | | |
41 | 2.65k | ZopfliCompress(&options, format, input.data(), input.size(), &outbuf, |
42 | 2.65k | &outsize); |
43 | | |
44 | 2.65k | if (outbuf != nullptr) { |
45 | 2.65k | free(outbuf); |
46 | 2.65k | } |
47 | | |
48 | 2.65k | return 0; |
49 | 2.66k | } |