Sri Lanka and New Zealand are set for an exciting finish to the first Test in Galle, with the home side best placed for victory after the fourth day.