From b92f2efd81fe41ac2f1e27fe9a2ee22fb26ef820 Mon Sep 17 00:00:00 2001
From: ssrlive <30760636+ssrlive@users.noreply.github.com>
Date: Tue, 8 Aug 2023 09:20:43 +0800
Subject: [PATCH] remove .expect call

---
 src/error.rs     | 3 +++
 src/tun2proxy.rs | 5 ++---
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/error.rs b/src/error.rs
index b0c916d..bae52ba 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -27,6 +27,9 @@ pub enum Error {
     #[error("smoltcp::socket::tcp::SendError {0:?}")]
     Send(#[from] smoltcp::socket::tcp::SendError),
 
+    #[error("smoltcp::socket::udp::SendError {0:?}")]
+    UdpSend(#[from] smoltcp::socket::udp::SendError),
+
     #[error("smoltcp::wire::Error {0:?}")]
     Wire(#[from] smoltcp::wire::Error),
 
diff --git a/src/tun2proxy.rs b/src/tun2proxy.rs
index bd68a76..b2dd44a 100644
--- a/src/tun2proxy.rs
+++ b/src/tun2proxy.rs
@@ -530,9 +530,8 @@ impl<'a> TunToProxy<'a> {
                             udp::PacketBuffer::new(vec![udp::PacketMetadata::EMPTY], vec![0; 4096]);
                         let mut socket = udp::Socket::new(rx_buffer, tx_buffer);
                         socket.bind(dst)?;
-                        socket
-                            .send_slice(response.as_slice(), UdpMetadata::from(connection_info.src))
-                            .expect("failed to send DNS response");
+                        let meta = UdpMetadata::from(connection_info.src);
+                        socket.send_slice(response.as_slice(), meta)?;
                         let handle = self.sockets.add(socket);
                         self.expect_smoltcp_send()?;
                         self.sockets.remove(handle);