thunderbolt: Make tb_switch_clx_disable() return CL states that were enabled

This allows us to disable all CL states temporarily when running lane
margining and then return back the previously enabled states.

Signed-off-by: Mika Westerberg <mika.westerberg@linux.intel.com>