spidey121 commited on
Commit
08690d1
·
1 Parent(s): 3ff9f99

Add results endpoint

Browse files
Files changed (2) hide show
  1. env/fake_server.py +5 -0
  2. inference.py +11 -3
env/fake_server.py CHANGED
@@ -94,3 +94,8 @@ def status():
94
  ],
95
  "status": "running"
96
  }
 
 
 
 
 
 
94
  ],
95
  "status": "running"
96
  }
97
+ RESULTS = {}
98
+
99
+ @app.route("/results")
100
+ def results():
101
+ return RESULTS
inference.py CHANGED
@@ -1,13 +1,11 @@
1
  import threading
2
  import time
3
- import sys
4
 
5
- from env.fake_server import run_server
6
  from tasks.easy import run as easy
7
  from tasks.medium import run as medium
8
  from tasks.hard import run as hard
9
 
10
-
11
  # Start fake server in background
12
  server_thread = threading.Thread(target=run_server, daemon=True)
13
  server_thread.start()
@@ -45,5 +43,15 @@ while True:
45
  flush=True
46
  )
47
 
 
 
 
 
 
 
 
 
 
 
48
  # wait before next attack cycle
49
  time.sleep(60)
 
1
  import threading
2
  import time
 
3
 
4
+ from env.fake_server import run_server, RESULTS
5
  from tasks.easy import run as easy
6
  from tasks.medium import run as medium
7
  from tasks.hard import run as hard
8
 
 
9
  # Start fake server in background
10
  server_thread = threading.Thread(target=run_server, daemon=True)
11
  server_thread.start()
 
43
  flush=True
44
  )
45
 
46
+ # Update results for judges
47
+ RESULTS["status"] = "completed"
48
+ RESULTS["steps"] = 3
49
+ RESULTS["rewards"] = [easy_score, medium_score, hard_score]
50
+ RESULTS["attacks"] = [
51
+ "brute_force",
52
+ "port_scan",
53
+ "credential_stuffing"
54
+ ]
55
+
56
  # wait before next attack cycle
57
  time.sleep(60)