From f5b8f3d41b533d51eb81e895ed9b6aa31d7aaaef Mon Sep 17 00:00:00 2001
From: obscuren <geffobscura@gmail.com>
Date: Fri, 19 Dec 2014 13:59:49 +0100
Subject: [PATCH] Removed OOG check. Revert should always happen.

---
 core/block_manager.go    |  2 +-
 core/execution.go        |  4 +---
 core/state_transition.go | 11 +++++++++--
 3 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/core/block_manager.go b/core/block_manager.go
index 8d319f84e5..20285f8f0e 100644
--- a/core/block_manager.go
+++ b/core/block_manager.go
@@ -129,7 +129,6 @@ done:
 				statelogger.Infoln(err)
 				erroneous = append(erroneous, tx)
 				err = nil
-				continue
 			}
 		}
 
@@ -215,6 +214,7 @@ func (sm *BlockManager) ProcessWithParent(block, parent *types.Block) (td *big.I
 
 	receiptSha := types.DeriveSha(receipts)
 	if bytes.Compare(receiptSha, block.ReceiptSha) != 0 {
+		chainlogger.Debugln(receipts)
 		err = fmt.Errorf("validating receipt root. received=%x got=%x", block.ReceiptSha, receiptSha)
 		return
 	}
diff --git a/core/execution.go b/core/execution.go
index 0b5e0558f1..44dbd3ace0 100644
--- a/core/execution.go
+++ b/core/execution.go
@@ -56,9 +56,7 @@ func (self *Execution) exec(code, contextAddr []byte, caller vm.ClosureRef) (ret
 
 	snapshot := env.State().Copy()
 	defer func() {
-		if vm.IsOOGErr(err) {
-			env.State().Set(snapshot)
-		}
+		env.State().Set(snapshot)
 		chainlogger.Debugf("post state %x\n", env.State().Root())
 	}()
 
diff --git a/core/state_transition.go b/core/state_transition.go
index a54246eba9..a60f31e3e6 100644
--- a/core/state_transition.go
+++ b/core/state_transition.go
@@ -189,12 +189,19 @@ func (self *StateTransition) TransitionState() (ret []byte, err error) {
 		self.rec = MakeContract(msg, self.state)
 
 		ret, err, ref = vmenv.Create(sender, self.rec.Address(), self.msg.Data(), self.gas, self.gasPrice, self.value)
-		ref.SetCode(ret)
+		if err == nil {
+			dataGas := big.NewInt(int64(len(ret)))
+			dataGas.Mul(dataGas, vm.GasCreateByte)
+			if err = self.UseGas(dataGas); err == nil {
+				ref.SetCode(ret)
+			}
+		}
 	} else {
 		ret, err = vmenv.Call(self.From(), self.To().Address(), self.msg.Data(), self.gas, self.gasPrice, self.value)
 	}
+
 	if err != nil {
-		statelogger.Debugln(err)
+		self.UseGas(self.gas)
 	}
 
 	return