Update the trace code to know about the key_share extension