mirror of
https://github.com/ethereum/solidity
synced 2023-10-03 13:03:40 +00:00
131 lines
4.0 KiB
C++
131 lines
4.0 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/>.
|
|
*/
|
|
/**
|
|
* @author Christian <c@ethdev.com>
|
|
* @date 2016
|
|
* Framework for executing Solidity contracts and testing them against C++ implementation.
|
|
*/
|
|
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <boost/test/framework.hpp>
|
|
#include <test/libsolidity/SolidityExecutionFramework.h>
|
|
|
|
using namespace solidity;
|
|
using namespace solidity::test;
|
|
using namespace solidity::frontend;
|
|
using namespace solidity::frontend::test;
|
|
using namespace std;
|
|
|
|
bytes SolidityExecutionFramework::multiSourceCompileContract(
|
|
map<string, string> const& _sourceCode,
|
|
string const& _contractName,
|
|
map<string, Address> const& _libraryAddresses
|
|
)
|
|
{
|
|
map<string, string> sourcesWithPreamble = _sourceCode;
|
|
for (auto& entry: sourcesWithPreamble)
|
|
entry.second = addPreamble(entry.second);
|
|
|
|
m_compiler.reset();
|
|
m_compiler.setSources(sourcesWithPreamble);
|
|
m_compiler.setLibraries(_libraryAddresses);
|
|
m_compiler.setRevertStringBehaviour(m_revertStrings);
|
|
m_compiler.setEVMVersion(m_evmVersion);
|
|
m_compiler.setOptimiserSettings(m_optimiserSettings);
|
|
m_compiler.enableIRGeneration(m_compileViaYul);
|
|
m_compiler.setRevertStringBehaviour(m_revertStrings);
|
|
if (!m_compiler.compile())
|
|
{
|
|
langutil::SourceReferenceFormatter formatter(std::cerr);
|
|
|
|
for (auto const& error: m_compiler.errors())
|
|
formatter.printErrorInformation(*error);
|
|
BOOST_ERROR("Compiling contract failed");
|
|
}
|
|
std::string contractName(_contractName.empty() ? m_compiler.lastContractName() : _contractName);
|
|
evmasm::LinkerObject obj;
|
|
if (m_compileViaYul)
|
|
{
|
|
// Try compiling twice: If the first run fails due to stack errors, forcefully enable
|
|
// the optimizer.
|
|
for (bool forceEnableOptimizer: {false, true})
|
|
{
|
|
OptimiserSettings optimiserSettings = m_optimiserSettings;
|
|
if (!forceEnableOptimizer && !optimiserSettings.runYulOptimiser)
|
|
{
|
|
// Enable some optimizations on the first run
|
|
optimiserSettings.runYulOptimiser = true;
|
|
optimiserSettings.yulOptimiserSteps = "uljmul jmul";
|
|
}
|
|
else if (forceEnableOptimizer)
|
|
optimiserSettings = OptimiserSettings::full();
|
|
|
|
yul::AssemblyStack asmStack(
|
|
m_evmVersion,
|
|
yul::AssemblyStack::Language::StrictAssembly,
|
|
optimiserSettings
|
|
);
|
|
bool analysisSuccessful = asmStack.parseAndAnalyze("", m_compiler.yulIROptimized(contractName));
|
|
solAssert(analysisSuccessful, "Code that passed analysis in CompilerStack can't have errors");
|
|
|
|
try
|
|
{
|
|
asmStack.optimize();
|
|
obj = std::move(*asmStack.assemble(yul::AssemblyStack::Machine::EVM).bytecode);
|
|
break;
|
|
}
|
|
catch (...)
|
|
{
|
|
if (forceEnableOptimizer || optimiserSettings == OptimiserSettings::full())
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
obj = m_compiler.object(contractName);
|
|
BOOST_REQUIRE(obj.linkReferences.empty());
|
|
if (m_showMetadata)
|
|
cout << "metadata: " << m_compiler.metadata(contractName) << endl;
|
|
return obj.bytecode;
|
|
}
|
|
|
|
bytes SolidityExecutionFramework::compileContract(
|
|
string const& _sourceCode,
|
|
string const& _contractName,
|
|
map<string, Address> const& _libraryAddresses
|
|
)
|
|
{
|
|
return multiSourceCompileContract(
|
|
{{"", _sourceCode}},
|
|
_contractName,
|
|
_libraryAddresses
|
|
);
|
|
}
|
|
|
|
string SolidityExecutionFramework::addPreamble(string const& _sourceCode)
|
|
{
|
|
// Silence compiler version warning
|
|
string preamble = "pragma solidity >=0.0;\n";
|
|
if (
|
|
solidity::test::CommonOptions::get().useABIEncoderV2 &&
|
|
_sourceCode.find("pragma experimental ABIEncoderV2;") == string::npos
|
|
)
|
|
preamble += "pragma experimental ABIEncoderV2;\n";
|
|
return preamble + _sourceCode;
|
|
}
|