Update sol.py

Author: as10968574

lab7/sol.py

@@ -1,9 +1,8 @@
 import angr
-
 import sys
-
 proj = angr.Project('./login')
 init_state = proj.factory.entry_state()
+
 simulation = proj.factory.simgr(init_state)
 
 def success_condition(state):
@@ -14,6 +13,9 @@ def fail_condition(state):
 
 simulation.explore(find=success_condition, avoid=fail_condition)
 
-solution = simulation.found[0]
-
-print(solution.posix.dumps(sys.stdin.fileno()))
+if simulation.found:
+    solution = simulation.found[0]
+    password = solution.posix.dumps(sys.stdin.fileno()).strip()
+    print(password.decode()) 
+else:
+    print("Password not found.")