Kusal Mendis has proved the hero Sri Lanka needed late in the first innings of their second Test against Australia in Galle.