Merge #610: Add timeout to UDP tracker requests

fec9716 feat: [#609] add timeout to UDP tracker requests (Jose Celano)

Pull request description:

  Add timeout to UDP tracker requests.

ACKs for top commit:
  josecelano:
    ACK fec9716

Tree-SHA512: 6f7f915007a6a1ba6f0bca512327e2760adb92d6af63ad5d96ab2d635569d85911dcce705acd9cc6b382ef71a57081c112b9ce3d3ef9f864a905bba7e4a7cdf7

Author: josecelano

src/servers/udp/server.rs

@@ -20,6 +20,7 @@
 use std::io::Cursor;
 use std::net::SocketAddr;
 use std::sync::Arc;
+use std::time::Duration;
 
 use aquatic_udp_protocol::Response;
 use derive_more::Constructor;
@@ -240,9 +241,16 @@ impl Udp {
                             debug!(target: "UDP Tracker", "From: {}", &remote_addr);
                             debug!(target: "UDP Tracker", "Payload: {:?}", payload);
 
-                            let response = handle_packet(remote_addr, payload, &tracker).await;
+                            let response_fut = handle_packet(remote_addr, payload, &tracker);
 
-                            Udp::send_response(socket_clone, remote_addr, response).await;
+                            match tokio::time::timeout(Duration::from_secs(5), response_fut).await {
+                                Ok(response) => {
+                                    Udp::send_response(socket_clone, remote_addr, response).await;
+                                }
+                                Err(_) => {
+                                    error!("Timeout occurred while processing the UDP request.");
+                                }
+                            }
                         }
                         Err(err) => {
                             error!("Error reading UDP datagram from socket. Error: {:?}", err);