stability fix

Author: cnbatch

src/main.cpp

@@ -16,15 +16,15 @@
 int main(int argc, char *argv[])
 {
 #ifdef __cpp_lib_format
-	std::cout << std::format("{} version 20241026\n", app_name);
+	std::cout << std::format("{} version 20241027\n", app_name);
 	if (argc <= 1)
 	{
 		std::cout << std::format("Usage: {} config1.conf\n", app_name);
 		std::cout << std::format("       {} config1.conf config2.conf...\n", (int)app_name.length(), app_name.data());
 		return 0;
 	}
 #else
-	std::cout << app_name << " version 20241026\n";
+	std::cout << app_name << " version 20241027\n";
 	if (argc <= 1)
 	{
 		std::cout << "Usage: " << app_name << " config1.conf\n";

src/networks/client.cpp

@@ -499,11 +499,11 @@ void client_mode::cleanup_expiring_forwarders()
 		int64_t time_elapsed = calculate_difference(time_right_now, expire_time);
 
 		if (time_elapsed > CLEANUP_WAITS / 3 &&
-			time_elapsed <= CLEANUP_WAITS / 3 * 2 &&
+			time_elapsed <= CLEANUP_WAITS * 2 / 3 &&
 			forwarder_ptr != nullptr)
 			forwarder_ptr->pause(true);
 
-		if (time_elapsed > CLEANUP_WAITS / 3 * 2 &&
+		if (time_elapsed > CLEANUP_WAITS * 2 / 3 &&
 			forwarder_ptr != nullptr)
 			forwarder_ptr->stop();
 

src/networks/connections.cpp

@@ -344,7 +344,7 @@ void udp_server::handle_receive(std::unique_ptr<uint8_t[]> buffer_cache, const a
 {
 	if (error)
 	{
-		if (!connection_socket.is_open())
+		if (error == asio::error::operation_aborted || !connection_socket.is_open())
 			return;
 	}
 
@@ -396,6 +396,9 @@ void udp_client::pause(bool set_as_pause)
 
 void udp_client::stop()
 {
+	bool expect = true;
+	if (stopped.compare_exchange_strong(expect, true))
+		return;
 	stopped.store(true);
 	callback = empty_udp_callback;
 	if (connection_socket.is_open())
@@ -544,7 +547,10 @@ void udp_client::start_receive()
 
 	std::unique_ptr<uint8_t[]> buffer_cache = std::make_unique<uint8_t[]>(BUFFER_SIZE);
 	uint8_t *buffer_raw_ptr = buffer_cache.get();
-	connection_socket.async_receive_from(asio::buffer(buffer_raw_ptr, BUFFER_SIZE), incoming_endpoint,
+	auto asio_buffer = asio::buffer(buffer_raw_ptr, BUFFER_SIZE);
+	if (!connection_socket.is_open())
+		return;
+	connection_socket.async_receive_from(asio_buffer, incoming_endpoint,
 		[buffer = std::move(buffer_cache), this](const asio::error_code &error, std::size_t bytes_transferred) mutable
 		{
 			handle_receive(std::move(buffer), error, bytes_transferred);
@@ -555,7 +561,7 @@ void udp_client::handle_receive(std::unique_ptr<uint8_t[]> buffer_cache, const a
 {
 	if (error)
 	{
-		if (!stopped.load() && !paused.load() && connection_socket.is_open())
+		if (error != asio::error::operation_aborted && !stopped.load() && !paused.load() && connection_socket.is_open())
 			start_receive();
 		return;
 	}

src/networks/relay.cpp

@@ -748,11 +748,11 @@ void relay_mode::cleanup_expiring_data_connections()
 		std::shared_ptr<forwarder> egress_forwarder = std::atomic_load(&udp_session_ptr->egress_forwarder);
 
 		if (time_elapsed > CLEANUP_WAITS / 3 &&
-			time_elapsed <= CLEANUP_WAITS / 3 * 2 &&
+			time_elapsed <= CLEANUP_WAITS * 2 / 3 &&
 			egress_forwarder != nullptr)
 			egress_forwarder->pause(true);
 
-		if (time_elapsed > CLEANUP_WAITS / 3 * 2 &&
+		if (time_elapsed > CLEANUP_WAITS * 2 / 3 &&
 			egress_forwarder != nullptr)
 			egress_forwarder->stop();