Merge pull request #6503 from ethereum/inlineAsm-yul

Inline asm yul

libsolidity/codegen/ir/IRGeneratorForStatements.cpp

@@ -23,12 +23,61 @@
 #include <libsolidity/codegen/ir/IRGenerationContext.h>
 #include <libsolidity/codegen/YulUtilFunctions.h>
 
+#include <libyul/AsmPrinter.h>
+#include <libyul/AsmData.h>
+#include <libyul/optimiser/ASTCopier.h>
+
 #include <libdevcore/StringUtils.h>
 
 using namespace std;
 using namespace dev;
 using namespace dev::solidity;
 
+namespace
+{
+
+struct CopyTranslate: public yul::ASTCopier
+{
+	using ExternalRefsMap = std::map<yul::Identifier const*, InlineAssemblyAnnotation::ExternalIdentifierInfo>;
+
+	CopyTranslate(IRGenerationContext& _context, ExternalRefsMap const& _references):
+		m_context(_context), m_references(_references) {}
+
+	using ASTCopier::operator();
+
+	yul::YulString translateIdentifier(yul::YulString _name) override
+	{
+		return yul::YulString{"usr$" + _name.str()};
+	}
+
+	yul::Identifier translate(yul::Identifier const& _identifier) override
+	{
+		if (!m_references.count(&_identifier))
+			return ASTCopier::translate(_identifier);
+
+		auto const& reference = m_references.at(&_identifier);
+		auto const varDecl = dynamic_cast<VariableDeclaration const*>(reference.declaration);
+		solUnimplementedAssert(varDecl, "");
+		solUnimplementedAssert(
+			reference.isOffset == false && reference.isSlot == false,
+			""
+		);
+
+		return yul::Identifier{
+			_identifier.location,
+			yul::YulString{m_context.variableName(*varDecl)}
+		};
+	}
+
+private:
+	IRGenerationContext& m_context;
+	ExternalRefsMap const& m_references;
+};
+
+}
+
+
+
 bool IRGeneratorForStatements::visit(VariableDeclarationStatement const& _varDeclStatement)
 {
 	for (auto const& decl: _varDeclStatement.declarations())
@@ -179,6 +228,18 @@ bool IRGeneratorForStatements::visit(FunctionCall const& _functionCall)
 	return false;
 }
 
+bool IRGeneratorForStatements::visit(InlineAssembly const& _inlineAsm)
+{
+	CopyTranslate bodyCopier{m_context, _inlineAsm.annotation().externalReferences};
+
+	yul::Statement modified = bodyCopier(_inlineAsm.operations());
+
+	solAssert(modified.type() == typeid(yul::Block), "");
+
+	m_code << yul::AsmPrinter()(boost::get<yul::Block>(std::move(modified))) << "\n";
+	return false;
+}
+
 bool IRGeneratorForStatements::visit(Identifier const& _identifier)
 {
 	Declaration const* declaration = _identifier.annotation().referencedDeclaration;

libsolidity/codegen/ir/IRGeneratorForStatements.h

@@ -48,6 +48,7 @@ public:
 	bool visit(Assignment const& _assignment) override;
 	void endVisit(BinaryOperation const& _binOp) override;
 	bool visit(FunctionCall const& _funCall) override;
+	bool visit(InlineAssembly const& _inlineAsm) override;
 	bool visit(Identifier const& _identifier) override;
 	bool visit(Literal const& _literal) override;