1 /*
2  * Copyright 2019 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 validate_bksv(struct mod_hdcp *hdcp)
29 {
30 	uint64_t n = 0;
31 	uint8_t count = 0;
32 
33 	memcpy(&n, hdcp->auth.msg.hdcp1.bksv, sizeof(uint64_t));
34 
35 	while (n) {
36 		count++;
37 		n &= (n - 1);
38 	}
39 	return (count == 20) ? MOD_HDCP_STATUS_SUCCESS :
40 			MOD_HDCP_STATUS_HDCP1_INVALID_BKSV;
41 }
42 
43 static inline enum mod_hdcp_status check_ksv_ready(struct mod_hdcp *hdcp)
44 {
45 	if (is_dp_hdcp(hdcp))
46 		return (hdcp->auth.msg.hdcp1.bstatus & DP_BSTATUS_READY) ?
47 				MOD_HDCP_STATUS_SUCCESS :
48 				MOD_HDCP_STATUS_HDCP1_KSV_LIST_NOT_READY;
49 	return (hdcp->auth.msg.hdcp1.bcaps & DRM_HDCP_DDC_BCAPS_KSV_FIFO_READY) ?
50 			MOD_HDCP_STATUS_SUCCESS :
51 			MOD_HDCP_STATUS_HDCP1_KSV_LIST_NOT_READY;
52 }
53 
54 static inline enum mod_hdcp_status check_hdcp_capable_dp(struct mod_hdcp *hdcp)
55 {
56 	return (hdcp->auth.msg.hdcp1.bcaps & DP_BCAPS_HDCP_CAPABLE) ?
57 			MOD_HDCP_STATUS_SUCCESS :
58 			MOD_HDCP_STATUS_HDCP1_NOT_CAPABLE;
59 }
60 
61 static inline enum mod_hdcp_status check_r0p_available_dp(struct mod_hdcp *hdcp)
62 {
63 	enum mod_hdcp_status status;
64 	if (is_dp_hdcp(hdcp)) {
65 		status = (hdcp->auth.msg.hdcp1.bstatus &
66 				DP_BSTATUS_R0_PRIME_READY) ?
67 			MOD_HDCP_STATUS_SUCCESS :
68 			MOD_HDCP_STATUS_HDCP1_R0_PRIME_PENDING;
69 	} else {
70 		status = MOD_HDCP_STATUS_INVALID_OPERATION;
71 	}
72 	return status;
73 }
74 
75 static inline enum mod_hdcp_status check_link_integrity_dp(
76 		struct mod_hdcp *hdcp)
77 {
78 	return (hdcp->auth.msg.hdcp1.bstatus &
79 			DP_BSTATUS_LINK_FAILURE) ?
80 			MOD_HDCP_STATUS_HDCP1_LINK_INTEGRITY_FAILURE :
81 			MOD_HDCP_STATUS_SUCCESS;
82 }
83 
84 static inline enum mod_hdcp_status check_no_reauthentication_request_dp(
85 		struct mod_hdcp *hdcp)
86 {
87 	return (hdcp->auth.msg.hdcp1.bstatus & DP_BSTATUS_REAUTH_REQ) ?
88 			MOD_HDCP_STATUS_HDCP1_REAUTH_REQUEST_ISSUED :
89 			MOD_HDCP_STATUS_SUCCESS;
90 }
91 
92 static inline enum mod_hdcp_status check_no_max_cascade(struct mod_hdcp *hdcp)
93 {
94 	enum mod_hdcp_status status;
95 
96 	if (is_dp_hdcp(hdcp))
97 		status = DRM_HDCP_MAX_CASCADE_EXCEEDED(hdcp->auth.msg.hdcp1.binfo_dp >> 8)
98 				 ? MOD_HDCP_STATUS_HDCP1_MAX_CASCADE_EXCEEDED_FAILURE
99 				 : MOD_HDCP_STATUS_SUCCESS;
100 	else
101 		status = DRM_HDCP_MAX_CASCADE_EXCEEDED(hdcp->auth.msg.hdcp1.bstatus >> 8)
102 				 ? MOD_HDCP_STATUS_HDCP1_MAX_CASCADE_EXCEEDED_FAILURE
103 				 : MOD_HDCP_STATUS_SUCCESS;
104 	return status;
105 }
106 
107 static inline enum mod_hdcp_status check_no_max_devs(struct mod_hdcp *hdcp)
108 {
109 	enum mod_hdcp_status status;
110 
111 	if (is_dp_hdcp(hdcp))
112 		status = DRM_HDCP_MAX_DEVICE_EXCEEDED(hdcp->auth.msg.hdcp1.binfo_dp) ?
113 				MOD_HDCP_STATUS_HDCP1_MAX_DEVS_EXCEEDED_FAILURE :
114 				MOD_HDCP_STATUS_SUCCESS;
115 	else
116 		status = DRM_HDCP_MAX_DEVICE_EXCEEDED(hdcp->auth.msg.hdcp1.bstatus) ?
117 				MOD_HDCP_STATUS_HDCP1_MAX_DEVS_EXCEEDED_FAILURE :
118 				MOD_HDCP_STATUS_SUCCESS;
119 	return status;
120 }
121 
122 static inline uint8_t get_device_count(struct mod_hdcp *hdcp)
123 {
124 	return is_dp_hdcp(hdcp) ?
125 			DRM_HDCP_NUM_DOWNSTREAM(hdcp->auth.msg.hdcp1.binfo_dp) :
126 			DRM_HDCP_NUM_DOWNSTREAM(hdcp->auth.msg.hdcp1.bstatus);
127 }
128 
129 static inline enum mod_hdcp_status check_device_count(struct mod_hdcp *hdcp)
130 {
131 	/* Some MST display may choose to report the internal panel as an HDCP RX.
132 	 * To update this condition with 1(because the immediate repeater's internal
133 	 * panel is possibly not included in DEVICE_COUNT) + get_device_count(hdcp).
134 	 * Device count must be greater than or equal to tracked hdcp displays.
135 	 */
136 	return ((1 + get_device_count(hdcp)) < get_active_display_count(hdcp)) ?
137 			MOD_HDCP_STATUS_HDCP1_DEVICE_COUNT_MISMATCH_FAILURE :
138 			MOD_HDCP_STATUS_SUCCESS;
139 }
140 
141 static enum mod_hdcp_status wait_for_active_rx(struct mod_hdcp *hdcp,
142 		struct mod_hdcp_event_context *event_ctx,
143 		struct mod_hdcp_transition_input_hdcp1 *input)
144 {
145 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
146 
147 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
148 		event_ctx->unexpected_event = 1;
149 		goto out;
150 	}
151 
152 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_bksv,
153 			&input->bksv_read, &status,
154 			hdcp, "bksv_read"))
155 		goto out;
156 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_bcaps,
157 			&input->bcaps_read, &status,
158 			hdcp, "bcaps_read"))
159 		goto out;
160 out:
161 	return status;
162 }
163 
164 static enum mod_hdcp_status exchange_ksvs(struct mod_hdcp *hdcp,
165 		struct mod_hdcp_event_context *event_ctx,
166 		struct mod_hdcp_transition_input_hdcp1 *input)
167 {
168 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
169 
170 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
171 		event_ctx->unexpected_event = 1;
172 		goto out;
173 	}
174 
175 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp1_create_session,
176 			&input->create_session, &status,
177 			hdcp, "create_session"))
178 		goto out;
179 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_an,
180 			&input->an_write, &status,
181 			hdcp, "an_write"))
182 		goto out;
183 	if (!mod_hdcp_execute_and_set(mod_hdcp_write_aksv,
184 			&input->aksv_write, &status,
185 			hdcp, "aksv_write"))
186 		goto out;
187 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_bksv,
188 			&input->bksv_read, &status,
189 			hdcp, "bksv_read"))
190 		goto out;
191 	if (!mod_hdcp_execute_and_set(validate_bksv,
192 			&input->bksv_validation, &status,
193 			hdcp, "bksv_validation"))
194 		goto out;
195 	if (hdcp->auth.msg.hdcp1.ainfo) {
196 		if (!mod_hdcp_execute_and_set(mod_hdcp_write_ainfo,
197 				&input->ainfo_write, &status,
198 				hdcp, "ainfo_write"))
199 			goto out;
200 	}
201 out:
202 	return status;
203 }
204 
205 static enum mod_hdcp_status computations_validate_rx_test_for_repeater(
206 		struct mod_hdcp *hdcp,
207 		struct mod_hdcp_event_context *event_ctx,
208 		struct mod_hdcp_transition_input_hdcp1 *input)
209 {
210 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
211 
212 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
213 		event_ctx->unexpected_event = 1;
214 		goto out;
215 	}
216 
217 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_r0p,
218 			&input->r0p_read, &status,
219 			hdcp, "r0p_read"))
220 		goto out;
221 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp1_validate_rx,
222 			&input->rx_validation, &status,
223 			hdcp, "rx_validation"))
224 		goto out;
225 	if (hdcp->connection.is_repeater) {
226 		if (!hdcp->connection.link.adjust.hdcp1.postpone_encryption)
227 			if (!mod_hdcp_execute_and_set(
228 					mod_hdcp_hdcp1_enable_encryption,
229 					&input->encryption, &status,
230 					hdcp, "encryption"))
231 				goto out;
232 	} else {
233 		if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp1_enable_encryption,
234 				&input->encryption, &status,
235 				hdcp, "encryption"))
236 			goto out;
237 		if (is_dp_mst_hdcp(hdcp))
238 			if (!mod_hdcp_execute_and_set(
239 					mod_hdcp_hdcp1_enable_dp_stream_encryption,
240 					&input->stream_encryption_dp, &status,
241 					hdcp, "stream_encryption_dp"))
242 				goto out;
243 	}
244 out:
245 	return status;
246 }
247 
248 static enum mod_hdcp_status authenticated(struct mod_hdcp *hdcp,
249 		struct mod_hdcp_event_context *event_ctx,
250 		struct mod_hdcp_transition_input_hdcp1 *input)
251 {
252 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
253 
254 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
255 		event_ctx->unexpected_event = 1;
256 		goto out;
257 	}
258 
259 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp1_link_maintenance,
260 			&input->link_maintenance, &status,
261 			hdcp, "link_maintenance"))
262 		goto out;
263 out:
264 	return status;
265 }
266 
267 static enum mod_hdcp_status wait_for_ready(struct mod_hdcp *hdcp,
268 		struct mod_hdcp_event_context *event_ctx,
269 		struct mod_hdcp_transition_input_hdcp1 *input)
270 {
271 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
272 
273 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK &&
274 			event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
275 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
276 		event_ctx->unexpected_event = 1;
277 		goto out;
278 	}
279 
280 	if (is_dp_hdcp(hdcp)) {
281 		if (!mod_hdcp_execute_and_set(mod_hdcp_read_bstatus,
282 				&input->bstatus_read, &status,
283 				hdcp, "bstatus_read"))
284 			goto out;
285 		if (!mod_hdcp_execute_and_set(check_link_integrity_dp,
286 				&input->link_integrity_check, &status,
287 				hdcp, "link_integrity_check"))
288 			goto out;
289 		if (!mod_hdcp_execute_and_set(check_no_reauthentication_request_dp,
290 				&input->reauth_request_check, &status,
291 				hdcp, "reauth_request_check"))
292 			goto out;
293 	} else {
294 		if (!mod_hdcp_execute_and_set(mod_hdcp_read_bcaps,
295 				&input->bcaps_read, &status,
296 				hdcp, "bcaps_read"))
297 			goto out;
298 	}
299 	if (!mod_hdcp_execute_and_set(check_ksv_ready,
300 			&input->ready_check, &status,
301 			hdcp, "ready_check"))
302 		goto out;
303 out:
304 	return status;
305 }
306 
307 static enum mod_hdcp_status read_ksv_list(struct mod_hdcp *hdcp,
308 		struct mod_hdcp_event_context *event_ctx,
309 		struct mod_hdcp_transition_input_hdcp1 *input)
310 {
311 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
312 	uint8_t device_count;
313 
314 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
315 		event_ctx->unexpected_event = 1;
316 		goto out;
317 	}
318 
319 	if (is_dp_hdcp(hdcp)) {
320 		if (!mod_hdcp_execute_and_set(mod_hdcp_read_binfo,
321 				&input->binfo_read_dp, &status,
322 				hdcp, "binfo_read_dp"))
323 			goto out;
324 	} else {
325 		if (!mod_hdcp_execute_and_set(mod_hdcp_read_bstatus,
326 				&input->bstatus_read, &status,
327 				hdcp, "bstatus_read"))
328 			goto out;
329 	}
330 	if (!mod_hdcp_execute_and_set(check_no_max_cascade,
331 			&input->max_cascade_check, &status,
332 			hdcp, "max_cascade_check"))
333 		goto out;
334 	if (!mod_hdcp_execute_and_set(check_no_max_devs,
335 			&input->max_devs_check, &status,
336 			hdcp, "max_devs_check"))
337 		goto out;
338 	if (!mod_hdcp_execute_and_set(check_device_count,
339 			&input->device_count_check, &status,
340 			hdcp, "device_count_check"))
341 		goto out;
342 	device_count = get_device_count(hdcp);
343 	hdcp->auth.msg.hdcp1.ksvlist_size = device_count*5;
344 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_ksvlist,
345 			&input->ksvlist_read, &status,
346 			hdcp, "ksvlist_read"))
347 		goto out;
348 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_vp,
349 			&input->vp_read, &status,
350 			hdcp, "vp_read"))
351 		goto out;
352 	if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp1_validate_ksvlist_vp,
353 			&input->ksvlist_vp_validation, &status,
354 			hdcp, "ksvlist_vp_validation"))
355 		goto out;
356 	if (input->encryption != PASS)
357 		if (!mod_hdcp_execute_and_set(mod_hdcp_hdcp1_enable_encryption,
358 				&input->encryption, &status,
359 				hdcp, "encryption"))
360 			goto out;
361 	if (is_dp_mst_hdcp(hdcp))
362 		if (!mod_hdcp_execute_and_set(
363 				mod_hdcp_hdcp1_enable_dp_stream_encryption,
364 				&input->stream_encryption_dp, &status,
365 				hdcp, "stream_encryption_dp"))
366 			goto out;
367 out:
368 	return status;
369 }
370 
371 static enum mod_hdcp_status determine_rx_hdcp_capable_dp(struct mod_hdcp *hdcp,
372 		struct mod_hdcp_event_context *event_ctx,
373 		struct mod_hdcp_transition_input_hdcp1 *input)
374 {
375 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
376 
377 	if (event_ctx->event != MOD_HDCP_EVENT_CALLBACK) {
378 		event_ctx->unexpected_event = 1;
379 		goto out;
380 	}
381 
382 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_bcaps,
383 			&input->bcaps_read, &status,
384 			hdcp, "bcaps_read"))
385 		goto out;
386 	if (!mod_hdcp_execute_and_set(check_hdcp_capable_dp,
387 			&input->hdcp_capable_dp, &status,
388 			hdcp, "hdcp_capable_dp"))
389 		goto out;
390 out:
391 	return status;
392 }
393 
394 static enum mod_hdcp_status wait_for_r0_prime_dp(struct mod_hdcp *hdcp,
395 		struct mod_hdcp_event_context *event_ctx,
396 		struct mod_hdcp_transition_input_hdcp1 *input)
397 {
398 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
399 
400 	if (event_ctx->event != MOD_HDCP_EVENT_CPIRQ &&
401 			event_ctx->event != MOD_HDCP_EVENT_WATCHDOG_TIMEOUT) {
402 		event_ctx->unexpected_event = 1;
403 		goto out;
404 	}
405 
406 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_bstatus,
407 			&input->bstatus_read, &status,
408 			hdcp, "bstatus_read"))
409 		goto out;
410 	if (!mod_hdcp_execute_and_set(check_r0p_available_dp,
411 			&input->r0p_available_dp, &status,
412 			hdcp, "r0p_available_dp"))
413 		goto out;
414 out:
415 	return status;
416 }
417 
418 static enum mod_hdcp_status authenticated_dp(struct mod_hdcp *hdcp,
419 		struct mod_hdcp_event_context *event_ctx,
420 		struct mod_hdcp_transition_input_hdcp1 *input)
421 {
422 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
423 
424 	if (event_ctx->event != MOD_HDCP_EVENT_CPIRQ) {
425 		event_ctx->unexpected_event = 1;
426 		goto out;
427 	}
428 
429 	if (!mod_hdcp_execute_and_set(mod_hdcp_read_bstatus,
430 			&input->bstatus_read, &status,
431 			hdcp, "bstatus_read"))
432 		goto out;
433 	if (!mod_hdcp_execute_and_set(check_link_integrity_dp,
434 			&input->link_integrity_check, &status,
435 			hdcp, "link_integrity_check"))
436 		goto out;
437 	if (!mod_hdcp_execute_and_set(check_no_reauthentication_request_dp,
438 			&input->reauth_request_check, &status,
439 			hdcp, "reauth_request_check"))
440 		goto out;
441 out:
442 	return status;
443 }
444 
445 uint8_t mod_hdcp_execute_and_set(
446 		mod_hdcp_action func, uint8_t *flag,
447 		enum mod_hdcp_status *status, struct mod_hdcp *hdcp, char *str)
448 {
449 	*status = func(hdcp);
450 	if (*status == MOD_HDCP_STATUS_SUCCESS && *flag != PASS) {
451 		HDCP_INPUT_PASS_TRACE(hdcp, str);
452 		*flag = PASS;
453 	} else if (*status != MOD_HDCP_STATUS_SUCCESS && *flag != FAIL) {
454 		HDCP_INPUT_FAIL_TRACE(hdcp, str);
455 		*flag = FAIL;
456 	}
457 	return (*status == MOD_HDCP_STATUS_SUCCESS);
458 }
459 
460 enum mod_hdcp_status mod_hdcp_hdcp1_execution(struct mod_hdcp *hdcp,
461 		struct mod_hdcp_event_context *event_ctx,
462 		struct mod_hdcp_transition_input_hdcp1 *input)
463 {
464 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
465 
466 	switch (current_state(hdcp)) {
467 	case H1_A0_WAIT_FOR_ACTIVE_RX:
468 		status = wait_for_active_rx(hdcp, event_ctx, input);
469 		break;
470 	case H1_A1_EXCHANGE_KSVS:
471 		status = exchange_ksvs(hdcp, event_ctx, input);
472 		break;
473 	case H1_A2_COMPUTATIONS_A3_VALIDATE_RX_A6_TEST_FOR_REPEATER:
474 		status = computations_validate_rx_test_for_repeater(hdcp,
475 				event_ctx, input);
476 		break;
477 	case H1_A45_AUTHENTICATED:
478 		status = authenticated(hdcp, event_ctx, input);
479 		break;
480 	case H1_A8_WAIT_FOR_READY:
481 		status = wait_for_ready(hdcp, event_ctx, input);
482 		break;
483 	case H1_A9_READ_KSV_LIST:
484 		status = read_ksv_list(hdcp, event_ctx, input);
485 		break;
486 	default:
487 		status = MOD_HDCP_STATUS_INVALID_STATE;
488 		break;
489 	}
490 
491 	return status;
492 }
493 
494 extern enum mod_hdcp_status mod_hdcp_hdcp1_dp_execution(struct mod_hdcp *hdcp,
495 		struct mod_hdcp_event_context *event_ctx,
496 		struct mod_hdcp_transition_input_hdcp1 *input)
497 {
498 	enum mod_hdcp_status status = MOD_HDCP_STATUS_SUCCESS;
499 
500 	switch (current_state(hdcp)) {
501 	case D1_A0_DETERMINE_RX_HDCP_CAPABLE:
502 		status = determine_rx_hdcp_capable_dp(hdcp, event_ctx, input);
503 		break;
504 	case D1_A1_EXCHANGE_KSVS:
505 		status = exchange_ksvs(hdcp, event_ctx, input);
506 		break;
507 	case D1_A23_WAIT_FOR_R0_PRIME:
508 		status = wait_for_r0_prime_dp(hdcp, event_ctx, input);
509 		break;
510 	case D1_A2_COMPUTATIONS_A3_VALIDATE_RX_A5_TEST_FOR_REPEATER:
511 		status = computations_validate_rx_test_for_repeater(
512 				hdcp, event_ctx, input);
513 		break;
514 	case D1_A4_AUTHENTICATED:
515 		status = authenticated_dp(hdcp, event_ctx, input);
516 		break;
517 	case D1_A6_WAIT_FOR_READY:
518 		status = wait_for_ready(hdcp, event_ctx, input);
519 		break;
520 	case D1_A7_READ_KSV_LIST:
521 		status = read_ksv_list(hdcp, event_ctx, input);
522 		break;
523 	default:
524 		status = MOD_HDCP_STATUS_INVALID_STATE;
525 		break;
526 	}
527 
528 	return status;
529 }
530