From 1aa5c94115977e7f619bf249312dfe8db9270077 Mon Sep 17 00:00:00 2001
From: kevin860104 <55239515+kevin860104@users.noreply.github.com>
Date: Tue, 28 May 2024 21:28:02 +0800
Subject: [PATCH] Update sol.py

---
 lab7/sol.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/lab7/sol.py b/lab7/sol.py
index e69de29b..f9ff28ae 100644
--- a/lab7/sol.py
+++ b/lab7/sol.py
@@ -0,0 +1,19 @@
+import angr
+
+import sys
+
+proj = angr.Project('./login')
+init_state = proj.factory.entry_state()
+simulation = proj.factory.simgr(init_state)
+
+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())
+
+simulation.explore(find=success_condition, avoid=fail_condition)
+
+solution = simulation.found[0]
+
+print(solution.posix.dumps(sys.stdin.fileno()))