diff --git a/src/main.rs b/src/main.rs index ab26691..8b2aea4 100644 --- a/src/main.rs +++ b/src/main.rs @@ -32,7 +32,10 @@ async fn connect_handler( Some(val) => val.to_str().unwrap_or("").split(" ").last().unwrap_or(""), None => match req.match_info().get("token") { Some(val) => val, - None => return Err(ErrorUnauthorized("Unauthorized")), + None => match req.query_string().split('=').last() { + Some(val) => val, + None => return Err(ErrorUnauthorized("Unauthorized")), + }, }, };