Wawrinka was full of energy early on, and a break of serve saw him take control of the opening set before clinching it 6-3.
The Australian Open champion then wasted two break points in a cagey second as Federer forced the tie-break, which he won 7-5 to level things up.
While Federer started to increase the tempo in the third, Wawrinka began to wane. Federer won 85% of points on his first serve during the set, allowing him to eventually come through 6-4.
The veteran was able to open a 3-1 lead in the fourth as Wawrinka continued to struggle, and from there he went on the close out the contest 3-6 7-6(5) 6-4 6-4.