Support sleeping until an irq fires, and use where applicable.

Support sleeping until an irq fires, and use where applicable.

Add wait_irq() - it's more efficient than looping with cpu_relax().
Also, move kbd irq enables down - only kbd_command needs it.
Also, make some minor code layout improvements to kbd.c.