Moved on to the state manager

3 files changed, 29 insertions, 27 deletions

diff --git a/ethchain/state_manager.go b/ethchain/state_manager.go
index 0923b017e..c2a31ff6c 100644
--- a/ethchain/state_manager.go
+++ b/ethchain/state_manager.go
@@ -62,6 +62,7 @@ func NewStateManager(ethereum EthManager) *StateManager {
 		addrStateStore: NewAddrStateStore(),
 		bc:             ethereum.BlockChain(),
 	}
+	sm.procState = ethereum.BlockChain().CurrentBlock.State()
 
 	return sm
 }
@@ -72,8 +73,8 @@ func (sm *StateManager) ProcState() *State {
 
 // Watches any given address and puts it in the address state store
 func (sm *StateManager) WatchAddr(addr []byte) *AccountState {
-	//FIXME account := sm.procState.GetAccount(addr)
-	account := sm.bc.CurrentBlock.state.GetAccount(addr)
+	//XXX account := sm.bc.CurrentBlock.state.GetAccount(addr)
+	account := sm.procState.GetAccount(addr)
 
 	return sm.addrStateStore.Add(addr, account)
 }
@@ -104,16 +105,16 @@ func (sm *StateManager) ApplyTransactions(block *Block, txs []*Transaction) {
 	for _, tx := range txs {
 		// If there's no recipient, it's a contract
 		if tx.IsContract() {
-			//FIXME sm.MakeContract(tx)
-			block.MakeContract(tx)
+			sm.MakeContract(tx)
+			//XXX block.MakeContract(tx)
 		} else {
-			//FIXME if contract := procState.GetContract(tx.Recipient); contract != nil {
-			if contract := block.state.GetContract(tx.Recipient); contract != nil {
+			if contract := sm.procState.GetContract(tx.Recipient); contract != nil {
+				//XXX if contract := block.state.GetContract(tx.Recipient); contract != nil {
 				sm.ProcessContract(contract, tx, block)
 			} else {
 				err := sm.Ethereum.TxPool().ProcessTransaction(tx, block)
 				if err != nil {
-					ethutil.Config.Log.Infoln("[smGR]", err)
+					ethutil.Config.Log.Infoln("[STATE]", err)
 				}
 			}
 		}
@@ -165,21 +166,21 @@ func (sm *StateManager) ProcessBlock(block *Block) error {
 
 	// I'm not sure, but I don't know if there should be thrown
 	// any errors at this time.
-	if err := sm.AccumelateRewards(sm.bc.CurrentBlock, block); err != nil {
+	if err := sm.AccumelateRewards(block); err != nil {
 		return err
 	}
 
 	// if !sm.compState.Cmp(sm.procState)
-	if !block.state.Cmp(sm.bc.CurrentBlock.state) {
-		return fmt.Errorf("Invalid merkle root. Expected %x, got %x", block.State().trie.Root, sm.bc.CurrentBlock.State().trie.Root)
-		//FIXME return fmt.Errorf("Invalid merkle root. Expected %x, got %x", sm.compState.trie.Root, sm.procState.trie.Root)
+	if !sm.compState.Cmp(sm.procState) {
+		//XXX return fmt.Errorf("Invalid merkle root. Expected %x, got %x", block.State().trie.Root, sm.bc.CurrentBlock.State().trie.Root)
+		return fmt.Errorf("Invalid merkle root. Expected %x, got %x", sm.compState.trie.Root, sm.procState.trie.Root)
 	}
 
 	// Calculate the new total difficulty and sync back to the db
 	if sm.CalculateTD(block) {
 		// Sync the current block's state to the database and cancelling out the deferred Undo
-		sm.bc.CurrentBlock.Sync()
-		//FIXME sm.procState.Sync()
+		//XXX sm.bc.CurrentBlock.Sync()
+		sm.procState.Sync()
 
 		// Broadcast the valid block back to the wire
 		//sm.Ethereum.Broadcast(ethwire.MsgBlockTy, []interface{}{block.Value().Val})
@@ -193,7 +194,7 @@ func (sm *StateManager) ProcessBlock(block *Block) error {
 			sm.SecondaryBlockProcessor.ProcessBlock(block)
 		}
 
-		ethutil.Config.Log.Infof("[smGR] Added block #%d (%x)\n", block.BlockInfo().Number, block.Hash())
+		ethutil.Config.Log.Infof("[STATE] Added block #%d (%x)\n", block.BlockInfo().Number, block.Hash())
 	} else {
 		fmt.Println("total diff failed")
 	}
@@ -270,22 +271,22 @@ func CalculateUncleReward(block *Block) *big.Int {
 	return UncleReward
 }
 
-func (sm *StateManager) AccumelateRewards(processor *Block, block *Block) error {
+func (sm *StateManager) AccumelateRewards(block *Block) error {
 	// Get the coinbase rlp data
-	addr := processor.state.GetAccount(block.Coinbase)
-	//FIXME addr := proc.GetAccount(block.Coinbase)
+	//XXX addr := processor.state.GetAccount(block.Coinbase)
+	addr := sm.procState.GetAccount(block.Coinbase)
 	// Reward amount of ether to the coinbase address
 	addr.AddFee(CalculateBlockReward(block, len(block.Uncles)))
 
-	processor.state.UpdateAccount(block.Coinbase, addr)
-	//FIXME proc.UpdateAccount(block.Coinbase, addr)
+	//XXX processor.state.UpdateAccount(block.Coinbase, addr)
+	sm.procState.UpdateAccount(block.Coinbase, addr)
 
 	for _, uncle := range block.Uncles {
-		uncleAddr := processor.state.GetAccount(uncle.Coinbase)
+		uncleAddr := sm.procState.GetAccount(uncle.Coinbase)
 		uncleAddr.AddFee(CalculateUncleReward(uncle))
 
-		processor.state.UpdateAccount(uncle.Coinbase, uncleAddr)
-		//FIXME proc.UpdateAccount(uncle.Coinbase, uncleAddr)
+		//processor.state.UpdateAccount(uncle.Coinbase, uncleAddr)
+		sm.procState.UpdateAccount(uncle.Coinbase, uncleAddr)
 	}
 
 	return nil
@@ -306,8 +307,8 @@ func (sm *StateManager) ProcessContract(contract *Contract, tx *Transaction, blo
 	*/
 
 	vm := &Vm{}
-	//vm.Process(contract, sm.procState, RuntimeVars{
-	vm.Process(contract, block.state, RuntimeVars{
+	//vm.Process(contract, block.state, RuntimeVars{
+	vm.Process(contract, sm.procState, RuntimeVars{
 		address:     tx.Hash()[12:],
 		blockNumber: block.BlockInfo().Number,
 		sender:      tx.Sender(),
diff --git a/ethchain/transaction_pool.go b/ethchain/transaction_pool.go
index 2c9a26936..fdc386303 100644
--- a/ethchain/transaction_pool.go
+++ b/ethchain/transaction_pool.go
@@ -233,11 +233,11 @@ func (pool *TxPool) Start() {
 }
 
 func (pool *TxPool) Stop() {
-	log.Println("[TXP] Stopping...")
-
 	close(pool.quit)
 
 	pool.Flush()
+
+	log.Println("[TXP] Stopped")
 }
 
 func (pool *TxPool) Subscribe(channel chan TxMsg) {
diff --git a/peer.go b/peer.go
index 22a4c32fd..89b567fb6 100644
--- a/peer.go
+++ b/peer.go
@@ -293,7 +293,8 @@ func (p *Peer) HandleInbound() {
 				var err error
 				for i := msg.Data.Len() - 1; i >= 0; i-- {
 					block = ethchain.NewBlockFromRlpValue(msg.Data.Get(i))
-					// FIXME p.ethereum.BlockManager.DefaultPrepare(block)
+
+					p.ethereum.StateManager().PrepareDefault(block)
 					err = p.ethereum.StateManager().ProcessBlock(block)
 
 					if err != nil {