fix memory leak

xtaci · Mar 13, 2017 · 1de9285 · 1de9285
1 parent 2f21a5b
commit 1de9285
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sess.go b/sess.go
@@ -291,7 +291,9 @@ func (s *UDPSession) Write(b []byte) (n int, err error) {
 // Close closes the connection.
 func (s *UDPSession) Close() error {
 	updater.removeSession(s)
-	s.l.closeSession(s.remote)
+	if s.l != nil { // notify listener
+		s.l.closeSession(s.remote)
+	}
 
 	s.mu.Lock()
 	defer s.mu.Unlock()