Iran to re-impose virus restrictions if rules violated