mirror of
https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git
synced 2025-10-13 20:11:19 -04:00
defer conn.Close for simplicity and remove unnecessary goroutines, improve error handling (close #12)
This commit is contained in:
parent
598e2a3bfb
commit
b8815627bd
2 changed files with 20 additions and 32 deletions
|
@ -2,6 +2,7 @@ package main
|
|||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/keroserene/go-webrtc"
|
||||
"io"
|
||||
|
@ -99,26 +100,24 @@ func NewWebRTCConnection(config *webrtc.Configuration,
|
|||
return connection
|
||||
}
|
||||
|
||||
// WebRTC re-establishment loop. Expected in own goroutine.
|
||||
func (c *webRTCConn) ConnectLoop() {
|
||||
// TODO: Multiplex.
|
||||
func (c *webRTCConn) Connect() error {
|
||||
log.Println("Establishing WebRTC connection...")
|
||||
// TODO: When go-webrtc is more stable, it's possible that a new
|
||||
// PeerConnection won't need to be re-prepared each time.
|
||||
err := c.preparePeerConnection()
|
||||
if err != nil {
|
||||
log.Println("WebRTC: Could not create PeerConnection.")
|
||||
return
|
||||
return err
|
||||
}
|
||||
err = c.establishDataChannel()
|
||||
if err != nil {
|
||||
log.Println("WebRTC: Could not establish DataChannel.")
|
||||
} else {
|
||||
go c.exchangeSDP()
|
||||
<-c.reset
|
||||
log.Println(" --- snowflake connection reset ---")
|
||||
return errors.New("WebRTC: Could not establish DataChannel.")
|
||||
}
|
||||
<-time.After(time.Second * 1)
|
||||
c.cleanup()
|
||||
err = c.exchangeSDP()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Create and prepare callbacks on a new WebRTC PeerConnection.
|
||||
|
@ -277,7 +276,6 @@ func (c *webRTCConn) exchangeSDP() error {
|
|||
answer = nil
|
||||
}
|
||||
}
|
||||
|
||||
log.Printf("Received Answer:\n\n%s\n", answer.Sdp)
|
||||
err := c.pc.SetRemoteDescription(answer)
|
||||
if nil != err {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue