xref: /aosp_15_r20/external/stg/fuzz/proto_reader_fuzzer.cc (revision 9e3b08ae94a55201065475453d799e8b1378bea6)
1*9e3b08aeSAndroid Build Coastguard Worker // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
2*9e3b08aeSAndroid Build Coastguard Worker // -*- mode: C++ -*-
3*9e3b08aeSAndroid Build Coastguard Worker //
4*9e3b08aeSAndroid Build Coastguard Worker // Copyright 2022 Google LLC
5*9e3b08aeSAndroid Build Coastguard Worker //
6*9e3b08aeSAndroid Build Coastguard Worker // Licensed under the Apache License v2.0 with LLVM Exceptions (the
7*9e3b08aeSAndroid Build Coastguard Worker // "License"); you may not use this file except in compliance with the
8*9e3b08aeSAndroid Build Coastguard Worker // License.  You may obtain a copy of the License at
9*9e3b08aeSAndroid Build Coastguard Worker //
10*9e3b08aeSAndroid Build Coastguard Worker //     https://llvm.org/LICENSE.txt
11*9e3b08aeSAndroid Build Coastguard Worker //
12*9e3b08aeSAndroid Build Coastguard Worker // Unless required by applicable law or agreed to in writing, software
13*9e3b08aeSAndroid Build Coastguard Worker // distributed under the License is distributed on an "AS IS" BASIS,
14*9e3b08aeSAndroid Build Coastguard Worker // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15*9e3b08aeSAndroid Build Coastguard Worker // See the License for the specific language governing permissions and
16*9e3b08aeSAndroid Build Coastguard Worker // limitations under the License.
17*9e3b08aeSAndroid Build Coastguard Worker //
18*9e3b08aeSAndroid Build Coastguard Worker // Author: Matthias Maennich
19*9e3b08aeSAndroid Build Coastguard Worker 
20*9e3b08aeSAndroid Build Coastguard Worker #include <cstddef>
21*9e3b08aeSAndroid Build Coastguard Worker #include <sstream>
22*9e3b08aeSAndroid Build Coastguard Worker #include <string_view>
23*9e3b08aeSAndroid Build Coastguard Worker 
24*9e3b08aeSAndroid Build Coastguard Worker #include "error.h"
25*9e3b08aeSAndroid Build Coastguard Worker #include "graph.h"
26*9e3b08aeSAndroid Build Coastguard Worker #include "proto_reader.h"
27*9e3b08aeSAndroid Build Coastguard Worker #include "runtime.h"
28*9e3b08aeSAndroid Build Coastguard Worker 
LLVMFuzzerTestOneInput(const char * data,size_t size)29*9e3b08aeSAndroid Build Coastguard Worker extern "C" int LLVMFuzzerTestOneInput(const char* data, size_t size) {
30*9e3b08aeSAndroid Build Coastguard Worker   try {
31*9e3b08aeSAndroid Build Coastguard Worker     std::ostringstream os;
32*9e3b08aeSAndroid Build Coastguard Worker     stg::Runtime runtime(os, false);
33*9e3b08aeSAndroid Build Coastguard Worker     stg::Graph graph;
34*9e3b08aeSAndroid Build Coastguard Worker     stg::proto::ReadFromString(runtime, graph, std::string_view(data, size));
35*9e3b08aeSAndroid Build Coastguard Worker   } catch (const stg::Exception&) {
36*9e3b08aeSAndroid Build Coastguard Worker     // Pass as this is us catching invalid proto properly.
37*9e3b08aeSAndroid Build Coastguard Worker   }
38*9e3b08aeSAndroid Build Coastguard Worker   return 0;
39*9e3b08aeSAndroid Build Coastguard Worker }
40