Steve Smith faces a race of the clock to be fit for the Sri Lanka Test series.