spinlock_t              list_lock;
 };
 
-struct s390_domain_device {
-       struct list_head        list;
-       struct zpci_dev         *zdev;
-};
-
 static struct s390_domain *to_s390_domain(struct iommu_domain *dom)
 {
        return container_of(dom, struct s390_domain, domain);
 static void __s390_iommu_detach_device(struct zpci_dev *zdev)
 {
        struct s390_domain *s390_domain = zdev->s390_domain;
-       struct s390_domain_device *domain_device, *tmp;
        unsigned long flags;
 
        if (!s390_domain)
                return;
 
        spin_lock_irqsave(&s390_domain->list_lock, flags);
-       list_for_each_entry_safe(domain_device, tmp, &s390_domain->devices,
-                                list) {
-               if (domain_device->zdev == zdev) {
-                       list_del(&domain_device->list);
-                       kfree(domain_device);
-                       break;
-               }
-       }
+       list_del_init(&zdev->iommu_list);
        spin_unlock_irqrestore(&s390_domain->list_lock, flags);
 
        zpci_unregister_ioat(zdev, 0);
 {
        struct s390_domain *s390_domain = to_s390_domain(domain);
        struct zpci_dev *zdev = to_zpci_dev(dev);
-       struct s390_domain_device *domain_device;
        unsigned long flags;
        int cc, rc = 0;
 
        if (!zdev)
                return -ENODEV;
 
-       domain_device = kzalloc(sizeof(*domain_device), GFP_KERNEL);
-       if (!domain_device)
-               return -ENOMEM;
-
        if (zdev->s390_domain)
                __s390_iommu_detach_device(zdev);
        else if (zdev->dma_table)
 
        cc = zpci_register_ioat(zdev, 0, zdev->start_dma, zdev->end_dma,
                                virt_to_phys(s390_domain->dma_table));
-       if (cc) {
-               rc = -EIO;
-               goto out_free;
-       }
+       if (cc)
+               return -EIO;
        zdev->dma_table = s390_domain->dma_table;
 
        spin_lock_irqsave(&s390_domain->list_lock, flags);
                rc = -EINVAL;
                goto out_unregister;
        }
-       domain_device->zdev = zdev;
        zdev->s390_domain = s390_domain;
-       list_add(&domain_device->list, &s390_domain->devices);
+       list_add(&zdev->iommu_list, &s390_domain->devices);
        spin_unlock_irqrestore(&s390_domain->list_lock, flags);
 
        return 0;
 out_unregister:
        zpci_unregister_ioat(zdev, 0);
        zdev->dma_table = NULL;
-out_free:
-       kfree(domain_device);
 
        return rc;
 }
                                   phys_addr_t pa, dma_addr_t dma_addr,
                                   size_t size, int flags)
 {
-       struct s390_domain_device *domain_device;
        phys_addr_t page_addr = pa & PAGE_MASK;
        dma_addr_t start_dma_addr = dma_addr;
        unsigned long irq_flags, nr_pages, i;
+       struct zpci_dev *zdev;
        unsigned long *entry;
        int rc = 0;
 
        }
 
        spin_lock(&s390_domain->list_lock);
-       list_for_each_entry(domain_device, &s390_domain->devices, list) {
-               rc = zpci_refresh_trans((u64) domain_device->zdev->fh << 32,
+       list_for_each_entry(zdev, &s390_domain->devices, iommu_list) {
+               rc = zpci_refresh_trans((u64)zdev->fh << 32,
                                        start_dma_addr, nr_pages * PAGE_SIZE);
                if (rc)
                        break;