Add some asserts to the tests

Author: mrexodia

tests/example.py

@@ -39,6 +39,7 @@ def old_test():
     print(icicle.MemoryProtection.ReadOnly)
     print(f"Architectures: {icicle.architectures()}")
     vm = icicle.Icicle("x86_64")
+    print(vm)
 
     for name, (offset, size) in vm.reg_list().items():
         value = vm.reg_read(name.lower())
@@ -59,11 +60,10 @@ def old_test():
     data = vm.mem_read(addr, 4)
     print(data, type(data))
 
-    print(vm)
-
 def main():
     old_test()
 
+    print("")
     vm = icicle.Icicle("x86_64", jit=False)
     print(vm)
 
@@ -86,9 +86,10 @@ def main():
     print(status)
     if status == icicle.RunStatus.UnhandledException:
         print(f"code: {vm.exception_code}, value: {hex(vm.exception_value)}")
-        print(f"rip: {hex(vm.reg_read('rip'))}")
+        print(f"rip: {hex(vm.reg_read('rip'))} (OK)")
     else:
         print(f"rax: {hex(vm.reg_read('rax'))}")
+        assert False, "Should be unreachable"
 
 if __name__ == "__main__":
     main()

tests/invalid.py

@@ -18,6 +18,9 @@ def nx():
     status = vm.run()
     print(status)
     print(vm.exception_code, hex(vm.exception_value))
+    # We expect an execution violation at rip
+    assert vm.exception_code == ExceptionCode.ExecViolation
+    assert vm.exception_value == page
 
 def inv_start():
     vm = Icicle("x86_64", jit=False)
@@ -28,6 +31,9 @@ def inv_start():
     status = vm.run()
     print(status)
     print(vm.exception_code, hex(vm.exception_value))
+    # We expect an invalid instruction at rip
+    assert vm.exception_code == ExceptionCode.InvalidInstruction
+    assert vm.exception_value == page
 
 def inv_middle():
     vm = Icicle("x86_64", jit=False)
@@ -38,6 +44,9 @@ def inv_middle():
     status = vm.run()
     print(status)
     print(vm.exception_code, hex(vm.exception_value))
+    # We expect an invalid instruction at rip+2
+    assert vm.exception_code == ExceptionCode.InvalidInstruction
+    assert vm.exception_value == page + 2
 
 def main():
     print("=== NX ===")
@@ -46,6 +55,7 @@ def main():
     inv_start()
     print("=== Invalid instruction (block middle) ===")
     inv_middle()
+    print("\nSUCCESS")
 
 if __name__ == "__main__":
     main()