So doing this and using freezable_schedule_timeout() a few lines below should do the trick AFAICT (at least it should match Dave Chinner's expectations of what try_to_freeze() is doing ;).