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

fix a race condition in quic.Server

Browse Source
This commit is contained in:
Lucas Clemente
2016-06-02 21:21:41 +02:00
parent bd228ffe27
commit 45554e67c4
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server.go
View File

@@ -24,7 +24,9 @@ type packetHandler interface {
// A Server of QUIC // A Server of QUIC
type Server struct { type Server struct {
addr *net.UDPAddr addr *net.UDPAddr
conn *net.UDPConn conn *net.UDPConn
connMutex sync.Mutex
signer crypto.Signer signer crypto.Signer
scfg *handshake.ServerConfig scfg *handshake.ServerConfig
@@ -74,7 +76,9 @@ func (s *Server) ListenAndServe() error {
if err != nil { if err != nil {
return err return err
} }
s.connMutex.Lock()
s.conn = conn s.conn = conn
s.connMutex.Unlock()
for { for {
data := make([]byte, protocol.MaxPacketSize) data := make([]byte, protocol.MaxPacketSize)
@@ -103,6 +107,10 @@ func (s *Server) Close() error {
} }
} }
s.sessionsMutex.Unlock() s.sessionsMutex.Unlock()
s.connMutex.Lock()
defer s.connMutex.Unlock()
if s.conn == nil { if s.conn == nil {
return nil return nil
} }