g3proxy: set expire info for socks5 peers in proxy_float

bytedance · Oct 25, 2024 · ab19798 · ab19798
1 parent 83919ec
commit ab19798
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/g3proxy/src/escape/proxy_float/peer/socks5/mod.rs b/g3proxy/src/escape/proxy_float/peer/socks5/mod.rs
@@ -272,6 +272,7 @@ impl NextProxyPeer for ProxyFloatSocks5Peer {
         task_notes: &ServerTaskNotes,
         task_stats: ArcUdpConnectTaskRemoteStats,
     ) -> UdpConnectResult {
+        udp_notes.expire = self.expire_datetime();
         self.udp_connect_to(escaper, udp_notes, task_notes, task_stats)
             .await
     }
@@ -283,6 +284,7 @@ impl NextProxyPeer for ProxyFloatSocks5Peer {
         task_notes: &ServerTaskNotes,
         task_stats: ArcUdpRelayTaskRemoteStats,
     ) -> UdpRelaySetupResult {
+        udp_notes.expire = self.expire_datetime();
         self.udp_setup_relay(escaper, udp_notes, task_notes, task_stats)
             .await
     }

diff --git a/g3proxy/src/escape/proxy_float/peer/socks5s/mod.rs b/g3proxy/src/escape/proxy_float/peer/socks5s/mod.rs
@@ -251,6 +251,7 @@ impl NextProxyPeer for ProxyFloatSocks5sPeer {
         task_notes: &ServerTaskNotes,
         task_stats: ArcUdpConnectTaskRemoteStats,
     ) -> UdpConnectResult {
+        udp_notes.expire = self.expire_datetime();
         self.udp_connect_to(escaper, udp_notes, task_notes, task_stats)
             .await
     }
@@ -262,6 +263,7 @@ impl NextProxyPeer for ProxyFloatSocks5sPeer {
         task_notes: &ServerTaskNotes,
         task_stats: ArcUdpRelayTaskRemoteStats,
     ) -> UdpRelaySetupResult {
+        udp_notes.expire = self.expire_datetime();
         self.udp_setup_relay(escaper, udp_notes, task_notes, task_stats)
             .await
     }