Compare commits
2 Commits
dc2fefdae2
...
59477c3ea7
| Author | SHA1 | Date | |
|---|---|---|---|
| 59477c3ea7 | |||
| 88b1cb8076 |
2
.gitignore
vendored
2
.gitignore
vendored
@@ -1 +1,3 @@
|
|||||||
/target
|
/target
|
||||||
|
**/.venv/**
|
||||||
|
|
||||||
|
|||||||
12
python/admin.py
Normal file
12
python/admin.py
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
import socket
|
||||||
|
|
||||||
|
SERVER_IP = "127.0.0.1"
|
||||||
|
PORT = 5901
|
||||||
|
|
||||||
|
# Manually set the starting "input" for the first team
|
||||||
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
s.connect((SERVER_IP, PORT))
|
||||||
|
# Using "0" as the ID for the very first team's source
|
||||||
|
s.send("ADMIN SET 0 80".encode()) # Sets the first clue as 80
|
||||||
|
print(s.recv(1024).decode())
|
||||||
|
s.close()
|
||||||
30
python/team.py
Normal file
30
python/team.py
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
import socket
|
||||||
|
|
||||||
|
# Config
|
||||||
|
SERVER_IP = "127.0.0.1" # Change to master server's IP
|
||||||
|
PORT = 5901
|
||||||
|
MY_ID = "1" # Team Number
|
||||||
|
PREV_ID = "0" # Team Number of Team Sending Clue (should be previous team, but in case of abscences)
|
||||||
|
|
||||||
|
# 1. GET: Wait for input
|
||||||
|
print(f"Waiting for Team {PREV_ID}...")
|
||||||
|
while True:
|
||||||
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
s.connect((SERVER_IP, PORT))
|
||||||
|
s.send(f"GET {PREV_ID}".encode())
|
||||||
|
input = s.recv(1024).decode().strip()
|
||||||
|
s.close()
|
||||||
|
|
||||||
|
if input != "-1": # -1 Means Clue Is Not Yet Sent
|
||||||
|
print(f"Received input: {input}")
|
||||||
|
break
|
||||||
|
|
||||||
|
# 2. TRANSFORM: Do your logic to convert input to output clue, which should also be an integer Number
|
||||||
|
output = int(input) + 42
|
||||||
|
|
||||||
|
# 3. PUT: Send output
|
||||||
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
s.connect((SERVER_IP, PORT))
|
||||||
|
s.send(f"PUT {MY_ID} {output}".encode())
|
||||||
|
s.close()
|
||||||
|
print(f"Sent output: {output}")
|
||||||
100
src/main.rs
100
src/main.rs
@@ -1,3 +1,99 @@
|
|||||||
fn main() {
|
use std::collections::HashMap;
|
||||||
println!("Hello, world!");
|
use std::sync::Arc;
|
||||||
|
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||||
|
use tokio::net::{TcpListener, TcpStream};
|
||||||
|
use tokio::sync::RwLock; // Use Tokio's async-aware lock
|
||||||
|
|
||||||
|
// Type alias for our shared thread-safe state
|
||||||
|
type Db = Arc<RwLock<HashMap<String, String>>>;
|
||||||
|
|
||||||
|
#[tokio::main]
|
||||||
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
|
// Bind to all interfaces on port 5901
|
||||||
|
let listener = TcpListener::bind("0.0.0.0:5901").await?;
|
||||||
|
let db: Db = Arc::new(RwLock::new(HashMap::new()));
|
||||||
|
|
||||||
|
println!("🚀 Escape Room Server active on port 5901");
|
||||||
|
println!("Waiting for teams to connect...");
|
||||||
|
|
||||||
|
loop {
|
||||||
|
let (socket, _) = listener.accept().await?;
|
||||||
|
let db_clone = Arc::clone(&db);
|
||||||
|
|
||||||
|
// Spawn a new async task for every connection
|
||||||
|
tokio::spawn(async move {
|
||||||
|
if let Err(e) = handle_connection(socket, db_clone).await {
|
||||||
|
eprintln!("[SOCKET ERROR] {}", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn handle_connection(mut socket: TcpStream, db: Db) -> std::io::Result<()> {
|
||||||
|
let mut buf = [0; 1024];
|
||||||
|
let n = socket.read(&mut buf).await?;
|
||||||
|
|
||||||
|
// Connection closed
|
||||||
|
if n == 0 {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
let request = String::from_utf8_lossy(&buf[..n]);
|
||||||
|
let parts: Vec<&str> = request.trim().split_whitespace().collect();
|
||||||
|
|
||||||
|
if parts.is_empty() {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
match parts[0].to_uppercase().as_str() {
|
||||||
|
// Teams submit their solution: PUT [team_id] [value]
|
||||||
|
"PUT" => {
|
||||||
|
if parts.len() == 3 {
|
||||||
|
let team_id = parts[1].to_string();
|
||||||
|
let val = parts[2].to_string();
|
||||||
|
|
||||||
|
let mut data = db.write().await; // Async lock
|
||||||
|
data.insert(team_id.clone(), val.clone());
|
||||||
|
|
||||||
|
socket.write_all(b"ACK\n").await?;
|
||||||
|
println!("[LOG] Team {} submitted solution: {}", team_id, val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Teams request previous solution: GET [prev_team_id]
|
||||||
|
"GET" => {
|
||||||
|
if parts.len() == 2 {
|
||||||
|
let target_id = parts[1];
|
||||||
|
let data = db.read().await; // Async lock
|
||||||
|
|
||||||
|
let response = data
|
||||||
|
.get(target_id)
|
||||||
|
.cloned()
|
||||||
|
.unwrap_or_else(|| "-1".to_string());
|
||||||
|
|
||||||
|
socket.write_all(response.as_bytes()).await?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Admin override: ADMIN SET [team_id] [value]
|
||||||
|
"ADMIN" => {
|
||||||
|
if parts.len() == 4 && parts[1].to_uppercase() == "SET" {
|
||||||
|
let team_id = parts[2].to_string();
|
||||||
|
let val = parts[3].to_string();
|
||||||
|
|
||||||
|
let mut data = db.write().await;
|
||||||
|
data.insert(team_id.clone(), val.clone());
|
||||||
|
|
||||||
|
socket.write_all(b"ADMIN_OK\n").await?;
|
||||||
|
println!("[ADMIN] Manual set: Team {} -> {}", team_id, val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_ => {
|
||||||
|
socket.write_all(b"ERR_INVALID_CMD\n").await?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user