When you get shot/punched by a mob,you'd expect to just take damage,but now you just get disconnected.

You have to do nothing to cause this bug. It just happens when you are hurt by a mob.