xref: /aosp_15_r20/external/flatbuffers/tests/fuzzer/flatbuffers_parser_fuzzer.cc (revision 890232f25432b36107d06881e0a25aaa6b473652)
1*890232f2SAndroid Build Coastguard Worker // Copyright 2015 The Chromium Authors. All rights reserved.
2*890232f2SAndroid Build Coastguard Worker // Use of this source code is governed by a BSD-style license that can be
3*890232f2SAndroid Build Coastguard Worker // found in the LICENSE file.
4*890232f2SAndroid Build Coastguard Worker #include <stddef.h>
5*890232f2SAndroid Build Coastguard Worker #include <stdint.h>
6*890232f2SAndroid Build Coastguard Worker #include <clocale>
7*890232f2SAndroid Build Coastguard Worker #include <string>
8*890232f2SAndroid Build Coastguard Worker 
9*890232f2SAndroid Build Coastguard Worker #include "flatbuffers/idl.h"
10*890232f2SAndroid Build Coastguard Worker #include "test_init.h"
11*890232f2SAndroid Build Coastguard Worker 
12*890232f2SAndroid Build Coastguard Worker static constexpr size_t kMinInputLength = 1;
13*890232f2SAndroid Build Coastguard Worker static constexpr size_t kMaxInputLength = 16384;
14*890232f2SAndroid Build Coastguard Worker 
15*890232f2SAndroid Build Coastguard Worker static constexpr uint8_t flags_strict_json = 0x80;
16*890232f2SAndroid Build Coastguard Worker static constexpr uint8_t flags_skip_unexpected_fields_in_json = 0x40;
17*890232f2SAndroid Build Coastguard Worker static constexpr uint8_t flags_allow_non_utf8 = 0x20;
18*890232f2SAndroid Build Coastguard Worker 
19*890232f2SAndroid Build Coastguard Worker // Utility for test run.
20*890232f2SAndroid Build Coastguard Worker OneTimeTestInit OneTimeTestInit::one_time_init_;
21*890232f2SAndroid Build Coastguard Worker 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)22*890232f2SAndroid Build Coastguard Worker extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
23*890232f2SAndroid Build Coastguard Worker   // Reserve one byte for Parser flags and one byte for repetition counter.
24*890232f2SAndroid Build Coastguard Worker   if (size < 3) return 0;
25*890232f2SAndroid Build Coastguard Worker   const uint8_t flags = data[0];
26*890232f2SAndroid Build Coastguard Worker   (void)data[1];  //  reserved
27*890232f2SAndroid Build Coastguard Worker   data += 2;
28*890232f2SAndroid Build Coastguard Worker   size -= 2;  // bypass
29*890232f2SAndroid Build Coastguard Worker 
30*890232f2SAndroid Build Coastguard Worker   const std::string original(reinterpret_cast<const char *>(data), size);
31*890232f2SAndroid Build Coastguard Worker   auto input = std::string(original.c_str());  // until '\0'
32*890232f2SAndroid Build Coastguard Worker   if (input.size() < kMinInputLength || input.size() > kMaxInputLength)
33*890232f2SAndroid Build Coastguard Worker     return 0;
34*890232f2SAndroid Build Coastguard Worker 
35*890232f2SAndroid Build Coastguard Worker   flatbuffers::IDLOptions opts;
36*890232f2SAndroid Build Coastguard Worker   opts.strict_json = (flags & flags_strict_json);
37*890232f2SAndroid Build Coastguard Worker   opts.skip_unexpected_fields_in_json =
38*890232f2SAndroid Build Coastguard Worker       (flags & flags_skip_unexpected_fields_in_json);
39*890232f2SAndroid Build Coastguard Worker   opts.allow_non_utf8 = (flags & flags_allow_non_utf8);
40*890232f2SAndroid Build Coastguard Worker 
41*890232f2SAndroid Build Coastguard Worker   flatbuffers::Parser parser(opts);
42*890232f2SAndroid Build Coastguard Worker 
43*890232f2SAndroid Build Coastguard Worker   // Guarantee 0-termination in the input.
44*890232f2SAndroid Build Coastguard Worker   auto parse_input = input.c_str();
45*890232f2SAndroid Build Coastguard Worker 
46*890232f2SAndroid Build Coastguard Worker   // Check Parser.
47*890232f2SAndroid Build Coastguard Worker   parser.Parse(parse_input);
48*890232f2SAndroid Build Coastguard Worker   // TODO:
49*890232f2SAndroid Build Coastguard Worker   // Need to add additional checks for inputs passed Parse(parse_input) successfully:
50*890232f2SAndroid Build Coastguard Worker   // 1. Serialization to bfbs.
51*890232f2SAndroid Build Coastguard Worker   // 2. Generation of a default object.
52*890232f2SAndroid Build Coastguard Worker   // 3. Verification of the object using reflection.
53*890232f2SAndroid Build Coastguard Worker   // 3. Printing to json.
54*890232f2SAndroid Build Coastguard Worker   return 0;
55*890232f2SAndroid Build Coastguard Worker }
56