Add validation for total account lamports

src/test_suite/test_suite.py

@@ -45,7 +45,7 @@ def execute_single_instruction(
     ),
 ):
     _, instruction_context = generate_test_case(file)
-    assert instruction_context is not None, f"Unable to read {file.name}"
+    assert instruction_context is not None, f"Unable to read/validate {file.name}"
 
     # Initialize output buffers and shared library
     initialize_process_output_buffers(randomize_output_buffer=randomize_output_buffer)
@@ -81,7 +81,7 @@ def debug_instruction(
 
     # Decode the file and pass it into GDB
     _, instruction_context = generate_test_case(file)
-    assert instruction_context is not None, f"Unable to read {file.name}"
+    assert instruction_context is not None, f"Unable to read/validate {file.name}"
     debug_host(shared_library, instruction_context, gdb=debugger)
 
 

src/test_suite/validation_utils.py

@@ -1,6 +1,9 @@
 import test_suite.invoke_pb2 as pb
 
 
+ULONG_MAX = 18446744073709551615
+
+
 def is_valid(instruction_context: pb.InstrContext) -> bool:
     """
     Checks whether an instruction context message is valid.
@@ -20,4 +23,15 @@ def is_valid(instruction_context: pb.InstrContext) -> bool:
         if not account.address or len(account.address) != 32:
             return False
 
+    # Sum of lamports should not exceed ulong max value
+    total_lamports = 0
+    for instruction_account in instruction_context.instr_accounts:
+        account_lamports = instruction_context.accounts[
+            instruction_account.index
+        ].lamports
+        total_lamports += account_lamports
+
+        if total_lamports > ULONG_MAX:
+            return False
+
     return instruction_context.data and instruction_context.instr_accounts