s3: Move the drain_socket on error to reply_write_and_X