lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2025-01-02 23:47:07 -05:00
Code Issues Releases Wiki Activity

refine kcp output

Browse Source
This commit is contained in:
Darien Raymond 2017-02-07 00:38:46 +01:00
parent 14bbb58b12
commit 3643dc37e0
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
transport/internet/kcp/output.go
View File

@ -28,8 +28,7 @@ func (v *SimpleSegmentWriter) Write(seg Segment) error {
v.Lock()
defer v.Unlock()
v.buffer.AppendSupplier(seg.Bytes())
v.buffer.Reset(seg.Bytes())
_, err := v.writer.Write(v.buffer.Bytes())
v.buffer.Clear()
return err
}

7
transport/internet/kcp/segment.go
View File

@ -61,12 +61,13 @@ func (v *DataSegment) Command() Command {
return CommandData
}
func (v *DataSegment) SetData(b []byte) {
func (v *DataSegment) SetData(data []byte) {
if v.Data == nil {
v.Data = buf.NewSmall()
}
v.Data.Clear()
v.Data.Append(b)
v.Data.Reset(func(b []byte) (int, error) {
return copy(b, data), nil
})
}
func (v *DataSegment) Bytes() buf.Supplier {