From e7528c6af8a9b313b4bb20358c939f16b693e9cb Mon Sep 17 00:00:00 2001
From: Christian <c@ethdev.com>
Date: Sun, 22 Feb 2015 18:38:32 +0100
Subject: Stack height checks and fix.

---
 Types.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'Types.cpp')

diff --git a/Types.cpp b/Types.cpp
index 55dedd92..ce51336d 100644
--- a/Types.cpp
+++ b/Types.cpp
@@ -833,10 +833,17 @@ string FunctionType::toString() const
 
 unsigned FunctionType::getSizeOnStack() const
 {
+	Location location = m_location;
+	if (m_location == Location::SetGas || m_location == Location::SetValue)
+	{
+		solAssert(m_returnParameterTypes.size() == 1, "");
+		location = dynamic_cast<FunctionType const&>(*m_returnParameterTypes.front()).m_location;
+	}
+
 	unsigned size = 0;
-	if (m_location == Location::External)
+	if (location == Location::External)
 		size = 2;
-	else if (m_location == Location::Internal || m_location == Location::Bare)
+	else if (location == Location::Internal || location == Location::Bare)
 		size = 1;
 	if (m_gasSet)
 		size++;
-- 
cgit