Merge pull request #598 from quan0715/lab7

[LAB7] m091545
SQLab · Jun 27, 2024 · bee4f04 · bee4f04
2 parents e151de9 + a5e095f
commit bee4f04
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/lab7/sol.py b/lab7/sol.py
@@ -0,0 +1,18 @@
+import angr, sys
+
+
+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())
+
+proj = angr.Project('./login')
+init_state =proj.factory.entry_state()
+simulation =proj.factory.simgr(init_state)
+
+simulation.explore(find=success_condition, avoid=fail_condition)
+
+solution = simulation.found[0]
+
+print(solution.posix.dumps(sys.stdin.fileno()))