Australia are on top in the second Test against Sri Lanka thanks to three wicket hauls from Mitchell Starc and Nathan Lyon.