sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

server: Prevent adding Listeners after Close

Browse Source 
Check done after calling trackListener to prevent the listener from
being registered after the server is closed.
This commit is contained in:
Adnan Maolood 2021-02-15 20:18:08 -05:00
parent a05fa6d6bd
commit 1d6cbddc5b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server.go
View File

@ -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 {