server: Prevent adding Listeners after Close

Check done after calling trackListener to prevent the listener from
being registered after the server is closed.

1 files changed, 3 insertions, 3 deletions

diff --git a/server.go b/server.go
index 32afe27..f461d4a 100644
--- a/server.go
+++ b/server.go
@@ -163,13 +163,13 @@ func (srv *Server) deleteListener(l *net.Listener) {
 func (srv *Server) Serve(l net.Listener) error {
 	defer l.Close()
 
+	srv.trackListener(&l)
+	defer srv.deleteListener(&l)
+
 	if atomic.LoadInt32(&srv.done) == 1 {
 		return ErrServerClosed
 	}
 
-	srv.trackListener(&l)
-	defer srv.deleteListener(&l)
-
 	var tempDelay time.Duration // how long to sleep on accept failure
 
 	for {