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 "amdgpu_dm_hdcp.h"
27 #include "amdgpu.h"
28 #include "amdgpu_dm.h"
29 #include "dm_helpers.h"
30 #include <drm/display/drm_hdcp_helper.h>
31 #include "hdcp_psp.h"
32 
33 /*
34  * If the SRM version being loaded is less than or equal to the
35  * currently loaded SRM, psp will return 0xFFFF as the version
36  */
37 #define PSP_SRM_VERSION_MAX 0xFFFF
38 
39 static bool
40 lp_write_i2c(void *handle, uint32_t address, const uint8_t *data, uint32_t size)
41 {
42 
43 	struct dc_link *link = handle;
44 	struct i2c_payload i2c_payloads[] = {{true, address, size, (void *)data} };
45 	struct i2c_command cmd = {i2c_payloads, 1, I2C_COMMAND_ENGINE_HW, link->dc->caps.i2c_speed_in_khz};
46 
47 	return dm_helpers_submit_i2c(link->ctx, link, &cmd);
48 }
49 
50 static bool
51 lp_read_i2c(void *handle, uint32_t address, uint8_t offset, uint8_t *data, uint32_t size)
52 {
53 	struct dc_link *link = handle;
54 
55 	struct i2c_payload i2c_payloads[] = {{true, address, 1, &offset}, {false, address, size, data} };
56 	struct i2c_command cmd = {i2c_payloads, 2, I2C_COMMAND_ENGINE_HW, link->dc->caps.i2c_speed_in_khz};
57 
58 	return dm_helpers_submit_i2c(link->ctx, link, &cmd);
59 }
60 
61 static bool
62 lp_write_dpcd(void *handle, uint32_t address, const uint8_t *data, uint32_t size)
63 {
64 	struct dc_link *link = handle;
65 
66 	return dm_helpers_dp_write_dpcd(link->ctx, link, address, data, size);
67 }
68 
69 static bool
70 lp_read_dpcd(void *handle, uint32_t address, uint8_t *data, uint32_t size)
71 {
72 	struct dc_link *link = handle;
73 
74 	return dm_helpers_dp_read_dpcd(link->ctx, link, address, data, size);
75 }
76 
77 static uint8_t *psp_get_srm(struct psp_context *psp, uint32_t *srm_version, uint32_t *srm_size)
78 {
79 
80 	struct ta_hdcp_shared_memory *hdcp_cmd;
81 
82 	if (!psp->hdcp_context.context.initialized) {
83 		DRM_WARN("Failed to get hdcp srm. HDCP TA is not initialized.");
84 		return NULL;
85 	}
86 
87 	hdcp_cmd = (struct ta_hdcp_shared_memory *)psp->hdcp_context.context.mem_context.shared_buf;
88 	memset(hdcp_cmd, 0, sizeof(struct ta_hdcp_shared_memory));
89 
90 	hdcp_cmd->cmd_id = TA_HDCP_COMMAND__HDCP_GET_SRM;
91 	psp_hdcp_invoke(psp, hdcp_cmd->cmd_id);
92 
93 	if (hdcp_cmd->hdcp_status != TA_HDCP_STATUS__SUCCESS)
94 		return NULL;
95 
96 	*srm_version = hdcp_cmd->out_msg.hdcp_get_srm.srm_version;
97 	*srm_size = hdcp_cmd->out_msg.hdcp_get_srm.srm_buf_size;
98 
99 
100 	return hdcp_cmd->out_msg.hdcp_get_srm.srm_buf;
101 }
102 
103 static int psp_set_srm(struct psp_context *psp, uint8_t *srm, uint32_t srm_size, uint32_t *srm_version)
104 {
105 
106 	struct ta_hdcp_shared_memory *hdcp_cmd;
107 
108 	if (!psp->hdcp_context.context.initialized) {
109 		DRM_WARN("Failed to get hdcp srm. HDCP TA is not initialized.");
110 		return -EINVAL;
111 	}
112 
113 	hdcp_cmd = (struct ta_hdcp_shared_memory *)psp->hdcp_context.context.mem_context.shared_buf;
114 	memset(hdcp_cmd, 0, sizeof(struct ta_hdcp_shared_memory));
115 
116 	memcpy(hdcp_cmd->in_msg.hdcp_set_srm.srm_buf, srm, srm_size);
117 	hdcp_cmd->in_msg.hdcp_set_srm.srm_buf_size = srm_size;
118 	hdcp_cmd->cmd_id = TA_HDCP_COMMAND__HDCP_SET_SRM;
119 
120 	psp_hdcp_invoke(psp, hdcp_cmd->cmd_id);
121 
122 	if (hdcp_cmd->hdcp_status != TA_HDCP_STATUS__SUCCESS || hdcp_cmd->out_msg.hdcp_set_srm.valid_signature != 1 ||
123 	    hdcp_cmd->out_msg.hdcp_set_srm.srm_version == PSP_SRM_VERSION_MAX)
124 		return -EINVAL;
125 
126 	*srm_version = hdcp_cmd->out_msg.hdcp_set_srm.srm_version;
127 	return 0;
128 }
129 
130 static void process_output(struct hdcp_workqueue *hdcp_work)
131 {
132 	struct mod_hdcp_output output = hdcp_work->output;
133 
134 	if (output.callback_stop)
135 		cancel_delayed_work(&hdcp_work->callback_dwork);
136 
137 	if (output.callback_needed)
138 		schedule_delayed_work(&hdcp_work->callback_dwork,
139 				      msecs_to_jiffies(output.callback_delay));
140 
141 	if (output.watchdog_timer_stop)
142 		cancel_delayed_work(&hdcp_work->watchdog_timer_dwork);
143 
144 	if (output.watchdog_timer_needed)
145 		schedule_delayed_work(&hdcp_work->watchdog_timer_dwork,
146 				      msecs_to_jiffies(output.watchdog_timer_delay));
147 
148 	schedule_delayed_work(&hdcp_work->property_validate_dwork, msecs_to_jiffies(0));
149 }
150 
151 static void link_lock(struct hdcp_workqueue *work, bool lock)
152 {
153 
154 	int i = 0;
155 
156 	for (i = 0; i < work->max_link; i++) {
157 		if (lock)
158 			mutex_lock(&work[i].mutex);
159 		else
160 			mutex_unlock(&work[i].mutex);
161 	}
162 }
163 void hdcp_update_display(struct hdcp_workqueue *hdcp_work,
164 			 unsigned int link_index,
165 			 struct amdgpu_dm_connector *aconnector,
166 			 uint8_t content_type,
167 			 bool enable_encryption)
168 {
169 	struct hdcp_workqueue *hdcp_w = &hdcp_work[link_index];
170 	struct mod_hdcp_display *display = &hdcp_work[link_index].display;
171 	struct mod_hdcp_link *link = &hdcp_work[link_index].link;
172 	struct mod_hdcp_display_query query;
173 	unsigned int conn_index = aconnector->base.index;
174 
175 	mutex_lock(&hdcp_w->mutex);
176 	hdcp_w->aconnector[conn_index] = aconnector;
177 
178 	query.display = NULL;
179 	mod_hdcp_query_display(&hdcp_w->hdcp, aconnector->base.index, &query);
180 
181 	if (query.display != NULL) {
182 		memcpy(display, query.display, sizeof(struct mod_hdcp_display));
183 		mod_hdcp_remove_display(&hdcp_w->hdcp, aconnector->base.index, &hdcp_w->output);
184 
185 		hdcp_w->link.adjust.hdcp2.force_type = MOD_HDCP_FORCE_TYPE_0;
186 
187 		if (enable_encryption) {
188 			/* Explicitly set the saved SRM as sysfs call will be after we already enabled hdcp
189 			 * (s3 resume case)
190 			 */
191 			if (hdcp_work->srm_size > 0)
192 				psp_set_srm(hdcp_work->hdcp.config.psp.handle, hdcp_work->srm, hdcp_work->srm_size,
193 					    &hdcp_work->srm_version);
194 
195 			display->adjust.disable = MOD_HDCP_DISPLAY_NOT_DISABLE;
196 			if (content_type == DRM_MODE_HDCP_CONTENT_TYPE0) {
197 				hdcp_w->link.adjust.hdcp1.disable = 0;
198 				hdcp_w->link.adjust.hdcp2.force_type = MOD_HDCP_FORCE_TYPE_0;
199 			} else if (content_type == DRM_MODE_HDCP_CONTENT_TYPE1) {
200 				hdcp_w->link.adjust.hdcp1.disable = 1;
201 				hdcp_w->link.adjust.hdcp2.force_type = MOD_HDCP_FORCE_TYPE_1;
202 			}
203 
204 			schedule_delayed_work(&hdcp_w->property_validate_dwork,
205 					      msecs_to_jiffies(DRM_HDCP_CHECK_PERIOD_MS));
206 		} else {
207 			display->adjust.disable = MOD_HDCP_DISPLAY_DISABLE_AUTHENTICATION;
208 			hdcp_w->encryption_status[conn_index] = MOD_HDCP_ENCRYPTION_STATUS_HDCP_OFF;
209 			cancel_delayed_work(&hdcp_w->property_validate_dwork);
210 		}
211 
212 		display->state = MOD_HDCP_DISPLAY_ACTIVE;
213 	}
214 
215 	mod_hdcp_add_display(&hdcp_w->hdcp, link, display, &hdcp_w->output);
216 
217 	process_output(hdcp_w);
218 	mutex_unlock(&hdcp_w->mutex);
219 }
220 
221 static void hdcp_remove_display(struct hdcp_workqueue *hdcp_work,
222 			 unsigned int link_index,
223 			 struct amdgpu_dm_connector *aconnector)
224 {
225 	struct hdcp_workqueue *hdcp_w = &hdcp_work[link_index];
226 	struct drm_connector_state *conn_state = aconnector->base.state;
227 	unsigned int conn_index = aconnector->base.index;
228 
229 	mutex_lock(&hdcp_w->mutex);
230 	hdcp_w->aconnector[conn_index] = aconnector;
231 
232 	/* the removal of display will invoke auth reset -> hdcp destroy and
233 	 * we'd expect the Content Protection (CP) property changed back to
234 	 * DESIRED if at the time ENABLED. CP property change should occur
235 	 * before the element removed from linked list.
236 	 */
237 	if (conn_state && conn_state->content_protection == DRM_MODE_CONTENT_PROTECTION_ENABLED) {
238 		conn_state->content_protection = DRM_MODE_CONTENT_PROTECTION_DESIRED;
239 
240 		DRM_DEBUG_DRIVER("[HDCP_DM] display %d, CP 2 -> 1, type %u, DPMS %u\n",
241 			 aconnector->base.index, conn_state->hdcp_content_type, aconnector->base.dpms);
242 	}
243 
244 	mod_hdcp_remove_display(&hdcp_w->hdcp, aconnector->base.index, &hdcp_w->output);
245 
246 	process_output(hdcp_w);
247 	mutex_unlock(&hdcp_w->mutex);
248 }
249 void hdcp_reset_display(struct hdcp_workqueue *hdcp_work, unsigned int link_index)
250 {
251 	struct hdcp_workqueue *hdcp_w = &hdcp_work[link_index];
252 	unsigned int conn_index;
253 
254 	mutex_lock(&hdcp_w->mutex);
255 
256 	mod_hdcp_reset_connection(&hdcp_w->hdcp,  &hdcp_w->output);
257 
258 	cancel_delayed_work(&hdcp_w->property_validate_dwork);
259 
260 	for (conn_index = 0; conn_index < AMDGPU_DM_MAX_DISPLAY_INDEX; conn_index++) {
261 		hdcp_w->encryption_status[conn_index] =
262 			MOD_HDCP_ENCRYPTION_STATUS_HDCP_OFF;
263 	}
264 
265 	process_output(hdcp_w);
266 
267 	mutex_unlock(&hdcp_w->mutex);
268 }
269 
270 void hdcp_handle_cpirq(struct hdcp_workqueue *hdcp_work, unsigned int link_index)
271 {
272 	struct hdcp_workqueue *hdcp_w = &hdcp_work[link_index];
273 
274 	schedule_work(&hdcp_w->cpirq_work);
275 }
276 
277 
278 
279 
280 static void event_callback(struct work_struct *work)
281 {
282 	struct hdcp_workqueue *hdcp_work;
283 
284 	hdcp_work = container_of(to_delayed_work(work), struct hdcp_workqueue,
285 				      callback_dwork);
286 
287 	mutex_lock(&hdcp_work->mutex);
288 
289 	cancel_delayed_work(&hdcp_work->callback_dwork);
290 
291 	mod_hdcp_process_event(&hdcp_work->hdcp, MOD_HDCP_EVENT_CALLBACK,
292 			       &hdcp_work->output);
293 
294 	process_output(hdcp_work);
295 
296 	mutex_unlock(&hdcp_work->mutex);
297 
298 
299 }
300 
301 static void event_property_update(struct work_struct *work)
302 {
303 	struct hdcp_workqueue *hdcp_work = container_of(work, struct hdcp_workqueue, property_update_work);
304 	struct amdgpu_dm_connector *aconnector = NULL;
305 	struct drm_device *dev;
306 	long ret;
307 	unsigned int conn_index;
308 	struct drm_connector *connector;
309 	struct drm_connector_state *conn_state;
310 
311 	for (conn_index = 0; conn_index < AMDGPU_DM_MAX_DISPLAY_INDEX; conn_index++) {
312 		aconnector = hdcp_work->aconnector[conn_index];
313 
314 		if (!aconnector)
315 			continue;
316 
317 		connector = &aconnector->base;
318 
319 		/* check if display connected */
320 		if (connector->status != connector_status_connected)
321 			continue;
322 
323 		conn_state = aconnector->base.state;
324 
325 		if (!conn_state)
326 			continue;
327 
328 		dev = connector->dev;
329 
330 		if (!dev)
331 			continue;
332 
333 		drm_modeset_lock(&dev->mode_config.connection_mutex, NULL);
334 		mutex_lock(&hdcp_work->mutex);
335 
336 		if (conn_state->commit) {
337 			ret = wait_for_completion_interruptible_timeout(
338 				&conn_state->commit->hw_done, 10 * HZ);
339 			if (ret == 0) {
340 				DRM_ERROR(
341 					"HDCP state unknown! Setting it to DESIRED");
342 				hdcp_work->encryption_status[conn_index] =
343 					MOD_HDCP_ENCRYPTION_STATUS_HDCP_OFF;
344 			}
345 		}
346 		if (hdcp_work->encryption_status[conn_index] !=
347 			MOD_HDCP_ENCRYPTION_STATUS_HDCP_OFF) {
348 			if (conn_state->hdcp_content_type ==
349 				DRM_MODE_HDCP_CONTENT_TYPE0 &&
350 				hdcp_work->encryption_status[conn_index] <=
351 				MOD_HDCP_ENCRYPTION_STATUS_HDCP2_TYPE0_ON) {
352 
353 				DRM_DEBUG_DRIVER("[HDCP_DM] DRM_MODE_CONTENT_PROTECTION_ENABLED\n");
354 				drm_hdcp_update_content_protection(
355 					connector,
356 					DRM_MODE_CONTENT_PROTECTION_ENABLED);
357 			} else if (conn_state->hdcp_content_type ==
358 					DRM_MODE_HDCP_CONTENT_TYPE1 &&
359 					hdcp_work->encryption_status[conn_index] ==
360 					MOD_HDCP_ENCRYPTION_STATUS_HDCP2_TYPE1_ON) {
361 				drm_hdcp_update_content_protection(
362 					connector,
363 					DRM_MODE_CONTENT_PROTECTION_ENABLED);
364 			}
365 		} else {
366 			DRM_DEBUG_DRIVER("[HDCP_DM] DRM_MODE_CONTENT_PROTECTION_DESIRED\n");
367 			drm_hdcp_update_content_protection(
368 				connector, DRM_MODE_CONTENT_PROTECTION_DESIRED);
369 
370 		}
371 		mutex_unlock(&hdcp_work->mutex);
372 		drm_modeset_unlock(&dev->mode_config.connection_mutex);
373 	}
374 }
375 
376 static void event_property_validate(struct work_struct *work)
377 {
378 	struct hdcp_workqueue *hdcp_work =
379 		container_of(to_delayed_work(work), struct hdcp_workqueue, property_validate_dwork);
380 	struct mod_hdcp_display_query query;
381 	struct amdgpu_dm_connector *aconnector;
382 	unsigned int conn_index;
383 
384 	mutex_lock(&hdcp_work->mutex);
385 
386 	for (conn_index = 0; conn_index < AMDGPU_DM_MAX_DISPLAY_INDEX;
387 	     conn_index++) {
388 		aconnector = hdcp_work->aconnector[conn_index];
389 
390 		if (!aconnector)
391 			continue;
392 
393 		/* check if display connected */
394 		if (aconnector->base.status != connector_status_connected)
395 			continue;
396 
397 		if (!aconnector->base.state)
398 			continue;
399 
400 		query.encryption_status = MOD_HDCP_ENCRYPTION_STATUS_HDCP_OFF;
401 		mod_hdcp_query_display(&hdcp_work->hdcp, aconnector->base.index,
402 				       &query);
403 
404 		DRM_DEBUG_DRIVER("[HDCP_DM] disp %d, connector->CP %u, (query, work): (%d, %d)\n",
405 			aconnector->base.index,
406 			aconnector->base.state->content_protection,
407 			query.encryption_status,
408 			hdcp_work->encryption_status[conn_index]);
409 
410 		if (query.encryption_status !=
411 		    hdcp_work->encryption_status[conn_index]) {
412 			DRM_DEBUG_DRIVER("[HDCP_DM] encryption_status change from %x to %x\n",
413 				hdcp_work->encryption_status[conn_index], query.encryption_status);
414 
415 			hdcp_work->encryption_status[conn_index] =
416 				query.encryption_status;
417 
418 			DRM_DEBUG_DRIVER("[HDCP_DM] trigger property_update_work\n");
419 
420 			schedule_work(&hdcp_work->property_update_work);
421 		}
422 	}
423 
424 	mutex_unlock(&hdcp_work->mutex);
425 }
426 
427 static void event_watchdog_timer(struct work_struct *work)
428 {
429 	struct hdcp_workqueue *hdcp_work;
430 
431 	hdcp_work = container_of(to_delayed_work(work),
432 				      struct hdcp_workqueue,
433 				      watchdog_timer_dwork);
434 
435 	mutex_lock(&hdcp_work->mutex);
436 
437 	cancel_delayed_work(&hdcp_work->watchdog_timer_dwork);
438 
439 	mod_hdcp_process_event(&hdcp_work->hdcp,
440 			       MOD_HDCP_EVENT_WATCHDOG_TIMEOUT,
441 			       &hdcp_work->output);
442 
443 	process_output(hdcp_work);
444 
445 	mutex_unlock(&hdcp_work->mutex);
446 
447 }
448 
449 static void event_cpirq(struct work_struct *work)
450 {
451 	struct hdcp_workqueue *hdcp_work;
452 
453 	hdcp_work = container_of(work, struct hdcp_workqueue, cpirq_work);
454 
455 	mutex_lock(&hdcp_work->mutex);
456 
457 	mod_hdcp_process_event(&hdcp_work->hdcp, MOD_HDCP_EVENT_CPIRQ, &hdcp_work->output);
458 
459 	process_output(hdcp_work);
460 
461 	mutex_unlock(&hdcp_work->mutex);
462 
463 }
464 
465 
466 void hdcp_destroy(struct kobject *kobj, struct hdcp_workqueue *hdcp_work)
467 {
468 	int i = 0;
469 
470 	for (i = 0; i < hdcp_work->max_link; i++) {
471 		cancel_delayed_work_sync(&hdcp_work[i].callback_dwork);
472 		cancel_delayed_work_sync(&hdcp_work[i].watchdog_timer_dwork);
473 	}
474 
475 	sysfs_remove_bin_file(kobj, &hdcp_work[0].attr);
476 	kfree(hdcp_work->srm);
477 	kfree(hdcp_work->srm_temp);
478 	kfree(hdcp_work);
479 }
480 
481 
482 static bool enable_assr(void *handle, struct dc_link *link)
483 {
484 
485 	struct hdcp_workqueue *hdcp_work = handle;
486 	struct mod_hdcp hdcp = hdcp_work->hdcp;
487 	struct psp_context *psp = hdcp.config.psp.handle;
488 	struct ta_dtm_shared_memory *dtm_cmd;
489 	bool res = true;
490 
491 	if (!psp->dtm_context.context.initialized) {
492 		DRM_INFO("Failed to enable ASSR, DTM TA is not initialized.");
493 		return false;
494 	}
495 
496 	dtm_cmd = (struct ta_dtm_shared_memory *)psp->dtm_context.context.mem_context.shared_buf;
497 
498 	mutex_lock(&psp->dtm_context.mutex);
499 	memset(dtm_cmd, 0, sizeof(struct ta_dtm_shared_memory));
500 
501 	dtm_cmd->cmd_id = TA_DTM_COMMAND__TOPOLOGY_ASSR_ENABLE;
502 	dtm_cmd->dtm_in_message.topology_assr_enable.display_topology_dig_be_index = link->link_enc_hw_inst;
503 	dtm_cmd->dtm_status = TA_DTM_STATUS__GENERIC_FAILURE;
504 
505 	psp_dtm_invoke(psp, dtm_cmd->cmd_id);
506 
507 	if (dtm_cmd->dtm_status != TA_DTM_STATUS__SUCCESS) {
508 		DRM_INFO("Failed to enable ASSR");
509 		res = false;
510 	}
511 
512 	mutex_unlock(&psp->dtm_context.mutex);
513 
514 	return res;
515 }
516 
517 static void update_config(void *handle, struct cp_psp_stream_config *config)
518 {
519 	struct hdcp_workqueue *hdcp_work = handle;
520 	struct amdgpu_dm_connector *aconnector = config->dm_stream_ctx;
521 	int link_index = aconnector->dc_link->link_index;
522 	struct mod_hdcp_display *display = &hdcp_work[link_index].display;
523 	struct mod_hdcp_link *link = &hdcp_work[link_index].link;
524 	struct drm_connector_state *conn_state;
525 	struct dc_sink *sink = NULL;
526 	bool link_is_hdcp14 = false;
527 
528 	if (config->dpms_off) {
529 		hdcp_remove_display(hdcp_work, link_index, aconnector);
530 		return;
531 	}
532 
533 	memset(display, 0, sizeof(*display));
534 	memset(link, 0, sizeof(*link));
535 
536 	display->index = aconnector->base.index;
537 	display->state = MOD_HDCP_DISPLAY_ACTIVE;
538 
539 	if (aconnector->dc_sink)
540 		sink = aconnector->dc_sink;
541 	else if (aconnector->dc_em_sink)
542 		sink = aconnector->dc_em_sink;
543 
544 	if (sink != NULL)
545 		link->mode = mod_hdcp_signal_type_to_operation_mode(sink->sink_signal);
546 
547 	display->controller = CONTROLLER_ID_D0 + config->otg_inst;
548 	display->dig_fe = config->dig_fe;
549 	link->dig_be = config->dig_be;
550 	link->ddc_line = aconnector->dc_link->ddc_hw_inst + 1;
551 	display->stream_enc_idx = config->stream_enc_idx;
552 	link->link_enc_idx = config->link_enc_idx;
553 	link->dio_output_id = config->dio_output_idx;
554 	link->phy_idx = config->phy_idx;
555 
556 	if (sink)
557 		link_is_hdcp14 = dc_link_is_hdcp14(aconnector->dc_link, sink->sink_signal);
558 	link->hdcp_supported_informational = link_is_hdcp14;
559 	link->dp.rev = aconnector->dc_link->dpcd_caps.dpcd_rev.raw;
560 	link->dp.assr_enabled = config->assr_enabled;
561 	link->dp.mst_enabled = config->mst_enabled;
562 	link->dp.usb4_enabled = config->usb4_enabled;
563 	display->adjust.disable = MOD_HDCP_DISPLAY_DISABLE_AUTHENTICATION;
564 	link->adjust.auth_delay = 2;
565 	link->adjust.hdcp1.disable = 0;
566 	conn_state = aconnector->base.state;
567 
568 	DRM_DEBUG_DRIVER("[HDCP_DM] display %d, CP %d, type %d\n", aconnector->base.index,
569 			(!!aconnector->base.state) ? aconnector->base.state->content_protection : -1,
570 			(!!aconnector->base.state) ? aconnector->base.state->hdcp_content_type : -1);
571 
572 	if (conn_state)
573 		hdcp_update_display(hdcp_work, link_index, aconnector,
574 			conn_state->hdcp_content_type, false);
575 }
576 
577 
578 /* NOTE: From the usermodes prospective you only need to call write *ONCE*, the kernel
579  *      will automatically call once or twice depending on the size
580  *
581  * call: "cat file > /sys/class/drm/card0/device/hdcp_srm" from usermode no matter what the size is
582  *
583  * The kernel can only send PAGE_SIZE at once and since MAX_SRM_FILE(5120) > PAGE_SIZE(4096),
584  * srm_data_write can be called multiple times.
585  *
586  * sysfs interface doesn't tell us the size we will get so we are sending partial SRMs to psp and on
587  * the last call we will send the full SRM. PSP will fail on every call before the last.
588  *
589  * This means we don't know if the SRM is good until the last call. And because of this limitation we
590  * cannot throw errors early as it will stop the kernel from writing to sysfs
591  *
592  * Example 1:
593  * 	Good SRM size = 5096
594  * 	first call to write 4096 -> PSP fails
595  * 	Second call to write 1000 -> PSP Pass -> SRM is set
596  *
597  * Example 2:
598  * 	Bad SRM size = 4096
599  * 	first call to write 4096 -> PSP fails (This is the same as above, but we don't know if this
600  * 	is the last call)
601  *
602  * Solution?:
603  * 	1: Parse the SRM? -> It is signed so we don't know the EOF
604  * 	2: We can have another sysfs that passes the size before calling set. -> simpler solution
605  * 	below
606  *
607  * Easy Solution:
608  * Always call get after Set to verify if set was successful.
609  * +----------------------+
610  * |   Why it works:      |
611  * +----------------------+
612  * PSP will only update its srm if its older than the one we are trying to load.
613  * Always do set first than get.
614  * 	-if we try to "1. SET" a older version PSP will reject it and we can "2. GET" the newer
615  * 	version and save it
616  *
617  * 	-if we try to "1. SET" a newer version PSP will accept it and we can "2. GET" the
618  * 	same(newer) version back and save it
619  *
620  * 	-if we try to "1. SET" a newer version and PSP rejects it. That means the format is
621  * 	incorrect/corrupted and we should correct our SRM by getting it from PSP
622  */
623 static ssize_t srm_data_write(struct file *filp, struct kobject *kobj, struct bin_attribute *bin_attr, char *buffer,
624 			      loff_t pos, size_t count)
625 {
626 	struct hdcp_workqueue *work;
627 	uint32_t srm_version = 0;
628 
629 	work = container_of(bin_attr, struct hdcp_workqueue, attr);
630 	link_lock(work, true);
631 
632 	memcpy(work->srm_temp + pos, buffer, count);
633 
634 	if (!psp_set_srm(work->hdcp.config.psp.handle, work->srm_temp, pos + count, &srm_version)) {
635 		DRM_DEBUG_DRIVER("HDCP SRM SET version 0x%X", srm_version);
636 		memcpy(work->srm, work->srm_temp, pos + count);
637 		work->srm_size = pos + count;
638 		work->srm_version = srm_version;
639 	}
640 
641 
642 	link_lock(work, false);
643 
644 	return count;
645 }
646 
647 static ssize_t srm_data_read(struct file *filp, struct kobject *kobj, struct bin_attribute *bin_attr, char *buffer,
648 			     loff_t pos, size_t count)
649 {
650 	struct hdcp_workqueue *work;
651 	uint8_t *srm = NULL;
652 	uint32_t srm_version;
653 	uint32_t srm_size;
654 	size_t ret = count;
655 
656 	work = container_of(bin_attr, struct hdcp_workqueue, attr);
657 
658 	link_lock(work, true);
659 
660 	srm = psp_get_srm(work->hdcp.config.psp.handle, &srm_version, &srm_size);
661 
662 	if (!srm) {
663 		ret = -EINVAL;
664 		goto ret;
665 	}
666 
667 	if (pos >= srm_size)
668 		ret = 0;
669 
670 	if (srm_size - pos < count) {
671 		memcpy(buffer, srm + pos, srm_size - pos);
672 		ret = srm_size - pos;
673 		goto ret;
674 	}
675 
676 	memcpy(buffer, srm + pos, count);
677 
678 ret:
679 	link_lock(work, false);
680 	return ret;
681 }
682 
683 /* From the hdcp spec (5.Renewability) SRM needs to be stored in a non-volatile memory.
684  *
685  * For example,
686  * 	if Application "A" sets the SRM (ver 2) and we reboot/suspend and later when Application "B"
687  * 	needs to use HDCP, the version in PSP should be SRM(ver 2). So SRM should be persistent
688  * 	across boot/reboots/suspend/resume/shutdown
689  *
690  * Currently when the system goes down (suspend/shutdown) the SRM is cleared from PSP. For HDCP we need
691  * to make the SRM persistent.
692  *
693  * -PSP owns the checking of SRM but doesn't have the ability to store it in a non-volatile memory.
694  * -The kernel cannot write to the file systems.
695  * -So we need usermode to do this for us, which is why an interface for usermode is needed
696  *
697  *
698  *
699  * Usermode can read/write to/from PSP using the sysfs interface
700  * For example:
701  * 	to save SRM from PSP to storage : cat /sys/class/drm/card0/device/hdcp_srm > srmfile
702  * 	to load from storage to PSP: cat srmfile > /sys/class/drm/card0/device/hdcp_srm
703  */
704 static const struct bin_attribute data_attr = {
705 	.attr = {.name = "hdcp_srm", .mode = 0664},
706 	.size = PSP_HDCP_SRM_FIRST_GEN_MAX_SIZE, /* Limit SRM size */
707 	.write = srm_data_write,
708 	.read = srm_data_read,
709 };
710 
711 
712 struct hdcp_workqueue *hdcp_create_workqueue(struct amdgpu_device *adev, struct cp_psp *cp_psp, struct dc *dc)
713 {
714 
715 	int max_caps = dc->caps.max_links;
716 	struct hdcp_workqueue *hdcp_work;
717 	int i = 0;
718 
719 	hdcp_work = kcalloc(max_caps, sizeof(*hdcp_work), GFP_KERNEL);
720 	if (ZERO_OR_NULL_PTR(hdcp_work))
721 		return NULL;
722 
723 	hdcp_work->srm = kcalloc(PSP_HDCP_SRM_FIRST_GEN_MAX_SIZE, sizeof(*hdcp_work->srm), GFP_KERNEL);
724 
725 	if (hdcp_work->srm == NULL)
726 		goto fail_alloc_context;
727 
728 	hdcp_work->srm_temp = kcalloc(PSP_HDCP_SRM_FIRST_GEN_MAX_SIZE, sizeof(*hdcp_work->srm_temp), GFP_KERNEL);
729 
730 	if (hdcp_work->srm_temp == NULL)
731 		goto fail_alloc_context;
732 
733 	hdcp_work->max_link = max_caps;
734 
735 	for (i = 0; i < max_caps; i++) {
736 		mutex_init(&hdcp_work[i].mutex);
737 
738 		INIT_WORK(&hdcp_work[i].cpirq_work, event_cpirq);
739 		INIT_WORK(&hdcp_work[i].property_update_work, event_property_update);
740 		INIT_DELAYED_WORK(&hdcp_work[i].callback_dwork, event_callback);
741 		INIT_DELAYED_WORK(&hdcp_work[i].watchdog_timer_dwork, event_watchdog_timer);
742 		INIT_DELAYED_WORK(&hdcp_work[i].property_validate_dwork, event_property_validate);
743 
744 		hdcp_work[i].hdcp.config.psp.handle = &adev->psp;
745 		if (dc->ctx->dce_version == DCN_VERSION_3_1 ||
746 		    dc->ctx->dce_version == DCN_VERSION_3_14 ||
747 		    dc->ctx->dce_version == DCN_VERSION_3_15 ||
748 		    dc->ctx->dce_version == DCN_VERSION_3_16)
749 			hdcp_work[i].hdcp.config.psp.caps.dtm_v3_supported = 1;
750 		hdcp_work[i].hdcp.config.ddc.handle = dc_get_link_at_index(dc, i);
751 		hdcp_work[i].hdcp.config.ddc.funcs.write_i2c = lp_write_i2c;
752 		hdcp_work[i].hdcp.config.ddc.funcs.read_i2c = lp_read_i2c;
753 		hdcp_work[i].hdcp.config.ddc.funcs.write_dpcd = lp_write_dpcd;
754 		hdcp_work[i].hdcp.config.ddc.funcs.read_dpcd = lp_read_dpcd;
755 
756 		memset(hdcp_work[i].aconnector, 0,
757 		       sizeof(struct amdgpu_dm_connector *) *
758 			       AMDGPU_DM_MAX_DISPLAY_INDEX);
759 		memset(hdcp_work[i].encryption_status, 0,
760 		       sizeof(enum mod_hdcp_encryption_status) *
761 			       AMDGPU_DM_MAX_DISPLAY_INDEX);
762 	}
763 
764 	cp_psp->funcs.update_stream_config = update_config;
765 	cp_psp->funcs.enable_assr = enable_assr;
766 	cp_psp->handle = hdcp_work;
767 
768 	/* File created at /sys/class/drm/card0/device/hdcp_srm*/
769 	hdcp_work[0].attr = data_attr;
770 	sysfs_bin_attr_init(&hdcp_work[0].attr);
771 
772 	if (sysfs_create_bin_file(&adev->dev->kobj, &hdcp_work[0].attr))
773 		DRM_WARN("Failed to create device file hdcp_srm");
774 
775 	return hdcp_work;
776 
777 fail_alloc_context:
778 	kfree(hdcp_work->srm);
779 	kfree(hdcp_work->srm_temp);
780 	kfree(hdcp_work);
781 
782 	return NULL;
783 
784 
785 
786 }
787 
788 
789 
790