An Iranian official said a "friendly country" had warned it of a potential Israeli attack.