server_common/wsproxy
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
azw 2023-05-07 07:42:01 +00:00
parent 6e8c4a1387
commit d8dcc89244
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/wsproxy/upstream.cc
View File

@ -106,7 +106,7 @@ void UpStream::SendStockMsg()
void UpStream::ForwardClientMsg(f8::MsgHdr& hdr) 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)); memset(buff, 0, sizeof(f8::WSProxyPackHead_C));
f8::WSProxyPackHead_C* head = (f8::WSProxyPackHead_C*)buff; f8::WSProxyPackHead_C* head = (f8::WSProxyPackHead_C*)buff;
head->packlen = hdr.buflen; head->packlen = hdr.buflen;
@ -124,7 +124,6 @@ void UpStream::ForwardClientMsg(f8::MsgHdr& hdr)
} }
tcp_client_->SendBuff(buff, sizeof(f8::WSProxyPackHead_C) + head->packlen); tcp_client_->SendBuff(buff, sizeof(f8::WSProxyPackHead_C) + head->packlen);
free(buff);
} }
void UpStream::ForwardClientMsgEx(f8::MsgHdr* hdr) void UpStream::ForwardClientMsgEx(f8::MsgHdr* hdr)

1
server/wsproxy/upstream.h
View File

@ -65,6 +65,7 @@ class UpStream
private: private:
std::array<char, PACK_MAX> recv_buff_; std::array<char, PACK_MAX> recv_buff_;
std::array<char, PACK_MAX> forward_buff_;
unsigned int recv_bufflen_ = 0; unsigned int recv_bufflen_ = 0;
std::shared_ptr<a8::TcpClient> tcp_client_; std::shared_ptr<a8::TcpClient> tcp_client_;
f8::TimerWp timer_wp_; f8::TimerWp timer_wp_;