psp->capability = val;
 
        /* Detect if TSME and SME are both enabled */
-       if (psp->capability & PSP_CAPABILITY_PSP_SECURITY_REPORTING &&
+       if (PSP_CAPABILITY(psp, PSP_SECURITY_REPORTING) &&
            psp->capability & (PSP_SECURITY_TSME_STATUS << PSP_CAPABILITY_PSP_SECURITY_OFFSET) &&
            cc_platform_has(CC_ATTR_HOST_MEM_ENCRYPT))
                dev_notice(psp->dev, "psp: Both TSME and SME are active, SME is unnecessary when TSME is active.\n");
 static int psp_check_sev_support(struct psp_device *psp)
 {
        /* Check if device supports SEV feature */
-       if (!(psp->capability & PSP_CAPABILITY_SEV)) {
+       if (!PSP_CAPABILITY(psp, SEV)) {
                dev_dbg(psp->dev, "psp does not support SEV\n");
                return -ENODEV;
        }
 static int psp_check_tee_support(struct psp_device *psp)
 {
        /* Check if device supports TEE feature */
-       if (!(psp->capability & PSP_CAPABILITY_TEE)) {
+       if (!PSP_CAPABILITY(psp, TEE)) {
                dev_dbg(psp->dev, "psp does not support TEE\n");
                return -ENODEV;
        }
 
 
 #define PLATFORM_FEATURE_DBC           0x1
 
+#define PSP_CAPABILITY(psp, cap) (psp->capability & PSP_CAPABILITY_##cap)
 #define PSP_FEATURE(psp, feat) (psp->vdata && psp->vdata->platform_features & PLATFORM_FEATURE_##feat)
 
 /* Structure to hold CCP device data */
 
        struct sp_device *sp = dev_get_drvdata(dev);
        struct psp_device *psp = sp->psp_data;
 
-       if (psp && (psp->capability & PSP_CAPABILITY_PSP_SECURITY_REPORTING))
+       if (psp && PSP_CAPABILITY(psp, PSP_SECURITY_REPORTING))
                return 0444;
 
        return 0;
                val = ioread32(psp->io_regs + psp->vdata->bootloader_info_reg);
 
        if (attr == &dev_attr_tee_version.attr &&
-           psp->capability & PSP_CAPABILITY_TEE &&
+           PSP_CAPABILITY(psp, TEE) &&
            psp->vdata->tee->info_reg)
                val = ioread32(psp->io_regs + psp->vdata->tee->info_reg);