From d8dcc892444315a1c28fce206a3a0691fc44c2d5 Mon Sep 17 00:00:00 2001 From: azw Date: Sun, 7 May 2023 07:42:01 +0000 Subject: [PATCH] 1 --- server/wsproxy/upstream.cc | 3 +-- server/wsproxy/upstream.h | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/server/wsproxy/upstream.cc b/server/wsproxy/upstream.cc index b76df15..f6bdbab 100644 --- a/server/wsproxy/upstream.cc +++ b/server/wsproxy/upstream.cc @@ -106,7 +106,7 @@ void UpStream::SendStockMsg() void UpStream::ForwardClientMsg(f8::MsgHdr& hdr) { - char* buff = (char*)malloc(sizeof(f8::WSProxyPackHead_C) + hdr.buflen); + char* buff = forward_buff_.data(); memset(buff, 0, sizeof(f8::WSProxyPackHead_C)); f8::WSProxyPackHead_C* head = (f8::WSProxyPackHead_C*)buff; head->packlen = hdr.buflen; @@ -124,7 +124,6 @@ void UpStream::ForwardClientMsg(f8::MsgHdr& hdr) } tcp_client_->SendBuff(buff, sizeof(f8::WSProxyPackHead_C) + head->packlen); - free(buff); } void UpStream::ForwardClientMsgEx(f8::MsgHdr* hdr) diff --git a/server/wsproxy/upstream.h b/server/wsproxy/upstream.h index b371546..28ce739 100644 --- a/server/wsproxy/upstream.h +++ b/server/wsproxy/upstream.h @@ -65,6 +65,7 @@ class UpStream private: std::array recv_buff_; + std::array forward_buff_; unsigned int recv_bufflen_ = 0; std::shared_ptr tcp_client_; f8::TimerWp timer_wp_;