Sri Lanka secured a 37-run lead by tea on the second day, at the P Sara Oval.
The lead was built further by Kusal Mendis and Kaushal Silva, who added 54 together, the highest partnership of the Test so far.
Sri Lanka were on 76 for 2 at stumps with a lead of 113.