static int acpi_nfit_add_dimm(struct acpi_nfit_desc *acpi_desc,
                struct nfit_mem *nfit_mem, u32 device_handle)
 {
+       struct nvdimm_bus_descriptor *nd_desc = &acpi_desc->nd_desc;
        struct acpi_device *adev, *adev_dimm;
        struct device *dev = acpi_desc->dev;
        unsigned long dsm_mask, label_mask;
        /* nfit test assumes 1:1 relationship between commands and dsms */
        nfit_mem->dsm_mask = acpi_desc->dimm_cmd_force_en;
        nfit_mem->family = NVDIMM_FAMILY_INTEL;
+       set_bit(NVDIMM_FAMILY_INTEL, &nd_desc->dimm_family_mask);
 
        if (dcr->valid_fields & ACPI_NFIT_CONTROL_MFG_INFO_VALID)
                sprintf(nfit_mem->id, "%04x-%02x-%04x-%08x",
         * Note, that checking for function0 (bit0) tells us if any commands
         * are reachable through this GUID.
         */
+       clear_bit(NVDIMM_FAMILY_INTEL, &nd_desc->dimm_family_mask);
        for (i = 0; i <= NVDIMM_FAMILY_MAX; i++)
-               if (acpi_check_dsm(adev_dimm->handle, to_nfit_uuid(i), 1, 1))
+               if (acpi_check_dsm(adev_dimm->handle, to_nfit_uuid(i), 1, 1)) {
+                       set_bit(i, &nd_desc->dimm_family_mask);
                        if (family < 0 || i == default_dsm_family)
                                family = i;
+               }
 
        /* limit the supported commands to those that are publicly documented */
        nfit_mem->family = family;
 
        nd_desc->cmd_mask = acpi_desc->bus_cmd_force_en;
        nd_desc->bus_dsm_mask = acpi_desc->bus_nfit_cmd_force_en;
+       set_bit(ND_CMD_CALL, &nd_desc->cmd_mask);
+       set_bit(NVDIMM_BUS_FAMILY_NFIT, &nd_desc->bus_family_mask);
+
        adev = to_acpi_dev(acpi_desc);
        if (!adev)
                return;
        for (i = ND_CMD_ARS_CAP; i <= ND_CMD_CLEAR_ERROR; i++)
                if (acpi_check_dsm(adev->handle, guid, 1, 1ULL << i))
                        set_bit(i, &nd_desc->cmd_mask);
-       set_bit(ND_CMD_CALL, &nd_desc->cmd_mask);
 
        dsm_mask =
                (1 << ND_CMD_ARS_CAP) |
 
                dimm_name = "bus";
        }
 
+       /* Validate command family support against bus declared support */
        if (cmd == ND_CMD_CALL) {
+               unsigned long *mask;
+
                if (copy_from_user(&pkg, p, sizeof(pkg)))
                        return -EFAULT;
+
+               if (nvdimm) {
+                       if (pkg.nd_family > NVDIMM_FAMILY_MAX)
+                               return -EINVAL;
+                       mask = &nd_desc->dimm_family_mask;
+               } else {
+                       if (pkg.nd_family > NVDIMM_BUS_FAMILY_MAX)
+                               return -EINVAL;
+                       mask = &nd_desc->bus_family_mask;
+               }
+
+               if (!test_bit(pkg.nd_family, mask))
+                       return -EINVAL;
        }
 
        if (!desc ||