Spaces:
Runtime error
Runtime error
Add results endpoint
Browse files- env/fake_server.py +5 -0
- 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)
|