Support multiple variables in a variable declaration in inline assembly

This commit is contained in:
Alex Beregszaszi 2017-05-05 18:35:40 +01:00
parent 74d2e7311a
commit 15b4d4def2
6 changed files with 71 additions and 15 deletions

View File

@ -181,12 +181,32 @@ bool AsmAnalyzer::operator()(FunctionalAssignment const& _assignment)
bool AsmAnalyzer::operator()(assembly::VariableDeclaration const& _varDecl) bool AsmAnalyzer::operator()(assembly::VariableDeclaration const& _varDecl)
{ {
// The number of variable names and values must match. One exception
// is a single value, where a tuple assignment is assumed from a function.
if (_varDecl.variables.size() != _varDecl.values.size())
{
if (_varDecl.values.size() != 1)
{
m_errors.push_back(make_shared<Error>(
Error::Type::DeclarationError,
"Variable declaration name and value count mismatch.",
_varDecl.location
));
return false;
}
}
int const expectedItems = _varDecl.variables.size();
int const stackHeight = m_stackHeight; int const stackHeight = m_stackHeight;
bool success = boost::apply_visitor(*this, *_varDecl.value); for (auto const& value: _varDecl.values)
solAssert(m_stackHeight - stackHeight == 1, "Invalid value size."); if (!boost::apply_visitor(*this, value))
boost::get<Scope::Variable>(m_currentScope->identifiers.at(_varDecl.variable.name)).active = true; return false;
solAssert(m_stackHeight - stackHeight == expectedItems, "Invalid value size.");
for (auto const& variable: _varDecl.variables)
boost::get<Scope::Variable>(m_currentScope->identifiers.at(variable.name)).active = true;
m_info.stackHeightInfo[&_varDecl] = m_stackHeight; m_info.stackHeightInfo[&_varDecl] = m_stackHeight;
return success; return true;
} }
bool AsmAnalyzer::operator()(assembly::FunctionDefinition const& _funDef) bool AsmAnalyzer::operator()(assembly::FunctionDefinition const& _funDef)

View File

@ -251,11 +251,16 @@ public:
void operator()(assembly::VariableDeclaration const& _varDecl) void operator()(assembly::VariableDeclaration const& _varDecl)
{ {
int height = m_assembly.stackHeight(); int height = m_assembly.stackHeight();
boost::apply_visitor(*this, *_varDecl.value); int expectedItems = _varDecl.variables.size();
expectDeposit(1, height); for (auto const& value: _varDecl.values)
auto& var = boost::get<Scope::Variable>(m_scope.identifiers.at(_varDecl.variable.name)); boost::apply_visitor(*this, value);
var.stackHeight = height; expectDeposit(expectedItems, height);
var.active = true; for (auto const& variable: _varDecl.variables)
{
auto& var = boost::get<Scope::Variable>(m_scope.identifiers.at(variable.name));
var.stackHeight = height++;
var.active = true;
}
} }
void operator()(assembly::Block const& _block) void operator()(assembly::Block const& _block)
{ {

View File

@ -65,7 +65,7 @@ struct FunctionalAssignment { SourceLocation location; Identifier variableName;
struct FunctionalInstruction { SourceLocation location; Instruction instruction; std::vector<Statement> arguments; }; struct FunctionalInstruction { SourceLocation location; Instruction instruction; std::vector<Statement> arguments; };
struct FunctionCall { SourceLocation location; Identifier functionName; std::vector<Statement> arguments; }; struct FunctionCall { SourceLocation location; Identifier functionName; std::vector<Statement> arguments; };
/// Block-scope variable declaration ("let x:u256 := mload(20:u256)"), non-hoisted /// Block-scope variable declaration ("let x:u256 := mload(20:u256)"), non-hoisted
struct VariableDeclaration { SourceLocation location; TypedName variable; std::shared_ptr<Statement> value; }; struct VariableDeclaration { SourceLocation location; TypedNameList variables; std::vector<Statement> values; };
/// Block that creates a scope (frees declared stack variables) /// Block that creates a scope (frees declared stack variables)
struct Block { SourceLocation location; std::vector<Statement> statements; }; struct Block { SourceLocation location; std::vector<Statement> statements; };
/// Function definition ("function f(a, b) -> (d, e) { ... }") /// Function definition ("function f(a, b) -> (d, e) { ... }")

View File

@ -258,11 +258,25 @@ assembly::VariableDeclaration Parser::parseVariableDeclaration()
{ {
VariableDeclaration varDecl = createWithLocation<VariableDeclaration>(); VariableDeclaration varDecl = createWithLocation<VariableDeclaration>();
expectToken(Token::Let); expectToken(Token::Let);
varDecl.variable = parseTypedName(); while (true)
{
varDecl.variables.push_back(parseTypedName());
if (m_scanner->currentToken() == Token::Comma)
expectToken(Token::Comma);
else
break;
}
expectToken(Token::Colon); expectToken(Token::Colon);
expectToken(Token::Assign); expectToken(Token::Assign);
varDecl.value.reset(new Statement(parseExpression())); while (true)
varDecl.location.end = locationOf(*varDecl.value).end; {
varDecl.values.emplace_back(new Statement(parseExpression()));
if (m_scanner->currentToken() == Token::Comma)
expectToken(Token::Comma);
else
break;
}
varDecl.location.end = locationOf(varDecl.values.back()).end;
return varDecl; return varDecl;
} }

View File

@ -121,7 +121,21 @@ string AsmPrinter::operator()(assembly::FunctionalAssignment const& _functionalA
string AsmPrinter::operator()(assembly::VariableDeclaration const& _variableDeclaration) string AsmPrinter::operator()(assembly::VariableDeclaration const& _variableDeclaration)
{ {
return "let " + _variableDeclaration.variable.name + appendTypeName(_variableDeclaration.variable.type) + " := " + boost::apply_visitor(*this, *_variableDeclaration.value); string out = "let ";
out += boost::algorithm::join(
_variableDeclaration.variables | boost::adaptors::transformed(
[this](TypedName variable) { return variable.name + appendTypeName(variable.type); }
),
", "
);
out += " := ";
out += boost::algorithm::join(
_variableDeclaration.values | boost::adaptors::transformed(
[this](Statement statement) { return boost::apply_visitor(*this, statement); }
),
", "
);
return out;
} }
string AsmPrinter::operator()(assembly::FunctionDefinition const& _functionDefinition) string AsmPrinter::operator()(assembly::FunctionDefinition const& _functionDefinition)

View File

@ -59,7 +59,10 @@ bool ScopeFiller::operator()(Label const& _item)
bool ScopeFiller::operator()(assembly::VariableDeclaration const& _varDecl) bool ScopeFiller::operator()(assembly::VariableDeclaration const& _varDecl)
{ {
return registerVariable(_varDecl.variable, _varDecl.location, *m_currentScope); for (auto const& variable: _varDecl.variables)
if (!registerVariable(variable, _varDecl.location, *m_currentScope))
return false;
return true;
} }
bool ScopeFiller::operator()(assembly::FunctionDefinition const& _funDef) bool ScopeFiller::operator()(assembly::FunctionDefinition const& _funDef)