diff --git a/server/wsproxy/target_conn.cc b/server/wsproxy/target_conn.cc index 8c4bda9..41a63bc 100644 --- a/server/wsproxy/target_conn.cc +++ b/server/wsproxy/target_conn.cc @@ -126,7 +126,7 @@ void TargetConn::on_socketread(a8::TcpClient* sender, char* buf, unsigned int le while (recv_bufflen_ - offset >= sizeof(f8::WSProxyPackHead_S)) { f8::WSProxyPackHead_S* p = (f8::WSProxyPackHead_S*) &recv_buff_[offset]; if (p->magic_code == f8::MAGIC_CODE) { - if (recv_bufflen_ - offset < sizeof(f8::PackHead) + p->packlen) { + if (recv_bufflen_ - offset < sizeof(f8::WSProxyPackHead_S) + p->packlen) { break; } App::Instance()->AddSocketMsg(SF_TargetServer, @@ -136,7 +136,7 @@ void TargetConn::on_socketread(a8::TcpClient* sender, char* buf, unsigned int le p->seqid, &recv_buff_[offset + sizeof(f8::WSProxyPackHead_S)], p->packlen); - offset += sizeof(f8::PackHead) + p->packlen; + offset += sizeof(f8::WSProxyPackHead_S) + p->packlen; } else { warning = true; offset++;