torrust_tracker/servers/udp/server/
processor.rs

1use std::io::Cursor;
2use std::net::SocketAddr;
3use std::sync::Arc;
4
5use aquatic_udp_protocol::Response;
6use tracing::{instrument, Level};
7
8use super::bound_socket::BoundSocket;
9use crate::core::Tracker;
10use crate::servers::udp::{handlers, RawRequest};
11
12pub struct Processor {
13    socket: Arc<BoundSocket>,
14    tracker: Arc<Tracker>,
15}
16
17impl Processor {
18    pub fn new(socket: Arc<BoundSocket>, tracker: Arc<Tracker>) -> Self {
19        Self { socket, tracker }
20    }
21
22    #[instrument(skip(self, request))]
23    pub async fn process_request(self, request: RawRequest) {
24        let from = request.from;
25        let response = handlers::handle_packet(request, &self.tracker, self.socket.address()).await;
26        self.send_response(from, response).await;
27    }
28
29    #[instrument(skip(self))]
30    async fn send_response(self, target: SocketAddr, response: Response) {
31        tracing::debug!("send response");
32
33        let response_type = match &response {
34            Response::Connect(_) => "Connect".to_string(),
35            Response::AnnounceIpv4(_) => "AnnounceIpv4".to_string(),
36            Response::AnnounceIpv6(_) => "AnnounceIpv6".to_string(),
37            Response::Scrape(_) => "Scrape".to_string(),
38            Response::Error(e) => format!("Error: {e:?}"),
39        };
40
41        let mut writer = Cursor::new(Vec::with_capacity(200));
42
43        match response.write_bytes(&mut writer) {
44            Ok(()) => {
45                let bytes_count = writer.get_ref().len();
46                let payload = writer.get_ref();
47
48                let () = match self.send_packet(&target, payload).await {
49                    Ok(sent_bytes) => {
50                        if tracing::event_enabled!(Level::TRACE) {
51                            tracing::debug!(%bytes_count, %sent_bytes, ?payload, "sent {response_type}");
52                        } else {
53                            tracing::debug!(%bytes_count, %sent_bytes, "sent {response_type}");
54                        }
55                    }
56                    Err(error) => tracing::warn!(%bytes_count, %error, ?payload, "failed to send"),
57                };
58            }
59            Err(e) => {
60                tracing::error!(%e, "error");
61            }
62        }
63    }
64
65    #[instrument(skip(self))]
66    async fn send_packet(&self, target: &SocketAddr, payload: &[u8]) -> std::io::Result<usize> {
67        tracing::trace!("send packet");
68
69        // doesn't matter if it reaches or not
70        self.socket.send_to(payload, target).await
71    }
72}
torrust_tracker/servers/udp/server/processor.rs

torrust_tracker/servers/udp/server/
processor.rs
