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