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

Add EventOnProxyConnectionOver Reporting

Browse source
This commit is contained in:
Shelikhoo 2021-12-20 13:36:22 +00:00
parent d64af31394
commit e4305a4d2b
No known key found for this signature in database
GPG key ID: C4D5E79D22B25316
2 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
proxy/lib/snowflake.go
View file

@ -40,6 +40,7 @@ import (
"sync"
"time"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/event"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/messages"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/task"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/util"
@ -350,6 +351,11 @@ func (sf *SnowflakeProxy) makePeerConnectionFromOffer(sdp *webrtc.SessionDescrip
defer conn.lock.Unlock()
log.Println("OnClose channel")
log.Println(conn.bytesLogger.ThroughputSummary())
in, out := conn.bytesLogger.GetStat()
conn.eventLogger.OnNewSnowflakeEvent(event.EventOnProxyConnectionOver{
InboundTraffic: in,
OutboundTraffic: out,
})
conn.dc = nil
dc.Close()
pw.Close()

2
proxy/lib/webrtcconn.go
View file

@ -9,6 +9,7 @@ import (
"sync"
"time"
"git.torproject.org/pluggable-transports/snowflake.git/v2/common/event"
"github.com/pion/ice/v2"
"github.com/pion/sdp/v3"
"github.com/pion/webrtc/v3"
@ -30,6 +31,7 @@ type webRTCConn struct {
once sync.Once // Synchronization for PeerConnection destruction
bytesLogger bytesLogger
eventLogger event.SnowflakeEventReceiver
}
func (c *webRTCConn) Read(b []byte) (int, error) {