1 /*
2  * Copyright 2018 Advanced Micro Devices, Inc.
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20  * OTHER DEALINGS IN THE SOFTWARE.
21  *
22  * Authors: AMD
23  *
24  */
25 
26 #include "hdcp.h"
27 
28 static inline enum mod_hdcp_status check_receiver_id_list_ready(struct mod_hdcp *hdcp)
29 {
30 	uint8_t is_ready = 0;
31 
32 	if (is_dp_hdcp(hdcp))
33 		is_ready = HDCP_2_2_DP_RXSTATUS_READY(hdcp->auth.msg.hdcp2.rxstatus_dp) ? 1 : 0;
34 	else
35 		is_ready = (HDCP_2_2_HDMI_RXSTATUS_READY(hdcp->auth.msg.hdcp2.rxstatus[0]) &&
36 				(HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
37 						hdcp->auth.msg.hdcp2.rxstatus[0])) ? 1 : 0;
38 	return is_ready ? MOD_HDCP_STATUS_SUCCESS :
39 			MOD_HDCP_STATUS_HDCP2_RX_ID_LIST_NOT_READY;
40 }
41 
42 static inline enum mod_hdcp_status check_hdcp2_capable(struct mod_hdcp *hdcp)
43 {
44 	enum mod_hdcp_status status;
45 
46 	if (is_dp_hdcp(hdcp))
47 		status = (hdcp->auth.msg.hdcp2.rxcaps_dp[2] & HDCP_2_2_RX_CAPS_VERSION_VAL) &&
48 				HDCP_2_2_DP_HDCP_CAPABLE(hdcp->auth.msg.hdcp2.rxcaps_dp[0]) ?
49 				MOD_HDCP_STATUS_SUCCESS :
50 				MOD_HDCP_STATUS_HDCP2_NOT_CAPABLE;
51 	else
52 		status = (hdcp->auth.msg.hdcp2.hdcp2version_hdmi & HDCP_2_2_HDMI_SUPPORT_MASK) ?
53 				MOD_HDCP_STATUS_SUCCESS :
54 				MOD_HDCP_STATUS_HDCP2_NOT_CAPABLE;
55 	return status;
56 }
57 
58 static inline enum mod_hdcp_status check_reauthentication_request(
59 		struct mod_hdcp *hdcp)
60 {
61 	uint8_t ret = 0;
62 
63 	if (is_dp_hdcp(hdcp))
64 		ret = HDCP_2_2_DP_RXSTATUS_REAUTH_REQ(hdcp->auth.msg.hdcp2.rxstatus_dp) ?
65 				MOD_HDCP_STATUS_HDCP2_REAUTH_REQUEST :
66 				MOD_HDCP_STATUS_SUCCESS;
67 	else
68 		ret = HDCP_2_2_HDMI_RXSTATUS_REAUTH_REQ(hdcp->auth.msg.hdcp2.rxstatus[0]) ?
69 				MOD_HDCP_STATUS_HDCP2_REAUTH_REQUEST :
70 				MOD_HDCP_STATUS_SUCCESS;
71 	return ret;
72 }
73 
74 static inline enum mod_hdcp_status check_link_integrity_failure_dp(
75 		struct mod_hdcp *hdcp)
76 {
77 	return HDCP_2_2_DP_RXSTATUS_LINK_FAILED(hdcp->auth.msg.hdcp2.rxstatus_dp) ?
78 			MOD_HDCP_STATUS_HDCP2_REAUTH_LINK_INTEGRITY_FAILURE :
79 			MOD_HDCP_STATUS_SUCCESS;
80 }
81 
82 static enum mod_hdcp_status check_ake_cert_available(struct mod_hdcp *hdcp)
83 {
84 	enum mod_hdcp_status status;
85 	uint16_t size;
86 
87 	if (is_dp_hdcp(hdcp)) {
88 		status = MOD_HDCP_STATUS_SUCCESS;
89 	} else {
90 		status = mod_hdcp_read_rxstatus(hdcp);
91 		if (status == MOD_HDCP_STATUS_SUCCESS) {
92 			size = HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
93 			       hdcp->auth.msg.hdcp2.rxstatus[0];
94 			status = (size == sizeof(hdcp->auth.msg.hdcp2.ake_cert)) ?
95 					MOD_HDCP_STATUS_SUCCESS :
96 					MOD_HDCP_STATUS_HDCP2_AKE_CERT_PENDING;
97 		}
98 	}
99 	return status;
100 }
101 
102 static enum mod_hdcp_status check_h_prime_available(struct mod_hdcp *hdcp)
103 {
104 	enum mod_hdcp_status status;
105 	uint8_t size;
106 
107 	status = mod_hdcp_read_rxstatus(hdcp);
108 	if (status != MOD_HDCP_STATUS_SUCCESS)
109 		goto out;
110 
111 	if (is_dp_hdcp(hdcp)) {
112 		status = HDCP_2_2_DP_RXSTATUS_H_PRIME(hdcp->auth.msg.hdcp2.rxstatus_dp) ?
113 				MOD_HDCP_STATUS_SUCCESS :
114 				MOD_HDCP_STATUS_HDCP2_H_PRIME_PENDING;
115 	} else {
116 		size = HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
117 		       hdcp->auth.msg.hdcp2.rxstatus[0];
118 		status = (size == sizeof(hdcp->auth.msg.hdcp2.ake_h_prime)) ?
119 				MOD_HDCP_STATUS_SUCCESS :
120 				MOD_HDCP_STATUS_HDCP2_H_PRIME_PENDING;
121 	}
122 out:
123 	return status;
124 }
125 
126 static enum mod_hdcp_status check_pairing_info_available(struct mod_hdcp *hdcp)
127 {
128 	enum mod_hdcp_status status;
129 	uint8_t size;
130 
131 	status = mod_hdcp_read_rxstatus(hdcp);
132 	if (status != MOD_HDCP_STATUS_SUCCESS)
133 		goto out;
134 
135 	if (is_dp_hdcp(hdcp)) {
136 		status = HDCP_2_2_DP_RXSTATUS_PAIRING(hdcp->auth.msg.hdcp2.rxstatus_dp) ?
137 				MOD_HDCP_STATUS_SUCCESS :
138 				MOD_HDCP_STATUS_HDCP2_PAIRING_INFO_PENDING;
139 	} else {
140 		size = HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
141 		       hdcp->auth.msg.hdcp2.rxstatus[0];
142 		status = (size == sizeof(hdcp->auth.msg.hdcp2.ake_pairing_info)) ?
143 				MOD_HDCP_STATUS_SUCCESS :
144 				MOD_HDCP_STATUS_HDCP2_PAIRING_INFO_PENDING;
145 	}
146 out:
147 	return status;
148 }
149 
150 static enum mod_hdcp_status poll_l_prime_available(struct mod_hdcp *hdcp)
151 {
152 	enum mod_hdcp_status status;
153 	uint8_t size;
154 	uint16_t max_wait = 20000; // units of us
155 	uint16_t num_polls = 5;
156 	uint16_t wait_time = max_wait / num_polls;
157 
158 	if (is_dp_hdcp(hdcp))
159 		status = MOD_HDCP_STATUS_INVALID_OPERATION;
160 	else
161 		for (; num_polls; num_polls--) {
162 			udelay(wait_time);
163 
164 			status = mod_hdcp_read_rxstatus(hdcp);
165 			if (status != MOD_HDCP_STATUS_SUCCESS)
166 				break;
167 
168 			size = HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
169 			       hdcp->auth.msg.hdcp2.rxstatus[0];
170 			status = (size == sizeof(hdcp->auth.msg.hdcp2.lc_l_prime)) ?
171 					MOD_HDCP_STATUS_SUCCESS :
172 					MOD_HDCP_STATUS_HDCP2_L_PRIME_PENDING;
173 			if (status == MOD_HDCP_STATUS_SUCCESS)
174 				break;
175 		}
176 	return status;
177 }
178 
179 static enum mod_hdcp_status check_stream_ready_available(struct mod_hdcp *hdcp)
180 {
181 	enum mod_hdcp_status status;
182 	uint8_t size;
183 
184 	if (is_dp_hdcp(hdcp)) {
185 		status = MOD_HDCP_STATUS_INVALID_OPERATION;
186 	} else {
187 		status = mod_hdcp_read_rxstatus(hdcp);
188 		if (status != MOD_HDCP_STATUS_SUCCESS)
189 			goto out;
190 		size = HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
191 		       hdcp->auth.msg.hdcp2.rxstatus[0];
192 		status = (size == sizeof(hdcp->auth.msg.hdcp2.repeater_auth_stream_ready)) ?
193 				MOD_HDCP_STATUS_SUCCESS :
194 				MOD_HDCP_STATUS_HDCP2_STREAM_READY_PENDING;
195 	}
196 out:
197 	return status;
198 }
199 
200 static inline uint8_t get_device_count(struct mod_hdcp *hdcp)
201 {
202 	return HDCP_2_2_DEV_COUNT_LO(hdcp->auth.msg.hdcp2.rx_id_list[2]) +
203 			(HDCP_2_2_DEV_COUNT_HI(hdcp->auth.msg.hdcp2.rx_id_list[1]) << 4);
204 }
205 
206 static enum mod_hdcp_status check_device_count(struct mod_hdcp *hdcp)
207 {
208 	/* device count must be greater than or equal to tracked hdcp displays */
209 	return (get_device_count(hdcp) < get_added_display_count(hdcp)) ?
210 			MOD_HDCP_STATUS_HDCP2_DEVICE_COUNT_MISMATCH_FAILURE :
211 			MOD_HDCP_STATUS_SUCCESS;
212 }
213 
214 static uint8_t process_rxstatus(struct mod_hdcp *hdcp,
215 		struct mod_hdcp_event_context *event_ctx,
216 		struct mod_hdcp_transition_input_hdcp2 *input,
217 		enum mod_hdcp_status *status)
218 {
219 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_rxstatus,
220 			&input->rxstatus_read, status,
221 			hdcp, "rxstatus_read"))
222 		goto out;
223 	if (!mod_hdcp_execute_and_set(check_reauthentication_request,
224 			&input->reauth_request_check, status,
225 			hdcp, "reauth_request_check"))
226 		goto out;
227 	if (is_dp_hdcp(hdcp)) {
228 		if (!mod_hdcp_execute_and_set(check_link_integrity_failure_dp,
229 				&input->link_integrity_check_dp, status,
230 				hdcp, "link_integrity_check_dp"))
231 			goto out;
232 	}
233 	if (hdcp->connection.is_repeater)
234 		if (check_receiver_id_list_ready(hdcp) ==
235 				MOD_HDCP_STATUS_SUCCESS) {
236 			HDCP_INPUT_PASS_TRACE(hdcp, "rx_id_list_ready");
237 			event_ctx->rx_id_list_ready = 1;
238 			if (is_dp_hdcp(hdcp))
239 				hdcp->auth.msg.hdcp2.rx_id_list_size =
240 						sizeof(hdcp->auth.msg.hdcp2.rx_id_list);
241 			else
242 				hdcp->auth.msg.hdcp2.rx_id_list_size =
243 					HDCP_2_2_HDMI_RXSTATUS_MSG_SZ_HI(hdcp->auth.msg.hdcp2.rxstatus[1]) << 8 |
244 					hdcp->auth.msg.hdcp2.rxstatus[0];
245 		}
246 out:
247 	return (*status == MOD_HDCP_STATUS_SUCCESS);
248 }
249 
250 static enum mod_hdcp_status known_hdcp2_capable_rx(struct mod_hdcp *hdcp,
251 		struct mod_hdcp_event_context *event_ctx,
252 		struct mod_hdcp_transition_input_hdcp2 *input)
253 {
254 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
255 
256 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
257 		event_ctx->unexpected_event = 1;
258 		goto out;
259 	}
260 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_hdcp2version,
261 			&input->hdcp2version_read, &status,
262 			hdcp, "hdcp2version_read"))
263 		goto out;
264 	if (!mod_hdcp_execute_and_set(check_hdcp2_capable,
265 			&input->hdcp2_capable_check, &status,
266 			hdcp, "hdcp2_capable"))
267 		goto out;
268 out:
269 	return status;
270 }
271 
272 static enum mod_hdcp_status send_ake_init(struct mod_hdcp *hdcp,
273 		struct mod_hdcp_event_context *event_ctx,
274 		struct mod_hdcp_transition_input_hdcp2 *input)
275 {
276 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
277 
278 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
279 		event_ctx->unexpected_event = 1;
280 		goto out;
281 	}
282 	if (!mod_hdcp_execute_and_set(mod_hdcp_add_display_topology,
283 			&input->add_topology, &status,
284 			hdcp, "add_topology"))
285 		goto out;
286 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_create_session,
287 			&input->create_session, &status,
288 			hdcp, "create_session"))
289 		goto out;
290 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_prepare_ake_init,
291 			&input->ake_init_prepare, &status,
292 			hdcp, "ake_init_prepare"))
293 		goto out;
294 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_ake_init,
295 			&input->ake_init_write, &status,
296 			hdcp, "ake_init_write"))
297 		goto out;
298 out:
299 	return status;
300 }
301 
302 static enum mod_hdcp_status validate_ake_cert(struct mod_hdcp *hdcp,
303 		struct mod_hdcp_event_context *event_ctx,
304 		struct mod_hdcp_transition_input_hdcp2 *input)
305 {
306 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
307 
308 
309 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
310 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
311 		event_ctx->unexpected_event = 1;
312 		goto out;
313 	}
314 
315 	if (is_hdmi_dvi_sl_hdcp(hdcp))
316 		if (!mod_hdcp_execute_and_set(check_ake_cert_available,
317 				&input->ake_cert_available, &status,
318 				hdcp, "ake_cert_available"))
319 			goto out;
320 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_ake_cert,
321 			&input->ake_cert_read, &status,
322 			hdcp, "ake_cert_read"))
323 		goto out;
324 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_validate_ake_cert,
325 			&input->ake_cert_validation, &status,
326 			hdcp, "ake_cert_validation"))
327 		goto out;
328 out:
329 	return status;
330 }
331 
332 static enum mod_hdcp_status send_no_stored_km(struct mod_hdcp *hdcp,
333 		struct mod_hdcp_event_context *event_ctx,
334 		struct mod_hdcp_transition_input_hdcp2 *input)
335 {
336 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
337 
338 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
339 		event_ctx->unexpected_event = 1;
340 		goto out;
341 	}
342 
343 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_no_stored_km,
344 			&input->no_stored_km_write, &status,
345 			hdcp, "no_stored_km_write"))
346 		goto out;
347 out:
348 	return status;
349 }
350 
351 static enum mod_hdcp_status read_h_prime(struct mod_hdcp *hdcp,
352 		struct mod_hdcp_event_context *event_ctx,
353 		struct mod_hdcp_transition_input_hdcp2 *input)
354 {
355 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
356 
357 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
358 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
359 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
360 		event_ctx->unexpected_event = 1;
361 		goto out;
362 	}
363 
364 	if (!mod_hdcp_execute_and_set(check_h_prime_available,
365 			&input->h_prime_available, &status,
366 			hdcp, "h_prime_available"))
367 		goto out;
368 
369 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_h_prime,
370 			&input->h_prime_read, &status,
371 			hdcp, "h_prime_read"))
372 		goto out;
373 out:
374 	return status;
375 }
376 
377 static enum mod_hdcp_status read_pairing_info_and_validate_h_prime(
378 		struct mod_hdcp *hdcp,
379 		struct mod_hdcp_event_context *event_ctx,
380 		struct mod_hdcp_transition_input_hdcp2 *input)
381 {
382 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
383 
384 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
385 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
386 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
387 		event_ctx->unexpected_event = 1;
388 		goto out;
389 	}
390 
391 	if (!mod_hdcp_execute_and_set(check_pairing_info_available,
392 			&input->pairing_available, &status,
393 			hdcp, "pairing_available"))
394 		goto out;
395 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_pairing_info,
396 			&input->pairing_info_read, &status,
397 			hdcp, "pairing_info_read"))
398 		goto out;
399 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_validate_h_prime,
400 			&input->h_prime_validation, &status,
401 			hdcp, "h_prime_validation"))
402 		goto out;
403 out:
404 	return status;
405 }
406 
407 static enum mod_hdcp_status send_stored_km(struct mod_hdcp *hdcp,
408 		struct mod_hdcp_event_context *event_ctx,
409 		struct mod_hdcp_transition_input_hdcp2 *input)
410 {
411 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
412 
413 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
414 		event_ctx->unexpected_event = 1;
415 		goto out;
416 	}
417 
418 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_stored_km,
419 			&input->stored_km_write, &status,
420 			hdcp, "stored_km_write"))
421 		goto out;
422 out:
423 	return status;
424 }
425 
426 static enum mod_hdcp_status validate_h_prime(struct mod_hdcp *hdcp,
427 		struct mod_hdcp_event_context *event_ctx,
428 		struct mod_hdcp_transition_input_hdcp2 *input)
429 {
430 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
431 
432 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
433 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
434 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
435 		event_ctx->unexpected_event = 1;
436 		goto out;
437 	}
438 
439 	if (!mod_hdcp_execute_and_set(check_h_prime_available,
440 			&input->h_prime_available, &status,
441 			hdcp, "h_prime_available"))
442 		goto out;
443 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_h_prime,
444 			&input->h_prime_read, &status,
445 			hdcp, "h_prime_read"))
446 		goto out;
447 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_validate_h_prime,
448 			&input->h_prime_validation, &status,
449 			hdcp, "h_prime_validation"))
450 		goto out;
451 out:
452 	return status;
453 }
454 
455 static enum mod_hdcp_status locality_check(struct mod_hdcp *hdcp,
456 		struct mod_hdcp_event_context *event_ctx,
457 		struct mod_hdcp_transition_input_hdcp2 *input)
458 {
459 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
460 
461 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
462 		event_ctx->unexpected_event = 1;
463 		goto out;
464 	}
465 
466 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_prepare_lc_init,
467 			&input->lc_init_prepare, &status,
468 			hdcp, "lc_init_prepare"))
469 		goto out;
470 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_lc_init,
471 			&input->lc_init_write, &status,
472 			 hdcp, "lc_init_write"))
473 		goto out;
474 	if (is_dp_hdcp(hdcp))
475 		udelay(16000);
476 	else
477 		if (!mod_hdcp_execute_and_set(poll_l_prime_available,
478 				&input->l_prime_available_poll, &status,
479 				hdcp, "l_prime_available_poll"))
480 			goto out;
481 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_l_prime,
482 			&input->l_prime_read, &status,
483 			hdcp, "l_prime_read"))
484 		goto out;
485 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_validate_l_prime,
486 			&input->l_prime_validation, &status,
487 			hdcp, "l_prime_validation"))
488 		goto out;
489 out:
490 	return status;
491 }
492 
493 static enum mod_hdcp_status exchange_ks_and_test_for_repeater(struct mod_hdcp *hdcp,
494 		struct mod_hdcp_event_context *event_ctx,
495 		struct mod_hdcp_transition_input_hdcp2 *input)
496 {
497 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
498 
499 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
500 		event_ctx->unexpected_event = 1;
501 		goto out;
502 	}
503 
504 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_prepare_eks,
505 			&input->eks_prepare, &status,
506 			hdcp, "eks_prepare"))
507 		goto out;
508 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_eks,
509 			&input->eks_write, &status,
510 			hdcp, "eks_write"))
511 		goto out;
512 out:
513 	return status;
514 }
515 
516 static enum mod_hdcp_status enable_encryption(struct mod_hdcp *hdcp,
517 		struct mod_hdcp_event_context *event_ctx,
518 		struct mod_hdcp_transition_input_hdcp2 *input)
519 {
520 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
521 
522 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
523 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ) {
524 		event_ctx->unexpected_event = 1;
525 		goto out;
526 	}
527 	if (event_ctx->event == MOD_HDCP_EVENT_CPIRQ) {
528 		process_rxstatus(hdcp, event_ctx, input, &status);
529 		goto out;
530 	}
531 
532 	if (is_hdmi_dvi_sl_hdcp(hdcp)) {
533 		if (!process_rxstatus(hdcp, event_ctx, input, &status))
534 			goto out;
535 		if (event_ctx->rx_id_list_ready)
536 			goto out;
537 	}
538 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_enable_encryption,
539 			&input->enable_encryption, &status,
540 			hdcp, "enable_encryption"))
541 		goto out;
542 	if (is_dp_mst_hdcp(hdcp)) {
543 		if (!mod_hdcp_execute_and_set(
544 				mod_hdcp_hdcp2_enable_dp_stream_encryption,
545 				&input->stream_encryption_dp, &status,
546 				hdcp, "stream_encryption_dp"))
547 			goto out;
548 	}
549 out:
550 	return status;
551 }
552 
553 static enum mod_hdcp_status authenticated(struct mod_hdcp *hdcp,
554 		struct mod_hdcp_event_context *event_ctx,
555 		struct mod_hdcp_transition_input_hdcp2 *input)
556 {
557 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
558 
559 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
560 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ) {
561 		event_ctx->unexpected_event = 1;
562 		goto out;
563 	}
564 
565 	if (!process_rxstatus(hdcp, event_ctx, input, &status))
566 		goto out;
567 	if (event_ctx->rx_id_list_ready)
568 		goto out;
569 out:
570 	return status;
571 }
572 
573 static enum mod_hdcp_status wait_for_rx_id_list(struct mod_hdcp *hdcp,
574 		struct mod_hdcp_event_context *event_ctx,
575 		struct mod_hdcp_transition_input_hdcp2 *input)
576 {
577 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
578 
579 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
580 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
581 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
582 		event_ctx->unexpected_event = 1;
583 		goto out;
584 	}
585 
586 	if (!process_rxstatus(hdcp, event_ctx, input, &status))
587 		goto out;
588 	if (!event_ctx->rx_id_list_ready) {
589 		status = MOD_HDCP_STATUS_HDCP2_RX_ID_LIST_NOT_READY;
590 		goto out;
591 	}
592 out:
593 	return status;
594 }
595 
596 static enum mod_hdcp_status verify_rx_id_list_and_send_ack(struct mod_hdcp *hdcp,
597 		struct mod_hdcp_event_context *event_ctx,
598 		struct mod_hdcp_transition_input_hdcp2 *input)
599 {
600 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
601 
602 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
603 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ) {
604 		event_ctx->unexpected_event = 1;
605 		goto out;
606 	}
607 	if (event_ctx->event == MOD_HDCP_EVENT_CPIRQ) {
608 		process_rxstatus(hdcp, event_ctx, input, &status);
609 		goto out;
610 	}
611 
612 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_rx_id_list,
613 			&input->rx_id_list_read,
614 			&status, hdcp, "receiver_id_list_read"))
615 		goto out;
616 	if (!mod_hdcp_execute_and_set(check_device_count,
617 			&input->device_count_check,
618 			&status, hdcp, "device_count_check"))
619 		goto out;
620 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_validate_rx_id_list,
621 			&input->rx_id_list_validation,
622 			&status, hdcp, "rx_id_list_validation"))
623 		goto out;
624 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_repeater_auth_ack,
625 			&input->repeater_auth_ack_write,
626 			&status, hdcp, "repeater_auth_ack_write"))
627 		goto out;
628 out:
629 	return status;
630 }
631 
632 static enum mod_hdcp_status send_stream_management(struct mod_hdcp *hdcp,
633 		struct mod_hdcp_event_context *event_ctx,
634 		struct mod_hdcp_transition_input_hdcp2 *input)
635 {
636 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
637 
638 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
639 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ) {
640 		event_ctx->unexpected_event = 1;
641 		goto out;
642 	}
643 	if (event_ctx->event == MOD_HDCP_EVENT_CPIRQ) {
644 		process_rxstatus(hdcp, event_ctx, input, &status);
645 		goto out;
646 	}
647 
648 	if (is_hdmi_dvi_sl_hdcp(hdcp)) {
649 		if (!process_rxstatus(hdcp, event_ctx, input, &status))
650 			goto out;
651 		if (event_ctx->rx_id_list_ready)
652 			goto out;
653 	}
654 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_prepare_stream_management,
655 			&input->prepare_stream_manage,
656 			&status, hdcp, "prepare_stream_manage"))
657 		goto out;
658 
659 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_stream_manage,
660 			&input->stream_manage_write,
661 			&status, hdcp, "stream_manage_write"))
662 		goto out;
663 out:
664 	return status;
665 }
666 
667 static enum mod_hdcp_status validate_stream_ready(struct mod_hdcp *hdcp,
668 		struct mod_hdcp_event_context *event_ctx,
669 		struct mod_hdcp_transition_input_hdcp2 *input)
670 {
671 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
672 
673 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
674 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
675 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
676 		event_ctx->unexpected_event = 1;
677 		goto out;
678 	}
679 	if (event_ctx->event == MOD_HDCP_EVENT_CPIRQ) {
680 		process_rxstatus(hdcp, event_ctx, input, &status);
681 		goto out;
682 	}
683 
684 	if (is_hdmi_dvi_sl_hdcp(hdcp)) {
685 		if (!process_rxstatus(hdcp, event_ctx, input, &status))
686 			goto out;
687 		if (event_ctx->rx_id_list_ready) {
688 			goto out;
689 		}
690 	}
691 	if (is_hdmi_dvi_sl_hdcp(hdcp))
692 		if (!mod_hdcp_execute_and_set(check_stream_ready_available,
693 				&input->stream_ready_available,
694 				&status, hdcp, "stream_ready_available"))
695 			goto out;
696 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_stream_ready,
697 			&input->stream_ready_read,
698 			&status, hdcp, "stream_ready_read"))
699 		goto out;
700 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp2_validate_stream_ready,
701 			&input->stream_ready_validation,
702 			&status, hdcp, "stream_ready_validation"))
703 		goto out;
704 
705 out:
706 	return status;
707 }
708 
709 static enum mod_hdcp_status determine_rx_hdcp_capable_dp(struct mod_hdcp *hdcp,
710 		struct mod_hdcp_event_context *event_ctx,
711 		struct mod_hdcp_transition_input_hdcp2 *input)
712 {
713 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
714 
715 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
716 		event_ctx->unexpected_event = 1;
717 		goto out;
718 	}
719 
720 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_rxcaps,
721 			&input->rx_caps_read_dp,
722 			&status, hdcp, "rx_caps_read_dp"))
723 		goto out;
724 	if (!mod_hdcp_execute_and_set(check_hdcp2_capable,
725 			&input->hdcp2_capable_check, &status,
726 			hdcp, "hdcp2_capable_check"))
727 		goto out;
728 out:
729 	return status;
730 }
731 
732 static enum mod_hdcp_status send_content_stream_type_dp(struct mod_hdcp *hdcp,
733 		struct mod_hdcp_event_context *event_ctx,
734 		struct mod_hdcp_transition_input_hdcp2 *input)
735 {
736 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
737 
738 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
739 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ) {
740 		event_ctx->unexpected_event = 1;
741 		goto out;
742 	}
743 
744 	if (!process_rxstatus(hdcp, event_ctx, input, &status))
745 		goto out;
746 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_content_type,
747 			&input->content_stream_type_write, &status,
748 			hdcp, "content_stream_type_write"))
749 		goto out;
750 out:
751 	return status;
752 }
753 
754 enum mod_hdcp_status mod_hdcp_hdcp2_execution(struct mod_hdcp *hdcp,
755 	struct mod_hdcp_event_context *event_ctx,
756 	struct mod_hdcp_transition_input_hdcp2 *input)
757 {
758 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
759 
760 	switch (current_state(hdcp)) {
761 	case H2_A0_KNOWN_HDCP2_CAPABLE_RX:
762 		status = known_hdcp2_capable_rx(hdcp, event_ctx, input);
763 		break;
764 	case H2_A1_SEND_AKE_INIT:
765 		status = send_ake_init(hdcp, event_ctx, input);
766 		break;
767 	case H2_A1_VALIDATE_AKE_CERT:
768 		status = validate_ake_cert(hdcp, event_ctx, input);
769 		break;
770 	case H2_A1_SEND_NO_STORED_KM:
771 		status = send_no_stored_km(hdcp, event_ctx, input);
772 		break;
773 	case H2_A1_READ_H_PRIME:
774 		status = read_h_prime(hdcp, event_ctx, input);
775 		break;
776 	case H2_A1_READ_PAIRING_INFO_AND_VALIDATE_H_PRIME:
777 		status = read_pairing_info_and_validate_h_prime(hdcp,
778 				event_ctx, input);
779 		break;
780 	case H2_A1_SEND_STORED_KM:
781 		status = send_stored_km(hdcp, event_ctx, input);
782 		break;
783 	case H2_A1_VALIDATE_H_PRIME:
784 		status = validate_h_prime(hdcp, event_ctx, input);
785 		break;
786 	case H2_A2_LOCALITY_CHECK:
787 		status = locality_check(hdcp, event_ctx, input);
788 		break;
789 	case H2_A3_EXCHANGE_KS_AND_TEST_FOR_REPEATER:
790 		status = exchange_ks_and_test_for_repeater(hdcp, event_ctx, input);
791 		break;
792 	case H2_ENABLE_ENCRYPTION:
793 		status = enable_encryption(hdcp, event_ctx, input);
794 		break;
795 	case H2_A5_AUTHENTICATED:
796 		status = authenticated(hdcp, event_ctx, input);
797 		break;
798 	case H2_A6_WAIT_FOR_RX_ID_LIST:
799 		status = wait_for_rx_id_list(hdcp, event_ctx, input);
800 		break;
801 	case H2_A78_VERIFY_RX_ID_LIST_AND_SEND_ACK:
802 		status = verify_rx_id_list_and_send_ack(hdcp, event_ctx, input);
803 		break;
804 	case H2_A9_SEND_STREAM_MANAGEMENT:
805 		status = send_stream_management(hdcp, event_ctx, input);
806 		break;
807 	case H2_A9_VALIDATE_STREAM_READY:
808 		status = validate_stream_ready(hdcp, event_ctx, input);
809 		break;
810 	default:
811 		status = MOD_HDCP_STATUS_INVALID_STATE;
812 		break;
813 	}
814 
815 	return status;
816 }
817 
818 enum mod_hdcp_status mod_hdcp_hdcp2_dp_execution(struct mod_hdcp *hdcp,
819 	struct mod_hdcp_event_context *event_ctx,
820 	struct mod_hdcp_transition_input_hdcp2 *input)
821 {
822 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
823 
824 	switch (current_state(hdcp)) {
825 	case D2_A0_DETERMINE_RX_HDCP_CAPABLE:
826 		status = determine_rx_hdcp_capable_dp(hdcp, event_ctx, input);
827 		break;
828 	case D2_A1_SEND_AKE_INIT:
829 		status = send_ake_init(hdcp, event_ctx, input);
830 		break;
831 	case D2_A1_VALIDATE_AKE_CERT:
832 		status = validate_ake_cert(hdcp, event_ctx, input);
833 		break;
834 	case D2_A1_SEND_NO_STORED_KM:
835 		status = send_no_stored_km(hdcp, event_ctx, input);
836 		break;
837 	case D2_A1_READ_H_PRIME:
838 		status = read_h_prime(hdcp, event_ctx, input);
839 		break;
840 	case D2_A1_READ_PAIRING_INFO_AND_VALIDATE_H_PRIME:
841 		status = read_pairing_info_and_validate_h_prime(hdcp,
842 				event_ctx, input);
843 		break;
844 	case D2_A1_SEND_STORED_KM:
845 		status = send_stored_km(hdcp, event_ctx, input);
846 		break;
847 	case D2_A1_VALIDATE_H_PRIME:
848 		status = validate_h_prime(hdcp, event_ctx, input);
849 		break;
850 	case D2_A2_LOCALITY_CHECK:
851 		status = locality_check(hdcp, event_ctx, input);
852 		break;
853 	case D2_A34_EXCHANGE_KS_AND_TEST_FOR_REPEATER:
854 		status = exchange_ks_and_test_for_repeater(hdcp,
855 				event_ctx, input);
856 		break;
857 	case D2_SEND_CONTENT_STREAM_TYPE:
858 		status = send_content_stream_type_dp(hdcp, event_ctx, input);
859 		break;
860 	case D2_ENABLE_ENCRYPTION:
861 		status = enable_encryption(hdcp, event_ctx, input);
862 		break;
863 	case D2_A5_AUTHENTICATED:
864 		status = authenticated(hdcp, event_ctx, input);
865 		break;
866 	case D2_A6_WAIT_FOR_RX_ID_LIST:
867 		status = wait_for_rx_id_list(hdcp, event_ctx, input);
868 		break;
869 	case D2_A78_VERIFY_RX_ID_LIST_AND_SEND_ACK:
870 		status = verify_rx_id_list_and_send_ack(hdcp, event_ctx, input);
871 		break;
872 	case D2_A9_SEND_STREAM_MANAGEMENT:
873 		status = send_stream_management(hdcp, event_ctx, input);
874 		break;
875 	case D2_A9_VALIDATE_STREAM_READY:
876 		status = validate_stream_ready(hdcp, event_ctx, input);
877 		break;
878 	default:
879 		status = MOD_HDCP_STATUS_INVALID_STATE;
880 		break;
881 	}
882 
883 	return status;
884 }
885