mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 04:37:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

call the ConnStateCallback in the client for encryption level changes

Browse source 
fixes #439
This commit is contained in:
Marten Seemann 2017-02-24 09:02:33 +07:00
parent 820eb2e6ee
commit 7109883a95
No known key found for this signature in database
GPG key ID: 3603F40B121FCDEA
2 changed files with 45 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

23
client_test.go
View file

@ -72,9 +72,8 @@ var _ = Describe("Client", func() {
Expect(err).To(MatchError(testErr))
})
// TODO: actually test this
// now we're only testing that Dial doesn't return directly after version negotiation
It("only returns once a forward-secure connection is established if no ConnState is defined", func() {
It("doesn't return after version negotiation is established if no ConnState is defined", func() {
packetConn.dataToRead = []byte{0x0, 0x1, 0x0}
config.ConnState = nil
var dialReturned bool
@ -86,6 +85,26 @@ var _ = Describe("Client", func() {
}()
Consistently(func() bool { return dialReturned }).Should(BeFalse())
})
It("only establishes a connection once it is forward-secure if no ConnState is defined", func() {
config.ConnState = nil
client := &client{conn: &conn{pconn: packetConn, currentAddr: addr}, config: config}
client.connStateChangeOrErrCond.L = &client.mutex
var returned bool
go func() {
defer GinkgoRecover()
_, err := client.establishConnection()
Expect(err).ToNot(HaveOccurred())
returned = true
}()
Consistently(func() bool { return returned }).Should(BeFalse())
// switch to a secure connection
client.cryptoChangeCallback(nil, false)
Consistently(func() bool { return returned }).Should(BeFalse())
// switch to a forward-secure connection
client.cryptoChangeCallback(nil, true)
Eventually(func() bool { return returned }).Should(BeFalse())
})
})
It("errors on invalid public header", func() {