Improve some compile error

Author: Jackarain

proxy/include/proxy/proxy_server.hpp

@@ -14,6 +14,8 @@
 
 #include "proxy/proxy_session.hpp"
 
+#include <atomic>
+
 
 namespace proxy {
 
@@ -368,6 +370,10 @@ namespace proxy {
 		void udp_tproxy_forward_packet(
 			udp_tproxy_flow_ptr flow, const char* data, std::size_t len);
 
+		// UDP TPROXY 使用 connect-udp 转发数据包 (RFC 9298 capsule).
+		void udp_tproxy_forward_packet_connect_udp(
+			udp_tproxy_flow_ptr flow, const char* data, std::size_t len);
+
 		// 启动 UDP TPROXY 监听协程.
 		net::awaitable<void> start_udp_tproxy_listen(udp::socket& udp_sock) noexcept;
 
@@ -436,7 +442,7 @@ namespace proxy {
 #endif // defined(__linux__)
 
 		// 当前服务是否中止标志.
-		bool m_abort{ false };
+		std::atomic<bool> m_abort{ false };
 	};
 
 }

proxy/include/proxy/proxy_session.hpp

@@ -120,6 +120,7 @@
 
 #include <cstdlib>
 #include <algorithm>
+#include <atomic>
 #include <cstddef>
 #include <cstring>
 #include <filesystem>
@@ -1521,7 +1522,7 @@ namespace proxy {
 		net::ssl::context m_ssl_cli_context{ net::ssl::context::sslv23_client };
 
 		// 当前 session 是否被中止的状态.
-		bool m_abort{ false };
+		std::atomic<bool> m_abort{ false };
 	};
 }
 

proxy/src/proxy_server.cpp

@@ -1922,30 +1922,31 @@ net::awaitable<void> proxy_server::start_udp_tproxy_listen(udp::socket& udp_sock
 
 		size_t flow_key = make_udp_flow_key(client_ep, original_dest);
 
-		// 查找或创建 flow.
+		auto scheme = boost::to_lower_copy(
+			std::string(m_option.proxy_pass_->scheme()));
+		bool using_connect_udp = scheme.starts_with("http");
+
+		// 查找或创建 flow (原子操作, 避免 TOCTOU 竞态条件).
 		std::shared_ptr<udp_tproxy_flow> flow;
+		bool is_new = false;
 		{
 			std::lock_guard<std::mutex> lock(m_udp_flows_mutex);
 
 			auto it = m_udp_tproxy_flows.find(flow_key);
 			if (it != m_udp_tproxy_flows.end())
+			{
 				flow = it->second;
-		}
-
-		auto scheme = boost::to_lower_copy(
-			std::string(m_option.proxy_pass_->scheme()));
-		bool using_connect_udp = scheme.starts_with("http");
-
-		if (!flow)
-		{
-			// 创建一个新的 flow 来处理这个客户端和原始目标地址的通信.
-			flow = std::make_shared<udp_tproxy_flow>(client_ep, original_dest, udp_sock, flow_key);
-
+			}
+			else
 			{
-				std::lock_guard<std::mutex> lock(m_udp_flows_mutex);
+				flow = std::make_shared<udp_tproxy_flow>(client_ep, original_dest, udp_sock, flow_key);
 				m_udp_tproxy_flows[flow_key] = flow;
+				is_new = true;
 			}
+		}
 
+		if (is_new)
+		{
 			if (using_connect_udp)
 			{
 				flow->using_connect_udp_ = true;