Warner continued pillaging the Sri Lankan attack until he was bowled on the last ball by paceman Dhammika Prasad.
Prasad had Dan Christian caught behind for 10 and Dave Hussey was out for one in the following over, caught and bowled by Herath, to leave Australia on 224-4 and give Sri Lanka hope of restricting them to an achievable total.
Michael Clarke righted the ship with a quickfire 37 on his return from injury, however, before Mike Hussey took over for an unbeaten 19 to see out the full quota of overs.
Sri Lanka's chase started brightly, with openers Mahela Jayawardene and Tillakaratne Dilshan belting Australia's pacemen to all corners of the ground, but Brett Lee removed both of them in successive overs to halt the charge.
In Image: Sri Lankan players celebrate the wicket of Australia's Matthew Wade