feat: 支持flash的lock/unlock功能

Author: xf-cai

cskburn/src/main.c

@@ -85,6 +85,8 @@ static struct option long_options[] = {
 		{"read", required_argument, NULL, 0},
 		{"erase", required_argument, NULL, 0},
 		{"erase-all", no_argument, NULL, 0},
+		{"lock", no_argument, NULL, 0},
+		{"unlock", no_argument, NULL, 0},
 		{"verify", required_argument, NULL, 0},
 		{"verify-all", no_argument, NULL, 0},
 		{"probe-timeout", required_argument, NULL, 0},
@@ -164,6 +166,8 @@ static struct {
 		uint32_t size;
 	} erase_parts[MAX_ERASE_PARTS];
 	bool erase_all;
+	bool lock;
+	bool unlock;
 	uint16_t verify_count;
 	struct {
 		uint32_t addr;
@@ -199,6 +203,8 @@ static struct {
 		.read_count = 0,
 		.erase_count = 0,
 		.erase_all = false,
+		.lock = false,
+		.unlock = false,
 		.verify_count = 0,
 		.verify_all = false,
 		.probe_timeout = DEFAULT_PROBE_TIMEOUT,
@@ -299,6 +305,10 @@ print_help(const char *progname)
 	LOGI("    erase specified flash region");
 	LOGI("  --erase-all");
 	LOGI("    erase the entire flash");
+	LOGI("  --lock");
+	LOGI("    lock the flash");
+	LOGI("  --unlock");
+	LOGI("    unlock the flash");
 	LOGI("  --verify <addr:size>");
 	LOGI("    verify specified flash region");
 	LOGI("");
@@ -527,6 +537,12 @@ main(int argc, char **argv)
 				} else if (strcmp(name, "emmc") == 0) {
 					options.target = TARGET_EMMC;
 					break;
+				} else if (strcmp(name, "lock") == 0) {
+					options.lock = true;
+					break;
+				} else if (strcmp(name, "unlock") == 0) {
+					options.unlock = true;
+					break;
 				} else {
 					print_help(argv[0]);
 					return 0;
@@ -924,6 +940,13 @@ serial_burn(cskburn_partition_t *parts, int parts_cnt)
 		LOGI("Detected eMMC size: %" PRIu64 " MB", flash_size >> 20);
 	}
 
+	if (options.unlock) {
+		if ((ret = cskburn_serial_unlock(dev, options.target)) != 0) {
+			LOGE_RET(ret, "ERROR: Failed locking device");
+			goto err_enter;
+		}
+	}
+
 	for (int i = 0; i < options.read_count; i++) {
 		if (options.read_parts[i].addr >= flash_size) {
 			LOGE("ERROR: The starting boundary of read address (0x%08X) exceeds the capacity of "
@@ -1108,6 +1131,13 @@ serial_burn(cskburn_partition_t *parts, int parts_cnt)
 		}
 	}
 
+	if (options.lock) {
+		if ((ret = cskburn_serial_lock(dev, options.target)) != 0) {
+			LOGE_RET(ret, "ERROR: Failed unlocking device");
+			goto err_enter;
+		}
+	}
+
 	if (jump_addr) {
 		LOGI("Jumping to 0x%08X...", jump_addr);
 	} else if (!options.no_reset) {

libcskburn_serial/include/cskburn_serial.h

@@ -122,6 +122,10 @@ int cskburn_serial_erase_all(cskburn_serial_device_t *dev, cskburn_serial_target
 int cskburn_serial_erase(
 		cskburn_serial_device_t *dev, cskburn_serial_target_t target, uint32_t addr, uint32_t size);
 
+int cskburn_serial_lock(cskburn_serial_device_t *dev, cskburn_serial_target_t target);
+
+int cskburn_serial_unlock(cskburn_serial_device_t *dev, cskburn_serial_target_t target);
+
 int cskburn_serial_verify(cskburn_serial_device_t *dev, cskburn_serial_target_t target,
 		uint32_t addr, uint32_t size, uint8_t *md5);
 

libcskburn_serial/src/cmd.c

@@ -46,6 +46,9 @@
 
 #define CMD_FLASH_ERASE_CHIP 0xD0
 #define CMD_FLASH_ERASE_REGION 0xD1
+#define CMD_FLASH_LOCK 0xD4
+#define CMD_FLASH_UNLOCK 0xD5
+
 #define CMD_READ_FLASH_ID 0xF3
 #define CMD_READ_CHIP_ID 0xF4
 
@@ -567,6 +570,18 @@ cmd_flash_erase_region(cskburn_serial_device_t *dev, uint32_t address, uint32_t
 			NULL, calc_timeout(size, TIMEOUT_FLASH_ERASE_PER_MB));
 }
 
+int
+cmd_flash_lock(cskburn_serial_device_t *dev)
+{
+	return check_command(dev, CMD_FLASH_LOCK, 0, CHECKSUM_NONE, NULL, TIMEOUT_FLASH_DATA);
+}
+
+int
+cmd_flash_unlock(cskburn_serial_device_t *dev)
+{
+	return check_command(dev, CMD_FLASH_UNLOCK, 0, CHECKSUM_NONE, NULL, TIMEOUT_FLASH_DATA);
+}
+
 int
 cmd_flash_md5sum(cskburn_serial_device_t *dev, uint32_t address, uint32_t size, uint8_t *md5)
 {

libcskburn_serial/src/cmd.h

@@ -72,6 +72,9 @@ int cmd_flash_finish(cskburn_serial_device_t *dev);
 int cmd_flash_erase_chip(cskburn_serial_device_t *dev);
 int cmd_flash_erase_region(cskburn_serial_device_t *dev, uint32_t address, uint32_t size);
 
+int cmd_flash_lock(cskburn_serial_device_t *dev);
+int cmd_flash_unlock(cskburn_serial_device_t *dev);
+
 int cmd_flash_md5sum(cskburn_serial_device_t *dev, uint32_t address, uint32_t size, uint8_t *md5);
 
 int cmd_read_flash(cskburn_serial_device_t *dev, uint32_t address, uint32_t size, uint8_t *data,

libcskburn_serial/src/core.c

@@ -543,6 +543,40 @@ cskburn_serial_erase(
 	return -EINVAL;
 }
 
+int
+cskburn_serial_lock(cskburn_serial_device_t *dev, cskburn_serial_target_t target)
+{
+	int ret;
+
+	if (target == TARGET_FLASH) {
+		if ((ret = cmd_flash_lock(dev)) != 0) {
+			return ret;
+		}
+
+		return 0;
+	}
+
+	LOGE("ERROR: LOCK is not supported, target: %d", target);
+	return -EINVAL;
+}
+
+int
+cskburn_serial_unlock(cskburn_serial_device_t *dev, cskburn_serial_target_t target)
+{
+	int ret;
+
+	if (target == TARGET_FLASH) {
+		if ((ret = cmd_flash_unlock(dev)) != 0) {
+			return ret;
+		}
+
+		return 0;
+	}
+
+	LOGE("ERROR: UNLOCK is not supported, target: %d", target);
+	return -EINVAL;
+}
+
 int
 cskburn_serial_verify(cskburn_serial_device_t *dev, cskburn_serial_target_t target, uint32_t addr,
 		uint32_t size, uint8_t *md5)