Sri Lanka landed in trouble during the 2nd Test in Colombo after umpires controversially kept play going despite fading light.