Eranga Jayawardena/Associated Press

South Africa's bid to return to the top of the world Test rankings were dented on Day 2 of the second Test against Sri Lanka in Colombo.

After the hosts extended their first-innings total to 421, the Proteas could only reach 98-3 in their reply at the Sinhalese Sports Club Ground.

South Africa crumbled to 23-2 at tea—Alviro Petersen offered Rangana Herath a return catch in the second over after making just two and Dean Elgar was caught at short leg off Dilruwan Perera, having made a single.

Faf du Plessis and captain Hashim Amla launched a revival, but debutant keeper Niroshan Dickwella took a superb catch down the leg-side off Suranga Lakmal to dismiss the former for 36.

Amla (46) and AB de Villiers (11) saw the tourists through to stumps without further loss.

Sri Lanka earlier added 116 runs after resuming on 305-5 with Mahela Jayawardene run out for 165 after a direct hit from Petersen.

Imran Tahir finally picked up a wicket when Perera (12) picked out mid-on. Dickwella then missed out on a debut century as he was also run out with a direct hit for 72.

The tail failed to wag with Vernon Philander mopping up the final two wickets.

SA, who won the first Test in Galle, need to avoid defeat to usurp Australia at the summit of the rankings.