Compare commits

..

3 Commits

Author SHA1 Message Date
patterniha
a316c75d9e set splice-timeout to 24 hours 2025-08-23 15:26:14 +03:30
patterniha
c527ff6749 fix splice-timeout 2025-08-23 15:26:14 +03:30
patterniha
3f3cf89ade Timer: prevent creating redundant check task 2025-08-23 15:26:14 +03:30
2 changed files with 32 additions and 14 deletions

View File

@@ -15,9 +15,10 @@ type ActivityUpdater interface {
type ActivityTimer struct {
sync.RWMutex
updated chan struct{}
checkTask *task.Periodic
onTimeout func()
updated chan struct{}
checkTask *task.Periodic
onTimeout func()
overridden bool
}
func (t *ActivityTimer) Update() {
@@ -31,14 +32,16 @@ func (t *ActivityTimer) check() error {
select {
case <-t.updated:
default:
t.finish()
t.finish(false)
}
return nil
}
func (t *ActivityTimer) finish() {
t.Lock()
defer t.Unlock()
func (t *ActivityTimer) finish(locked bool) {
if !locked {
t.Lock()
defer t.Unlock()
}
if t.onTimeout != nil {
t.onTimeout()
@@ -50,9 +53,12 @@ func (t *ActivityTimer) finish() {
}
}
func (t *ActivityTimer) SetTimeout(timeout time.Duration) {
func (t *ActivityTimer) setTimeout(timeout time.Duration) {
if t.onTimeout == nil {
return
}
if timeout == 0 {
t.finish()
t.finish(true)
return
}
@@ -61,14 +67,26 @@ func (t *ActivityTimer) SetTimeout(timeout time.Duration) {
Execute: t.check,
}
t.Lock()
if t.checkTask != nil {
t.checkTask.Close()
t.overridden = true
}
t.checkTask = checkTask
t.Update()
common.Must(checkTask.Start())
}
func (t *ActivityTimer) SetTimeout(timeout time.Duration) {
t.Lock()
t.setTimeout(timeout)
t.Unlock()
}
func (t *ActivityTimer) SetTimeoutIfNotOverridden(timeout time.Duration) {
t.Lock()
if !t.overridden {
t.setTimeout(timeout)
}
t.Unlock()
}

View File

@@ -595,10 +595,10 @@ func CopyRawConnIfExist(ctx context.Context, readerConn net.Conn, writerConn net
errors.LogInfo(ctx, "CopyRawConn splice")
statWriter, _ := writer.(*dispatcher.SizeStatWriter)
//runtime.Gosched() // necessary
time.Sleep(time.Millisecond) // without this, there will be a rare ssl error for freedom splice
timer.SetTimeout(8 * time.Hour) // prevent leak, just in case
time.Sleep(time.Millisecond) // without this, there will be a rare ssl error for freedom splice
timer.SetTimeoutIfNotOverridden(24 * time.Hour) // prevent leak, just in case
if inTimer != nil {
inTimer.SetTimeout(8 * time.Hour)
inTimer.SetTimeoutIfNotOverridden(24 * time.Hour)
}
w, err := tc.ReadFrom(readerConn)
if readCounter != nil {