Move preemption handling out of critical_exit.
In preperataion for making the enter/exit pair inline.
Reviewed by: kib