Fix noise key

Jackarain · Nov 25, 2023 · deaa31d · deaa31d
1 parent 20313dc
commit deaa31d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/proxy/include/proxy/proxy_server.hpp b/proxy/include/proxy/proxy_server.hpp
@@ -578,7 +578,7 @@ R"x*x*x(<html>
 
 				fvalue = cvalue;
 
-				co_await net::async_read(
+				recv_length = co_await net::async_read(
 					socket,
 					net::buffer(bufs, recv_length),
 					net_awaitable[error]);
@@ -597,8 +597,8 @@ R"x*x*x(<html>
 					static_cast<uint16_t>(bufs[1]) |
 					(static_cast<uint16_t>(bufs[0]) << 8);
 
-				noise.push_back(bufs[0]);
-				noise.push_back(bufs[1]);
+				for (int i = 0; i < recv_length; i++)
+					noise.push_back(bufs[i]);
 
 				len += recv_length;
 				if (len == 1)
@@ -2459,7 +2459,7 @@ R"x*x*x(<html>
 
 				fvalue = cvalue;
 
-				co_await net::async_read(
+				recv_length = co_await net::async_read(
 					socket,
 					net::buffer(bufs, recv_length),
 					net_awaitable[error]);
@@ -2478,8 +2478,8 @@ R"x*x*x(<html>
 					static_cast<uint16_t>(bufs[1]) |
 					(static_cast<uint16_t>(bufs[0]) << 8);
 
-				noise.push_back(bufs[0]);
-				noise.push_back(bufs[1]);
+				for (int i = 0; i < recv_length; i++)
+					noise.push_back(bufs[i]);
 
 				len += recv_length;
 				if (len == 1)