nik/chorus
1
0
Fork 0
mirror of https://github.com/mikedilger/chorus.git synced 2026-03-04 06:36:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Log user agent

Browse Source
This commit is contained in:
Mike Dilger 2024-02-20 20:45:54 +13:00
parent 5e7c1c386e
commit c0f77caf58
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.rs
View File

@ -230,6 +230,11 @@ async fn handle_http_request(
peer: SocketAddr,
mut request: Request<Body>,
) -> Result<Response<Body>, Error> {
let ua = match request.headers().get("user-agent") {
Some(ua) => ua.to_str().unwrap_or("NON-UTF8-HEADER").to_owned(),
None => "".to_owned(),
};
if hyper_tungstenite::is_upgrade_request(&request) {
let web_socket_config = WebSocketConfig {
max_write_buffer_size: 1024 * 1024, // 1 MB
@ -259,9 +264,10 @@ async fn handle_http_request(
let old_num_websockets = GLOBALS.num_clients.fetch_add(1, Ordering::SeqCst);
log::info!(
"{}: websocket started (making {} active websockets)",
"{}: TOTAL={}, Connection: {}",
peer,
old_num_websockets + 1
old_num_websockets + 1,
ua
);
// Everybody gets a 4-second ban on disconnect to prevent
@ -290,11 +296,7 @@ async fn handle_http_request(
// Decrement count of active websockets
let old_num_websockets = GLOBALS.num_clients.fetch_sub(1, Ordering::SeqCst);
log::info!(
"{}: websocket ended (making {} active websockets)",
peer,
old_num_websockets - 1
);
log::info!("{}: TOTAL={}, Disconnection", peer, old_num_websockets - 1);
// Ban for the appropriate duration
Globals::ban(peer.ip(), ban_seconds).await;