Australia have taken two key wickets in the first session on day three of their first Test match against Sri Lanka before showers hovered over Galle.