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

use the net.PacketConn everywhere in the server

Browse Source
This commit is contained in:
Marten Seemann
2017-02-16 22:30:17 +07:00
parent 5b42675da2
commit e924f0ecb3
3 changed files with 61 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
server.go
View File

@@ -28,7 +28,7 @@ type packetHandler interface {
type Server struct {
addr *net.UDPAddr
conn *net.UDPConn
conn net.PacketConn
connMutex sync.Mutex
certChain crypto.CertChain
@@ -81,8 +81,8 @@ func (s *Server) ListenAndServe() error {
return s.Serve(conn)
}
// Serve on an existing UDP connection.
func (s *Server) Serve(conn *net.UDPConn) error {
// Serve on an existing PacketConn
func (s *Server) Serve(conn net.PacketConn) error {
s.connMutex.Lock()
s.conn = conn
s.connMutex.Unlock()
@@ -90,7 +90,7 @@ func (s *Server) Serve(conn *net.UDPConn) error {
for {
data := getPacketBuffer()
data = data[:protocol.MaxPacketSize]
n, remoteAddr, err := conn.ReadFromUDP(data)
n, remoteAddr, err := conn.ReadFrom(data)
if err != nil {
if strings.HasSuffix(err.Error(), "use of closed network connection") {
return nil
@@ -132,7 +132,7 @@ func (s *Server) Addr() net.Addr {
return s.addr
}
func (s *Server) handlePacket(pconn net.PacketConn, remoteAddr *net.UDPAddr, packet []byte) error {
func (s *Server) handlePacket(pconn net.PacketConn, remoteAddr net.Addr, packet []byte) error {
if protocol.ByteCount(len(packet)) > protocol.MaxPacketSize {
return qerr.PacketTooLarge
}