solidity/test/tools/ossfuzz/SolProtoAdaptor.cpp
Bhargava Shastry 92590a13f8 WIP
2020-04-16 01:02:48 +02:00

1549 lines
42 KiB
C++

/*
This file is part of solidity.
solidity is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
solidity is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with solidity. If not, see <http://www.gnu.org/licenses/>.
*/
#include <test/tools/ossfuzz/SolProtoAdaptor.h>
#include <liblangutil/Exceptions.h>
#include <libsolutil/Whiskers.h>
#include <sstream>
using namespace solidity::test::solprotofuzzer::adaptor;
using namespace solidity::test::solprotofuzzer;
using namespace std;
using namespace solidity::util;
namespace
{
SolFunctionStateMutability mutabilityConverter(InterfaceFunction_StateMutability _mut)
{
switch (_mut)
{
case InterfaceFunction_StateMutability_PURE:
return SolFunctionStateMutability::PURE;
case InterfaceFunction_StateMutability_VIEW:
return SolFunctionStateMutability::VIEW;
case InterfaceFunction_StateMutability_PAYABLE:
return SolFunctionStateMutability::PAYABLE;
}
}
SolFunctionStateMutability mutabilityConverter(ContractFunction_StateMutability _mut)
{
switch (_mut)
{
case ContractFunction_StateMutability_PURE:
return SolFunctionStateMutability::PURE;
case ContractFunction_StateMutability_VIEW:
return SolFunctionStateMutability::VIEW;
case ContractFunction_StateMutability_PAYABLE:
return SolFunctionStateMutability::PAYABLE;
}
}
SolLibraryFunctionStateMutability mutabilityConverter(LibraryFunction_StateMutability _mut)
{
switch (_mut)
{
case LibraryFunction_StateMutability_PURE:
return SolLibraryFunctionStateMutability::PURE;
case LibraryFunction_StateMutability_VIEW:
return SolLibraryFunctionStateMutability::VIEW;
}
}
SolFunctionVisibility visibilityConverter(ContractFunction_Visibility _vis)
{
switch (_vis)
{
case ContractFunction_Visibility_PUBLIC:
return SolFunctionVisibility::PUBLIC;
case ContractFunction_Visibility_PRIVATE:
return SolFunctionVisibility::PRIVATE;
case ContractFunction_Visibility_EXTERNAL:
return SolFunctionVisibility::EXTERNAL;
case ContractFunction_Visibility_INTERNAL:
return SolFunctionVisibility::INTERNAL;
}
}
SolFunctionVisibility visibilityConverter(LibraryFunction_Visibility _vis)
{
switch (_vis)
{
case LibraryFunction_Visibility_PUBLIC:
return SolFunctionVisibility::PUBLIC;
case LibraryFunction_Visibility_PRIVATE:
return SolFunctionVisibility::PRIVATE;
case LibraryFunction_Visibility_EXTERNAL:
return SolFunctionVisibility::EXTERNAL;
case LibraryFunction_Visibility_INTERNAL:
return SolFunctionVisibility::INTERNAL;
}
}
string functionVisibility(SolFunctionVisibility _vis)
{
switch (_vis)
{
case SolFunctionVisibility::PUBLIC:
return "public";
case SolFunctionVisibility::PRIVATE:
return "private";
case SolFunctionVisibility::EXTERNAL:
return "external";
case SolFunctionVisibility::INTERNAL:
return "internal";
}
}
string functionMutability(SolFunctionStateMutability _mut)
{
switch (_mut)
{
case SolFunctionStateMutability::PURE:
return "pure";
case SolFunctionStateMutability::VIEW:
return "view";
case SolFunctionStateMutability::PAYABLE:
return "payable";
}
}
string libraryFunctionMutability(SolLibraryFunctionStateMutability _mut)
{
switch (_mut)
{
case SolLibraryFunctionStateMutability::PURE:
return "pure";
case SolLibraryFunctionStateMutability::VIEW:
return "view";
}
}
}
void SolInterfaceFunction::markExplicitOverride(string _contractName)
{
m_type = Type::EXPLICITOVERRIDE;
m_overriddenFrom.clear();
m_contractName = _contractName;
}
SolInterfaceFunction::SolInterfaceFunction(
string _functionName,
SolFunctionStateMutability _mutability,
Type _type,
string _contractName
)
{
m_functionName = _functionName;
m_mutability = _mutability;
m_type = _type;
m_overriddenFrom = {_contractName};
m_contractName = _contractName;
}
bool SolInterfaceFunction::namesake(SolInterfaceFunction const& _rhs) const
{
// TODO: Change this once we permit arbitrary function parameter types
return name() == _rhs.name();
}
bool SolInterfaceFunction::operator==(SolInterfaceFunction const& _rhs) const
{
solAssert(namesake(_rhs), "Sol proto adaptor: Cannot compare two interface functions with different names");
return m_mutability == _rhs.m_mutability;
}
bool SolInterfaceFunction::operator!=(SolInterfaceFunction const& _rhs) const
{
solAssert(namesake(_rhs), "Sol proto adaptor: Cannot compare two interface functions with different names");
return m_mutability != _rhs.m_mutability;
}
void SolInterfaceFunction::merge(SolInterfaceFunction const& _rhs)
{
assertThrow(
this->operator==(_rhs),
langutil::FuzzerError,
"Sol proto adaptor: Invalid inheritance hierarchy"
);
m_type = Type::EXPLICITOVERRIDE;
for (auto &b: _rhs.m_overriddenFrom)
m_overriddenFrom.push_back(b);
}
bool SolInterfaceFunction::operator==(SolContractFunction const& _rhs) const
{
// TODO: Change this once we permit arbitrary function parameter types
return name() == _rhs.name();
}
bool SolInterfaceFunction::operator!=(SolContractFunction const& _rhs) const
{
// TODO: Change this once we permit arbitrary function parameter types
return name() != _rhs.name();
}
string SolInterfaceFunction::overriddenFromBaseNames() const
{
ostringstream nameStr;
string separator{};
for (auto &b: m_overriddenFrom)
{
nameStr << separator << b;
if (separator.empty())
separator = ", ";
}
return nameStr.str();
}
string SolInterfaceFunction::str() const
{
if (explicitOverride())
return Whiskers(R"(
function <functionName>() override<?isMultiple>(<baseNames>)</isMultiple>
external <stateMutability> returns (uint);)")
("functionName", name())
("isMultiple", multipleBases())
("baseNames", overriddenFromBaseNames())
("stateMutability", functionMutability(mutability()))
.render();
else if (memberFunction())
return Whiskers(R"(
function <functionName>() external <stateMutability> returns (uint);)")
("functionName", name())
("stateMutability", functionMutability(mutability()))
.render();
else
return "";
}
SolContractFunction::SolContractFunction(
ContractFunction const& _function,
std::string _contractName,
std::string _functionName,
bool _implemented,
std::string _returnValue
)
{
m_contractName = _contractName;
m_functionName = _functionName;
m_visibility = visibilityConverter(_function.vis());
m_mutability = mutabilityConverter(_function.mut());
m_returnValue = _returnValue;
m_implemented = _implemented;
// Unimplemented contract functions must be marked virtual
m_virtual = _implemented ? _function.virtualfunc() : true;
}
bool SolContractFunction::operator==(SolContractFunction const& _rhs) const
{
// TODO: Consider function parameters in addition to name once they are
// implemented.
return name() == _rhs.name();
}
bool SolContractFunction::operator!=(SolContractFunction const& _rhs) const
{
// TODO: Consider function parameters in addition to name once they are
// implemented.
return name() != _rhs.name();
}
bool SolContractFunction::operator==(SolInterfaceFunction const& _rhs) const
{
// TODO: Consider function parameters in addition to name once they are
// implemented.
return name() == _rhs.name();
}
bool SolContractFunction::operator!=(SolInterfaceFunction const& _rhs) const
{
// TODO: Consider function parameters in addition to name once they are
// implemented.
return name() != _rhs.name();
}
bool SolContractFunction::disallowed() const
{
// Private virtual functions are disallowed
if (visibility() == SolFunctionVisibility::PRIVATE && isVirtual())
return true;
// Private payable functions are disallowed
else if (visibility() == SolFunctionVisibility::PRIVATE && mutability() == SolFunctionStateMutability::PAYABLE)
return true;
// Internal payable functions are disallowed
else if (visibility() == SolFunctionVisibility::INTERNAL && mutability() == SolFunctionStateMutability::PAYABLE)
return true;
return false;
}
string SolContractFunction::str() const
{
if (disallowed())
return "";
string bodyStr = Whiskers(R"(
{
return <uint>;
})")
("uint", returnValue())
.render();
return Whiskers(R"(
function <functionName>()<?isVirtual> virtual</isVirtual> <visibility> <stateMutability>
returns (uint)<?isImplemented><body><!isImplemented>;</isImplemented>)")
("functionName", name())
("isVirtual", isVirtual())
("visibility", functionVisibility(visibility()))
("stateMutability", functionMutability(mutability()))
("body", bodyStr)
("isImplemented", implemented())
.render();
}
SolLibraryFunction::SolLibraryFunction(
LibraryFunction const& _function,
std::string _libraryName,
std::string _functionName,
std::string _returnValue
)
{
m_libraryName = _libraryName;
m_functionName = _functionName;
m_visibility = visibilityConverter(_function.vis());
m_mutability = mutabilityConverter(_function.mut());
m_returnValue = _returnValue;
}
string SolLibraryFunction::str() const
{
string bodyStr = Whiskers(R"(
{
return <uint>;
})")
("uint", returnValue())
.render();
return Whiskers(R"(
function <functionName>(uint) <visibility> <stateMutability> returns (uint)<body>)")
("functionName", name())
("visibility", functionVisibility(visibility()))
("stateMutability", libraryFunctionMutability(mutability()))
("body", bodyStr)
.render();
}
#if 0
unsigned SolBaseContract::functionIndex()
{
if (type() == BaseType::INTERFACE)
return interface()->functionIndex();
else
{
solAssert(type() == BaseType::CONTRACT, "Sol proto adaptor: Invalid base contract");
return contract()->functionIndex();
}
}
string SolBaseContract::lastBaseName()
{
if (type() == BaseType::INTERFACE)
return interface()->lastBaseName();
else
{
solAssert(type() == BaseType::CONTRACT, "Sol proto adaptor: Invalid base contract");
return contract()->lastBaseName();
}
}
SolBaseContract::BaseType SolBaseContract::type() const
{
if (holds_alternative<shared_ptr<SolInterface>>(m_base))
return BaseType::INTERFACE;
else
{
solAssert(holds_alternative<shared_ptr<SolContract>>(m_base), "Sol proto fuzzer: Invalid base contract");
return BaseType::CONTRACT;
}
}
string SolBaseContract::str()
{
switch (type())
{
case BaseType::INTERFACE:
return interface()->str();
case BaseType::CONTRACT:
return contract()->str();
}
}
string SolBaseContract::name()
{
if (type() == BaseType::INTERFACE)
return interface()->name();
else
{
solAssert(type() == BaseType::CONTRACT, "Sol proto adaptor: Invalid base contract");
return contract()->name();
}
}
SolBaseContract::SolBaseContract(
ProtoBaseContract _base,
string _name,
shared_ptr<SolRandomNumGenerator> _prng
)
{
if (holds_alternative<Contract const*>(_base))
m_base = make_shared<SolContract>(
SolContract(*get<Contract const*>(_base), _name, _prng)
);
else
{
solAssert(holds_alternative<Interface const*>(_base), "Sol proto adaptor: Invalid base contract");
m_base = make_shared<SolInterface>(
SolInterface(*get<Interface const*>(_base), _name, _prng)
);
}
}
void InterfaceFunctionAttributes::merge(InterfaceFunctionAttributes const& _rhs)
{
solAssert(namesake(_rhs), "Sol proto adaptor: Merge called on differently named interface functions");
assertThrow(
this->operator==(_rhs),
langutil::FuzzerError,
"Sol proto adaptor: Invalid interface inheritance hierarchy"
);
// Must override
m_override = true;
for (auto &b: _rhs.m_baseNames)
m_baseNames.push_back(b);
}
bool InterfaceFunctionAttributes::namesake(InterfaceFunctionAttributes const& _rhs) const
{
return m_name == _rhs.m_name;
}
bool InterfaceFunctionAttributes::operator==(InterfaceFunctionAttributes const& _rhs) const
{
return namesake(_rhs) && (m_mutability == _rhs.m_mutability);
}
bool InterfaceFunctionAttributes::operator!=(InterfaceFunctionAttributes const& _rhs) const
{
return !namesake(_rhs) || (m_mutability != _rhs.m_mutability);
}
#endif
void SolInterface::merge()
{
/* Merge algorithm:
* 1. Deep copy all base interface functions (local) into a single list (global)
* 2. Mark all of these as implicit overrides
* 3. Iterate list of implicit overrides
* 3a. If n-way merge is necessary, do so and mark a two-base explicit override and add to contract
* 3b. If n-way merge is not possible, add as implicit override to contract
* 4. Update ownership of n-way merges
* 5. Iterate list of contract implicit and explicit (2-way) overrides
* 5a. If implicit, pseudo randomly mark it explicit
*/
// Step 1-2
vector<shared_ptr<SolInterfaceFunction>> global{};
for (auto &base: m_baseInterfaces)
{
vector<shared_ptr<SolInterfaceFunction>> local{};
for (auto &bf: base->m_functions)
local.push_back(make_shared<SolInterfaceFunction>(*bf));
for (auto &l: local)
{
// Reset override history for past n-way merge
if (l->explicitOverride() && l->numOverriddenFromBases() > 1)
l->resetOverriddenBases();
// Mark all as implicit overrides
l->markImplicitOverride();
global.push_back(l);
}
}
// Step 3
vector<shared_ptr<SolInterfaceFunction>> updateList;
for (auto &f: global)
{
bool merged = false;
for (auto &e: m_functions)
{
if (e->namesake(*f))
{
e->merge(*f);
updateList.push_back(e);
merged = true;
break;
}
}
if (!merged)
m_functions.push_back(f);
}
// Step 4
for (auto &u: updateList)
u->m_contractName = name();
// Step 5
for (auto &e: m_functions)
if (e->implicitOverride() && coinToss())
e->markExplicitOverride(name());
}
void SolInterface::addBases(Interface const& _interface)
{
for (auto &b: _interface.bases())
{
auto base = make_shared<SolInterface>(SolInterface(b, newBaseName(), m_prng));
m_baseInterfaces.push_back(base);
// Do bookkeeping to keep function and base numbering
// consistent
m_functionIndex += base->functionIndex();
m_lastBaseName = base->lastBaseName();
}
merge();
}
void SolInterface::addFunctions(Interface const& _interface)
{
for (auto &f: _interface.funcdef())
m_functions.push_back(
make_shared<SolInterfaceFunction>(
SolInterfaceFunction(
newFunctionName(),
mutabilityConverter(f.mut()),
SolInterfaceFunction::Type::MEMBERFUNCTION,
name()
)
)
);
}
SolInterface::SolInterface(
Interface const& _interface,
string _name,
shared_ptr<SolRandomNumGenerator> _prng
)
{
m_prng = _prng;
m_interfaceName = _name;
m_lastBaseName = m_interfaceName;
addBases(_interface);
addFunctions(_interface);
}
string SolInterface::baseNames() const
{
ostringstream bases;
string separator{};
for (auto &b: m_baseInterfaces)
{
bases << separator << b->name();
if (separator.empty())
separator = ", ";
}
return bases.str();
}
string SolInterface::baseInterfaceStr() const
{
ostringstream baseInterfaces;
for (auto &b: m_baseInterfaces)
baseInterfaces << b->str();
return baseInterfaces.str();
}
string SolInterface::functionStr() const
{
ostringstream functions;
for (auto &f: m_functions)
functions << f->str();
return functions.str();
}
string SolInterface::str() const
{
return Whiskers(R"(
<bases>
interface <programName><?inheritance> is <baseNames></inheritance> {
<functionDefs>
})")
("bases", baseInterfaceStr())
("programName", name())
("inheritance", bases())
("baseNames", baseNames())
("functionDefs", functionStr())
.render();
}
#if 0
SolFunction::Type SolFunction::operator()(FunctionVariant _var) const
{
if (holds_alternative<shared_ptr<SolInterfaceFunction>>(_var))
return SolFunction::Type::INTERFACE;
else
{
solAssert(holds_alternative<shared_ptr<SolContractFunction>>(_var), "Sol proto adaptor: Invalid Sol function");
return SolFunction::Type::CONTRACT;
}
}
bool ContractFunctionAttributes::namesake(ContractFunctionAttributes const& _attr) const
{
return m_name == _attr.m_name;
}
bool ContractFunctionAttributes::operator==(ContractFunctionAttributes const& _rhs) const
{
return m_mutability == _rhs.m_mutability && m_visibility == _rhs.m_visibility;
}
bool ContractFunctionAttributes::operator!=(ContractFunctionAttributes const& _rhs) const
{
return m_mutability != _rhs.m_mutability || m_visibility != _rhs.m_visibility;
}
void ContractFunctionAttributes::merge(ContractFunctionAttributes const& _rhs)
{
assertThrow(
namesake(_rhs),
langutil::FuzzerError,
"Sol proto adaptor: Differently named functions cannot be merged"
);
assertThrow(
this->operator==(_rhs),
langutil::FuzzerError,
"Sol proto adaptor: Namesake functions with different visibility and/or state mutability cannot be merged"
);
// Must override since more than one inherited bases defines
// identical function with identical visibility and/or state
// mutability.
m_override = true;
// Add base to override base list
for (auto &b: _rhs.m_bases)
m_bases.push_back(b);
}
void SolContract::disallowedBase(shared_ptr<SolBaseContract> _base1, shared_ptr<SolBaseContract> _base2)
{
auto base1Type = _base1->type();
auto base2Type = _base2->type();
if (base1Type == SolBaseContract::BaseType::INTERFACE && base2Type == SolBaseContract::BaseType::INTERFACE)
{
for (auto &bf: _base1->interface()->m_functions)
for (auto &lbf: _base2->interface()->m_functions)
assertThrow(
(bf != lbf) || (bf->mutability() == lbf->mutability()),
langutil::FuzzerError,
"Sol proto adaptor: New base defines namesake function with different "
"visibility and/or state mutability"
);
}
else if (base1Type == SolBaseContract::BaseType::INTERFACE && base2Type == SolBaseContract::BaseType::CONTRACT)
{
for (auto &bf: _base1->interface()->m_functions)
for (auto &lbf: _base2->contract()->m_contractFunctions)
if (bf != lbf ||
((bf->mutability() == lbf->mutability()) && (lbf->visibility() == SolFunctionVisibility::EXTERNAL)))
assertThrow(
false,
langutil::FuzzerError,
"Sol proto adaptor: New base defines namesake function with different "
"visibility and/or state mutability"
);
}
else if (base1Type == SolBaseContract::BaseType::CONTRACT && base2Type == SolBaseContract::BaseType::INTERFACE)
{
for (auto &bf: _base1->contract()->m_contractFunctions)
for (auto &lbf: _base2->interface()->m_functions)
if (bf->name() != lbf->name() ||
((bf->mutability() == lbf->mutability()) || (bf->visibility() == SolFunctionVisibility::EXTERNAL)))
assertThrow(
false,
langutil::FuzzerError,
"Sol proto adaptor: New base defines namesake function with different "
"visibility and/or state mutability"
);
}
else
{
for (auto &bf: _base1->contract()->m_contractFunctions)
for (auto &lbf: _base2->contract()->m_contractFunctions)
assertThrow(
bf != lbf || (bf->mutability() == lbf->mutability() && bf->visibility() == lbf->visibility()),
langutil::FuzzerError,
"Sol proto adaptor: New base defines namesake function with different "
"visibility and/or state mutability"
);
}
}
string SolContract::baseNames() const
{
ostringstream bases;
string separator{};
for (auto &b: m_baseContracts)
{
bases << separator << b->name();
if (separator.empty())
separator = ", ";
}
return bases.str();
}
bool SolContract::validTest() const
{
// Check if at least one contract has one valid test function
for (auto &c: m_contractFunctionMap)
if (c.second.size() > 1)
return true;
return false;
}
tuple<string, string, string> SolContract::validContractTest()
{
string chosenContractName{};
string chosenFunctionName{};
string expectedOutput{};
unsigned numContracts = m_contractFunctionMap.size();
unsigned contractIdx = random() % numContracts;
unsigned functionIdx = 0;
unsigned mapIdx = 0;
for (auto &e: m_contractFunctionMap)
{
if (contractIdx == mapIdx)
{
// Recurse if chosen contract has no valid test cases
// We can be sure there is at least one contract with
// a valid test case because validTest() has been
// asserted by caller of this function.
if (e.second.size() == 0)
return validContractTest();
else
{
chosenContractName = e.first;
functionIdx = random() % e.second.size();
unsigned functionMapIdx = 0;
for (auto &f: e.second)
{
if (functionIdx == functionMapIdx)
{
chosenFunctionName = f.first;
expectedOutput = f.second;
break;
}
functionMapIdx++;
}
break;
}
}
mapIdx++;
}
solAssert(m_contractFunctionMap.count(chosenContractName), "Sol proto adaptor: Invalid contract chosen");
solAssert(m_contractFunctionMap[chosenContractName].count(chosenFunctionName), "Sol proto adaptor: Invalid contract function chosen");
return tuple(chosenContractName, chosenFunctionName, expectedOutput);
}
tuple<string, string, string> SolContract::pseudoRandomTest()
{
solAssert(validTest(), "Sol proto adaptor: No valid contract test cases");
return validContractTest();
}
void SolContract::interfaceFunctionOverride(
std::shared_ptr<SolInterface> _base,
std::shared_ptr<SolInterfaceFunction> _function
)
{
string functionName = _function->name();
auto mutability = _function->mutability();
// Check if two or more bases define this function
bool multipleOverride = false;
// If function has already been overridden, add
// new base to list of overridden bases
for (auto &m: m_overriddenFunctions)
{
if (holds_alternative<shared_ptr<SolContractFunction>>(m.first))
// Must override if two or more bases define the
// same function
if (m.first->operator==(*_function))
{
// Report error if state mutability of identically
// named functions differ
if (m.first->mutability() != mutability)
assertThrow(
false,
langutil::FuzzerError,
"Input specifies multiple function overrides with identical names"
" and parameter types but different mutability."
);
// Should interface function be implemented: May be but if not it must be marked virtual
// Should it be explicitly overridden: Yes
// Should it be marked virtual: May be
bool implement = abstract() ? coinToss() : true;
bool virtualize = coinToss();
if (abstract() && !implement)
virtualize = true;
// Add new base to list of overridden bases
m_overriddenInterfaceFunctions[m.first].push_back(
shared_ptr<IFunctionOverride>(
make_shared<IFunctionOverride>(
IFunctionOverride(
_base,
_function,
this,
implement,
virtualize,
true,
newReturnValue()
)
)
)
);
multipleOverride = true;
#if 1
std::cout << "Explicitly overriding interface " <<
_function->name() <<
" of " <<
_base->name() <<
" from " <<
name() <<
std::endl;
#endif
break;
}
}
// Use a pseudo-random coin toss to decide whether to override explicitly
// or not. Implicit override means that the overridden function is not
// redeclared with the override keyword.
bool explicitOverride = abstract() ? coinToss() : true;
// If function has not been overridden, add new override pseudo-randomly
// Should it be virtual: May be but only matters for explicit overrides
// Should it be implemented: If non abstract, otherwise may be
bool virtualize = explicitOverride ? coinToss() : false;
bool implement = abstract() ? coinToss() : true;
if (abstract() && explicitOverride && !implement)
virtualize = true;
if (!multipleOverride)
{
m_overriddenFunctions.insert(
pair(
_function,
vector<shared_ptr<CFunctionOverride>>{
make_shared<CFunctionOverride>(
CFunctionOverride(
_base,
_function,
this,
implement,
virtualize,
explicitOverride,
newReturnValue()
)
)
}
)
);
#if 1
std::cout << (explicitOverride ? "Explicitly" : "Implicitly") <<
" overriding interface " <<
_function->name() <<
" of " <<
_base->name() <<
" from " <<
name() <<
std::endl;
#endif
}
}
void SolContract::contractFunctionOverride(
std::shared_ptr<SolContract> _base,
std::shared_ptr<SolContractFunction> _function
)
{
string functionName = _function->name();
auto mutability = _function->mutability();
auto visibility = _function->visibility();
// Check if two or more bases define this function
bool multipleOverride = false;
// If function has already been overridden, add
// new base to list of overridden bases
for (auto &m: m_overriddenContractFunctions)
{
// Must override if two or more bases define the
// same function
if (m.first->operator==(*_function))
{
// Report error if state mutability of identically
// named functions differ
if (m.first->mutability() != mutability || m.first->visibility() != visibility)
assertThrow(
false,
langutil::FuzzerError,
"Input specifies multiple contract function overrides with identical names"
" and parameter types but different mutability and/or visibility."
);
/* Case 1: Base and derived are abstract
* Case 2: Base and derived not abstract
* Case 3: Derived abstract, base not
* Case 4: Derived non abstract, base abstract
*/
bool implement = abstract() ? coinToss() : true;
bool virtualize = coinToss();
if (abstract() && !implement)
virtualize = true;
// Add new base to list of overridden bases
m_overriddenContractFunctions[m.first].push_back(
shared_ptr<CFunctionOverride>(
make_shared<CFunctionOverride>(
CFunctionOverride(
_base,
_function,
this,
implement,
virtualize,
true,
newReturnValue()
)
)
)
);
#if 1
std::cout << "Explicitly overriding contract " <<
_function->name() <<
" of " <<
_base->name() <<
" from " <<
name() <<
std::endl;
#endif
multipleOverride = true;
break;
}
}
bool implement;
if (_function->implemented())
implement = true;
else
implement = abstract() ? coinToss() : true;
bool virtualize = coinToss();
if (!implement && abstract())
virtualize = true;
bool explicitOverride = true;
if (_base->abstract() && !implement && abstract())
explicitOverride = coinToss();
if (!multipleOverride)
{
m_overriddenContractFunctions.insert(
pair(
_function,
vector<shared_ptr<CFunctionOverride>>{
make_shared<CFunctionOverride>(
CFunctionOverride(
_base,
_function,
this,
implement,
virtualize,
explicitOverride,
newReturnValue()
)
)
}
)
);
#if 1
std::cout << (explicitOverride ? "Explicitly" : "Implicitly") <<
" overriding contract " <<
_function->name() <<
" of " <<
_base->name() <<
" from " <<
name() <<
std::endl;
#endif
}
}
void SolContract::merge(shared_ptr<SolInterface> _interface)
{
/* If abstract contract, we may override interface functions
* If non abstract contract, we must override them
*/
if (m_abstract)
{
for (_interface->m_)
}
else
{
}
}
void SolContract::merge(shared_ptr<SolBaseContract> _base)
{
if (_base->type() == SolBaseContract::INTERFACE)
merge(_base->interface());
else
merge(_base->contract());
}
void SolContract::addOverrides()
{
for (auto &base: m_baseContracts)
{
// Check if base is contract or interface
if (base->type() == SolBaseContract::BaseType::INTERFACE)
{
// Override interface functions
for (auto &f: base->interface()->m_functions)
{
interfaceFunctionOverride(base->interface(), f);
}
// Override interface implicit and explicit overrides
for (auto &m: base->interface()->m_overrideMap)
{
solAssert(m.second.size() >= 1, "Sol proto adaptor: Inconsistent interface override map");
if (m.second.size() == 1)
{
if (m.second[0]->explicitlyInherited())
interfaceFunctionOverride(base->interface(), m.first);
else
interfaceFunctionOverride(m.second[0]->m_baseInterface, m.first);
}
else
{
interfaceFunctionOverride(base->interface(), m.first);
}
}
}
else
{
solAssert(base->type() == SolBaseContract::BaseType::CONTRACT, "Sol proto fuzzer: Base contract neither interface nor contract");
// Override virtual contract functions
for (auto &f: base->contract()->m_contractFunctions)
{
// Override well-defined virtual functions only
// Well defined means that they are not explicitly disallowed
// See SolContractFunction::disallowed() for what is disallowed.
if (!f->disallowed() && f->isVirtual())
contractFunctionOverride(base->contract(), f);
}
// Override contract overrides only if they are virtual.
for (auto &m: base->contract()->m_overriddenContractFunctions)
{
if (!m.first->disallowed())
for (auto &b: m.second)
if (b->virtualized())
contractFunctionOverride(b->m_baseContract, m.first);
}
// Override implicit contract overrides from base interface only if
// - They have been implicitly overridden by base contract OR
// - They have been explicitly overridden and virtualized but unimplemented
for (auto &m: base->contract()->m_overriddenInterfaceFunctions)
{
std::cout << "Overriding " << m.first->name() << " from " << name() << std::endl;
if (m.second.size() == 1)
{
if (!m.second[0]->explicitlyInherited())
interfaceFunctionOverride(m.second[0]->m_baseInterface, m.first);
// else if (m.second[0]->virtualized() && !m.second[0]->implemented())
// interfaceFunctionOverride(base, m.first);
}
}
}
}
// If any of the interface overrides has same name as one of the contract overrides
// but they differ in state mutability and/or visibility, we throw to signal invalid
// contract.
// for (auto &c: m_overriddenContractFunctions)
// for (auto &i: m_overriddenInterfaceFunctions)
// if (i.first->name() == c.first->name() && ((i.first->mutability() != c.first->mutability()) ||
// c.first->visibility() != SolFunctionVisibility::EXTERNAL))
// assertThrow(
// false,
// langutil::FuzzerError,
// "Sol proto fuzzer: Interface and contract overrides have same name but "
// "different visibility and/or state mutability."
// );
}
void SolContract::addFunctions(Contract const& _contract)
{
bool abs = abstract();
string contractName = name();
// Add contract to contract function map only if the contract
// is not abstract.
if (!abs)
m_contractFunctionMap.insert(pair(contractName, map<string, string>{}));
// Add functions
for (auto &f: _contract.funcdef())
{
auto function = make_shared<SolContractFunction>(
SolContractFunction(
f,
contractName,
newFunctionName(),
(abs ? coinToss() : true),
newReturnValue()
)
);
m_contractFunctions.push_back(function);
// If contract is not abstract, add its public and external
// functions to contract function map.
if (!abs)
{
auto visibility = function->visibility();
string functionName = function->name();
string expectedOutput = function->returnValue();
// Register only public and external contract functions because only they can
// be called from a different contract.
if (visibility == SolFunctionVisibility::PUBLIC || visibility == SolFunctionVisibility::EXTERNAL)
{
solAssert(!m_contractFunctionMap[contractName].count(functionName), "Sol proto adaptor: Duplicate contract function");
m_contractFunctionMap[contractName].insert(pair(functionName, expectedOutput));
}
}
}
}
void SolContract::addBases(Contract const& _contract)
{
shared_ptr<SolBaseContract> base;
for (auto &b: _contract.bases())
{
switch (b.contract_or_interface_oneof_case())
{
case ContractOrInterface::kC:
base = make_shared<SolBaseContract>(
SolBaseContract(&b.c(), newBaseName(), m_prng)
);
break;
case ContractOrInterface::kI:
base = make_shared<SolBaseContract>(
SolBaseContract(&b.i(), newBaseName(), m_prng)
);
break;
case ContractOrInterface::CONTRACT_OR_INTERFACE_ONEOF_NOT_SET:
continue;
}
// Check for namesake functions with different visibility and/or
// state mutability. Bail out by throwing exception
if (m_baseContracts.size() > 1)
for (auto &existingBase: m_baseContracts)
disallowedBase(base, existingBase);
m_baseContracts.push_back(base);
// Worst case, we override all base functions so we
// increment derived contract's function index by
// this amount.
m_functionIndex += base->functionIndex();
m_lastBaseName = base->lastBaseName();
}
}
string SolContract::contractOverrideStr()
{
ostringstream overriddenFunctions;
for (auto &f: m_overriddenContractFunctions)
{
string returnValue = f.second[0]->returnValue();
string bodyStr = Whiskers(R"(
{
return <uint>;
})")
("uint", returnValue)
.render();
bool implemented = f.second[0]->implemented();
bool virtualized = f.second[0]->virtualized();
ostringstream overriddenBaseNames;
if (f.second.size() > 1)
{
string sep{};
for (auto &b: f.second)
{
overriddenBaseNames << Whiskers(R"(<sep><name>)")
("sep", sep)
("name", b->baseName())
.render();
if (sep.empty())
sep = ", ";
}
}
else
{
solAssert(f.second.size() == 1, "Inconsistent override map");
if (!f.second[0]->explicitlyInherited())
continue;
}
string functionName = f.first->name();
overriddenFunctions << Whiskers(R"(
function <functionName>() <visibility> <stateMutability><?isVirtual> virtual</isVirtual>
override<?multiple>(<baseNames>)</multiple> returns (uint)<?isImplemented><body><!isImplemented>;</isImplemented>)")
("functionName", f.first->name())
("visibility", functionVisibility(f.first->visibility()))
("stateMutability", functionMutability(f.first->mutability()))
("isVirtual", virtualized)
("multiple", f.second.size() > 1)
("baseNames", overriddenBaseNames.str())
("isImplemented", implemented)
("body", bodyStr)
.render();
if (!abstract())
m_contractFunctionMap[name()].insert(pair(functionName, returnValue));
}
return overriddenFunctions.str();
}
string SolContract::interfaceOverrideStr()
{
ostringstream overriddenFunctions;
for (auto &f: m_overriddenInterfaceFunctions)
{
string returnValue = f.second[0]->returnValue();
string bodyStr = Whiskers(R"(
{
return <uint>;
})")
("uint", returnValue)
.render();
// Check override class for implementation and virtualized
// flags and print appropriately.
bool implemented = f.second[0]->implemented();
bool virtualized = f.second[0]->virtualized();
ostringstream overriddenBaseNames;
if (f.second.size() > 1)
{
string sep{};
for (auto &b: f.second)
{
overriddenBaseNames << Whiskers(R"(<sep><name>)")
("sep", sep)
("name", b->baseName())
.render();
if (sep.empty())
sep = ", ";
}
}
else
{
solAssert(f.second.size() == 1, "Inconsistent override map");
if (!f.second[0]->explicitlyInherited())
{
continue;
}
}
string functionName = f.first->name();
overriddenFunctions << Whiskers(R"(
function <functionName>() external <stateMutability><?isVirtual> virtual</isVirtual>
override<?multiple>(<baseNames>)</multiple> returns (uint)<?isImplemented><body><!isImplemented>;</isImplemented>)")
("functionName", f.first->name())
("stateMutability", functionMutability(f.first->mutability()))
("isVirtual", virtualized)
("multiple", f.second.size() > 1)
("baseNames", overriddenBaseNames.str())
("isImplemented", implemented)
("body", bodyStr)
.render();
if (!abstract())
m_contractFunctionMap[name()].insert(pair(functionName, returnValue));
}
return overriddenFunctions.str();
}
string SolContract::str()
{
ostringstream bases;
for (auto &b: m_baseContracts)
bases << b->str();
ostringstream functions;
// Print overridden functions
functions << interfaceOverrideStr() << contractOverrideStr();
// Print non-overridden functions
for (auto &f: m_contractFunctions)
functions << f->str();
return Whiskers(R"(
<bases>
<?isAbstract>abstract </isAbstract>contract <contractName><?inheritance> is <baseNames></inheritance> {
<functions>
})")
("bases", bases.str())
("isAbstract", abstract())
("contractName", name())
("inheritance", m_baseContracts.size() > 0)
("baseNames", baseNames())
("functions", functions.str())
.render();
}
SolContract::SolContract(
Contract const& _contract,
std::string _name,
shared_ptr<SolRandomNumGenerator> _prng
)
{
m_prng = _prng;
m_contractName = _name;
m_lastBaseName = m_contractName;
m_abstract = _contract.abstract();
addBases(_contract);
addOverrides();
addFunctions(_contract);
}
#endif
void SolLibrary::addFunction(LibraryFunction const& _function)
{
// Register function name and return value
string functionName = newFunctionName();
string outputStr = newReturnValue();
SolFunctionVisibility visibility = visibilityConverter(_function.vis());
if (visibility == SolFunctionVisibility::PUBLIC || visibility == SolFunctionVisibility::EXTERNAL)
{
solAssert(!m_publicFunctionMap.count(functionName), "Sol proto adapter: Duplicate library function");
m_publicFunctionMap.insert(pair(functionName, outputStr));
}
// Create and add function to library
m_functions.push_back(
make_unique<SolLibraryFunction>(
SolLibraryFunction(_function,
name(),
functionName,
outputStr
)
)
);
}
SolLibrary::SolLibrary(Library const& _library, string _name, shared_ptr<SolRandomNumGenerator> _prng)
{
m_libraryName = _name;
m_prng = _prng;
for (LibraryFunction const& f: _library.funcdef())
addFunction(f);
}
string SolLibrary::str() const
{
ostringstream functions;
for (auto &f: m_functions)
functions << f->str();
return Whiskers(R"(
library <name> {
<functions>
})")
("name", name())
("functions", functions.str())
.render();
}
bool SolLibrary::validTest() const
{
return m_publicFunctionMap.size() > 0;
}
pair<string, string> SolLibrary::pseudoRandomTest()
{
solAssert(m_publicFunctionMap.size() > 0, "Sol proto adaptor: Empty library map");
string chosenFunction;
unsigned numFunctions = m_publicFunctionMap.size();
unsigned functionIndex = randomNumber() % numFunctions;
unsigned mapIdx = 0;
for (auto &e: m_publicFunctionMap)
{
if (functionIndex == mapIdx)
{
chosenFunction = e.first;
break;
}
mapIdx++;
}
solAssert(m_publicFunctionMap.count(chosenFunction), "Sol proto adaptor: Invalid library function chosen");
return pair(chosenFunction, m_publicFunctionMap[chosenFunction]);
}
#if 0
string CFunctionOverride::name() const
{
return m_baseFunction->name();
}
SolFunctionVisibility CFunctionOverride::visibility() const
{
return m_baseFunction->visibility();
}
SolFunctionStateMutability CFunctionOverride::mutability() const
{
return m_baseFunction->mutability();
}
string CFunctionOverride::str() const
{
solAssert(virtualized() || !implemented(), "Sol proto fuzzer: Invalid virtualization of contract function override");
string bodyStr = Whiskers(R"(
{
return <uint>;
})")
("uint", returnValue())
.render();
return Whiskers(R"(
function <functionName>() override <?isVirtual> virtual</isVirtual> <visibility> <stateMutability>
returns (uint)<?isImplemented><body><!isImplemented>;</isImplemented>)")
("functionName", name())
("isVirtual", virtualized())
("visibility", functionVisibility(visibility()))
("stateMutability", functionMutability(mutability()))
("body", bodyStr)
("isImplemented", implemented())
.render();
}
string CFunctionOverride::baseName() const
{
return m_baseContract->name();
}
IFunctionOverride::IFunctionOverride(
std::shared_ptr<SolInterface> _baseInterface,
std::shared_ptr<SolInterfaceFunction const> _baseFunction,
std::variant<SolInterface*, SolContract*> _derivedProgram,
bool _implement,
bool _virtual,
bool _explicitInherit,
std::string _returnValue
)
{
if (std::holds_alternative<SolContract*>(_derivedProgram))
{
auto derived = std::get<SolContract*>(_derivedProgram);
m_derivedType = derived->abstract() ? DerivedType::ABSTRACTCONTRACT : DerivedType::CONTRACT;
if (!derived->abstract())
solAssert(
_explicitInherit && !_returnValue.empty() && _implement,
"Contract overrides base interface function either without"
" implementing it or implictly overrides."
);
else
if (_explicitInherit)
solAssert(
_virtual || (_implement && !_returnValue.empty()),
"Abstract contract overrides base interface function either"
" without implementing it or without marking it virtual."
);
}
else
{
solAssert(
holds_alternative<SolInterface*>(_derivedProgram),
"Derived program neither an interface nor a contract"
);
m_derivedType = DerivedType::INTERFACE;
if (_explicitInherit)
solAssert(
!_virtual && !_implement && _returnValue.empty(),
"Interface overrides base interface function with invalid parameters."
);
}
m_baseInterface = _baseInterface;
m_baseFunction = _baseFunction;
m_derivedProgram = _derivedProgram;
m_implemented = _implement;
m_virtualized = _virtual;
m_explicitlyInherited = _explicitInherit;
m_returnValue = _returnValue;
}
std::string IFunctionOverride::str() const
{
switch (m_derivedType)
{
case DerivedType::INTERFACE:
return interfaceStr();
case DerivedType::ABSTRACTCONTRACT:
case DerivedType::CONTRACT:
return contractStr();
}
}
string IFunctionOverride::interfaceStr() const
{
return Whiskers(R"(
function <functionName>() external <mutability> override returns(uint);
)")
("functionName", m_baseFunction->name())
("mutability", functionMutability(m_baseFunction->mutability()))
.render();
}
string IFunctionOverride::contractStr() const
{
string bodyStr = Whiskers(R"(
{
return <uint>;
})")
("uint", returnValue())
.render();
return Whiskers(R"(
function <functionName>() external <mutability> override
<?isVirtual> virtual</isVirtual> returns(uint)<?isImplemented><body><!isImplemented>;</isImplemented>
)")
("functionName", m_baseFunction->name())
("mutability", functionMutability(m_baseFunction->mutability()))
("isVirtual", virtualized())
("isImplemented", implemented())
("body", bodyStr)
.render();
}
#endif