Fix les server handler for registered peers

les/peer.go

@@ -1429,6 +1429,21 @@ func (s *serverSet) register(peer *clientPeer) error {
 	return nil
 }
 
+func (s *serverSet) unregister(peer *clientPeer) error {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+
+	if s.closed {
+		return errClosed
+	}
+	if _, exist := s.set[peer.id]; !exist {
+		return errNotRegistered
+	}
+	delete(s.set, peer.id)
+	peer.Peer.Disconnect(p2p.DiscQuitting)
+	return nil
+}
+
 func (s *serverSet) close() {
 	s.lock.Lock()
 	defer s.lock.Unlock()

les/server_handler.go

@@ -138,6 +138,7 @@ func (h *serverHandler) handle(p *clientPeer) error {
 		}
 		// connected to another server, no messages expected, just wait for disconnection
 		_, err := p.rw.ReadMsg()
+		h.server.serverset.unregister(p)
 		return err
 	}
 	// Reject light clients if server is not synced.