India needs six wickets on the final day of the Galle Test to secure a 1-0 series lead after setting Sri Lanka a target of 372 runs.
Third umpire Raveendra Wimalasiri sparked debate by ruling Lahiru Udara out after a catch by Dhruv Jurel, despite replays suggesting the ball may have touched the ground.
Sri Lanka ended the fourth day at 84/4, with Manav Suthar taking two wickets and Mohammed Siraj and Prasidh Krishna claiming one each to maintain India's pressure.