/src/brpc/test/fuzzing/fuzz_shead.cpp
Line  | Count  | Source  | 
1  |  | // Licensed to the Apache Software Foundation (ASF) under one  | 
2  |  | // or more contributor license agreements.  See the NOTICE file  | 
3  |  | // distributed with this work for additional information  | 
4  |  | // regarding copyright ownership.  The ASF licenses this file  | 
5  |  | // to you under the Apache License, Version 2.0 (the  | 
6  |  | // "License"); you may not use this file except in compliance  | 
7  |  | // with the License.  You may obtain a copy of the License at  | 
8  |  | //  | 
9  |  | //   http://www.apache.org/licenses/LICENSE-2.0  | 
10  |  | //  | 
11  |  | // Unless required by applicable law or agreed to in writing,  | 
12  |  | // software distributed under the License is distributed on an  | 
13  |  | // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY  | 
14  |  | // KIND, either express or implied.  See the License for the  | 
15  |  | // specific language governing permissions and limitations  | 
16  |  | // under the License.  | 
17  |  | #include "brpc/policy/public_pbrpc_meta.pb.h"  | 
18  |  | #include "brpc/policy/public_pbrpc_protocol.h"  | 
19  |  | #include "brpc/policy/most_common_message.h"  | 
20  |  |  | 
21  | 556  | #define kMinInputLength 5  | 
22  | 274  | #define kMaxInputLength 1024  | 
23  |  |  | 
24  |  | extern "C" int  | 
25  |  | LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)  | 
26  | 278  | { | 
27  | 278  |     if (size < kMinInputLength || size > kMaxInputLength){ | 
28  | 28  |         return 1;  | 
29  | 28  |     }  | 
30  |  |  | 
31  | 250  |     std::string input(reinterpret_cast<const char*>(data), size);  | 
32  | 250  |     butil::IOBuf buf;  | 
33  | 250  |     buf.append(input);  | 
34  |  |  | 
35  | 250  |     brpc::policy::ParseNsheadMessage(&buf, NULL, false, NULL);  | 
36  |  |  | 
37  | 250  |     return 0;  | 
38  | 278  | }  |