Update sol.py

lab7/sol.py

@@ -0,0 +1,16 @@
+import angr, sys
+
+proj = angr.Project("./login")
+
+init_state = proj.factory.entry_state()
+simulation = proj.factory.simgr(init_state)
+simulation.explore(find=success_condition, avoid=fail_condition)
+
+def success_condition(state):
+    return b"Login successful" in state.posix.dumps(sys.stdout.fileno())
+
+def fail_condition(state):
+    return b"Login failed" in state.posix.dumps(sys.stdout.fileno())
+
+solution = simulation.found[0]
+print(solution.posix.dumps(sys.stdin.fileno()))