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 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)21 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
22   ZopfliOptions options;
23   ZopfliInitOptions(&options);
24 
25   FuzzedDataProvider stream(data, size);
26 
27   // From documentation: valid values for btype are 0, 1, or 2.
28   const int btype = stream.PickValueInArray({0, 1, 2});
29   // The final parameter is an int but it is used as a bool.
30   const int is_final = stream.ConsumeIntegralInRange(0, 1);
31   const std::string input = stream.ConsumeRemainingBytesAsString();
32 
33   unsigned char* out = nullptr;
34   size_t outsize = 0;
35   unsigned char bp = 0;  // Apparently must be zero.
36   ZopfliDeflate(&options, btype, is_final,
37                 reinterpret_cast<const unsigned char*>(input.data()),
38                 input.size(), &bp, &out, &outsize);
39 
40   if (out != nullptr) {
41     free(out);
42   }
43 
44   return 0;
45 }
46