[PATCH v3 10/12] ufs: host: support for generate, import and prepare key

[Gaurav Kashyap]

Implements the vops for generate, prepare and import key
apis in ufs qcom and call the respective ICE module apis
for them.

Signed-off-by: Gaurav Kashyap <quic_gaurk...@quicinc.com>
---
 drivers/ufs/host/ufs-qcom.c | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/drivers/ufs/host/ufs-qcom.c b/drivers/ufs/host/ufs-qcom.c
index 6b09e09b1a30..b4644d07d6f3 100644
--- a/drivers/ufs/host/ufs-qcom.c
+++ b/drivers/ufs/host/ufs-qcom.c
@@ -191,10 +191,41 @@ int ufs_qcom_ice_derive_sw_secret(struct ufs_hba *hba, 
const u8 wkey[],
        return qcom_ice_derive_sw_secret(host->ice, wkey, wkey_size, sw_secret);
 }
 
+int ufs_qcom_ice_generate_key(struct ufs_hba *hba,
+                             u8 lt_key[BLK_CRYPTO_MAX_HW_WRAPPED_KEY_SIZE])
+{
+       struct ufs_qcom_host *host = ufshcd_get_variant(hba);
+
+       return qcom_ice_generate_key(host->ice, lt_key);
+}
+
+int ufs_qcom_ice_prepare_key(struct ufs_hba *hba,
+                            const u8 *lt_key, size_t lt_key_size,
+                            u8 eph_key[BLK_CRYPTO_MAX_HW_WRAPPED_KEY_SIZE])
+{
+       struct ufs_qcom_host *host = ufshcd_get_variant(hba);
+
+       return qcom_ice_prepare_key(host->ice, lt_key, lt_key_size,
+                                   eph_key);
+}
+
+int ufs_qcom_ice_import_key(struct ufs_hba *hba,
+                           const u8 *imp_key, size_t imp_key_size,
+                           u8 lt_key[BLK_CRYPTO_MAX_HW_WRAPPED_KEY_SIZE])
+{
+       struct ufs_qcom_host *host = ufshcd_get_variant(hba);
+
+       return qcom_ice_import_key(host->ice, imp_key, imp_key_size,
+                                  lt_key);
+}
+
 #else
 
 #define ufs_qcom_ice_program_key NULL
 #define ufs_qcom_ice_derive_sw_secret NULL
+#define ufs_qcom_ice_generate_key NULL
+#define ufs_qcom_ice_prepare_key NULL
+#define ufs_qcom_ice_import_key NULL
 
 static inline void ufs_qcom_ice_enable(struct ufs_qcom_host *host)
 {
@@ -2017,6 +2048,9 @@ static const struct ufs_hba_variant_ops ufs_hba_qcom_vops 
= {
        .config_scaling_param = ufs_qcom_config_scaling_param,
        .program_key            = ufs_qcom_ice_program_key,
        .derive_sw_secret       = ufs_qcom_ice_derive_sw_secret,
+       .generate_key           = ufs_qcom_ice_generate_key,
+       .prepare_key            = ufs_qcom_ice_prepare_key,
+       .import_key             = ufs_qcom_ice_import_key,
        .reinit_notify          = ufs_qcom_reinit_notify,
        .mcq_config_resource    = ufs_qcom_mcq_config_resource,
        .get_hba_mac            = ufs_qcom_get_hba_mac,
-- 
2.25.1