My impression was that it was indeed in his head; the prison break happened, but he wasn't on scene that whole time.

I could be wrong of course, and that's what I like about this show: The viewer never truly knows WTF is going on.