driver: flash: Add Set/ Get write protect function

Add Set_WP function to set SPI flash WP line to low
Add Get_WP function to obtain status of the SPI flash WP line

Signed-off-by: Benson Huang <[email protected]>

Author: benson0715

drivers/flash/flash_realtek_rts5912.c

@@ -427,6 +427,34 @@ static int flash_read_sr2(const struct device *dev, uint8_t *val)
 }
 #endif
 
+#ifdef CONFIG_FLASH_EX_OP_ENABLED
+#define SPIC_BASE 0x40010200ul
+static int flash_set_wp(const struct device *dev, uint8_t *val, uint8_t cnt)
+{
+	volatile uint32_t *reg = (volatile uint32_t *)(SPIC_BASE + 0x110u);
+
+	if (!val) {
+		return -EINVAL;
+	}
+
+	if (*val) {
+		*reg |= (1u << 1);
+	}
+
+	return 0;
+}
+
+static int flash_get_wp(const struct device *dev, uint8_t *val)
+{
+	volatile uint32_t *reg = (volatile uint32_t *)(SPIC_BASE + 0x110u);
+	uint32_t tmp = *reg;
+
+	*val = (uint8_t)(tmp & 0x02u);
+
+	return 0;
+}
+#endif
+
 static int flash_wait_till_ready(const struct device *dev)
 {
 	int ret;
@@ -438,9 +466,6 @@ static int flash_wait_till_ready(const struct device *dev)
 	 */
 	do {
 		ret = flash_read_sr(dev, &sr);
-		if (ret < 0) {
-			return ret;
-		}
 		if (!(sr & SPI_NOR_WIP_BIT)) {
 			return 0;
 		}
@@ -751,6 +776,12 @@ static int flash_rts5912_ex_op(const struct device *dev, uint16_t opcode, const
 	case FLASH_RTS5912_EX_OP_RD_SR2:
 		ret = flash_read_sr2(dev, (uint8_t *)in);
 		break;
+	case FLASH_RTS5912_EX_OP_SET_WP:
+		ret = flash_set_wp(dev, (uint8_t *)out, 1);
+		break;
+	case FLASH_RTS5912_EX_OP_GET_WP:
+		ret = flash_get_wp(dev, (uint8_t *)in);
+		break;
 	}
 
 	k_sem_give(&dev_data->sem);

include/zephyr/drivers/flash/rts5912_flash_api_ex.h

@@ -13,6 +13,8 @@ enum flash_rts5912_ex_ops {
 	FLASH_RTS5912_EX_OP_WR_SR2,
 	FLASH_RTS5912_EX_OP_RD_SR,
 	FLASH_RTS5912_EX_OP_RD_SR2,
+	FLASH_RTS5912_EX_OP_SET_WP,
+	FLASH_RTS5912_EX_OP_GET_WP,
 };
 
 #endif /* __ZEPHYR_INCLUDE_DRIVERS_RTS5912_FLASH_API_EX_H__ */