Take special functions that require literals into account.

10 files changed, 51 insertions, 10 deletions

diff --git a/libyul/AsmAnalysis.cpp b/libyul/AsmAnalysis.cpp
index 0ecc5a30..6a7b2b61 100644
--- a/libyul/AsmAnalysis.cpp
+++ b/libyul/AsmAnalysis.cpp
@@ -299,11 +299,14 @@ bool AsmAnalyzer::operator()(FunctionCall const& _funCall)
 	bool success = true;
 	size_t parameters = 0;
 	size_t returns = 0;
+	bool needsLiteralArguments = false;
 	if (BuiltinFunction const* f = m_dialect->builtin(_funCall.functionName.name))
 	{
 		// TODO: compare types, too
 		parameters = f->parameters.size();
 		returns = f->returns.size();
+		if (f->literalArguments)
+			needsLiteralArguments = true;
 	}
 	else if (!m_currentScope->lookup(_funCall.functionName.name, Scope::Visitor(
 		[&](Scope::Variable const&)
@@ -347,8 +350,15 @@ bool AsmAnalyzer::operator()(FunctionCall const& _funCall)
 		}
 
 	for (auto const& arg: _funCall.arguments | boost::adaptors::reversed)
+	{
 		if (!expectExpression(arg))
 			success = false;
+		else if (needsLiteralArguments && arg.type() != typeid(Literal))
+			m_errorReporter.typeError(
+				_funCall.functionName.location,
+				"Function expects direct literals as arguments."
+			);
+	}
 	// Use argument size instead of parameter count to avoid misleading errors.
 	m_stackHeight += int(returns) - int(_funCall.arguments.size());
 	m_info.stackHeightInfo[&_funCall] = m_stackHeight;
diff --git a/libyul/Dialect.h b/libyul/Dialect.h
index 01fd98df..e06a6826 100644
--- a/libyul/Dialect.h
+++ b/libyul/Dialect.h
@@ -44,7 +44,13 @@ struct BuiltinFunction
 	YulString name;
 	std::vector<Type> parameters;
 	std::vector<Type> returns;
-	bool movable;
+	/// If true, calls to this function can be freely moved and copied (as long as their
+	/// arguments are either variables or also movable) without altering the semantics.
+	/// This means the function cannot depend on storage or memory, cannot have any side-effects,
+	/// but it can depend on state that is constant across an EVM-call.
+	bool movable = false;
+	/// If true, can only accept literals as arguments and they cannot be moved to voriables.
+	bool literalArguments = false;
 };
 
 struct Dialect: boost::noncopyable
diff --git a/libyul/backends/evm/EVMDialect.cpp b/libyul/backends/evm/EVMDialect.cpp
index 935f05c6..18502117 100644
--- a/libyul/backends/evm/EVMDialect.cpp
+++ b/libyul/backends/evm/EVMDialect.cpp
@@ -44,7 +44,7 @@ EVMDialect::EVMDialect(AsmFlavour _flavour, bool _objectAccess):
 	if (!m_objectAccess)
 		return;
 
-	addFunction("datasize", 1, 1, true, [this](
+	addFunction("datasize", 1, 1, true, true, [this](
 		FunctionCall const& _call,
 		AbstractAssembly& _assembly,
 		std::function<void()>
@@ -58,7 +58,7 @@ EVMDialect::EVMDialect(AsmFlavour _flavour, bool _objectAccess):
 		else
 			_assembly.appendDataSize(m_subIDs.at(dataName));
 	});
-	addFunction("dataoffset", 1, 1, true, [this](
+	addFunction("dataoffset", 1, 1, true, true, [this](
 		FunctionCall const& _call,
 		AbstractAssembly& _assembly,
 		std::function<void()>
@@ -72,7 +72,7 @@ EVMDialect::EVMDialect(AsmFlavour _flavour, bool _objectAccess):
 		else
 			_assembly.appendDataOffset(m_subIDs.at(dataName));
 	});
-	addFunction("datacopy", 3, 0, false, [](
+	addFunction("datacopy", 3, 0, false, false, [](
 		FunctionCall const&,
 		AbstractAssembly& _assembly,
 		std::function<void()> _visitArguments
@@ -128,6 +128,7 @@ void EVMDialect::addFunction(
 	size_t _params,
 	size_t _returns,
 	bool _movable,
+	bool _literalArguments,
 	std::function<void(FunctionCall const&, AbstractAssembly&, std::function<void()>)> _generateCode
 )
 {
@@ -137,5 +138,6 @@ void EVMDialect::addFunction(
 	f.parameters.resize(_params);
 	f.returns.resize(_returns);
 	f.movable = _movable;
+	f.literalArguments = _literalArguments;
 	f.generateCode = std::move(_generateCode);
 }
diff --git a/libyul/backends/evm/EVMDialect.h b/libyul/backends/evm/EVMDialect.h
index feb00b03..c23833bc 100644
--- a/libyul/backends/evm/EVMDialect.h
+++ b/libyul/backends/evm/EVMDialect.h
@@ -72,6 +72,7 @@ private:
 		size_t _params,
 		size_t _returns,
 		bool _movable,
+		bool _literalArguments,
 		std::function<void(FunctionCall const&, AbstractAssembly&, std::function<void()>)> _generateCode
 	);
 
diff --git a/libyul/optimiser/CommonSubexpressionEliminator.cpp b/libyul/optimiser/CommonSubexpressionEliminator.cpp
index 9b851333..8ce003e7 100644
--- a/libyul/optimiser/CommonSubexpressionEliminator.cpp
+++ b/libyul/optimiser/CommonSubexpressionEliminator.cpp
@@ -25,6 +25,7 @@
 #include <libyul/optimiser/SyntacticalEquality.h>
 #include <libyul/Exceptions.h>
 #include <libyul/AsmData.h>
+#include <libyul/Dialect.h>
 
 using namespace std;
 using namespace dev;
@@ -32,12 +33,24 @@ using namespace yul;
 
 void CommonSubexpressionEliminator::visit(Expression& _e)
 {
+	bool descend = true;
+	// If this is a function call to a function that requires literal arguments,
+	// do not try to simplify there.
+	if (_e.type() == typeid(FunctionCall))
+		if (BuiltinFunction const* builtin = m_dialect.builtin(boost::get<FunctionCall>(_e).functionName.name))
+			if (builtin->literalArguments)
+				// We should not modify function arguments that have to be literals
+				// Note that replacing the function call entirely is fine,
+				// if the function call is movable.
+				descend = false;
+
 	// We visit the inner expression first to first simplify inner expressions,
 	// which hopefully allows more matches.
 	// Note that the DataFlowAnalyzer itself only has code for visiting Statements,
 	// so this basically invokes the AST walker directly and thus post-visiting
 	// is also fine with regards to data flow analysis.
-	DataFlowAnalyzer::visit(_e);
+	if (descend)
+		DataFlowAnalyzer::visit(_e);
 
 	if (_e.type() == typeid(Identifier))
 	{
diff --git a/libyul/optimiser/CommonSubexpressionEliminator.h b/libyul/optimiser/CommonSubexpressionEliminator.h
index 4b7973e2..9f416d9f 100644
--- a/libyul/optimiser/CommonSubexpressionEliminator.h
+++ b/libyul/optimiser/CommonSubexpressionEliminator.h
@@ -26,6 +26,8 @@
 namespace yul
 {
 
+struct Dialect;
+
 /**
  * Optimisation stage that replaces expressions known to be the current value of a variable
  * in scope by a reference to that variable.
diff --git a/libyul/optimiser/ExpressionSplitter.cpp b/libyul/optimiser/ExpressionSplitter.cpp
index a3b2dc11..334e33f4 100644
--- a/libyul/optimiser/ExpressionSplitter.cpp
+++ b/libyul/optimiser/ExpressionSplitter.cpp
@@ -24,6 +24,7 @@
 #include <libyul/optimiser/ASTWalker.h>
 
 #include <libyul/AsmData.h>
+#include <libyul/Dialect.h>
 
 #include <libdevcore/CommonData.h>
 
@@ -43,6 +44,11 @@ void ExpressionSplitter::operator()(FunctionalInstruction& _instruction)
 
 void ExpressionSplitter::operator()(FunctionCall& _funCall)
 {
+	if (BuiltinFunction const* builtin = m_dialect.builtin(_funCall.functionName.name))
+		if (builtin->literalArguments)
+			// We cannot outline function arguments that have to be literals
+			return;
+
 	for (auto& arg: _funCall.arguments | boost::adaptors::reversed)
 		outlineExpression(arg);
 }
diff --git a/libyul/optimiser/ExpressionSplitter.h b/libyul/optimiser/ExpressionSplitter.h
index d4d2b3f6..a72d14ba 100644
--- a/libyul/optimiser/ExpressionSplitter.h
+++ b/libyul/optimiser/ExpressionSplitter.h
@@ -31,6 +31,7 @@ namespace yul
 {
 
 class NameCollector;
+struct Dialect;
 
 
 /**
@@ -57,8 +58,8 @@ class NameCollector;
 class ExpressionSplitter: public ASTModifier
 {
 public:
-	explicit ExpressionSplitter(NameDispenser& _nameDispenser):
-		m_nameDispenser(_nameDispenser)
+	explicit ExpressionSplitter(Dialect const& _dialect, NameDispenser& _nameDispenser):
+		m_dialect(_dialect), m_nameDispenser(_nameDispenser)
 	{ }
 
 	void operator()(FunctionalInstruction&) override;
@@ -77,6 +78,7 @@ private:
 	/// List of statements that should go in front of the currently visited AST element,
 	/// at the statement level.
 	std::vector<Statement> m_statementsToPrefix;
+	Dialect const& m_dialect;
 	NameDispenser& m_nameDispenser;
 };
 
diff --git a/libyul/optimiser/Suite.cpp b/libyul/optimiser/Suite.cpp
index 9b6e1337..e10916ea 100644
--- a/libyul/optimiser/Suite.cpp
+++ b/libyul/optimiser/Suite.cpp
@@ -67,7 +67,7 @@ void OptimiserSuite::run(
 
 	for (size_t i = 0; i < 4; i++)
 	{
-		ExpressionSplitter{dispenser}(ast);
+		ExpressionSplitter{_dialect, dispenser}(ast);
 		SSATransform::run(ast, dispenser);
 		RedundantAssignEliminator::run(_dialect, ast);
 		RedundantAssignEliminator::run(_dialect, ast);
@@ -90,7 +90,7 @@ void OptimiserSuite::run(
 		ExpressionInliner(_dialect, ast).run();
 		UnusedPruner::runUntilStabilised(_dialect, ast);
 
-		ExpressionSplitter{dispenser}(ast);
+		ExpressionSplitter{_dialect, dispenser}(ast);
 		SSATransform::run(ast, dispenser);
 		RedundantAssignEliminator::run(_dialect, ast);
 		RedundantAssignEliminator::run(_dialect, ast);
diff --git a/libyul/optimiser/Suite.h b/libyul/optimiser/Suite.h
index ce40b204..376e9889 100644
--- a/libyul/optimiser/Suite.h
+++ b/libyul/optimiser/Suite.h
@@ -41,7 +41,6 @@ public:
 		Dialect const& _dialect,
 		Block& _ast,
 		AsmAnalysisInfo const& _analysisInfo,
-
 		std::set<YulString> const& _externallyUsedIdentifiers = {}
 	);
 };