github
/
v2ray-core
mirror of https://github.com/v2ray/v2ray-core
1
0
Fork 0
Code Issues Releases Wiki Activity

safe release buffer

pull/700/head
Darien Raymond 2017-11-07 11:58:43 +01:00
parent ccf1fb6616
commit c1eccb3a45
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app/proxyman/mux/writer.go
View File

@ -82,23 +82,21 @@ func (w *Writer) writeData(mb buf.MultiBuffer) error {
// Write implements buf.MultiBufferWriter.
func (w *Writer) Write(mb buf.MultiBuffer) error {
defer mb.Release()
if mb.IsEmpty() {
return w.writeMetaOnly()
}
if w.transferType == protocol.TransferTypeStream {
const chunkSize = 8 * 1024
for !mb.IsEmpty() {
slice := mb.SliceBySize(chunkSize)
if err := w.writeData(slice); err != nil {
return err
}
}
var chunk buf.MultiBuffer
if w.transferType == protocol.TransferTypeStream {
chunk = mb.SliceBySize(8 * 1024)
} else {
for _, b := range mb {
if err := w.writeData(buf.NewMultiBufferValue(b)); err != nil {
return err
chunk = buf.NewMultiBufferValue(mb.SplitFirst())
}
if err := w.writeData(chunk); err != nil {
return err
}
}