mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.1 KiB
73 lines
2.1 KiB
3 years ago
|
// Copyright 2021 gRPC authors.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
#include "src/core/lib/avl/avl.h"
|
||
|
|
#include "src/libfuzzer/libfuzzer_macro.h"
|
||
|
|
#include "test/core/avl/avl_fuzzer.pb.h"
|
||
|
|
|
||
|
|
bool squelch = true;
|
||
|
|
bool leak_check = true;
|
||
|
|
|
||
|
|
namespace grpc_core {
|
||
|
|
|
||
|
|
class Fuzzer {
|
||
|
|
public:
|
||
|
|
void Run(const avl_fuzzer::Msg& msg) {
|
||
|
|
CheckEqual();
|
||
|
|
for (const auto& action : msg.actions()) {
|
||
|
|
switch (action.action_case()) {
|
||
|
|
case avl_fuzzer::Action::kSet:
|
||
|
|
avl_ = avl_.Add(action.key(), action.set());
|
||
|
|
map_[action.key()] = action.set();
|
||
|
|
break;
|
||
|
|
case avl_fuzzer::Action::kDel:
|
||
|
|
avl_ = avl_.Remove(action.key());
|
||
|
|
map_.erase(action.key());
|
||
|
|
break;
|
||
|
|
case avl_fuzzer::Action::kGet: {
|
||
|
|
auto* p = avl_.Lookup(action.key());
|
||
|
|
auto it = map_.find(action.key());
|
||
|
|
if (it == map_.end() && p != nullptr) abort();
|
||
|
|
if (it != map_.end() && p == nullptr) abort();
|
||
|
|
if (it != map_.end() && it->second != *p) abort();
|
||
|
|
} break;
|
||
|
|
case avl_fuzzer::Action::ACTION_NOT_SET:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
CheckEqual();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
void CheckEqual() {
|
||
|
|
auto it = map_.begin();
|
||
|
|
avl_.ForEach([&](int key, int value) {
|
||
|
|
if (it == map_.end()) abort();
|
||
|
|
if (it->first != key) abort();
|
||
|
|
if (it->second != value) abort();
|
||
|
|
++it;
|
||
|
|
});
|
||
|
|
if (it != map_.end()) abort();
|
||
|
|
}
|
||
|
|
|
||
|
|
AVL<int, int> avl_;
|
||
|
|
std::map<int, int> map_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace grpc_core
|
||
|
|
|
||
|
|
DEFINE_PROTO_FUZZER(const avl_fuzzer::Msg& msg) {
|
||
|
|
grpc_core::Fuzzer().Run(msg);
|
||
|
|
}
|