Accessors for multiple mappings implemented

2 files changed, 33 insertions, 34 deletions

diff --git a/ExpressionCompiler.cpp b/ExpressionCompiler.cpp
index 722ac989..45e0e80b 100644
--- a/ExpressionCompiler.cpp
+++ b/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,
diff --git a/Types.cpp b/Types.cpp
index 7fa4561e..cfb852c2 100644
--- a/Types.cpp
+++ b/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);