On Mon, Feb 24, 2025 at 08:35:56PM +0000, Pranjal Shrivastava wrote: > oN sAt, Feb 22, 2025 at 07:54:09AM -0800, Nicolin Chen wrote: > > diff --git a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3-iommufd.c b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3-iommufd.c > > index 5aa2e7af58b4..364d8469a480 100644 > > --- a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3-iommufd.c > > +++ b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3-iommufd.c > > @@ -85,6 +85,59 @@ static void arm_smmu_make_nested_domain_ste( > > } > > } > > > > +int arm_smmu_attach_prepare_vmaster(struct arm_smmu_attach_state *state, > > + struct iommu_domain *domain) > > +{ > > + struct arm_smmu_nested_domain *nested_domain; > > + struct arm_smmu_vmaster *vmaster; > > + unsigned long vsid; > > + int ret; > > + > > + iommu_group_mutex_assert(state->master->dev); > > + > > + if (domain->type != IOMMU_DOMAIN_NESTED) > > + return 0; > > + nested_domain = to_smmu_nested_domain(domain); > > + > > + /* Skip invalid vSTE */ > > + if (!(nested_domain->ste[0] & cpu_to_le64(STRTAB_STE_0_V))) > > + return 0; > > + > > + ret = iommufd_viommu_get_vdev_id(&nested_domain->vsmmu->core, > > + state->master->dev, &vsid); > > + if (ret) > > + return ret; > > + > > + vmaster = kzalloc(sizeof(*vmaster), GFP_KERNEL); > > + if (!vmaster) > > + return -ENOMEM; > > + vmaster->vsmmu = nested_domain->vsmmu; > > + vmaster->vsid = vsid; > > + state->vmaster = vmaster; > > + > > + return 0; > > +} > > + > > +void arm_smmu_attach_commit_vmaster(struct arm_smmu_attach_state *state) > > +{ > > + struct arm_smmu_master *master = state->master; > > + > > + mutex_lock(&master->smmu->streams_mutex); > > + if (state->vmaster != master->vmaster) { > > + kfree(master->vmaster); > > + master->vmaster = state->vmaster; > > + } > > Does this condition suggest that we might end up calling > `arm_smmu_attach_prepare_vmaster()` multiple times before __actually__ > commiting to a vmaster? No. prepare() and commit() are 1:1. How is it interpreted to have "multiple times"? > > + mutex_unlock(&master->smmu->streams_mutex); > > +} > > + > > +void arm_smmu_master_clear_vmaster(struct arm_smmu_master *master) > > +{ > > + mutex_lock(&master->smmu->streams_mutex); > > + kfree(master->vmaster); > > + master->vmaster = NULL; > > + mutex_unlock(&master->smmu->streams_mutex); > > +} > > + > > static int arm_smmu_attach_dev_nested(struct iommu_domain *domain, > > struct device *dev) > > { > > diff --git a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c > > index 358072b4e293..9e50bcee69d1 100644 > > --- a/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c > > +++ b/drivers/iommu/arm/arm-smmu-v3/arm-smmu-v3.c > > @@ -2803,6 +2803,7 @@ int arm_smmu_attach_prepare(struct arm_smmu_attach_state *state, > > struct arm_smmu_domain *smmu_domain = > > to_smmu_domain_devices(new_domain); > > unsigned long flags; > > + int ret; > > > > /* > > * arm_smmu_share_asid() must not see two domains pointing to the same > > @@ -2832,9 +2833,15 @@ int arm_smmu_attach_prepare(struct arm_smmu_attach_state *state, > > } > > > > if (smmu_domain) { > > + ret = arm_smmu_attach_prepare_vmaster(state, new_domain); > > IMO, this adds a little confusion for folks not using iommufd. > > I guess it'd be cleaner if we invoke this below within the: > `if (new_domain->type == IOMMU_DOMAIN_NESTED)` condition instead of > simply returning from the function if the new_domain->type isn't NESTED. But the arm_smmu_attach_commit_vmaster() still has to be unconditional as !NESTED domain should clean the vamster away.. Nicolin