South Africa lost two late wickets to end day four on 138-2 after being set the small matter of 458 by India to win the first Test in Johannesburg.