Australia are in the box seat for a series win in Sri Lanka but may have cause for concern after Steve Smith appeared to hurt his elbow in the field.