vecna
/
lox
mirror of https://gitlab.torproject.org/vecna/lox.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simple lox server 

Right now only issues open invite tokens
This commit is contained in:
Cecylia Bocovich 2022-11-15 19:11:16 -05:00
commit 9992e3c73c
3 changed files with 88 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
crates/lox-distributor/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
/target

16
crates/lox-distributor/Cargo.toml Normal file
View File

@ -0,0 +1,16 @@
[package]
name = "lox-server"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
hyper = "0.13"
tokio = { version = "0.2", features = ["macros", "signal"] }
serde = "1"
serde_with = "1.9.1"
serde_json = "1.0.87"
lox = { git = "https://git-crysp.uwaterloo.ca/iang/lox.git" }

71
crates/lox-distributor/src/main.rs Normal file
View File

@ -0,0 +1,71 @@
use std::{
net::SocketAddr,
sync::{Arc, Mutex},
convert::Infallible,
};
use hyper::{
service::{make_service_fn, service_fn},
Body, Request, Server, Response,
};
use serde::{Deserialize, Serialize};
use serde_with::serde_as;
use serde_json;
use lox::{BridgeDb, OPENINV_LENGTH};
#[serde_as]
#[derive(Serialize, Deserialize)]
pub struct Invite {
#[serde_as(as = "[_; OPENINV_LENGTH]")]
invite: [u8; OPENINV_LENGTH],
}
#[tokio::main]
async fn main() {
// Create and initialize a new db
let mut bridgedb = BridgeDb::new();
for i in &[1u32, 5, 7, 12, 19, 20, 22] {
bridgedb.insert_openinv(*i);
}
let db = Arc::new(Mutex::new(bridgedb));
let new_service = make_service_fn(move |_conn| {
let db = db.clone();
async move {
Ok::<_, Infallible>(service_fn(move |_req: Request<Body>| {
let db = db.clone();
async move {
Ok::<_, Infallible>(generate_invite(db))
}
}))
}
});
let addr = SocketAddr::from(([127, 0, 0, 1], 8001));
let server = Server::bind(&addr).serve(new_service);
let graceful = server.with_graceful_shutdown(shutdown_signal());
println!("Listening on {}", addr);
if let Err(e) = graceful.await {
eprintln!("server error: {}", e);
}
}
async fn shutdown_signal() {
tokio::signal::ctrl_c()
.await
.expect("failed to listen for ctrl+c signal");
}
fn generate_invite(db: Arc<Mutex<lox::BridgeDb>>) -> Response<Body> {
let obj = db.lock().unwrap();
let invite = Invite {
invite: obj.invite(),
};
let token = serde_json::to_string(&invite).unwrap();
Response::new(Body::from(token))
}