tcp: improve waitclose in closing states