.irq_unmask             = pci_msi_unmask_irq,
        .irq_mask               = pci_msi_mask_irq,
        .irq_ack                = irq_chip_ack_parent,
-       .irq_set_affinity       = msi_domain_set_affinity,
        .irq_retrigger          = irq_chip_retrigger_hierarchy,
        .irq_compose_msi_msg    = irq_msi_compose_msg,
-       .irq_write_msi_msg      = pci_msi_domain_write_msg,
        .flags                  = IRQCHIP_SKIP_SET_WAKE,
 };
 
        .irq_unmask             = pci_msi_unmask_irq,
        .irq_mask               = pci_msi_mask_irq,
        .irq_ack                = irq_chip_ack_parent,
-       .irq_set_affinity       = msi_domain_set_affinity,
        .irq_retrigger          = irq_chip_retrigger_hierarchy,
-       .irq_write_msi_msg      = pci_msi_domain_write_msg,
        .flags                  = IRQCHIP_SKIP_SET_WAKE,
 };
 
 #endif
 
 #ifdef CONFIG_DMAR_TABLE
-static int
-dmar_msi_set_affinity(struct irq_data *data, const struct cpumask *mask,
-                     bool force)
-{
-       struct irq_data *parent = data->parent_data;
-       struct msi_msg msg;
-       int ret;
-
-       ret = parent->chip->irq_set_affinity(parent, mask, force);
-       if (ret >= 0) {
-               irq_chip_compose_msi_msg(data, &msg);
-               dmar_msi_write(data->irq, &msg);
-       }
-
-       return ret;
-}
-
 static void dmar_msi_write_msg(struct irq_data *data, struct msi_msg *msg)
 {
        dmar_msi_write(data->irq, msg);
        .irq_unmask             = dmar_msi_unmask,
        .irq_mask               = dmar_msi_mask,
        .irq_ack                = irq_chip_ack_parent,
-       .irq_set_affinity       = dmar_msi_set_affinity,
+       .irq_set_affinity       = msi_domain_set_affinity,
        .irq_retrigger          = irq_chip_retrigger_hierarchy,
        .irq_compose_msi_msg    = irq_msi_compose_msg,
        .irq_write_msi_msg      = dmar_msi_write_msg,
        .flags                  = IRQCHIP_SKIP_SET_WAKE,
 };
 
-static int dmar_domain_alloc(struct irq_domain *domain, unsigned int virq,
-                            unsigned int nr_irqs, void *arg)
+static irq_hw_number_t dmar_msi_get_hwirq(struct msi_domain_info *info,
+                                         msi_alloc_info_t *arg)
 {
-       struct irq_alloc_info *info = arg;
-       int ret;
-
-       if (nr_irqs > 1 || !info || info->type != X86_IRQ_ALLOC_TYPE_DMAR)
-               return -EINVAL;
-       if (irq_find_mapping(domain, info->dmar_id)) {
-               pr_warn("IRQ for DMAR%d already exists.\n", info->dmar_id);
-               return -EEXIST;
-       }
-
-       ret = irq_domain_alloc_irqs_parent(domain, virq, nr_irqs, arg);
-       if (ret >= 0) {
-               irq_domain_set_hwirq_and_chip(domain, virq, info->dmar_id,
-                                             &dmar_msi_controller, NULL);
-               irq_set_handler_data(virq, info->dmar_data);
-               __irq_set_handler(virq, handle_edge_irq, 0, "edge");
-       }
-
-       return ret;
+       return arg->dmar_id;
 }
 
-static void dmar_domain_free(struct irq_domain *domain, unsigned int virq,
-                            unsigned int nr_irqs)
+static int dmar_msi_init(struct irq_domain *domain,
+                        struct msi_domain_info *info, unsigned int virq,
+                        irq_hw_number_t hwirq, msi_alloc_info_t *arg)
 {
-       BUG_ON(nr_irqs > 1);
-       irq_domain_free_irqs_top(domain, virq, nr_irqs);
-}
+       irq_domain_set_info(domain, virq, arg->dmar_id, info->chip, NULL,
+                           handle_edge_irq, arg->dmar_data, "edge");
 
-static void dmar_domain_activate(struct irq_domain *domain,
-                                struct irq_data *irq_data)
-{
-       struct msi_msg msg;
-
-       BUG_ON(irq_chip_compose_msi_msg(irq_data, &msg));
-       dmar_msi_write(irq_data->irq, &msg);
+       return 0;
 }
 
-static void dmar_domain_deactivate(struct irq_domain *domain,
-                                  struct irq_data *irq_data)
-{
-       struct msi_msg msg;
-
-       memset(&msg, 0, sizeof(msg));
-       dmar_msi_write(irq_data->irq, &msg);
-}
+static struct msi_domain_ops dmar_msi_domain_ops = {
+       .get_hwirq      = dmar_msi_get_hwirq,
+       .msi_init       = dmar_msi_init,
+};
 
-static struct irq_domain_ops dmar_domain_ops = {
-       .alloc = dmar_domain_alloc,
-       .free = dmar_domain_free,
-       .activate = dmar_domain_activate,
-       .deactivate = dmar_domain_deactivate,
+static struct msi_domain_info dmar_msi_domain_info = {
+       .ops            = &dmar_msi_domain_ops,
+       .chip           = &dmar_msi_controller,
 };
 
 static struct irq_domain *dmar_get_irq_domain(void)
        static DEFINE_MUTEX(dmar_lock);
 
        mutex_lock(&dmar_lock);
-       if (dmar_domain == NULL) {
-               dmar_domain = irq_domain_add_tree(NULL, &dmar_domain_ops, NULL);
-               if (dmar_domain)
-                       dmar_domain->parent = x86_vector_domain;
-       }
+       if (dmar_domain == NULL)
+               dmar_domain = msi_create_irq_domain(NULL, &dmar_msi_domain_info,
+                                                   x86_vector_domain);
        mutex_unlock(&dmar_lock);
 
        return dmar_domain;
 #ifdef CONFIG_HPET_TIMER
 static inline int hpet_dev_id(struct irq_domain *domain)
 {
-       return (int)(long)domain->host_data;
-}
+       struct msi_domain_info *info = msi_get_domain_info(domain);
 
-static int hpet_msi_set_affinity(struct irq_data *data,
-                                const struct cpumask *mask, bool force)
-{
-       struct irq_data *parent = data->parent_data;
-       struct msi_msg msg;
-       int ret;
-
-       ret = parent->chip->irq_set_affinity(parent, mask, force);
-       if (ret >= 0 && ret != IRQ_SET_MASK_OK_DONE) {
-               irq_chip_compose_msi_msg(data, &msg);
-               hpet_msi_write(data->handler_data, &msg);
-       }
-
-       return ret;
+       return (int)(long)info->data;
 }
 
 static void hpet_msi_write_msg(struct irq_data *data, struct msi_msg *msg)
        .irq_unmask = hpet_msi_unmask,
        .irq_mask = hpet_msi_mask,
        .irq_ack = irq_chip_ack_parent,
-       .irq_set_affinity = hpet_msi_set_affinity,
+       .irq_set_affinity = msi_domain_set_affinity,
        .irq_retrigger = irq_chip_retrigger_hierarchy,
        .irq_compose_msi_msg = irq_msi_compose_msg,
        .irq_write_msi_msg = hpet_msi_write_msg,
        .flags = IRQCHIP_SKIP_SET_WAKE,
 };
 
-static int hpet_domain_alloc(struct irq_domain *domain, unsigned int virq,
-                            unsigned int nr_irqs, void *arg)
-{
-       struct irq_alloc_info *info = arg;
-       int ret;
-
-       if (nr_irqs > 1 || !info || info->type != X86_IRQ_ALLOC_TYPE_HPET)
-               return -EINVAL;
-       if (irq_find_mapping(domain, info->hpet_index)) {
-               pr_warn("IRQ for HPET%d already exists.\n", info->hpet_index);
-               return -EEXIST;
-       }
-
-       ret = irq_domain_alloc_irqs_parent(domain, virq, nr_irqs, arg);
-       if (ret >= 0) {
-               irq_set_status_flags(virq, IRQ_MOVE_PCNTXT);
-               irq_domain_set_hwirq_and_chip(domain, virq, info->hpet_index,
-                                             &hpet_msi_controller, NULL);
-               irq_set_handler_data(virq, info->hpet_data);
-               __irq_set_handler(virq, handle_edge_irq, 0, "edge");
-       }
-
-       return ret;
-}
-
-static void hpet_domain_free(struct irq_domain *domain, unsigned int virq,
-                            unsigned int nr_irqs)
+static irq_hw_number_t hpet_msi_get_hwirq(struct msi_domain_info *info,
+                                         msi_alloc_info_t *arg)
 {
-       BUG_ON(nr_irqs > 1);
-       irq_clear_status_flags(virq, IRQ_MOVE_PCNTXT);
-       irq_domain_free_irqs_top(domain, virq, nr_irqs);
+       return arg->hpet_index;
 }
 
-static void hpet_domain_activate(struct irq_domain *domain,
-                               struct irq_data *irq_data)
+static int hpet_msi_init(struct irq_domain *domain,
+                        struct msi_domain_info *info, unsigned int virq,
+                        irq_hw_number_t hwirq, msi_alloc_info_t *arg)
 {
-       struct msi_msg msg;
+       irq_set_status_flags(virq, IRQ_MOVE_PCNTXT);
+       irq_domain_set_info(domain, virq, arg->hpet_index, info->chip, NULL,
+                           handle_edge_irq, arg->hpet_data, "edge");
 
-       BUG_ON(irq_chip_compose_msi_msg(irq_data, &msg));
-       hpet_msi_write(irq_get_handler_data(irq_data->irq), &msg);
+       return 0;
 }
 
-static void hpet_domain_deactivate(struct irq_domain *domain,
-                                 struct irq_data *irq_data)
+static void hpet_msi_free(struct irq_domain *domain,
+                         struct msi_domain_info *info, unsigned int virq)
 {
-       struct msi_msg msg;
-
-       memset(&msg, 0, sizeof(msg));
-       hpet_msi_write(irq_get_handler_data(irq_data->irq), &msg);
+       irq_clear_status_flags(virq, IRQ_MOVE_PCNTXT);
 }
 
-static struct irq_domain_ops hpet_domain_ops = {
-       .alloc = hpet_domain_alloc,
-       .free = hpet_domain_free,
-       .activate = hpet_domain_activate,
-       .deactivate = hpet_domain_deactivate,
+static struct msi_domain_ops hpet_msi_domain_ops = {
+       .get_hwirq      = hpet_msi_get_hwirq,
+       .msi_init       = hpet_msi_init,
+       .msi_free       = hpet_msi_free,
+};
+
+static struct msi_domain_info hpet_msi_domain_info = {
+       .ops            = &hpet_msi_domain_ops,
+       .chip           = &hpet_msi_controller,
 };
 
 struct irq_domain *hpet_create_irq_domain(int hpet_id)
 {
        struct irq_domain *parent;
        struct irq_alloc_info info;
+       struct msi_domain_info *domain_info;
 
        if (x86_vector_domain == NULL)
                return NULL;
 
+       domain_info = kzalloc(sizeof(*domain_info), GFP_KERNEL);
+       if (!domain_info)
+               return NULL;
+
+       *domain_info = hpet_msi_domain_info;
+       domain_info->data = (void *)(long)hpet_id;
+
        init_irq_alloc_info(&info, NULL);
        info.type = X86_IRQ_ALLOC_TYPE_HPET;
        info.hpet_id = hpet_id;
        else
                hpet_msi_controller.name = "IR-HPET-MSI";
 
-       return irq_domain_add_hierarchy(parent, 0, 0, NULL, &hpet_domain_ops,
-                                       (void *)(long)hpet_id);
+       return msi_create_irq_domain(NULL, domain_info, parent);
 }
 
 int hpet_assign_irq(struct irq_domain *domain, struct hpet_dev *dev,