Accessors for multiple mappings implemented

libsolidity/ExpressionCompiler.cpp

@@ -22,6 +22,7 @@
 
 #include <utility>
 #include <numeric>
+#include <boost/range/adaptor/reversed.hpp>
 #include <libdevcore/Common.h>
 #include <libdevcrypto/SHA3.h>
 #include <libsolidity/AST.h>
@@ -847,36 +848,34 @@ unsigned ExpressionCompiler::appendExpressionCopyToMemory(Type const& _expectedT
 
 void ExpressionCompiler::appendStateVariableAccessor(VariableDeclaration const& _varDecl)
 {
-	TypePointer resultType = _varDecl.getType();
-	auto mappingType = dynamic_cast<const MappingType*>(resultType.get());
+	FunctionType thisType(_varDecl);
+	solAssert(thisType.getReturnParameterTypes().size() == 1, "");
+	TypePointer const& resultType = thisType.getReturnParameterTypes().front();
 	unsigned sizeOnStack;
 
-	if (mappingType != nullptr)
-	{
-		// this copies from ExpressionCompiler::visit(IndexAccess .. ) for mapping access
-		TypePointer const& keyType = mappingType->getKeyType();
-		unsigned length = appendTypeConversionAndMoveToMemory(*keyType, *keyType, Location());
-		solAssert(length == 32, "Mapping key has to take 32 bytes in memory (for now).");
-		// @todo move this once we actually use memory
-		m_context << m_context.getStorageLocationOfVariable(_varDecl);
-		length += CompilerUtils(m_context).storeInMemory(length);
-		m_context << u256(length) << u256(0) << eth::Instruction::SHA3;
-		m_currentLValue = LValue(m_context, LValue::STORAGE, *mappingType->getValueType());
-		m_currentLValue.retrieveValue(mappingType->getValueType(), Location(), true);
-		m_currentLValue.reset();
+	unsigned length = 0;
+	TypePointers const& params = thisType.getParameterTypes();
+	// move arguments to memory
+	for (TypePointer const& param: boost::adaptors::reverse(params))
+		length += appendTypeConversionAndMoveToMemory(*param, *param, Location(), length);
 
-		resultType = mappingType->getValueType();
-	}
-	else
+	// retrieve the position of the mapping
+	m_context << m_context.getStorageLocationOfVariable(_varDecl);
+
+	for (TypePointer const& param: params)
 	{
-		m_currentLValue.fromStateVariable(_varDecl, _varDecl.getType());
-		solAssert(m_currentLValue.isInStorage(), "");
-		m_currentLValue.retrieveValue(_varDecl.getType(), Location(), true);
+		// move offset to memory
+		CompilerUtils(m_context).storeInMemory(length);
+		unsigned argLen = CompilerUtils::getPaddedSize(param->getCalldataEncodedSize());
+		length -= argLen;
+		m_context << u256(argLen + 32) << u256(length) << eth::Instruction::SHA3;
 	}
-	sizeOnStack = _varDecl.getType()->getSizeOnStack();
+
+	m_currentLValue = LValue(m_context, LValue::STORAGE, *resultType);
+	m_currentLValue.retrieveValue(resultType, Location(), true);
+	sizeOnStack = resultType->getSizeOnStack();
 	solAssert(sizeOnStack <= 15, "Stack too deep.");
 	m_context << eth::dupInstruction(sizeOnStack + 1) << eth::Instruction::JUMP;
-
 }
 
 ExpressionCompiler::LValue::LValue(CompilerContext& _compilerContext, LValueType _type, Type const& _dataType,

libsolidity/Types.cpp

@@ -626,20 +626,20 @@ FunctionType::FunctionType(VariableDeclaration const& _varDecl):
 	TypePointers retParams;
 	vector<string> retParamNames;
 	TypePointer varDeclType = _varDecl.getType();
-	auto mappingType = dynamic_cast<const MappingType*>(varDeclType.get());
-	if (mappingType!= nullptr)
-	{
-		params.push_back(mappingType->getKeyType());
-		paramNames.push_back(mappingType->getKeyType()->toString());
+	auto mappingType = dynamic_cast<MappingType const*>(varDeclType.get());
+	auto returnType = varDeclType;
 
-		retParams.push_back(mappingType->getValueType());
-		retParamNames.push_back(mappingType->getValueType()->toString());
-	}
-	else // elelemntary type
+	while (mappingType!= nullptr)
 	{
-		retParams.push_back(varDeclType);
-		retParamNames.push_back(_varDecl.getName());
+		params.push_back(mappingType->getKeyType());
+		paramNames.push_back("");
+		returnType = mappingType->getValueType();
+		mappingType = dynamic_cast<MappingType const*>(mappingType->getValueType().get());
 	}
+
+	retParams.push_back(returnType);
+	retParamNames.push_back("");
+
 	swap(params, m_parameterTypes);
 	swap(paramNames, m_parameterNames);
 	swap(retParams, m_returnParameterTypes);

test/SolidityEndToEndTest.cpp

@@ -925,16 +925,19 @@ BOOST_AUTO_TEST_CASE(complex_accessors)
 							 "  mapping(uint256 => string4) to_string_map;\n"
 							 "  mapping(uint256 => bool) to_bool_map;\n"
 							 "  mapping(uint256 => uint256) to_uint_map;\n"
+							 "  mapping(uint256 => mapping(uint256 => uint256)) to_multiple_map;\n"
 							 "  function test() {\n"
 							 "    to_string_map[42] = \"24\";\n"
 							 "    to_bool_map[42] = false;\n"
 							 "    to_uint_map[42] = 12;\n"
+							 "    to_multiple_map[42][23] = 31;\n"
 							 "  }\n"
 							 "}\n";
 	compileAndRun(sourceCode);
 	BOOST_CHECK(callContractFunction("to_string_map(uint256)", 42) == encodeArgs("24"));
 	BOOST_CHECK(callContractFunction("to_bool_map(uint256)", 42) == encodeArgs(false));
 	BOOST_CHECK(callContractFunction("to_uint_map(uint256)", 42) == encodeArgs(12));
+	BOOST_CHECK(callContractFunction("to_multiple_map(uint256,uint256)", 42, 23) == encodeArgs(31));
 }
 
 BOOST_AUTO_TEST_CASE(balance)

test/SolidityNameAndTypeResolution.cpp

@@ -638,6 +638,7 @@ BOOST_AUTO_TEST_CASE(state_variable_accessors)
 					   "  }\n"
 					   "uint256 foo;\n"
 					   "mapping(uint=>string4) map;\n"
+					   "mapping(uint=>mapping(uint=>string4)) multiple_map;\n"
 					   "}\n";
 
 	ASTPointer<SourceUnit> source;
@@ -649,10 +650,20 @@ BOOST_AUTO_TEST_CASE(state_variable_accessors)
 	auto returnParams = function->getReturnParameterTypeNames();
 	BOOST_CHECK_EQUAL(returnParams.at(0), "uint256");
 	BOOST_CHECK(function->isConstant());
+
 	function = retrieveFunctionBySignature(contract, "map(uint256)");
 	BOOST_REQUIRE(function && function->hasDeclaration());
-	auto Params = function->getParameterTypeNames();
-	BOOST_CHECK_EQUAL(returnParams.at(0), "uint256");
+	auto params = function->getParameterTypeNames();
+	BOOST_CHECK_EQUAL(params.at(0), "uint256");
+	returnParams = function->getReturnParameterTypeNames();
+	BOOST_CHECK_EQUAL(returnParams.at(0), "string4");
+	BOOST_CHECK(function->isConstant());
+
+	function = retrieveFunctionBySignature(contract, "multiple_map(uint256,uint256)");
+	BOOST_REQUIRE(function && function->hasDeclaration());
+	params = function->getParameterTypeNames();
+	BOOST_CHECK_EQUAL(params.at(0), "uint256");
+	BOOST_CHECK_EQUAL(params.at(1), "uint256");
 	returnParams = function->getReturnParameterTypeNames();
 	BOOST_CHECK_EQUAL(returnParams.at(0), "string4");
 	BOOST_CHECK(function->isConstant());