From 7dbabffa096b0834bbe87b7457cea4cf4d101461 Mon Sep 17 00:00:00 2001
From: azw <vcl_delphi@126.com>
Date: Mon, 8 May 2023 05:03:30 +0000
Subject: [PATCH] 1

---
 server/wsproxy/mmcache.cc | 10 ++++++++--
 server/wsproxy/mmcache.h  |  2 ++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/server/wsproxy/mmcache.cc b/server/wsproxy/mmcache.cc
index 65bf884..78d0858 100644
--- a/server/wsproxy/mmcache.cc
+++ b/server/wsproxy/mmcache.cc
@@ -4,12 +4,18 @@
 
 void MMCache::Init()
 {
-
+    socket_msg_buf_len_ = 1024 * 1024;
+    udp_msg_buf_len_ = 1024 * 1024;
+    socket_msg_buf_ = (char*)malloc(socket_msg_buf_len_);
+    udp_msg_buf_ = (char*)malloc(udp_msg_buf_len_);
 }
 
 void MMCache::UnInit()
 {
-
+    free(socket_msg_buf_);
+    socket_msg_buf_ = nullptr;
+    free(udp_msg_buf_);
+    udp_msg_buf_ = nullptr;
 }
 
 char* MMCache::AllocSocketMsg(int len)
diff --git a/server/wsproxy/mmcache.h b/server/wsproxy/mmcache.h
index 748ee1d..d2a355b 100644
--- a/server/wsproxy/mmcache.h
+++ b/server/wsproxy/mmcache.h
@@ -22,6 +22,8 @@ public:
 private:
 
     char* socket_msg_buf_ = nullptr;
+    int socket_msg_buf_len_ = 1024;
     char* udp_msg_buf_ = nullptr;
+    int udp_msg_buf_len_ = 1024;
 
 };