archives/snowflake
2
0
Fork 0
mirror of https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git synced 2025-10-13 20:11:19 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add timeout for webRTCConn

Browse source
This commit is contained in:
Cecylia Bocovich 2022-10-18 15:43:30 -04:00
parent 6007d5e08e
commit 5c23fcf14a
No known key found for this signature in database
GPG key ID: 009DE379FD9B7B90
1 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
proxy/lib/webrtcconn.go
View file

@ -32,20 +32,41 @@ type webRTCConn struct {
bytesLogger bytesLogger
eventLogger event.SnowflakeEventReceiver
inactivityTimeout time.Duration
activity chan struct{}
}
func newWebRTCConn(pc *webrtc.PeerConnection, dc *webrtc.DataChannel, pr *io.PipeReader, eventLogger event.SnowflakeEventReceiver) (*webRTCConn) {
func newWebRTCConn(pc *webrtc.PeerConnection, dc *webrtc.DataChannel, pr *io.PipeReader, eventLogger event.SnowflakeEventReceiver) *webRTCConn {
conn := &webRTCConn{pc: pc, dc: dc, pr: pr, eventLogger: eventLogger}
conn.bytesLogger = newBytesSyncLogger()
conn.activity = make(chan struct{}, 100)
conn.inactivityTimeout = 30 * time.Second
go conn.timeoutLoop()
return conn
}
func (c *webRTCConn) timeoutLoop() {
for {
select {
case <-time.After(c.inactivityTimeout):
c.Close()
log.Println("Closed connection due to inactivity")
return
case <-c.activity:
continue
}
}
}
func (c *webRTCConn) Read(b []byte) (int, error) {
return c.pr.Read(b)
}
func (c *webRTCConn) Write(b []byte) (int, error) {
c.bytesLogger.AddInbound(int64(len(b)))
c.activity <- struct{}{}
c.lock.Lock()
defer c.lock.Unlock()
if c.dc != nil {