Sri Lanka took two key wickets as Australia set out to make headway on the hosts' first-innings 257 in sunny Galle.