solidity/libyul/backends/evm/EVMCodeTransform.h

253 lines
8.2 KiB
C
Raw Normal View History

2017-05-23 17:11:14 +00:00
/*
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/>.
*/
/**
2018-06-18 21:26:31 +00:00
* Common code generator for translating Yul / inline assembly to EVM and EVM1.5.
2017-05-23 17:11:14 +00:00
*/
2018-12-12 15:47:05 +00:00
#pragma once
2018-10-15 09:52:35 +00:00
#include <libyul/backends/evm/EVMAssembly.h>
2017-05-23 17:21:14 +00:00
2018-12-06 23:56:16 +00:00
#include <libyul/backends/evm/EVMDialect.h>
#include <libyul/optimiser/ASTWalker.h>
#include <libyul/AsmDataForward.h>
#include <libyul/AsmScope.h>
2017-05-23 17:21:14 +00:00
#include <boost/variant.hpp>
#include <boost/optional.hpp>
2017-05-23 17:21:14 +00:00
#include <stack>
namespace langutil
2017-05-23 17:21:14 +00:00
{
class ErrorReporter;
2017-05-23 17:21:14 +00:00
}
namespace yul
2017-05-23 17:21:14 +00:00
{
struct AsmAnalysisInfo;
class EVMAssembly;
2017-05-23 17:11:14 +00:00
2019-01-29 09:51:25 +00:00
struct StackTooDeepError: virtual YulException
{
StackTooDeepError(YulString _variable, int _depth): variable(_variable), depth(_depth) {}
StackTooDeepError(YulString _functionName, YulString _variable, int _depth):
functionName(_functionName), variable(_variable), depth(_depth)
{}
YulString functionName;
YulString variable;
int depth;
};
struct CodeTransformContext
{
std::map<Scope::Label const*, AbstractAssembly::LabelID> labelIDs;
std::map<Scope::Function const*, AbstractAssembly::LabelID> functionEntryIDs;
std::map<Scope::Variable const*, int> variableStackHeights;
std::map<Scope::Variable const*, unsigned> variableReferences;
struct JumpInfo
{
AbstractAssembly::LabelID label; ///< Jump's LabelID to jump to.
int targetStackHeight; ///< Stack height after the jump.
};
struct ForLoopLabels
{
JumpInfo post; ///< Jump info for jumping to post branch.
JumpInfo done; ///< Jump info for jumping to done branch.
};
std::stack<ForLoopLabels> forLoopStack;
};
/**
* Counts the number of references to a variable. This includes actual (read) references
* but also assignments to the variable. It does not include the declaration itself or
* function parameters, but it does include function return parameters.
*
* This component can handle multiple variables of the same name.
*
* Can only be applied to strict assembly.
*/
class VariableReferenceCounter: public yul::ASTWalker
{
public:
explicit VariableReferenceCounter(
CodeTransformContext& _context,
AsmAnalysisInfo const& _assemblyInfo
): m_context(_context), m_info(_assemblyInfo)
{}
public:
void operator()(Identifier const& _identifier);
void operator()(FunctionDefinition const&);
void operator()(ForLoop const&);
void operator()(Block const& _block);
private:
void increaseRefIfFound(YulString _variableName);
CodeTransformContext& m_context;
AsmAnalysisInfo const& m_info;
Scope* m_scope = nullptr;
};
2017-05-23 17:11:14 +00:00
class CodeTransform: public boost::static_visitor<>
{
public:
/// Create the code transformer.
2017-05-23 17:11:14 +00:00
/// @param _identifierAccess used to resolve identifiers external to the inline assembly
/// As a side-effect of its construction, translates the Yul code and appends it to the
/// given assembly.
/// Throws StackTooDeepError if a variable is not accessible or if a function has too
/// many parameters.
2017-05-23 17:21:14 +00:00
CodeTransform(
AbstractAssembly& _assembly,
AsmAnalysisInfo& _analysisInfo,
Block const& _block,
2018-12-06 23:56:16 +00:00
EVMDialect const& _dialect,
BuiltinContext& _builtinContext,
bool _allowStackOpt = false,
bool _evm15 = false,
2017-08-25 15:04:31 +00:00
ExternalIdentifierAccess const& _identifierAccess = ExternalIdentifierAccess(),
bool _useNamedLabelsForFunctions = false
2017-06-13 19:58:25 +00:00
): CodeTransform(
_assembly,
_analysisInfo,
_block,
_allowStackOpt,
2018-12-06 17:07:08 +00:00
_dialect,
_builtinContext,
2017-06-13 19:58:25 +00:00
_evm15,
_identifierAccess,
2017-08-25 15:04:31 +00:00
_useNamedLabelsForFunctions,
2017-06-13 19:58:25 +00:00
_assembly.stackHeight(),
nullptr
2017-06-13 19:58:25 +00:00
)
2017-05-23 17:11:14 +00:00
{
}
2019-01-29 09:51:25 +00:00
std::vector<StackTooDeepError> const& stackErrors() const { return m_stackErrors; }
protected:
using Context = CodeTransformContext;
2017-06-13 19:58:25 +00:00
2017-05-23 17:11:14 +00:00
CodeTransform(
AbstractAssembly& _assembly,
AsmAnalysisInfo& _analysisInfo,
Block const& _block,
bool _allowStackOpt,
2018-12-06 23:56:16 +00:00
EVMDialect const& _dialect,
BuiltinContext& _builtinContext,
bool _evm15,
2017-05-23 17:21:14 +00:00
ExternalIdentifierAccess const& _identifierAccess,
2017-08-25 15:04:31 +00:00
bool _useNamedLabelsForFunctions,
2017-06-13 19:58:25 +00:00
int _stackAdjustment,
std::shared_ptr<Context> _context
);
void decreaseReference(YulString _name, Scope::Variable const& _var);
bool unreferenced(Scope::Variable const& _var) const;
/// Marks slots of variables that are not used anymore
/// and were defined in the current scope for reuse.
/// Also POPs unused topmost stack slots.
void freeUnusedVariables();
/// Marks the stack slot of @a _var to be reused.
void deleteVariable(Scope::Variable const& _var);
2017-05-23 17:11:14 +00:00
public:
void operator()(Instruction const& _instruction);
void operator()(Literal const& _literal);
void operator()(Identifier const& _identifier);
void operator()(FunctionalInstruction const& _instr);
void operator()(FunctionCall const&);
2017-12-08 13:01:22 +00:00
void operator()(ExpressionStatement const& _statement);
void operator()(Label const& _label);
void operator()(StackAssignment const& _assignment);
void operator()(Assignment const& _assignment);
void operator()(VariableDeclaration const& _varDecl);
void operator()(If const& _if);
void operator()(Switch const& _switch);
void operator()(FunctionDefinition const&);
void operator()(ForLoop const&);
void operator()(Break const&);
void operator()(Continue const&);
void operator()(Block const& _block);
2017-06-01 16:16:38 +00:00
private:
AbstractAssembly::LabelID labelFromIdentifier(Identifier const& _identifier);
2017-06-13 19:58:25 +00:00
/// @returns the label ID corresponding to the given label, allocating a new one if
/// necessary.
AbstractAssembly::LabelID labelID(Scope::Label const& _label);
AbstractAssembly::LabelID functionEntryID(YulString _name, Scope::Function const& _function);
/// Generates code for an expression that is supposed to return a single value.
2017-12-08 13:01:22 +00:00
void visitExpression(Expression const& _expression);
2017-05-23 17:11:14 +00:00
void visitStatements(std::vector<Statement> const& _statements);
2017-06-14 13:35:51 +00:00
/// Pops all variables declared in the block and checks that the stack height is equal
/// to @a _blackStartStackHeight.
void finalizeBlock(Block const& _block, int _blockStartStackHeight);
void generateMultiAssignment(std::vector<Identifier> const& _variableNames);
void generateAssignment(Identifier const& _variableName);
2017-05-23 17:11:14 +00:00
/// Determines the stack height difference to the given variables. Throws
/// if it is not yet in scope or the height difference is too large. Returns
/// the (positive) stack height difference otherwise.
int variableHeightDiff(Scope::Variable const& _var, YulString _name, bool _forSwap);
2017-05-23 17:11:14 +00:00
2017-09-20 07:52:10 +00:00
void expectDeposit(int _deposit, int _oldHeight) const;
2017-05-23 17:11:14 +00:00
2017-09-20 07:52:10 +00:00
void checkStackHeight(void const* _astElement) const;
2017-05-23 17:11:14 +00:00
2019-01-29 09:51:25 +00:00
/// Stores the stack error in the list of errors, appends an invalid opcode
/// and corrects the stack height to the target stack height.
void stackError(StackTooDeepError _error, int _targetStackSize);
/// Ensures stack height is down to @p _targetDepth by appending POP instructions to the output assembly.
/// Returns the number of POP statements that have been appended.
int appendPopUntil(int _targetDepth);
AbstractAssembly& m_assembly;
AsmAnalysisInfo& m_info;
Scope* m_scope = nullptr;
2018-12-06 23:56:16 +00:00
EVMDialect const& m_dialect;
BuiltinContext& m_builtinContext;
bool const m_allowStackOpt = true;
2018-12-06 23:56:16 +00:00
bool const m_evm15 = false;
bool const m_useNamedLabelsForFunctions = false;
2017-05-23 17:11:14 +00:00
ExternalIdentifierAccess m_identifierAccess;
/// Adjustment between the stack height as determined during the analysis phase
/// and the stack height in the assembly. This is caused by an initial stack being present
/// for inline assembly and different stack heights depending on the EVM backend used
/// (EVM 1.0 or 1.5).
int m_stackAdjustment = 0;
2017-06-13 19:58:25 +00:00
std::shared_ptr<Context> m_context;
/// Set of variables whose reference counter has reached zero,
/// and whose stack slot will be marked as unused once we reach
/// statement level in the scope where the variable was defined.
std::set<Scope::Variable const*> m_variablesScheduledForDeletion;
std::set<int> m_unusedStackSlots;
2019-01-29 09:51:25 +00:00
std::vector<StackTooDeepError> m_stackErrors;
2017-05-23 17:11:14 +00:00
};
2017-05-23 17:21:14 +00:00
}