net/tls: simplify driver context retrieval