diff --git a/cpp/protoutils.cc b/cpp/protoutils.cc index 2c72117..b07d9e3 100644 --- a/cpp/protoutils.cc +++ b/cpp/protoutils.cc @@ -66,6 +66,7 @@ void Net_PackMsg(unsigned short msgid, ::google::protobuf::Message& msg, std::st int Net_SendMsg(a8::TcpClient* tcp_client, unsigned seqid, unsigned short msgid, ::google::protobuf::Message& msg) { int packlen = msg.ByteSize(); + assert(packlen < 1024 * 60); char* buff = (char*)malloc(sizeof(PackHead) + packlen); PackHead* head = (PackHead*)buff; @@ -83,6 +84,7 @@ int Net_SendMsg(a8::TcpListener* tcp_listener, unsigned short socket_handle, uns unsigned short msgid, ::google::protobuf::Message& msg) { int packlen = msg.ByteSize(); + assert(packlen < 1024 * 60); char* buff = (char*)malloc(sizeof(PackHead) + packlen); PackHead* head = (PackHead*)buff; @@ -100,6 +102,7 @@ int Net_BroadcastMsg(a8::TcpListener* tcp_listener, unsigned int seqid, unsigned short msgid, ::google::protobuf::Message& msg) { int packlen = msg.ByteSize(); + assert(packlen < 1024 * 60); char* buff = (char*)malloc(sizeof(PackHead) + packlen); PackHead* head = (PackHead*)buff; @@ -116,6 +119,7 @@ int Net_BroadcastMsg(a8::TcpListener* tcp_listener, unsigned int seqid, int Net_SendProxyCMsg(a8::TcpClient* tcp_client, unsigned short msgid, ::google::protobuf::Message& msg) { int packlen = msg.ByteSize(); + assert(packlen < 1024 * 60); char* buff = (char*)malloc(sizeof(WSProxyPackHead_S) + packlen); WSProxyPackHead_C* head = (WSProxyPackHead_C*)buff; @@ -149,6 +153,7 @@ int Net_SendProxyMsg(a8::TcpListener* tcp_listener, int socket_handle, unsigned unsigned short child_socket_handle = socket_handle & 0xFFFF; int packlen = msg.ByteSize(); + assert(packlen < 1024 * 60); char* buff = (char*)malloc(sizeof(WSProxyPackHead_S) + packlen); WSProxyPackHead_S* head = (WSProxyPackHead_S*)buff;