LF-12610: drivers: se_fw: fix signed message support

signed message ioctl() is not available due to which
test will fail on i.MX8X SECO that have signed message
 not null.

Signed-off-by: Vabhav Sharma <vabhav.sharma@nxp.com>
Reviewed-by: Pankaj Gupta <pankaj.gupta@nxp.com>
Reviewed-by: Meenakshi Aggarwal <meenakshi.aggarwal@nxp.com>
Acked-by: Jason Liu <jason.hui.liu@nxp.com>

Author: vabhavnxp

drivers/firmware/imx/se_fw.c

@@ -48,6 +48,7 @@ static uint32_t v2x_fw_state;
 #define SOC_ID_MASK			0x0000FFFF
 #define RESERVED_DMA_POOL		BIT(1)
 #define SCU_MEM_CFG			BIT(2)
+#define SCU_SIGNED_MSG_CFG		BIT(3)
 #define IMX_ELE_FW_DIR                 "/lib/firmware/imx/ele/"
 #define SECURE_RAM_BASE_ADDRESS_SCU	(0x20800000u)
 #define V2X_NON_FIPS			0x00000c00
@@ -1228,6 +1229,36 @@ static int ele_mu_ioctl_get_mu_info(struct ele_mu_device_ctx *dev_ctx,
 	return err;
 }
 
+static int ele_mu_ioctl_signed_msg_handler(struct file *fp,
+					   struct ele_mu_device_ctx *dev_ctx,
+					   unsigned long arg)
+{
+	struct ele_mu_ioctl_signed_message msg;
+	int err;
+
+	err = copy_from_user(&msg, (u8 *)arg, sizeof(msg));
+	if (err) {
+		dev_err(dev_ctx->priv->dev, "Failed to copy from user: %d\n", err);
+		return -EFAULT;
+	}
+
+	err = imx_scu_signed_msg(fp, msg.message, msg.msg_size, &msg.error_code);
+	if (err) {
+		dev_err(dev_ctx->priv->dev,
+			"Failed to send signed message: %d\n",
+			err);
+		return err;
+	}
+
+	err = copy_to_user((u8 *)arg, &msg, sizeof(msg));
+	if (err) {
+		dev_err(dev_ctx->priv->dev, "Failed to copy to user: %d\n", err);
+		return -EFAULT;
+	}
+
+	return err;
+}
+
 /*
  * Copy a buffer of data to/from the user and return the address to use in
  * messages
@@ -1645,6 +1676,10 @@ static long ele_mu_ioctl(struct file *fp, unsigned int cmd, unsigned long arg)
 	case ELE_MU_IOCTL_GET_TIMER:
 		err = ele_mu_ioctl_get_time(dev_ctx, arg);
 		break;
+	case ELE_MU_IOCTL_SIGNED_MESSAGE:
+		if (ele_mu_priv->flags & SCU_SIGNED_MSG_CFG)
+			err = ele_mu_ioctl_signed_msg_handler(fp, dev_ctx, arg);
+		break;
 	default:
 		err = -EINVAL;
 		dev_dbg(ele_mu_priv->dev,
@@ -1919,6 +1954,7 @@ static int se_fw_probe(struct platform_device *pdev)
 	struct imx_info *info = NULL;
 	int ret;
 	struct device_node *np;
+	struct imx_info_list *info_list;
 
 	info = get_imx_info((struct imx_info_list *)of_id->data,
 			    pdev->name, strlen(pdev->name) + 1);
@@ -1966,15 +2002,27 @@ static int se_fw_probe(struct platform_device *pdev)
 		goto exit;
 	}
 
+	info_list = (struct imx_info_list *)of_id->data;
+
 	if (info->pre_if_config) {
 		/* start initializing ele fw */
 		ret = info->pre_if_config(dev);
 		if (ret) {
 			dev_err(dev, "Failed to initialize scu config.\n");
 			priv->flags &= (~SCU_MEM_CFG);
+			if ((info_list->soc_id == SOC_ID_OF_IMX8DXL) &&
+			    ((memcmp(info->se_name, "she1", 5)) ||
+			    (memcmp(info->se_name, "hsm1", 5)))) {
+				priv->flags &= (~SCU_SIGNED_MSG_CFG);
+			}
 			goto exit;
 		}
 		priv->flags |= SCU_MEM_CFG;
+		if ((info_list->soc_id == SOC_ID_OF_IMX8DXL) &&
+		    ((memcmp(info->se_name, "she1", 5)) ||
+		    (memcmp(info->se_name, "hsm1", 5)))) {
+			priv->flags |= SCU_SIGNED_MSG_CFG;
+		}
 	}
 
 	/* Initialize the mutex. */

drivers/firmware/imx/seco.c

@@ -344,3 +344,51 @@ int imx_scu_mem_access(struct file *fp)
 	return ret;
 }
 EXPORT_SYMBOL(imx_scu_mem_access);
+
+int imx_scu_signed_msg(struct file *fp,
+		       uint8_t *msg,
+		       uint32_t size,
+		       uint32_t *error)
+{
+	struct ele_mu_device_ctx *dev_ctx
+		= container_of(fp->private_data,
+			       struct ele_mu_device_ctx,
+			       miscdev);
+	struct ele_mu_priv *priv = dev_ctx->priv;
+	struct ele_shared_mem *shared_mem = &dev_ctx->non_secure_mem;
+	int err;
+	u64 addr;
+	u32 pos;
+
+	/* Check there is enough space in the shared memory. */
+	if (size >= shared_mem->size - shared_mem->pos) {
+		dev_err(dev_ctx->priv->dev,
+			"Not enough mem: %d left, %d required\n",
+			shared_mem->size - shared_mem->pos, size);
+		return -ENOMEM;
+	}
+
+	/* Allocate space in shared memory. 8 bytes aligned. */
+	pos = shared_mem->pos;
+
+	/* get physical address from the pos */
+	addr = (u64)shared_mem->dma_addr + pos;
+
+	/* copy signed message from user space to this allocated buffer */
+	err = copy_from_user(shared_mem->ptr + pos, msg, size);
+	if (err) {
+		dev_err(dev_ctx->priv->dev,
+			"Failed to signed message from user: %d\n",
+			err);
+		return -EFAULT;
+	}
+
+	*error = imx_sc_seco_sab_msg(priv->ipc_scu, addr);
+	err = *error;
+	if (err) {
+		dev_err(dev_ctx->priv->dev, "Failt to send signed message\n");
+	}
+
+	return err;
+}
+EXPORT_SYMBOL(imx_scu_signed_msg);

include/linux/firmware/imx/ele_mu_ioctl.h

@@ -91,6 +91,13 @@ struct ele_time_frame {
  */
 #define ELE_MU_IOCTL_GET_MU_INFO	_IOR(ELE_MU_IOCTL, 0x04, \
 					struct ele_mu_ioctl_get_mu_info)
+
+/*
+ * ioctl to send signed message to SE.
+ */
+#define ELE_MU_IOCTL_SIGNED_MESSAGE     _IOR(ELE_MU_IOCTL, 0x05, \
+					struct ele_mu_ioctl_signed_message)
+
 /*
  * ioctl to get SoC Info from user-space.
  */

include/linux/firmware/imx/svc/seco.h

@@ -45,6 +45,7 @@ int imx_sc_seco_secvio_dgo_config(struct imx_sc_ipc *ipc, u8 id, u8 access,
 int imx_scu_init_fw(struct device *dev);
 int imx_scu_sec_mem_cfg(struct file *fp, uint32_t offset, uint32_t size);
 int imx_scu_mem_access(struct file *fp);
+int imx_scu_signed_msg(struct file *fp, uint8_t *msg, uint32_t size, uint32_t *error);
 #else /* IS_ENABLED(CONFIG_IMX_SCU) */
 static inline
 int imx_sc_seco_build_info(struct imx_sc_ipc *ipc, uint32_t *version,
@@ -92,6 +93,12 @@ int imx_scu_sec_mem_cfg(struct file *fp, uint32_t offset, uint32_t size)
 	return -EOPNOTSUPP;
 }
 
+static inline
+int imx_scu_signed_msg(struct file *fp, uint8_t *msg, uint32_t size, uint32_t *error)
+{
+	return -EOPNOTSUPP;
+}
+
 static inline
 int imx_scu_mem_access(struct file *fp)//device *dev)
 {