Removed regular ints from the virtual machine and closures

1 files changed, 7 insertions, 2 deletions

diff --git a/ethchain/contract.go b/ethchain/contract.go
index 93d2b68ba..f7ae01753 100644
--- a/ethchain/contract.go
+++ b/ethchain/contract.go
@@ -39,12 +39,17 @@ func (c *Contract) State() *State {
 	return c.state
 }
 
-func (c *Contract) GetMem(num int64) *ethutil.Value {
-	nb := ethutil.BigToBytes(big.NewInt(num), 256)
+func (c *Contract) GetMem(num *big.Int) *ethutil.Value {
+	nb := ethutil.BigToBytes(num, 256)
 
 	return c.Addr(nb)
 }
 
+func (c *Contract) SetMem(num *big.Int, val *ethutil.Value) {
+	addr := ethutil.BigToBytes(num, 256)
+	c.state.trie.Update(string(addr), string(val.Encode()))
+}
+
 // 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)