当收到错误报文时，会产生fatal error: runtime: out of memory致命错误

func LengthOf(stream []byte) (int32, error) {
    if len(stream) < 4 {
        return 0, errors.New(fmt.Sprintf("stream lenth should be bigger than 4"))
    }
    length := binary.BigEndian.Uint32(stream[0:4])
    return int32(length), nil
}

当收到任意报文时，没有对length上限进行判断，可能会开辟过大的内存。测试代码

func main() {
    var tcpAddr *net.TCPAddr
    tcpAddr, _ = net.ResolveTCPAddr("tcp", "127.0.0.1:7170")

    conn, _ := net.DialTCP("tcp", nil, tcpAddr)
    b := []byte("time\n")
    conn.Write(b)

    time.Sleep(time.Second * 1)
}

测试机内存为2G，当收到错误的报文时，会产生fatal error: runtime: out of memory致命错误

fwhezfwhez / tcpx

当收到错误报文时，会产生fatal error: runtime: out of memory致命错误 #41