kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix termination of long downloads after StreamTimeout seconds. 

- Even if not broadcasting in a loop, we still need to update the read deadline.
- Don't enforce the timeout after the first data is written.
This commit is contained in:
notsure2 2020-12-06 05:31:50 +02:00
parent 46f4235ccb
commit ee5968afee
2 changed files with 19 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
internal/multiplex/datagramBufferedPipe.go
View File

@ -72,6 +72,7 @@ func (d *datagramBufferedPipe) Read(target []byte) (int, error) {
func (d *datagramBufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
d.rwCond.L.Lock()
defer d.rwCond.L.Unlock()
enforceTimeout := true
if d.buf == nil {
d.buf = new(bytes.Buffer)
}
@ -80,11 +81,12 @@ func (d *datagramBufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
return 0, io.EOF
}
hasRDeadline := !d.rDeadline.IsZero()
if hasRDeadline {
if time.Until(d.rDeadline) <= 0 {
if !d.rDeadline.IsZero() && enforceTimeout && time.Until(d.rDeadline) <= 0 {
return 0, ErrTimeout
}
if d.wtTimeout != 0 {
d.rDeadline = time.Now().Add(d.wtTimeout)
}
if len(d.pLens) > 0 {
@ -96,14 +98,14 @@ func (d *datagramBufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
d.rwCond.Broadcast()
return n, er
}
enforceTimeout = false
d.rwCond.Broadcast()
} else {
if d.wtTimeout == 0 {
if hasRDeadline {
if !d.rDeadline.IsZero() {
d.broadcastAfter(time.Until(d.rDeadline))
}
} else {
d.rDeadline = time.Now().Add(d.wtTimeout)
d.broadcastAfter(d.wtTimeout)
}

13
internal/multiplex/streamBufferedPipe.go
View File

@ -64,6 +64,7 @@ func (p *streamBufferedPipe) Read(target []byte) (int, error) {
func (p *streamBufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
p.rwCond.L.Lock()
defer p.rwCond.L.Unlock()
enforceTimeout := true
if p.buf == nil {
p.buf = new(bytes.Buffer)
}
@ -72,12 +73,14 @@ func (p *streamBufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
return 0, io.EOF
}
hasRDeadline := !p.rDeadline.IsZero()
if hasRDeadline {
if time.Until(p.rDeadline) <= 0 {
if !p.rDeadline.IsZero() && enforceTimeout && time.Until(p.rDeadline) <= 0 {
return 0, ErrTimeout
}
if p.wtTimeout != 0 {
p.rDeadline = time.Now().Add(p.wtTimeout)
}
if p.buf.Len() > 0 {
written, er := p.buf.WriteTo(w)
n += written
@ -85,14 +88,14 @@ func (p *streamBufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
p.rwCond.Broadcast()
return n, er
}
enforceTimeout = false
p.rwCond.Broadcast()
} else {
if p.wtTimeout == 0 {
if hasRDeadline {
if !p.rDeadline.IsZero() {
p.broadcastAfter(time.Until(p.rDeadline))
}
} else {
p.rDeadline = time.Now().Add(p.wtTimeout)
p.broadcastAfter(p.wtTimeout)
}