Function selector and variable (un)packing.

1 files changed, 8 insertions, 0 deletions

diff --git a/Types.h b/Types.h
index 828f4809..d7c3b241 100644
--- a/Types.h
+++ b/Types.h
@@ -68,6 +68,10 @@ public:
 	virtual bool operator==(Type const& _other) const { return getCategory() == _other.getCategory(); }
 	virtual bool operator!=(Type const& _other) const { return !this->operator ==(_other); }
 
+	/// @returns number of bytes used by this type when encoded for CALL, or 0 if the encoding
+	/// is not a simple big-endian encoding or the type cannot be stored on the stack.
+	virtual unsigned getCalldataEncodedSize() const { return 0; }
+
 	virtual std::string toString() const = 0;
 	virtual u256 literalValue(Literal const&) const { assert(false); }
 };
@@ -93,6 +97,8 @@ public:
 
 	virtual bool operator==(Type const& _other) const override;
 
+	virtual unsigned getCalldataEncodedSize() const { return m_bits / 8; }
+
 	virtual std::string toString() const override;
 	virtual u256 literalValue(Literal const& _literal) const override;
 
@@ -121,6 +127,8 @@ public:
 		return _operator == Token::NOT || _operator == Token::DELETE;
 	}
 
+	virtual unsigned getCalldataEncodedSize() const { return 1; }
+
 	virtual std::string toString() const override { return "bool"; }
 	virtual u256 literalValue(Literal const& _literal) const override;
 };