624 	struct dmar_domain **domains;  in get_iommu_domain()  local
627 	domains = iommu->domains[idx];  in get_iommu_domain()
628 	if (!domains)  in get_iommu_domain()
631 	return domains[did & 0xff];  in get_iommu_domain()
637 	struct dmar_domain **domains;  in set_iommu_domain()  local
640 	if (!iommu->domains[idx]) {  in set_iommu_domain()
642 		iommu->domains[idx] = kzalloc(size, GFP_ATOMIC);  in set_iommu_domain()
645 	domains = iommu->domains[idx];  in set_iommu_domain()
646 	if (WARN_ON(!domains))  in set_iommu_domain()
649 		domains[did & 0xff] = domain;  in set_iommu_domain()
1689 	iommu->domains = kzalloc(size, GFP_KERNEL);  in iommu_init_domains()
1691 	if (iommu->domains) {  in iommu_init_domains()
1693 		iommu->domains[0] = kzalloc(size, GFP_KERNEL);  in iommu_init_domains()
1696 	if (!iommu->domains || !iommu->domains[0]) {  in iommu_init_domains()
1700 		kfree(iommu->domains);  in iommu_init_domains()
1702 		iommu->domains    = NULL;  in iommu_init_domains()
1724 	if (!iommu->domains || !iommu->domain_ids)  in disable_dmar_iommu()
1762 	if ((iommu->domains) && (iommu->domain_ids)) {  in free_dmar_iommu()
1767 			kfree(iommu->domains[i]);  in free_dmar_iommu()
1768 		kfree(iommu->domains);  in free_dmar_iommu()
1770 		iommu->domains = NULL;  in free_dmar_iommu()