Heavily armed police are at a prison on the outskirts of Melbourne following a major riot reportedly linked to a smoking ban in prisons. A ban is due to come into effect on Wednesday.