vmess: fixed a bug in chunked writer

This commit is contained in:
nadoo 2018-07-10 09:44:51 +08:00
parent fd298207c5
commit dc46705f3f

View File

@ -84,6 +84,7 @@ func (w *chunkedWriter) ReadFrom(r io.Reader) (n int64, err error) {
nr, er := r.Read(payloadBuf)
if nr > 0 {
n += int64(nr)
buf = buf[:2+nr]
payloadBuf = payloadBuf[:nr]
binary.BigEndian.PutUint16(buf[:2], uint16(nr))
@ -102,6 +103,5 @@ func (w *chunkedWriter) ReadFrom(r io.Reader) (n int64, err error) {
}
}
w.Writer.Write([]byte{0, 0})
return n, err
}