Add drain_and_close_socket()