From 82d0f65dab253e215349ea685382bca9672378d8 Mon Sep 17 00:00:00 2001
From: obscuren <geffobscura@gmail.com>
Date: Thu, 20 Mar 2014 17:25:11 +0100
Subject: [PATCH] Comply to Callee structure

---
 ethchain/contract.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/ethchain/contract.go b/ethchain/contract.go
index 21ac828fe9..10c5e2df67 100644
--- a/ethchain/contract.go
+++ b/ethchain/contract.go
@@ -43,12 +43,17 @@ func (c *Contract) State() *State {
 	return c.state
 }
 
-func (c *Contract) GetMem(num int) *ethutil.Value {
-	nb := ethutil.BigToBytes(big.NewInt(int64(num)), 256)
+func (c *Contract) GetMem(num int64) *ethutil.Value {
+	nb := ethutil.BigToBytes(big.NewInt(num), 256)
 
 	return c.Addr(nb)
 }
 
+// Return the gas back to the origin. Used by the Virtual machine or Closures
+func (c *Contract) ReturnGas(val *big.Int, state *State) {
+	c.Amount.Add(c.Amount, val)
+}
+
 func MakeContract(tx *Transaction, state *State) *Contract {
 	// Create contract if there's no recipient
 	if tx.IsContract() {