From: Eric Biggers <[email protected]>

Add a helper function that encapsulates a container_of expression.  For
now there are two users but soon there will be more.

Tested-by: Bartosz Golaszewski <[email protected]> # sm8650
Signed-off-by: Eric Biggers <[email protected]>
---
 drivers/ufs/core/ufshcd-crypto.c | 6 ++----
 include/ufs/ufshcd.h             | 8 ++++++++
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/drivers/ufs/core/ufshcd-crypto.c b/drivers/ufs/core/ufshcd-crypto.c
index a714dad82cd1..0cb425ef618e 100644
--- a/drivers/ufs/core/ufshcd-crypto.c
+++ b/drivers/ufs/core/ufshcd-crypto.c
@@ -50,12 +50,11 @@ static int ufshcd_program_key(struct ufs_hba *hba,
 
 static int ufshcd_crypto_keyslot_program(struct blk_crypto_profile *profile,
                                         const struct blk_crypto_key *key,
                                         unsigned int slot)
 {
-       struct ufs_hba *hba =
-               container_of(profile, struct ufs_hba, crypto_profile);
+       struct ufs_hba *hba = ufs_hba_from_crypto_profile(profile);
        const union ufs_crypto_cap_entry *ccap_array = hba->crypto_cap_array;
        const struct ufs_crypto_alg_entry *alg =
                        &ufs_crypto_algs[key->crypto_cfg.crypto_mode];
        u8 data_unit_mask = key->crypto_cfg.data_unit_size / 512;
        int i;
@@ -97,12 +96,11 @@ static int ufshcd_crypto_keyslot_program(struct 
blk_crypto_profile *profile,
 
 static int ufshcd_crypto_keyslot_evict(struct blk_crypto_profile *profile,
                                       const struct blk_crypto_key *key,
                                       unsigned int slot)
 {
-       struct ufs_hba *hba =
-               container_of(profile, struct ufs_hba, crypto_profile);
+       struct ufs_hba *hba = ufs_hba_from_crypto_profile(profile);
        /*
         * Clear the crypto cfg on the device. Clearing CFGE
         * might not be sufficient, so just clear the entire cfg.
         */
        union ufs_crypto_cfg_entry cfg = {};
diff --git a/include/ufs/ufshcd.h b/include/ufs/ufshcd.h
index 4ff117fd27cd..55b81996b6e1 100644
--- a/include/ufs/ufshcd.h
+++ b/include/ufs/ufshcd.h
@@ -1211,10 +1211,18 @@ static inline size_t ufshcd_sg_entry_size(const struct 
ufs_hba *hba)
 
 #define ufshcd_set_sg_entry_size(hba, sg_entry_size)                   \
        ({ (void)(hba); BUILD_BUG_ON(sg_entry_size != sizeof(struct 
ufshcd_sg_entry)); })
 #endif
 
+#ifdef CONFIG_SCSI_UFS_CRYPTO
+static inline struct ufs_hba *
+ufs_hba_from_crypto_profile(struct blk_crypto_profile *profile)
+{
+       return container_of(profile, struct ufs_hba, crypto_profile);
+}
+#endif
+
 static inline size_t ufshcd_get_ucd_size(const struct ufs_hba *hba)
 {
        return sizeof(struct utp_transfer_cmd_desc) + SG_ALL * 
ufshcd_sg_entry_size(hba);
 }
 
-- 
2.47.1


Reply via email to