return memcmp(&old->msi, &new->msi, sizeof(new->msi));
 }
 
-int kvm_arch_update_irqfd_routing(struct kvm *kvm, unsigned int host_irq,
-                                 uint32_t guest_irq, bool set)
+int kvm_arch_update_irqfd_routing(struct kvm_kernel_irqfd *irqfd,
+                                 struct kvm_kernel_irq_routing_entry *old,
+                                 struct kvm_kernel_irq_routing_entry *new)
 {
        /*
         * Remapping the vLPI requires taking the its_lock mutex to resolve
         *
         * Unmap the vLPI and fall back to software LPI injection.
         */
-       return kvm_vgic_v4_unset_forwarding(kvm, host_irq);
+       return kvm_vgic_v4_unset_forwarding(irqfd->kvm, irqfd->producer->irq);
 }
 
 void kvm_arch_irq_bypass_stop(struct irq_bypass_consumer *cons)
 
  */
 #define KVM_APIC_PV_EOI_PENDING        1
 
+struct kvm_kernel_irqfd;
 struct kvm_kernel_irq_routing_entry;
 
 /*
        void (*vcpu_blocking)(struct kvm_vcpu *vcpu);
        void (*vcpu_unblocking)(struct kvm_vcpu *vcpu);
 
-       int (*pi_update_irte)(struct kvm *kvm, unsigned int host_irq,
-                             uint32_t guest_irq, bool set);
+       int (*pi_update_irte)(struct kvm_kernel_irqfd *irqfd, struct kvm *kvm,
+                             unsigned int host_irq, uint32_t guest_irq,
+                             struct kvm_kernel_irq_routing_entry *new);
        void (*pi_start_assignment)(struct kvm *kvm);
        void (*apicv_pre_state_restore)(struct kvm_vcpu *vcpu);
        void (*apicv_post_state_restore)(struct kvm_vcpu *vcpu);
 
 #include <linux/hashtable.h>
 #include <linux/amd-iommu.h>
 #include <linux/kvm_host.h>
+#include <linux/kvm_irqfd.h>
 
 #include <asm/irq_remapping.h>
 #include <asm/msr.h>
        return 0;
 }
 
-/*
- * avic_pi_update_irte - set IRTE for Posted-Interrupts
- *
- * @kvm: kvm
- * @host_irq: host irq of the interrupt
- * @guest_irq: gsi of the interrupt
- * @set: set or unset PI
- * returns 0 on success, < 0 on failure
- */
-int avic_pi_update_irte(struct kvm *kvm, unsigned int host_irq,
-                       uint32_t guest_irq, bool set)
+int avic_pi_update_irte(struct kvm_kernel_irqfd *irqfd, struct kvm *kvm,
+                       unsigned int host_irq, uint32_t guest_irq,
+                       struct kvm_kernel_irq_routing_entry *new)
 {
        struct kvm_kernel_irq_routing_entry *e;
        struct kvm_irq_routing_table *irq_rt;
        bool enable_remapped_mode = true;
+       bool set = !!new;
        int idx, ret = 0;
 
        if (!kvm_arch_has_assigned_device(kvm) || !kvm_arch_has_irq_bypass())
                if (e->type != KVM_IRQ_ROUTING_MSI)
                        continue;
 
+               WARN_ON_ONCE(new && memcmp(e, new, sizeof(*new)));
+
                /**
                 * Here, we setup with legacy mode in the following cases:
                 * 1. When cannot target interrupt to a specific vcpu.
 
 void avic_vcpu_put(struct kvm_vcpu *vcpu);
 void avic_apicv_post_state_restore(struct kvm_vcpu *vcpu);
 void avic_refresh_apicv_exec_ctrl(struct kvm_vcpu *vcpu);
-int avic_pi_update_irte(struct kvm *kvm, unsigned int host_irq,
-                       uint32_t guest_irq, bool set);
+int avic_pi_update_irte(struct kvm_kernel_irqfd *irqfd, struct kvm *kvm,
+                       unsigned int host_irq, uint32_t guest_irq,
+                       struct kvm_kernel_irq_routing_entry *new);
 void avic_vcpu_blocking(struct kvm_vcpu *vcpu);
 void avic_vcpu_unblocking(struct kvm_vcpu *vcpu);
 void avic_ring_doorbell(struct kvm_vcpu *vcpu);
 
 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
 
 #include <linux/kvm_host.h>
+#include <linux/kvm_irqfd.h>
 
 #include <asm/irq_remapping.h>
 #include <asm/cpu.h>
        kvm_make_all_cpus_request(kvm, KVM_REQ_UNBLOCK);
 }
 
-/*
- * vmx_pi_update_irte - set IRTE for Posted-Interrupts
- *
- * @kvm: kvm
- * @host_irq: host irq of the interrupt
- * @guest_irq: gsi of the interrupt
- * @set: set or unset PI
- * returns 0 on success, < 0 on failure
- */
-int vmx_pi_update_irte(struct kvm *kvm, unsigned int host_irq,
-                      uint32_t guest_irq, bool set)
+int vmx_pi_update_irte(struct kvm_kernel_irqfd *irqfd, struct kvm *kvm,
+                      unsigned int host_irq, uint32_t guest_irq,
+                      struct kvm_kernel_irq_routing_entry *new)
 {
        struct kvm_kernel_irq_routing_entry *e;
        struct kvm_irq_routing_table *irq_rt;
        struct kvm_lapic_irq irq;
        struct kvm_vcpu *vcpu;
        struct vcpu_data vcpu_info;
+       bool set = !!new;
        int idx, ret = 0;
 
        if (!vmx_can_use_vtd_pi(kvm))
        hlist_for_each_entry(e, &irq_rt->map[guest_irq], link) {
                if (e->type != KVM_IRQ_ROUTING_MSI)
                        continue;
+
+               WARN_ON_ONCE(new && memcmp(e, new, sizeof(*new)));
+
                /*
                 * VT-d PI cannot support posting multicast/broadcast
                 * interrupts to a vCPU, we still use interrupt remapping
 
 #define __KVM_X86_VMX_POSTED_INTR_H
 
 #include <linux/bitmap.h>
+#include <linux/find.h>
+#include <linux/kvm_host.h>
+
 #include <asm/posted_intr.h>
 
 void vmx_vcpu_pi_load(struct kvm_vcpu *vcpu, int cpu);
 void __init pi_init_cpu(int cpu);
 void pi_apicv_pre_state_restore(struct kvm_vcpu *vcpu);
 bool pi_has_pending_interrupt(struct kvm_vcpu *vcpu);
-int vmx_pi_update_irte(struct kvm *kvm, unsigned int host_irq,
-                      uint32_t guest_irq, bool set);
+int vmx_pi_update_irte(struct kvm_kernel_irqfd *irqfd, struct kvm *kvm,
+                      unsigned int host_irq, uint32_t guest_irq,
+                      struct kvm_kernel_irq_routing_entry *new);
 void vmx_pi_start_assignment(struct kvm *kvm);
 
 static inline int pi_find_highest_vector(struct pi_desc *pi_desc)
 
        struct kvm_kernel_irqfd *irqfd =
                container_of(cons, struct kvm_kernel_irqfd, consumer);
        struct kvm *kvm = irqfd->kvm;
-       int ret;
+       int ret = 0;
 
        kvm_arch_start_assignment(irqfd->kvm);
 
        spin_lock_irq(&kvm->irqfds.lock);
        irqfd->producer = prod;
 
-       ret = kvm_x86_call(pi_update_irte)(irqfd->kvm,
-                                          prod->irq, irqfd->gsi, 1);
-       if (ret)
-               kvm_arch_end_assignment(irqfd->kvm);
-
+       if (irqfd->irq_entry.type == KVM_IRQ_ROUTING_MSI) {
+               ret = kvm_x86_call(pi_update_irte)(irqfd, irqfd->kvm, prod->irq,
+                                                  irqfd->gsi, &irqfd->irq_entry);
+               if (ret)
+                       kvm_arch_end_assignment(irqfd->kvm);
+       }
        spin_unlock_irq(&kvm->irqfds.lock);
 
-
        return ret;
 }
 
 void kvm_arch_irq_bypass_del_producer(struct irq_bypass_consumer *cons,
                                      struct irq_bypass_producer *prod)
 {
-       int ret;
        struct kvm_kernel_irqfd *irqfd =
                container_of(cons, struct kvm_kernel_irqfd, consumer);
        struct kvm *kvm = irqfd->kvm;
+       int ret;
 
        WARN_ON(irqfd->producer != prod);
 
        spin_lock_irq(&kvm->irqfds.lock);
        irqfd->producer = NULL;
 
-       ret = kvm_x86_call(pi_update_irte)(irqfd->kvm,
-                                          prod->irq, irqfd->gsi, 0);
-       if (ret)
-               printk(KERN_INFO "irq bypass consumer (eventfd %p) unregistration"
-                      " fails: %d\n", irqfd->consumer.eventfd, ret);
+       if (irqfd->irq_entry.type == KVM_IRQ_ROUTING_MSI) {
+               ret = kvm_x86_call(pi_update_irte)(irqfd, irqfd->kvm, prod->irq,
+                                                  irqfd->gsi, NULL);
+               if (ret)
+                       pr_info("irq bypass consumer (eventfd %p) unregistration fails: %d\n",
+                               irqfd->consumer.eventfd, ret);
+       }
 
        spin_unlock_irq(&kvm->irqfds.lock);
 
        kvm_arch_end_assignment(irqfd->kvm);
 }
 
-int kvm_arch_update_irqfd_routing(struct kvm *kvm, unsigned int host_irq,
-                                  uint32_t guest_irq, bool set)
+int kvm_arch_update_irqfd_routing(struct kvm_kernel_irqfd *irqfd,
+                                 struct kvm_kernel_irq_routing_entry *old,
+                                 struct kvm_kernel_irq_routing_entry *new)
 {
-       return kvm_x86_call(pi_update_irte)(kvm, host_irq, guest_irq, set);
+       return kvm_x86_call(pi_update_irte)(irqfd, irqfd->kvm, irqfd->producer->irq,
+                                           irqfd->gsi, new);
 }
 
 bool kvm_arch_irqfd_route_changed(struct kvm_kernel_irq_routing_entry *old,
 
 struct kvm_vcpu * __percpu *kvm_get_running_vcpus(void);
 
 #if IS_ENABLED(CONFIG_HAVE_KVM_IRQ_BYPASS)
+struct kvm_kernel_irqfd;
+
 bool kvm_arch_has_irq_bypass(void);
 int kvm_arch_irq_bypass_add_producer(struct irq_bypass_consumer *,
                           struct irq_bypass_producer *);
                           struct irq_bypass_producer *);
 void kvm_arch_irq_bypass_stop(struct irq_bypass_consumer *);
 void kvm_arch_irq_bypass_start(struct irq_bypass_consumer *);
-int kvm_arch_update_irqfd_routing(struct kvm *kvm, unsigned int host_irq,
-                                 uint32_t guest_irq, bool set);
+int kvm_arch_update_irqfd_routing(struct kvm_kernel_irqfd *irqfd,
+                                 struct kvm_kernel_irq_routing_entry *old,
+                                 struct kvm_kernel_irq_routing_entry *new);
 bool kvm_arch_irqfd_route_changed(struct kvm_kernel_irq_routing_entry *,
                                  struct kvm_kernel_irq_routing_entry *);
 #endif /* CONFIG_HAVE_KVM_IRQ_BYPASS */
 
 {
 }
 
-int  __attribute__((weak)) kvm_arch_update_irqfd_routing(
-                               struct kvm *kvm, unsigned int host_irq,
-                               uint32_t guest_irq, bool set)
+int __weak kvm_arch_update_irqfd_routing(struct kvm_kernel_irqfd *irqfd,
+                                        struct kvm_kernel_irq_routing_entry *old,
+                                        struct kvm_kernel_irq_routing_entry *new)
 {
        return 0;
 }
 #if IS_ENABLED(CONFIG_HAVE_KVM_IRQ_BYPASS)
                if (irqfd->producer &&
                    kvm_arch_irqfd_route_changed(&old, &irqfd->irq_entry)) {
-                       int ret = kvm_arch_update_irqfd_routing(
-                                       irqfd->kvm, irqfd->producer->irq,
-                                       irqfd->gsi, 1);
+                       int ret = kvm_arch_update_irqfd_routing(irqfd, &old, &irqfd->irq_entry);
+
                        WARN_ON(ret);
                }
 #endif