gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

test version negotiation packet

Browse Source
This commit is contained in:
Lucas Clemente
2016-04-21 09:21:21 +02:00
parent bb3c8b707b
commit fd28921bdd
2 changed files with 29 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
server.go
View File

@@ -68,11 +68,13 @@ func (s *Server) ListenAndServe(address string) error {
continue
}
fmt.Printf("<- Got packet %d (%d bytes) from %v\n", publicHeader.PacketNumber, n, remoteAddr)
// fmt.Printf("<- Got packet %d (%d bytes) from %v\n", publicHeader.PacketNumber, n, remoteAddr)
// Send Version Negotiation Packet if the client is speaking a different protocol version
if publicHeader.VersionFlag && !protocol.IsSupportedVersion(publicHeader.VersionNumber) {
if err := sendVersionNegotiation(conn, remoteAddr, publicHeader); err != nil {
fmt.Println("Sending VersionNegotiationPacket")
_, err = conn.WriteToUDP(composeVersionNegotiation(publicHeader.ConnectionID), remoteAddr)
if err != nil {
fmt.Printf("Error sending version negotiation: %s", err.Error())
}
continue
@@ -91,18 +93,17 @@ func (s *Server) ListenAndServe(address string) error {
}
}
func sendVersionNegotiation(conn *net.UDPConn, remoteAddr *net.UDPAddr, publicHeader *PublicHeader) error {
fmt.Println("Sending VersionNegotiationPacket")
func composeVersionNegotiation(connectionID protocol.ConnectionID) []byte {
fullReply := &bytes.Buffer{}
responsePublicHeader := PublicHeader{ConnectionID: publicHeader.ConnectionID, PacketNumber: 1, VersionFlag: true}
responsePublicHeader := PublicHeader{
ConnectionID: connectionID,
PacketNumber: 1,
VersionFlag: true,
}
err := responsePublicHeader.WritePublicHeader(fullReply)
if err != nil {
return err
panic(err) // Should not happen ;)
}
fullReply.Write(protocol.SupportedVersionsAsTags)
_, err = conn.WriteToUDP(fullReply.Bytes(), remoteAddr)
if err != nil {
return err
}
return nil
return fullReply.Bytes()
}

17
server_test.go Normal file
View File

@@ -0,0 +1,17 @@
package quic
import (
"github.com/lucas-clemente/quic-go/protocol"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Server", func() {
It("composes version negotiation packets", func() {
expected := append(
[]byte{0x2d, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0},
protocol.SupportedVersionsAsTags...,
)
Expect(composeVersionNegotiation(1)).To(Equal(expected))
})
})