mirror of
https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git
synced 2025-10-13 20:11:19 -04:00
Start refactoring out a client and library
This commit is contained in:
Arlo Breault
parent
7662ccb00c
commit
cce7ee64a7
8 changed files with 120 additions and 105 deletions
319
client/lib/lib_test.go
Normal file
319
client/lib/lib_test.go
Normal file
|
|
@ -0,0 +1,319 @@
|
|||
package lib
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net"
|
||||
"net/http"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/keroserene/go-webrtc"
|
||||
. "github.com/smartystreets/goconvey/convey"
|
||||
)
|
||||
|
||||
type MockDataChannel struct {
|
||||
destination bytes.Buffer
|
||||
done chan bool
|
||||
}
|
||||
|
||||
func (m *MockDataChannel) Send(data []byte) {
|
||||
m.destination.Write(data)
|
||||
m.done <- true
|
||||
}
|
||||
|
||||
func (*MockDataChannel) Close() error { return nil }
|
||||
|
||||
type MockResponse struct{}
|
||||
|
||||
func (m *MockResponse) Read(p []byte) (int, error) {
|
||||
p = []byte(`{"type":"answer","sdp":"fake"}`)
|
||||
return 0, nil
|
||||
}
|
||||
func (m *MockResponse) Close() error { return nil }
|
||||
|
||||
type MockTransport struct{ statusOverride int }
|
||||
|
||||
// Just returns a response with fake SDP answer.
|
||||
func (m *MockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
||||
s := ioutil.NopCloser(strings.NewReader(`{"type":"answer","sdp":"fake"}`))
|
||||
r := &http.Response{
|
||||
StatusCode: m.statusOverride,
|
||||
Body: s,
|
||||
}
|
||||
return r, nil
|
||||
}
|
||||
|
||||
type FakeDialer struct{}
|
||||
|
||||
func (w FakeDialer) Catch() (Snowflake, error) {
|
||||
fmt.Println("Caught a dummy snowflake.")
|
||||
return &WebRTCPeer{}, nil
|
||||
}
|
||||
|
||||
type FakeSocksConn struct {
|
||||
net.Conn
|
||||
rejected bool
|
||||
}
|
||||
|
||||
func (f FakeSocksConn) Reject() error {
|
||||
f.rejected = true
|
||||
return nil
|
||||
}
|
||||
func (f FakeSocksConn) Grant(addr *net.TCPAddr) error { return nil }
|
||||
|
||||
type FakePeers struct{ toRelease *WebRTCPeer }
|
||||
|
||||
func (f FakePeers) Collect() (Snowflake, error) { return &WebRTCPeer{}, nil }
|
||||
func (f FakePeers) Pop() Snowflake { return nil }
|
||||
func (f FakePeers) Melted() <-chan struct{} { return nil }
|
||||
|
||||
func TestSnowflakeClient(t *testing.T) {
|
||||
|
||||
Convey("Peers", t, func() {
|
||||
Convey("Can construct", func() {
|
||||
p := NewPeers(1)
|
||||
So(p.capacity, ShouldEqual, 1)
|
||||
So(p.snowflakeChan, ShouldNotBeNil)
|
||||
So(cap(p.snowflakeChan), ShouldEqual, 1)
|
||||
})
|
||||
|
||||
Convey("Collecting a Snowflake requires a Tongue.", func() {
|
||||
p := NewPeers(1)
|
||||
_, err := p.Collect()
|
||||
So(err, ShouldNotBeNil)
|
||||
So(p.Count(), ShouldEqual, 0)
|
||||
// Set the dialer so that collection is possible.
|
||||
p.Tongue = FakeDialer{}
|
||||
_, err = p.Collect()
|
||||
So(err, ShouldBeNil)
|
||||
So(p.Count(), ShouldEqual, 1)
|
||||
// S
|
||||
_, err = p.Collect()
|
||||
})
|
||||
|
||||
Convey("Collection continues until capacity.", func() {
|
||||
c := 5
|
||||
p := NewPeers(c)
|
||||
p.Tongue = FakeDialer{}
|
||||
// Fill up to capacity.
|
||||
for i := 0; i < c; i++ {
|
||||
fmt.Println("Adding snowflake ", i)
|
||||
_, err := p.Collect()
|
||||
So(err, ShouldBeNil)
|
||||
So(p.Count(), ShouldEqual, i+1)
|
||||
}
|
||||
// But adding another gives an error.
|
||||
So(p.Count(), ShouldEqual, c)
|
||||
_, err := p.Collect()
|
||||
So(err, ShouldNotBeNil)
|
||||
So(p.Count(), ShouldEqual, c)
|
||||
|
||||
// But popping and closing allows it to continue.
|
||||
s := p.Pop()
|
||||
s.Close()
|
||||
So(s, ShouldNotBeNil)
|
||||
So(p.Count(), ShouldEqual, c-1)
|
||||
|
||||
_, err = p.Collect()
|
||||
So(err, ShouldBeNil)
|
||||
So(p.Count(), ShouldEqual, c)
|
||||
})
|
||||
|
||||
Convey("Count correctly purges peers marked for deletion.", func() {
|
||||
p := NewPeers(4)
|
||||
p.Tongue = FakeDialer{}
|
||||
p.Collect()
|
||||
p.Collect()
|
||||
p.Collect()
|
||||
p.Collect()
|
||||
So(p.Count(), ShouldEqual, 4)
|
||||
s := p.Pop()
|
||||
s.Close()
|
||||
So(p.Count(), ShouldEqual, 3)
|
||||
s = p.Pop()
|
||||
s.Close()
|
||||
So(p.Count(), ShouldEqual, 2)
|
||||
})
|
||||
|
||||
Convey("End Closes all peers.", func() {
|
||||
cnt := 5
|
||||
p := NewPeers(cnt)
|
||||
for i := 0; i < cnt; i++ {
|
||||
p.activePeers.PushBack(&WebRTCPeer{})
|
||||
}
|
||||
So(p.Count(), ShouldEqual, cnt)
|
||||
p.End()
|
||||
<-p.Melted()
|
||||
So(p.Count(), ShouldEqual, 0)
|
||||
})
|
||||
|
||||
Convey("Pop skips over closed peers.", func() {
|
||||
p := NewPeers(4)
|
||||
p.Tongue = FakeDialer{}
|
||||
wc1, _ := p.Collect()
|
||||
wc2, _ := p.Collect()
|
||||
wc3, _ := p.Collect()
|
||||
So(wc1, ShouldNotBeNil)
|
||||
So(wc2, ShouldNotBeNil)
|
||||
So(wc3, ShouldNotBeNil)
|
||||
wc1.Close()
|
||||
r := p.Pop()
|
||||
So(p.Count(), ShouldEqual, 2)
|
||||
So(r, ShouldEqual, wc2)
|
||||
wc4, _ := p.Collect()
|
||||
wc2.Close()
|
||||
wc3.Close()
|
||||
r = p.Pop()
|
||||
So(r, ShouldEqual, wc4)
|
||||
})
|
||||
|
||||
})
|
||||
|
||||
Convey("Snowflake", t, func() {
|
||||
|
||||
SkipConvey("Handler Grants correctly", func() {
|
||||
socks := &FakeSocksConn{}
|
||||
snowflakes := &FakePeers{}
|
||||
|
||||
So(socks.rejected, ShouldEqual, false)
|
||||
snowflakes.toRelease = nil
|
||||
Handler(socks, snowflakes)
|
||||
So(socks.rejected, ShouldEqual, true)
|
||||
})
|
||||
|
||||
Convey("WebRTC Connection", func() {
|
||||
c := NewWebRTCPeer(nil, nil)
|
||||
So(c.buffer.Bytes(), ShouldEqual, nil)
|
||||
|
||||
Convey("Can construct a WebRTCConn", func() {
|
||||
s := NewWebRTCPeer(nil, nil)
|
||||
So(s, ShouldNotBeNil)
|
||||
So(s.offerChannel, ShouldNotBeNil)
|
||||
So(s.answerChannel, ShouldNotBeNil)
|
||||
s.Close()
|
||||
})
|
||||
|
||||
Convey("Write buffers when datachannel is nil", func() {
|
||||
c.Write([]byte("test"))
|
||||
c.transport = nil
|
||||
So(c.buffer.Bytes(), ShouldResemble, []byte("test"))
|
||||
})
|
||||
|
||||
Convey("Write sends to datachannel when not nil", func() {
|
||||
mock := new(MockDataChannel)
|
||||
c.transport = mock
|
||||
mock.done = make(chan bool, 1)
|
||||
c.Write([]byte("test"))
|
||||
<-mock.done
|
||||
So(c.buffer.Bytes(), ShouldEqual, nil)
|
||||
So(mock.destination.Bytes(), ShouldResemble, []byte("test"))
|
||||
})
|
||||
|
||||
Convey("Exchange SDP sets remote description", func() {
|
||||
c.offerChannel = make(chan *webrtc.SessionDescription, 1)
|
||||
c.answerChannel = make(chan *webrtc.SessionDescription, 1)
|
||||
|
||||
c.config = webrtc.NewConfiguration()
|
||||
c.preparePeerConnection()
|
||||
|
||||
c.offerChannel <- nil
|
||||
answer := webrtc.DeserializeSessionDescription(
|
||||
`{"type":"answer","sdp":""}`)
|
||||
c.answerChannel <- answer
|
||||
c.exchangeSDP()
|
||||
})
|
||||
|
||||
SkipConvey("Exchange SDP fails on nil answer", func() {
|
||||
c.reset = make(chan struct{})
|
||||
c.offerChannel = make(chan *webrtc.SessionDescription, 1)
|
||||
c.answerChannel = make(chan *webrtc.SessionDescription, 1)
|
||||
c.offerChannel <- nil
|
||||
c.answerChannel <- nil
|
||||
c.exchangeSDP()
|
||||
<-c.reset
|
||||
})
|
||||
|
||||
})
|
||||
})
|
||||
|
||||
Convey("Dialers", t, func() {
|
||||
Convey("Can construct WebRTCDialer.", func() {
|
||||
broker := &BrokerChannel{Host: "test"}
|
||||
d := NewWebRTCDialer(broker, nil)
|
||||
So(d, ShouldNotBeNil)
|
||||
So(d.BrokerChannel, ShouldNotBeNil)
|
||||
So(d.BrokerChannel.Host, ShouldEqual, "test")
|
||||
})
|
||||
Convey("WebRTCDialer cannot Catch a snowflake with nil broker.", func() {
|
||||
d := NewWebRTCDialer(nil, nil)
|
||||
conn, err := d.Catch()
|
||||
So(conn, ShouldBeNil)
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
SkipConvey("WebRTCDialer can Catch a snowflake.", func() {
|
||||
broker := &BrokerChannel{Host: "test"}
|
||||
d := NewWebRTCDialer(broker, nil)
|
||||
conn, err := d.Catch()
|
||||
So(conn, ShouldBeNil)
|
||||
So(err, ShouldNotBeNil)
|
||||
})
|
||||
})
|
||||
|
||||
Convey("Rendezvous", t, func() {
|
||||
webrtc.SetLoggingVerbosity(0)
|
||||
transport := &MockTransport{http.StatusOK}
|
||||
fakeOffer := webrtc.DeserializeSessionDescription("test")
|
||||
|
||||
Convey("Construct BrokerChannel with no front domain", func() {
|
||||
b := NewBrokerChannel("test.broker", "", transport)
|
||||
So(b.url, ShouldNotBeNil)
|
||||
So(b.url.Path, ShouldResemble, "test.broker")
|
||||
So(b.transport, ShouldNotBeNil)
|
||||
})
|
||||
|
||||
Convey("Construct BrokerChannel *with* front domain", func() {
|
||||
b := NewBrokerChannel("test.broker", "front", transport)
|
||||
So(b.url, ShouldNotBeNil)
|
||||
So(b.url.Path, ShouldResemble, "test.broker")
|
||||
So(b.url.Host, ShouldResemble, "front")
|
||||
So(b.transport, ShouldNotBeNil)
|
||||
})
|
||||
|
||||
Convey("BrokerChannel.Negotiate responds with answer", func() {
|
||||
b := NewBrokerChannel("test.broker", "", transport)
|
||||
answer, err := b.Negotiate(fakeOffer)
|
||||
So(err, ShouldBeNil)
|
||||
So(answer, ShouldNotBeNil)
|
||||
So(answer.Sdp, ShouldResemble, "fake")
|
||||
})
|
||||
|
||||
Convey("BrokerChannel.Negotiate fails with 503", func() {
|
||||
b := NewBrokerChannel("test.broker", "",
|
||||
&MockTransport{http.StatusServiceUnavailable})
|
||||
answer, err := b.Negotiate(fakeOffer)
|
||||
So(err, ShouldNotBeNil)
|
||||
So(answer, ShouldBeNil)
|
||||
So(err.Error(), ShouldResemble, BrokerError503)
|
||||
})
|
||||
|
||||
Convey("BrokerChannel.Negotiate fails with 400", func() {
|
||||
b := NewBrokerChannel("test.broker", "",
|
||||
&MockTransport{http.StatusBadRequest})
|
||||
answer, err := b.Negotiate(fakeOffer)
|
||||
So(err, ShouldNotBeNil)
|
||||
So(answer, ShouldBeNil)
|
||||
So(err.Error(), ShouldResemble, BrokerError400)
|
||||
})
|
||||
|
||||
Convey("BrokerChannel.Negotiate fails with unexpected error", func() {
|
||||
b := NewBrokerChannel("test.broker", "",
|
||||
&MockTransport{123})
|
||||
answer, err := b.Negotiate(fakeOffer)
|
||||
So(err, ShouldNotBeNil)
|
||||
So(answer, ShouldBeNil)
|
||||
So(err.Error(), ShouldResemble, BrokerErrorUnexpected)
|
||||
})
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue