He was mostly confused, but that's got nothing on his surprise when he's hugged. He actually freezes up for a second out of shock.

"...Heiji. Wake up." His arm is still free, so he'll try shaking him again, if a bit less gently this time.