/* * Copyright 2022 Advanced Micro Devices, Inc. * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR * OTHER DEALINGS IN THE SOFTWARE. * */ #include "amdgpu.h"#include "amdgpu_psp_ta.h" #if defined(CONFIG_DEBUG_FS) static ssize_t ta_if_load_debugfs_write(struct file *fp, const char *buf,					    size_t len, loff_t *off);static ssize_t ta_if_unload_debugfs_write(struct file *fp, const char *buf,					    size_t len, loff_t *off);static ssize_t ta_if_invoke_debugfs_write(struct file *fp, const char *buf,					    size_t len, loff_t *off); static uint32_t get_bin_version(const uint8_t *bin){	const struct common_firmware_header *hdr =			     (const struct common_firmware_header *)bin; 	return hdr->ucode_version;} static int prep_ta_mem_context(struct ta_mem_context *mem_context,					     uint8_t *shared_buf,					     uint32_t shared_buf_len){	if (mem_context->shared_mem_size < shared_buf_len)		return -EINVAL;	memset(mem_context->shared_buf, 0, mem_context->shared_mem_size);	memcpy((void *)mem_context->shared_buf, shared_buf, shared_buf_len); 	return 0;} static bool is_ta_type_valid(enum ta_type_id ta_type){	switch (ta_type) {	case TA_TYPE_RAS:		return true;	default:		return false;	}} static const struct ta_funcs ras_ta_funcs = {	.fn_ta_initialize = psp_ras_initialize,	.fn_ta_invoke    = psp_ras_invoke,	.fn_ta_terminate = psp_ras_terminate}; static void set_ta_context_funcs(struct psp_context *psp,						      enum ta_type_id ta_type,						      struct ta_context **pcontext){	switch (ta_type) {	case TA_TYPE_RAS:		*pcontext = &psp->ras_context.context;		psp->ta_funcs = &ras_ta_funcs;		break;	default:		break;	}} static const struct file_operations ta_load_debugfs_fops = {	.write  = ta_if_load_debugfs_write,	.llseek = default_llseek,	.owner  = THIS_MODULE}; static const struct file_operations ta_unload_debugfs_fops = {	.write  = ta_if_unload_debugfs_write,	.llseek = default_llseek,	.owner  = THIS_MODULE}; static const struct file_operations ta_invoke_debugfs_fops = {	.write  = ta_if_invoke_debugfs_write,	.llseek = default_llseek,	.owner  = THIS_MODULE}; /* * DOC: AMDGPU TA debugfs interfaces * * Three debugfs interfaces can be opened by a program to * load/invoke/unload TA, * * - /sys/kernel/debug/dri/<N>/ta_if/ta_load * - /sys/kernel/debug/dri/<N>/ta_if/ta_invoke * - /sys/kernel/debug/dri/<N>/ta_if/ta_unload * * How to use the interfaces in a program? * * A program needs to provide transmit buffer to the interfaces * and will receive buffer from the interfaces below, * * - For TA load debugfs interface: *   Transmit buffer: *    - TA type (4bytes) *    - TA bin length (4bytes) *    - TA bin *   Receive buffer: *    - TA ID (4bytes) * * - For TA invoke debugfs interface: *   Transmit buffer: *    - TA type (4bytes) *    - TA ID (4bytes) *    - TA CMD ID (4bytes) *    - TA shard buf length *      (4bytes, value not beyond TA shared memory size) *    - TA shared buf *   Receive buffer: *    - TA shared buf * * - For TA unload debugfs interface: *   Transmit buffer: *    - TA type (4bytes) *    - TA ID (4bytes) */ static ssize_t ta_if_load_debugfs_write(struct file *fp, const char *buf, size_t len, loff_t *off){	uint32_t ta_type    = 0;	uint32_t ta_bin_len = 0;	uint8_t  *ta_bin    = NULL;	uint32_t copy_pos   = 0;	int      ret        = 0; 	struct amdgpu_device *adev    = (struct amdgpu_device *)file_inode(fp)->i_private;	struct psp_context   *psp     = &adev->psp;	struct ta_context    *context = NULL; 	if (!buf)		return -EINVAL; 	ret = copy_from_user((void *)&ta_type, &buf[copy_pos], sizeof(uint32_t));	if (ret || (!is_ta_type_valid(ta_type)))		return -EFAULT; 	copy_pos += sizeof(uint32_t); 	ret = copy_from_user((void *)&ta_bin_len, &buf[copy_pos], sizeof(uint32_t));	if (ret)		return -EFAULT; 	if (ta_bin_len < sizeof(struct common_firmware_header) ||	    ta_bin_len > PSP_1_MEG)		return -EINVAL; 	copy_pos += sizeof(uint32_t); 	ta_bin = memdup_user(&buf[copy_pos], ta_bin_len);	if (IS_ERR(ta_bin))		return PTR_ERR(ta_bin); 	/* Set TA context and functions */	set_ta_context_funcs(psp, ta_type, &context); 	if (!psp->ta_funcs || !psp->ta_funcs->fn_ta_terminate) {		dev_err(adev->dev, "Unsupported function to terminate TA\n");		ret = -EOPNOTSUPP;		goto err_free_bin;	} 	/*	 * Allocate TA shared buf in case shared buf was freed	 * due to loading TA failed before.	 */	if (!context->mem_context.shared_buf) {		ret = psp_ta_init_shared_buf(psp, &context->mem_context);		if (ret) {			ret = -ENOMEM;			goto err_free_bin;		}	} 	ret = psp_fn_ta_terminate(psp);	if (ret || context->resp_status) {		dev_err(adev->dev,			"Failed to unload embedded TA (%d) and status (0x%X)\n",			ret, context->resp_status);		if (!ret)			ret = -EINVAL;		goto err_free_ta_shared_buf;	} 	/* Prepare TA context for TA initialization */	context->ta_type                     = ta_type;	context->bin_desc.fw_version         = get_bin_version(ta_bin);	context->bin_desc.size_bytes         = ta_bin_len;	context->bin_desc.start_addr         = ta_bin; 	if (!psp->ta_funcs->fn_ta_initialize) {		dev_err(adev->dev, "Unsupported function to initialize TA\n");		ret = -EOPNOTSUPP;		goto err_free_ta_shared_buf;	} 	ret = psp_fn_ta_initialize(psp);	if (ret || context->resp_status) {		dev_err(adev->dev, "Failed to load TA via debugfs (%d) and status (0x%X)\n",			ret, context->resp_status);		if (!ret)			ret = -EINVAL;		goto err_free_ta_shared_buf;	} 	if (copy_to_user((char *)buf, (void *)&context->session_id, sizeof(uint32_t)))		ret = -EFAULT; err_free_ta_shared_buf:	/* Only free TA shared buf when returns error code */	if (ret && context->mem_context.shared_buf)		psp_ta_free_shared_buf(&context->mem_context);err_free_bin:	kfree(ta_bin); 	return ret;} static ssize_t ta_if_unload_debugfs_write(struct file *fp, const char *buf, size_t len, loff_t *off){	uint32_t ta_type    = 0;	uint32_t ta_id      = 0;	uint32_t copy_pos   = 0;	int      ret        = 0; 	struct amdgpu_device *adev    = (struct amdgpu_device *)file_inode(fp)->i_private;	struct psp_context   *psp     = &adev->psp;	struct ta_context    *context = NULL; 	if (!buf)		return -EINVAL; 	ret = copy_from_user((void *)&ta_type, &buf[copy_pos], sizeof(uint32_t));	if (ret || (!is_ta_type_valid(ta_type)))		return -EFAULT; 	copy_pos += sizeof(uint32_t); 	ret = copy_from_user((void *)&ta_id, &buf[copy_pos], sizeof(uint32_t));	if (ret)		return -EFAULT; 	set_ta_context_funcs(psp, ta_type, &context);	context->session_id = ta_id; 	if (!psp->ta_funcs || !psp->ta_funcs->fn_ta_terminate) {		dev_err(adev->dev, "Unsupported function to terminate TA\n");		return -EOPNOTSUPP;	} 	ret = psp_fn_ta_terminate(psp);	if (ret || context->resp_status) {		dev_err(adev->dev, "Failed to unload TA via debugfs (%d) and status (0x%X)\n",			ret, context->resp_status);		if (!ret)			ret = -EINVAL;	} 	if (context->mem_context.shared_buf)		psp_ta_free_shared_buf(&context->mem_context); 	return ret;} static ssize_t ta_if_invoke_debugfs_write(struct file *fp, const char *buf, size_t len, loff_t *off){	uint32_t ta_type        = 0;	uint32_t ta_id          = 0;	uint32_t cmd_id         = 0;	uint32_t shared_buf_len = 0;	uint8_t *shared_buf     = NULL;	uint32_t copy_pos       = 0;	int      ret            = 0; 	struct amdgpu_device *adev    = (struct amdgpu_device *)file_inode(fp)->i_private;	struct psp_context   *psp     = &adev->psp;	struct ta_context    *context = NULL; 	if (!buf)		return -EINVAL; 	ret = copy_from_user((void *)&ta_type, &buf[copy_pos], sizeof(uint32_t));	if (ret)		return -EFAULT;	copy_pos += sizeof(uint32_t); 	ret = copy_from_user((void *)&ta_id, &buf[copy_pos], sizeof(uint32_t));	if (ret)		return -EFAULT;	copy_pos += sizeof(uint32_t); 	ret = copy_from_user((void *)&cmd_id, &buf[copy_pos], sizeof(uint32_t));	if (ret)		return -EFAULT;	copy_pos += sizeof(uint32_t); 	ret = copy_from_user((void *)&shared_buf_len, &buf[copy_pos], sizeof(uint32_t));	if (ret)		return -EFAULT;	if (!shared_buf_len || shared_buf_len > PSP_1_MEG)		return -EINVAL;	copy_pos += sizeof(uint32_t); 	shared_buf = memdup_user(&buf[copy_pos], shared_buf_len);	if (IS_ERR(shared_buf))		return PTR_ERR(shared_buf); 	set_ta_context_funcs(psp, ta_type, &context); 	if (!context || !context->initialized) {		dev_err(adev->dev, "TA is not initialized\n");		ret = -EINVAL;		goto free_shared_buf;	} 	if (!psp->ta_funcs || !psp->ta_funcs->fn_ta_invoke) {		dev_err(adev->dev, "Unsupported function to invoke TA\n");		ret = -EOPNOTSUPP;		goto free_shared_buf;	} 	context->session_id = ta_id; 	mutex_lock(&psp->ras_context.mutex);	ret = prep_ta_mem_context(&context->mem_context, shared_buf, shared_buf_len);	if (ret)		goto unlock; 	ret = psp_fn_ta_invoke(psp, cmd_id);	if (ret || context->resp_status) {		dev_err(adev->dev, "Failed to invoke TA via debugfs (%d) and status (0x%X)\n",			ret, context->resp_status);		if (!ret) {			ret = -EINVAL;			goto unlock;		}	} 	if (copy_to_user((char *)&buf[copy_pos], context->mem_context.shared_buf, shared_buf_len))		ret = -EFAULT; unlock:	mutex_unlock(&psp->ras_context.mutex); free_shared_buf:	kfree(shared_buf); 	return ret;} void amdgpu_ta_if_debugfs_init(struct amdgpu_device *adev){	struct drm_minor *minor = adev_to_drm(adev)->primary; 	struct dentry *dir = debugfs_create_dir("ta_if", minor->debugfs_root); 	debugfs_create_file("ta_load", 0200, dir, adev,				     &ta_load_debugfs_fops); 	debugfs_create_file("ta_unload", 0200, dir,				     adev, &ta_unload_debugfs_fops); 	debugfs_create_file("ta_invoke", 0200, dir,				     adev, &ta_invoke_debugfs_fops);} #elsevoid amdgpu_ta_if_debugfs_init(struct amdgpu_device *adev){ }#endif