cerc-io/solidity
10
0
Fork 0
mirror of https://github.com/ethereum/solidity synced 2023-10-03 13:03:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add/remove function call in another function's body

Browse Source
This commit is contained in:
Bhargava Shastry 2020-01-07 15:40:46 +01:00
parent c9da41743e
commit 98ce79dd0c
1 changed files with 52 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
test/tools/ossfuzz/protomutators/YulProtoMutator.cpp
View File

@ -823,7 +823,6 @@ static YulProtoMutator addFuncDef(
std::cout << "YULMUTATOR: Add function def" << std::endl;
#endif
auto block = static_cast<Block*>(_message);
auto stmt = block->add_statements();
auto funcDef = new FunctionDef();
funcDef->set_num_input_params(_seed);
funcDef->set_num_output_params(_seed + block->ByteSizeLong());
@ -831,6 +830,7 @@ static YulProtoMutator addFuncDef(
auto funcBlock = new Block();
funcBlock->CopyFrom(*block);
funcDef->set_allocated_block(funcBlock);
auto stmt = block->add_statements();
stmt->set_allocated_funcdef(funcDef);
#ifdef DEBUG
std::cout << protobuf_mutator::SaveMessageAsText(*_message) << std::endl;
@ -1224,6 +1224,57 @@ static YulProtoMutator addPopUserFunction(
}
);
/// Add function call in another function's body
static YulProtoMutator addFuncCallInFuncBody(
FunctionDef::descriptor(),
[](google::protobuf::Message* _message, unsigned int _seed)
{
if (_seed % YulProtoMutator::s_mediumIP == 0)
{
#ifdef DEBUG
std::cout << "----------------------------------" << std::endl;
std::cout << protobuf_mutator::SaveMessageAsText(*_message) << std::endl;
std::cout << "YULMUTATOR: Add function call in func body" << std::endl;
#endif
auto functioncall = new FunctionCall();
YulProtoMutator::configureCall(functioncall, _seed);
auto block = static_cast<FunctionDef*>(_message)->mutable_block();
auto stmt = block->add_statements();
stmt->set_allocated_functioncall(functioncall);
#ifdef DEBUG
std::cout << protobuf_mutator::SaveMessageAsText(*_message) << std::endl;
std::cout << "----------------------------------" << std::endl;
#endif
}
}
);
/// Remove function call from another function's body
static YulProtoMutator removeFuncCallInFuncBody(
FunctionDef::descriptor(),
[](google::protobuf::Message* _message, unsigned int _seed)
{
if (_seed % YulProtoMutator::s_mediumIP == 1)
{
#ifdef DEBUG
std::cout << "----------------------------------" << std::endl;
std::cout << protobuf_mutator::SaveMessageAsText(*_message) << std::endl;
std::cout << "YULMUTATOR: Remove function call in func body" << std::endl;
#endif
for (auto &stmt: *static_cast<FunctionDef*>(_message)->mutable_block()->mutable_statements())
if (stmt.has_functioncall())
{
stmt.clear_functioncall();
break;
}
#ifdef DEBUG
std::cout << protobuf_mutator::SaveMessageAsText(*_message) << std::endl;
std::cout << "----------------------------------" << std::endl;
#endif
}
}
);
Literal* YulProtoMutator::intLiteral(unsigned _value)
{
auto lit = new Literal();