privisioners/shell: wait for reboot if we can