Masked AES implementation code or file


I want to capture the power traces of masked AES implementation from the Atmega328p and the SAKURA-G respectively to carry out some experiments. Where could I get the corresponding codes or files that I can load into them and then capture the traces.