Temple dress code in Sri Lanka - what do I need to cover?

Planning to visit several temples. Need special clothes or can I buy a sarong at the entrance?