Add timeout and rate limit for http server

Jackarain · Sep 4, 2024 · 8793e16 · 8793e16
1 parent fd42ca7
commit 8793e16
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/proxy/include/proxy/proxy_server.hpp b/proxy/include/proxy/proxy_server.hpp
@@ -4235,6 +4235,8 @@ R"x*x*x(<html>
 			char* buf = bufs.get();
 			std::streamsize total = 0;
 
+			stream_rate_limit(m_local_socket, m_option.tcp_rate_limit_);
+
 			do
 			{
 				auto bytes_transferred = fileop::read(file, std::span<char>(buf, buf_size));
@@ -4255,6 +4257,8 @@ R"x*x*x(<html>
 					res.body().more = true;
 				}
 
+				stream_expires_after(m_local_socket, std::chrono::seconds(m_option.tcp_timeout_));
+
 				co_await http::async_write(
 					m_local_socket,
 					sr,