Security forces in Afghanistan have arrested five men suspected of being involved in the massacre last month at a military school in Pakistan in which 148 people were killed, mostly children.