add missing exceptions to prague

Author: souradeep-das

src/ethereum/exceptions.py

@@ -39,3 +39,13 @@ class InvalidSignatureError(InvalidTransaction):
     """
     Thrown when a transaction has an invalid signature.
     """
+
+class InsufficientBalanceError(InvalidTransaction):
+    """
+    Thrown when a transaction cannot be executed due to insufficient sender funds.
+    """
+
+class NonceMismatchError(InvalidTransaction):
+    """
+    Thrown when a transaction's nonce does not match the expected nonce for the sender.
+    """

src/ethereum/prague/exceptions.py

@@ -22,3 +22,52 @@ class TransactionTypeError(InvalidTransaction):
     def __init__(self, transaction_type: int):
         super().__init__(f"unknown transaction type `{transaction_type}`")
         self.transaction_type = transaction_type
+
+class TransactionTypeContractCreationError(InvalidTransaction):
+    """
+    Transaction type is not allowed for contract creation.
+    """
+
+    transaction_type: Final[int]
+    """
+    The type byte of the transaction that caused the error.
+    """
+
+    def __init__(self, transaction_type: int):
+        super().__init__(f"transaction type `{transaction_type}` not allowed for contract creation")
+        self.transaction_type = transaction_type
+
+class BlobGasLimitExceededError(InvalidTransaction):
+    """
+    The blob gas limit for the transaction exceeds the maximum allowed.
+    """
+
+class InsufficientMaxFeePerBlobGasError(InvalidTransaction):
+    """
+    The maximum fee per blob gas is insufficient for the transaction.
+    """
+
+class InsufficientMaxFeePerGasError(InvalidTransaction):
+    """
+    The maximum fee per gas is insufficient for the transaction.
+    """
+
+class InvalidBlobVersionedHashError(InvalidTransaction):
+    """
+    The versioned hash of the blob is invalid.
+    """
+
+class NoBlobDataError(InvalidTransaction):
+    """
+    The transaction does not contain any blob data.
+    """
+
+class PriorityFeeGreaterThanMaxFeeError(InvalidTransaction):
+    """
+    The priority fee is greater than the maximum fee per gas.
+    """
+
+class EmptyAuthorizationListError(InvalidTransaction):
+    """
+    The authorization list in the transaction is empty.
+    """

src/ethereum/prague/fork.py

@@ -24,11 +24,23 @@
     EthereumException,
     InvalidBlock,
     InvalidSenderError,
+    InsufficientBalanceError,
+    NonceMismatchError
 )
 
 from . import vm
 from .blocks import Block, Header, Log, Receipt, Withdrawal, encode_receipt
 from .bloom import logs_bloom
+from .exceptions import (
+    TransactionTypeContractCreationError,
+    BlobGasLimitExceededError,
+    InsufficientMaxFeePerBlobGasError,
+    InsufficientMaxFeePerGasError,
+    InvalidBlobVersionedHashError,
+    NoBlobDataError,
+    PriorityFeeGreaterThanMaxFeeError,
+    EmptyAuthorizationListError
+)
 from .fork_types import Account, Address, Authorization, VersionedHash
 from .requests import (
     CONSOLIDATION_REQUEST_TYPE,
@@ -409,7 +421,7 @@ def check_transaction(
 
     tx_blob_gas_used = calculate_total_blob_gas(tx)
     if tx_blob_gas_used > blob_gas_available:
-        raise InvalidBlock
+        raise BlobGasLimitExceededError("blob gas limit exceeded")
 
     sender_address = recover_sender(block_env.chain_id, tx)
     sender_account = get_account(block_env.state, sender_address)
@@ -418,9 +430,9 @@ def check_transaction(
         tx, (FeeMarketTransaction, BlobTransaction, SetCodeTransaction)
     ):
         if tx.max_fee_per_gas < tx.max_priority_fee_per_gas:
-            raise InvalidBlock
+            raise PriorityFeeGreaterThanMaxFeeError("priority fee greater than max fee")
         if tx.max_fee_per_gas < block_env.base_fee_per_gas:
-            raise InvalidBlock
+            raise InsufficientMaxFeePerGasError("insufficient max fee per gas")
 
         priority_fee_per_gas = min(
             tx.max_priority_fee_per_gas,
@@ -436,14 +448,14 @@ def check_transaction(
 
     if isinstance(tx, BlobTransaction):
         if len(tx.blob_versioned_hashes) == 0:
-            raise InvalidBlock
+            raise NoBlobDataError("no blob data in transaction")
         for blob_versioned_hash in tx.blob_versioned_hashes:
             if blob_versioned_hash[0:1] != VERSIONED_HASH_VERSION_KZG:
-                raise InvalidBlock
+                raise InvalidBlobVersionedHashError("invalid blob versioned hash")
 
         blob_gas_price = calculate_blob_gas_price(block_env.excess_blob_gas)
         if Uint(tx.max_fee_per_blob_gas) < blob_gas_price:
-            raise InvalidBlock
+            raise InsufficientMaxFeePerBlobGasError("insufficient max fee per blob gas")
 
         max_gas_fee += Uint(calculate_total_blob_gas(tx)) * Uint(
             tx.max_fee_per_blob_gas
@@ -454,16 +466,19 @@ def check_transaction(
 
     if isinstance(tx, (BlobTransaction, SetCodeTransaction)):
         if not isinstance(tx.to, Address):
-            raise InvalidBlock
+            raise TransactionTypeContractCreationError(tx[0])
 
     if isinstance(tx, SetCodeTransaction):
         if not any(tx.authorizations):
-            raise InvalidBlock
+            raise EmptyAuthorizationListError("empty authorization list")
 
     if sender_account.nonce != tx.nonce:
-        raise InvalidBlock
+        if sender_account.nonce > tx.nonce:
+            raise NonceMismatchError("nonce too low")
+        else:
+            raise NonceMismatchError("nonce too high")
     if Uint(sender_account.balance) < max_gas_fee + Uint(tx.value):
-        raise InvalidBlock
+        raise InsufficientBalanceError("insufficient sender balance")
     if sender_account.code and not is_valid_delegation(sender_account.code):
         raise InvalidSenderError("not EOA")