solidity/libyul/AssemblyStack.cpp

228 lines
6.6 KiB
C++
Raw Normal View History

/*
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/>.
*/
/**
* Full assembly stack that can support EVM-assembly and Yul as input and EVM, EVM1.5 and
* eWasm as output.
*/
2019-02-13 11:07:20 +00:00
#include <libyul/AssemblyStack.h>
2018-12-17 18:24:42 +00:00
#include <libyul/AsmAnalysis.h>
#include <libyul/AsmAnalysisInfo.h>
2018-12-17 18:24:42 +00:00
#include <libyul/AsmParser.h>
#include <libyul/AsmPrinter.h>
2019-02-13 10:35:49 +00:00
#include <libyul/backends/evm/AsmCodeGen.h>
2018-10-15 09:52:35 +00:00
#include <libyul/backends/evm/EVMAssembly.h>
2018-12-17 18:24:42 +00:00
#include <libyul/backends/evm/EVMCodeTransform.h>
2018-12-06 23:56:16 +00:00
#include <libyul/backends/evm/EVMDialect.h>
2018-12-17 18:24:42 +00:00
#include <libyul/backends/evm/EVMObjectCompiler.h>
2019-05-28 10:57:15 +00:00
#include <libyul/backends/evm/EVMMetrics.h>
#include <libyul/backends/wasm/WasmDialect.h>
#include <libyul/backends/wasm/EWasmObjectCompiler.h>
2019-05-15 08:33:35 +00:00
#include <libyul/optimiser/Metrics.h>
2018-11-07 11:01:43 +00:00
#include <libyul/ObjectParser.h>
#include <libyul/optimiser/Suite.h>
#include <libsolidity/interface/OptimiserSettings.h>
2019-02-13 11:07:20 +00:00
#include <libevmasm/Assembly.h>
#include <liblangutil/Scanner.h>
using namespace std;
using namespace langutil;
2019-02-13 11:07:20 +00:00
using namespace yul;
namespace
{
Dialect const& languageToDialect(AssemblyStack::Language _language, EVMVersion _version)
{
switch (_language)
{
case AssemblyStack::Language::Assembly:
2019-02-21 20:56:30 +00:00
return EVMDialect::looseAssemblyForEVM(_version);
case AssemblyStack::Language::StrictAssembly:
2019-02-21 20:56:30 +00:00
return EVMDialect::strictAssemblyForEVMObjects(_version);
case AssemblyStack::Language::Yul:
2019-02-13 11:07:20 +00:00
return Dialect::yul();
case AssemblyStack::Language::EWasm:
return WasmDialect::instance();
}
solAssert(false, "");
2019-02-13 11:07:20 +00:00
return Dialect::yul();
}
}
2017-05-23 17:21:14 +00:00
Scanner const& AssemblyStack::scanner() const
{
solAssert(m_scanner, "");
return *m_scanner;
}
2017-05-23 17:21:14 +00:00
bool AssemblyStack::parseAndAnalyze(std::string const& _sourceName, std::string const& _source)
{
2017-06-07 13:41:44 +00:00
m_errors.clear();
m_analysisSuccessful = false;
m_scanner = make_shared<Scanner>(CharStream(_source, _sourceName));
2019-02-21 20:56:30 +00:00
m_parserResult = ObjectParser(m_errorReporter, languageToDialect(m_language, m_evmVersion)).parse(m_scanner, false);
if (!m_errorReporter.errors().empty())
return false;
solAssert(m_parserResult, "");
2018-11-07 11:01:43 +00:00
solAssert(m_parserResult->code, "");
2017-06-02 20:52:48 +00:00
return analyzeParsed();
}
void AssemblyStack::optimize()
{
if (!m_optimiserSettings.runYulOptimiser)
return;
2018-12-04 17:57:32 +00:00
solAssert(m_analysisSuccessful, "Analysis was not successful.");
2018-12-04 17:57:32 +00:00
m_analysisSuccessful = false;
solAssert(m_parserResult, "");
2019-05-15 08:33:35 +00:00
optimize(*m_parserResult, true);
solAssert(analyzeParsed(), "Invalid source code after optimization.");
}
2017-06-02 20:52:48 +00:00
bool AssemblyStack::analyzeParsed()
{
2018-11-07 11:01:43 +00:00
solAssert(m_parserResult, "");
2018-12-04 17:57:32 +00:00
m_analysisSuccessful = analyzeParsed(*m_parserResult);
2017-05-28 13:13:41 +00:00
return m_analysisSuccessful;
}
2019-02-13 11:07:20 +00:00
bool AssemblyStack::analyzeParsed(Object& _object)
2018-12-04 17:57:32 +00:00
{
solAssert(_object.code, "");
2019-02-13 11:07:20 +00:00
_object.analysisInfo = make_shared<AsmAnalysisInfo>();
AsmAnalyzer analyzer(
*_object.analysisInfo,
m_errorReporter,
std::nullopt,
languageToDialect(m_language, m_evmVersion),
{},
_object.dataNames()
);
2018-12-04 17:57:32 +00:00
bool success = analyzer.analyze(*_object.code);
for (auto& subNode: _object.subObjects)
2019-02-13 11:07:20 +00:00
if (auto subObject = dynamic_cast<Object*>(subNode.get()))
2018-12-04 17:57:32 +00:00
if (!analyzeParsed(*subObject))
success = false;
return success;
}
2019-02-13 11:07:20 +00:00
void AssemblyStack::compileEVM(AbstractAssembly& _assembly, bool _evm15, bool _optimize) const
2018-12-06 23:56:16 +00:00
{
EVMDialect const* dialect = nullptr;
2018-12-06 23:56:16 +00:00
if (m_language == Language::Assembly)
dialect = &EVMDialect::looseAssemblyForEVM(m_evmVersion);
2018-12-06 23:56:16 +00:00
else if (m_language == AssemblyStack::Language::StrictAssembly)
dialect = &EVMDialect::strictAssemblyForEVMObjects(m_evmVersion);
2018-12-06 23:56:16 +00:00
else if (m_language == AssemblyStack::Language::Yul)
dialect = &EVMDialect::yulForEVM(m_evmVersion);
2018-12-06 23:56:16 +00:00
else
solAssert(false, "Invalid language.");
2019-02-13 11:07:20 +00:00
EVMObjectCompiler::compile(*m_parserResult, _assembly, *dialect, _evm15, _optimize);
2018-12-06 23:56:16 +00:00
}
2019-05-15 08:33:35 +00:00
void AssemblyStack::optimize(Object& _object, bool _isCreation)
2018-12-04 17:57:32 +00:00
{
solAssert(_object.code, "");
solAssert(_object.analysisInfo, "");
for (auto& subNode: _object.subObjects)
2019-02-13 11:07:20 +00:00
if (auto subObject = dynamic_cast<Object*>(subNode.get()))
2019-05-15 08:33:35 +00:00
optimize(*subObject, false);
2019-05-29 21:14:06 +00:00
Dialect const& dialect = languageToDialect(m_language, m_evmVersion);
unique_ptr<GasMeter> meter;
if (EVMDialect const* evmDialect = dynamic_cast<EVMDialect const*>(&dialect))
meter = make_unique<GasMeter>(*evmDialect, _isCreation, m_optimiserSettings.expectedExecutionsPerDeployment);
OptimiserSuite::run(
dialect,
meter.get(),
_object,
2019-05-29 21:14:06 +00:00
m_optimiserSettings.optimizeStackAllocation
);
2018-12-04 17:57:32 +00:00
}
MachineAssemblyObject AssemblyStack::assemble(Machine _machine) const
{
solAssert(m_analysisSuccessful, "");
solAssert(m_parserResult, "");
2018-11-07 11:01:43 +00:00
solAssert(m_parserResult->code, "");
solAssert(m_parserResult->analysisInfo, "");
2017-05-23 17:21:14 +00:00
switch (_machine)
{
case Machine::EVM:
{
2017-05-29 22:50:46 +00:00
MachineAssemblyObject object;
2019-02-13 11:07:20 +00:00
dev::eth::Assembly assembly;
EthAssemblyAdapter adapter(assembly);
compileEVM(adapter, false, m_optimiserSettings.optimizeStackAllocation);
2019-02-13 11:07:20 +00:00
object.bytecode = make_shared<dev::eth::LinkerObject>(assembly.assemble());
2017-08-30 01:17:15 +00:00
object.assembly = assembly.assemblyString();
2017-05-29 22:50:46 +00:00
return object;
}
case Machine::EVM15:
{
2017-05-29 22:50:46 +00:00
MachineAssemblyObject object;
2019-02-13 11:07:20 +00:00
EVMAssembly assembly(true);
compileEVM(assembly, true, m_optimiserSettings.optimizeStackAllocation);
2019-02-13 11:07:20 +00:00
object.bytecode = make_shared<dev::eth::LinkerObject>(assembly.finalize());
/// TODO: fill out text representation
2017-05-29 22:50:46 +00:00
return object;
}
case Machine::eWasm:
{
solAssert(m_language == Language::EWasm, "");
Dialect const& dialect = languageToDialect(m_language, EVMVersion{});
MachineAssemblyObject object;
2019-10-31 17:42:14 +00:00
auto result = EWasmObjectCompiler::compile(*m_parserResult, dialect);
object.assembly = std::move(result.first);
object.bytecode = make_shared<dev::eth::LinkerObject>();
object.bytecode->bytecode = std::move(result.second);
return object;
}
}
// unreachable
2017-05-29 22:50:46 +00:00
return MachineAssemblyObject();
}
string AssemblyStack::print() const
{
solAssert(m_parserResult, "");
2018-11-07 11:01:43 +00:00
solAssert(m_parserResult->code, "");
return m_parserResult->toString(m_language == Language::Yul) + "\n";
}
2019-02-13 11:21:53 +00:00
shared_ptr<Object> AssemblyStack::parserResult() const
{
solAssert(m_analysisSuccessful, "Analysis was not successful.");
solAssert(m_parserResult, "");
solAssert(m_parserResult->code, "");
return m_parserResult;
}