efuse: add mrk checksum commands in kernel [1/1]

PD#OTT-66795

Problem:
No efuse mrk commands in kernel

Solution:
add it

Verify:
s7d

Change-Id: I0393e26ce0cea637b40ec37d00825fd8d6f68798
Signed-off-by: Tao Zeng <tao.zeng@amlogic.com>

drivers/efuse_unifykey/efuse.c

@@ -34,6 +34,8 @@ static struct aml_efuse_lockable_check efusecheck = {
 };
 
 struct efuse_obj_field_t efuse_field;
+static char mrk_name[16] = {};
+static unsigned int mrk_checksum[6] = {-1, -1, -1, -1, -1, -1};
 
 struct aml_efuse_cmd efuse_cmd;
 unsigned int efuse_pattern_size;
@@ -1153,12 +1155,70 @@ static ssize_t efuse_obj_show(struct class *class,
 	return len;
 }
 
+static ssize_t efuse_mrk_store(struct class *cla,
+	struct class_attribute *attr, const char *buf, size_t count)
+{
+	int ret = 0;
+
+	mutex_lock(&efuse_obj_mutex);
+	memset(mrk_name, 0, sizeof(mrk_name));
+	memset(mrk_checksum, 0, sizeof(mrk_checksum));
+	strncpy(mrk_name, buf, count);
+	for (ret = 0; ret < sizeof(mrk_name); ret++)
+		if (mrk_name[ret] == '\n') {	/* remove enter key */
+			mrk_name[ret] = '\0';
+			break;
+		}
+
+	/* short checknum */
+	ret = efuse_mrk_get_checknum(mrk_name, 0, mrk_checksum);
+	if (ret)
+		goto err;
+
+	/* long checknum */
+	ret = efuse_mrk_get_checknum(mrk_name, 1, mrk_checksum + 1);
+err:
+	mrk_checksum[5] = ret;
+	mutex_unlock(&efuse_obj_mutex);
+	return count;
+}
+
+static ssize_t efuse_mrk_show(struct class *class,
+	struct class_attribute *attr, char *buf)
+{
+	int ret;
+	int count = 0;
+
+	mutex_lock(&efuse_obj_mutex);
+	ret = mrk_checksum[5];
+	switch (ret) {
+	case 0:
+		count = sprintf(buf, "%s Short checknum:%08x, long checknum:%08x %08x %08x %08x\n",
+				mrk_name, mrk_checksum[0], mrk_checksum[1],
+				mrk_checksum[2], mrk_checksum[3], mrk_checksum[4]);
+		break;
+	case EFUSE_MRK_CHECKNUM_NOT_SUPPORTED:
+		count = sprintf(buf, "MRK field %s not supported\n", mrk_name);
+		break;
+	case EFUSE_MRK_CHECKNUM_INVALID_ARGUMENT:
+		count = sprintf(buf, "get mrk checknum for %s failed, MRK field may not be written\n",
+				mrk_name);
+		break;
+	default:
+		count = -1;
+		break;
+	}
+	mutex_unlock(&efuse_obj_mutex);
+	return count;
+}
+
 static EFUSE_CLASS_ATTR(userdata);
 static EFUSE_CLASS_ATTR(mac);
 static EFUSE_CLASS_ATTR(mac_bt);
 static EFUSE_CLASS_ATTR(mac_wifi);
 static EFUSE_CLASS_ATTR(usid);
 static EFUSE_CLASS_ATTR(efuse_obj);
+static EFUSE_CLASS_ATTR(efuse_mrk);
 static CLASS_ATTR_WO(amlogic_set);
 static CLASS_ATTR_RO(secureboot_check);
 static EFUSE_CLASS_ATTR(checkburn);
@@ -1171,6 +1231,7 @@ static struct attribute *efuse_class_attrs[] = {
 	&class_attr_mac_wifi.attr,
 	&class_attr_usid.attr,
 	&class_attr_efuse_obj.attr,
+	&class_attr_efuse_mrk.attr,
 	&class_attr_amlogic_set.attr,
 	&class_attr_secureboot_check.attr,
 	&class_attr_checkburn.attr,

drivers/efuse_unifykey/efuse.h

@@ -42,6 +42,7 @@ struct aml_efuse_key {
 #define EFUSE_PATTERN_SIZE      (0x400)
 #define EFUSE_OBJ_READ          (0x8200003B)
 #define EFUSE_OBJ_WRITE         (0x8200003C)
+#define EFUSE_MRK_GET_CHECKNUM	(0x8200003F)
 
 enum efuse_obj_status_e {
 	EFUSE_OBJ_SUCCESS		= 0,
@@ -69,6 +70,12 @@ enum efuse_obj_info_e {
 	EFUSE_OBJ_EFUSE_ENC_DATA,
 };
 
+typedef enum efuse_mrk_checknum_status_s {
+	EFUSE_MRK_CHECKNUM_SUCCESS             = 0,
+	EFUSE_MRK_CHECKNUM_NOT_SUPPORTED       = 1,
+	EFUSE_MRK_CHECKNUM_INVALID_ARGUMENT    = 2,
+} efuse_mrk_checknum_status_e;
+
 struct efuse_obj_field_t {
 	char name[48];
 	unsigned char data[32];
@@ -130,6 +137,8 @@ uint32_t efuse_obj_set_license(char *name);
 uint32_t efuse_obj_lock(char *name);
 uint32_t efuse_obj_get_data(char *name);
 uint32_t efuse_obj_get_lock(char *name);
+uint32_t efuse_mrk_get_checknum(const char *name,
+				uint32_t longmrk, uint32_t *checknum);
 
 /*return: 0:is configurated, -1: don't cfg*/
 int efuse_burn_lockable_is_cfg(char *itemname);

drivers/efuse_unifykey/efuse_api.c

@@ -573,3 +573,29 @@ ssize_t efuse_write_usr(char *buf, size_t count, loff_t *ppos)
 
 	return ret;
 }
+
+uint32_t efuse_mrk_get_checknum(const char *name, uint32_t longmrk, uint32_t *result)
+{
+	u32 rc = EFUSE_OBJ_ERR_UNKNOWN;
+	struct arm_smccc_res res;
+
+	meson_sm_mutex_lock();
+
+	strncpy((void *)sharemem_input_base, name, 16);
+	do {
+		arm_smccc_smc((unsigned long)EFUSE_MRK_GET_CHECKNUM,
+			      (unsigned long)longmrk,
+			      0, 0, 0, 0, 0, 0, &res);
+
+	} while (0);
+
+	rc = res.a0;
+	if (longmrk == 1)
+		memcpy((void *)result, (void *)sharemem_output_base, 16);
+	else
+		*result = res.a1;
+
+	meson_sm_mutex_unlock();
+
+	return rc;
+}