rust: simplify loop code

I may be missing something, but I think this is equivalent to the previous code (except we don’t do an extra recv when shutting down).