diff --git a/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs b/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs index b928e217f5..466d6e182e 100644 --- a/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs +++ b/gateway/src/node/client_handling/websocket/connection_handler/fresh.rs @@ -26,6 +26,7 @@ use std::time::Duration; use thiserror::Error; use tokio::io::{AsyncRead, AsyncWrite}; use tokio_tungstenite::tungstenite::{protocol::Message, Error as WsError}; +use tracing::field::debug; use tracing::*; use crate::node::client_handling::websocket::common_state::CommonHandlerState; @@ -298,6 +299,7 @@ where /// /// * `client_address`: address of the client that is going to receive the messages. /// * `shared_keys`: shared keys derived between the client and the gateway used to encrypt and tag the messages. + #[instrument(skip_all)] async fn push_stored_messages_to_client( &mut self, client_address: DestinationAddressBytes, @@ -306,6 +308,7 @@ where where S: AsyncRead + AsyncWrite + Unpin, { + debug!("attempting to push stored messages to client"); let mut start_next_after = None; loop { // retrieve some messages @@ -521,6 +524,7 @@ where /// * `client_address`: address of the client wishing to authenticate. /// * `encrypted_address`: ciphertext of the address of the client wishing to authenticate. /// * `iv`: fresh IV received with the request. + #[instrument(skip_all)] async fn handle_authenticate( &mut self, client_protocol_version: Option, @@ -531,6 +535,7 @@ where where S: AsyncRead + AsyncWrite + Unpin, { + debug("handling client authentication"); let negotiated_protocol = self.negotiate_client_protocol(client_protocol_version)?; // populate the negotiated protocol for future uses self.negotiated_protocol = Some(negotiated_protocol); @@ -625,6 +630,7 @@ where /// # Arguments /// /// * `init_data`: init payload of the registration handshake. + #[instrument(skip_all)] async fn handle_register( &mut self, client_protocol_version: Option, @@ -633,6 +639,7 @@ where where S: AsyncRead + AsyncWrite + Unpin + Send, { + debug!("handling client registration"); let negotiated_protocol = self.negotiate_client_protocol(client_protocol_version)?; // populate the negotiated protocol for future uses self.negotiated_protocol = Some(negotiated_protocol); @@ -645,7 +652,9 @@ where } let shared_keys = self.perform_registration_handshake(init_data).await?; + debug!("managed to derived shared keys"); let client_id = self.register_client(remote_address, &shared_keys).await?; + event!(Level::DEBUG, client_id, protocol = negotiated_protocol); let client_details = ClientDetails::new(client_id, remote_address, shared_keys);