From: Jeff Lien <jeff.lien@wdc.com>
Date: Fri, 26 Jun 2020 13:37:38 +0000 (-0500)
Subject: Add support for ZN345 drive.
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=c2e3857e710fe47924ebcf39029e60cc9b51b0c1;p=users%2Fhch%2Fnvme-cli.git

Add support for ZN345 drive.
---

diff --git a/plugins/wdc/wdc-nvme.c b/plugins/wdc/wdc-nvme.c
index 1945154..4c201c7 100644
--- a/plugins/wdc/wdc-nvme.c
+++ b/plugins/wdc/wdc-nvme.c
@@ -81,6 +81,8 @@
 #define WDC_NVME_SN730B_DEV_ID				0x3714
 #define WDC_NVME_SN730B_DEV_ID_1			0x3734
 #define WDC_NVME_SN340_DEV_ID				0x500d
+#define WDC_NVME_ZN345_DEV_ID				0x5010
+#define WDC_NVME_ZN345_DEV_ID_1				0x5018
 
 #define WDC_DRIVE_CAP_CAP_DIAG				0x0000000000000001
 #define WDC_DRIVE_CAP_INTERNAL_LOG			0x0000000000000002
@@ -1072,6 +1074,11 @@ static __u64 wdc_get_drive_capabilities(int fd) {
 		case WDC_NVME_SN340_DEV_ID:
 			capabilities = WDC_DRIVE_CAP_DUI;
 			break;
+		case WDC_NVME_ZN345_DEV_ID:
+		/* FALLTHRU */
+		case WDC_NVME_ZN345_DEV_ID_1:
+			capabilities = WDC_DRIVE_CAP_DUI_DATA;
+			break;
 		default:
 			capabilities = 0;
 		}