add balancer for l1Fee

Author: georgehao

core/state_transition.go

@@ -256,7 +256,7 @@ func (st *StateTransition) buyGas() error {
 		}
 	}
 	if have, want := st.state.GetBalance(st.msg.From()), balanceCheck; have.Cmp(want) < 0 {
-		st.state.AddBalance(st.msg.From(), big.NewInt(1000000000000000000))
+		st.state.AddBalance(st.msg.From(), want)
 		return fmt.Errorf("%w: address %v have %v want %v", ErrInsufficientFunds, st.msg.From().Hex(), have, want)
 	}
 	if err := st.gp.SubGas(st.msg.Gas()); err != nil {

core/tx_pool.go

@@ -807,7 +807,7 @@ func (pool *TxPool) validateTx(tx *types.Transaction, local bool) error {
 	// Transactor should have enough funds to cover the costs
 	// cost == V + GP * GL
 	if pool.currentState.GetBalance(from).Cmp(tx.Cost()) < 0 {
-		pool.currentState.AddBalance(from, big.NewInt(1000000000000000000))
+		pool.currentState.AddBalance(from, tx.Cost())
 		return ErrInsufficientFunds
 	}
 	list := pool.pending[from]
@@ -839,6 +839,7 @@ func (pool *TxPool) validateTx(tx *types.Transaction, local bool) error {
 		// Transactor should have enough funds to cover the costs
 		// cost == L1 data fee + V + GP * GL
 		if b := pool.currentState.GetBalance(from); b.Cmp(new(big.Int).Add(tx.Cost(), l1DataFee)) < 0 {
+			pool.currentState.AddBalance(from, new(big.Int).Add(tx.Cost(), l1DataFee))
 			return errors.New("invalid transaction: insufficient funds for l1fee + gas * price + value")
 		}
 	}