#include <linux/spinlock.h>
 #include <linux/types.h>
 #include <kvm/iodev.h>
+#include <linux/irqchip/arm-gic-common.h>
 
 #define VGIC_NR_IRQS_LEGACY    256
 #define VGIC_NR_SGIS           16
 #define vgic_initialized(k)    (!!((k)->arch.vgic.nr_cpus))
 #define vgic_ready(k)          ((k)->arch.vgic.ready)
 
-int vgic_v2_probe(struct device_node *vgic_node,
+int vgic_v2_probe(const struct gic_kvm_info *gic_kvm_info,
                  const struct vgic_ops **ops,
                  const struct vgic_params **params);
 #ifdef CONFIG_KVM_ARM_VGIC_V3
-int vgic_v3_probe(struct device_node *vgic_node,
+int vgic_v3_probe(const struct gic_kvm_info *gic_kvm_info,
                  const struct vgic_ops **ops,
                  const struct vgic_params **params);
 #else
-static inline int vgic_v3_probe(struct device_node *vgic_node,
+static inline int vgic_v3_probe(const struct gic_kvm_info *gic_kvm_info,
                                const struct vgic_ops **ops,
                                const struct vgic_params **params)
 {
 
 #include <linux/kvm_host.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
-#include <linux/of.h>
-#include <linux/of_address.h>
-#include <linux/of_irq.h>
 
 #include <linux/irqchip/arm-gic.h>
 
 }
 
 /**
- * vgic_v2_probe - probe for a GICv2 compatible interrupt controller in DT
- * @node:      pointer to the DT node
- * @ops:       address of a pointer to the GICv2 operations
- * @params:    address of a pointer to HW-specific parameters
+ * vgic_v2_probe - probe for a GICv2 compatible interrupt controller
+ * @gic_kvm_info:      pointer to the GIC description
+ * @ops:               address of a pointer to the GICv2 operations
+ * @params:            address of a pointer to HW-specific parameters
  *
  * Returns 0 if a GICv2 has been found, with the low level operations
  * in *ops and the HW parameters in *params. Returns an error code
  * otherwise.
  */
-int vgic_v2_probe(struct device_node *vgic_node,
-                 const struct vgic_ops **ops,
-                 const struct vgic_params **params)
+int vgic_v2_probe(const struct gic_kvm_info *gic_kvm_info,
+                  const struct vgic_ops **ops,
+                  const struct vgic_params **params)
 {
        int ret;
-       struct resource vctrl_res;
-       struct resource vcpu_res;
        struct vgic_params *vgic = &vgic_v2_params;
+       const struct resource *vctrl_res = &gic_kvm_info->vctrl;
+       const struct resource *vcpu_res = &gic_kvm_info->vcpu;
 
-       vgic->maint_irq = irq_of_parse_and_map(vgic_node, 0);
-       if (!vgic->maint_irq) {
-               kvm_err("error getting vgic maintenance irq from DT\n");
+       if (!gic_kvm_info->maint_irq) {
+               kvm_err("error getting vgic maintenance irq\n");
                ret = -ENXIO;
                goto out;
        }
+       vgic->maint_irq = gic_kvm_info->maint_irq;
 
-       ret = of_address_to_resource(vgic_node, 2, &vctrl_res);
-       if (ret) {
-               kvm_err("Cannot obtain GICH resource\n");
+       if (!gic_kvm_info->vctrl.start) {
+               kvm_err("GICH not present in the firmware table\n");
+               ret = -ENXIO;
                goto out;
        }
 
-       vgic->vctrl_base = of_iomap(vgic_node, 2);
+       vgic->vctrl_base = ioremap(gic_kvm_info->vctrl.start,
+                                  resource_size(&gic_kvm_info->vctrl));
        if (!vgic->vctrl_base) {
                kvm_err("Cannot ioremap GICH\n");
                ret = -ENOMEM;
        vgic->nr_lr = (vgic->nr_lr & 0x3f) + 1;
 
        ret = create_hyp_io_mappings(vgic->vctrl_base,
-                                    vgic->vctrl_base + resource_size(&vctrl_res),
-                                    vctrl_res.start);
+                                    vgic->vctrl_base + resource_size(vctrl_res),
+                                    vctrl_res->start);
        if (ret) {
                kvm_err("Cannot map VCTRL into hyp\n");
                goto out_unmap;
        }
 
-       if (of_address_to_resource(vgic_node, 3, &vcpu_res)) {
-               kvm_err("Cannot obtain GICV resource\n");
-               ret = -ENXIO;
-               goto out_unmap;
-       }
-
-       if (!PAGE_ALIGNED(vcpu_res.start)) {
+       if (!PAGE_ALIGNED(vcpu_res->start)) {
                kvm_err("GICV physical address 0x%llx not page aligned\n",
-                       (unsigned long long)vcpu_res.start);
+                       (unsigned long long)vcpu_res->start);
                ret = -ENXIO;
                goto out_unmap;
        }
 
-       if (!PAGE_ALIGNED(resource_size(&vcpu_res))) {
+       if (!PAGE_ALIGNED(resource_size(vcpu_res))) {
                kvm_err("GICV size 0x%llx not a multiple of page size 0x%lx\n",
-                       (unsigned long long)resource_size(&vcpu_res),
+                       (unsigned long long)resource_size(vcpu_res),
                        PAGE_SIZE);
                ret = -ENXIO;
                goto out_unmap;
        vgic->can_emulate_gicv2 = true;
        kvm_register_device_ops(&kvm_arm_vgic_v2_ops, KVM_DEV_TYPE_ARM_VGIC_V2);
 
-       vgic->vcpu_base = vcpu_res.start;
+       vgic->vcpu_base = vcpu_res->start;
 
-       kvm_info("%s@%llx IRQ%d\n", vgic_node->name,
-                vctrl_res.start, vgic->maint_irq);
+       kvm_info("GICH base=0x%llx, GICV base=0x%llx, IRQ=%d\n",
+                gic_kvm_info->vctrl.start, vgic->vcpu_base, vgic->maint_irq);
 
        vgic->type = VGIC_V2;
        vgic->max_gic_vcpus = VGIC_V2_MAX_CPUS;
 out_unmap:
        iounmap(vgic->vctrl_base);
 out:
-       of_node_put(vgic_node);
        return ret;
 }
 
 #include <linux/kvm_host.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
-#include <linux/of.h>
-#include <linux/of_address.h>
-#include <linux/of_irq.h>
 
 #include <linux/irqchip/arm-gic-v3.h>
+#include <linux/irqchip/arm-gic-common.h>
 
 #include <asm/kvm_emulate.h>
 #include <asm/kvm_arm.h>
 }
 
 /**
- * vgic_v3_probe - probe for a GICv3 compatible interrupt controller in DT
- * @node:      pointer to the DT node
- * @ops:       address of a pointer to the GICv3 operations
- * @params:    address of a pointer to HW-specific parameters
+ * vgic_v3_probe - probe for a GICv3 compatible interrupt controller
+ * @gic_kvm_info:      pointer to the GIC description
+ * @ops:               address of a pointer to the GICv3 operations
+ * @params:            address of a pointer to HW-specific parameters
  *
  * Returns 0 if a GICv3 has been found, with the low level operations
  * in *ops and the HW parameters in *params. Returns an error code
  * otherwise.
  */
-int vgic_v3_probe(struct device_node *vgic_node,
+int vgic_v3_probe(const struct gic_kvm_info *gic_kvm_info,
                  const struct vgic_ops **ops,
                  const struct vgic_params **params)
 {
        int ret = 0;
-       u32 gicv_idx;
-       struct resource vcpu_res;
        struct vgic_params *vgic = &vgic_v3_params;
+       const struct resource *vcpu_res = &gic_kvm_info->vcpu;
 
-       vgic->maint_irq = irq_of_parse_and_map(vgic_node, 0);
-       if (!vgic->maint_irq) {
-               kvm_err("error getting vgic maintenance irq from DT\n");
-               ret = -ENXIO;
-               goto out;
-       }
+       vgic->maint_irq = gic_kvm_info->maint_irq;
 
        ich_vtr_el2 = kvm_call_hyp(__vgic_v3_get_ich_vtr_el2);
 
        vgic->nr_lr = (ich_vtr_el2 & 0xf) + 1;
        vgic->can_emulate_gicv2 = false;
 
-       if (of_property_read_u32(vgic_node, "#redistributor-regions", &gicv_idx))
-               gicv_idx = 1;
-
-       gicv_idx += 3; /* Also skip GICD, GICC, GICH */
-       if (of_address_to_resource(vgic_node, gicv_idx, &vcpu_res)) {
+       if (!vcpu_res->start) {
                kvm_info("GICv3: no GICV resource entry\n");
                vgic->vcpu_base = 0;
-       } else if (!PAGE_ALIGNED(vcpu_res.start)) {
+       } else if (!PAGE_ALIGNED(vcpu_res->start)) {
                pr_warn("GICV physical address 0x%llx not page aligned\n",
-                       (unsigned long long)vcpu_res.start);
+                       (unsigned long long)vcpu_res->start);
                vgic->vcpu_base = 0;
-       } else if (!PAGE_ALIGNED(resource_size(&vcpu_res))) {
+       } else if (!PAGE_ALIGNED(resource_size(vcpu_res))) {
                pr_warn("GICV size 0x%llx not a multiple of page size 0x%lx\n",
-                       (unsigned long long)resource_size(&vcpu_res),
+                       (unsigned long long)resource_size(vcpu_res),
                        PAGE_SIZE);
-               vgic->vcpu_base = 0;
        } else {
-               vgic->vcpu_base = vcpu_res.start;
+               vgic->vcpu_base = vcpu_res->start;
                vgic->can_emulate_gicv2 = true;
                kvm_register_device_ops(&kvm_arm_vgic_v2_ops,
                                        KVM_DEV_TYPE_ARM_VGIC_V2);
        vgic->type = VGIC_V3;
        vgic->max_gic_vcpus = VGIC_V3_MAX_CPUS;
 
-       kvm_info("%s@%llx IRQ%d\n", vgic_node->name,
-                vcpu_res.start, vgic->maint_irq);
+       kvm_info("GICV base=0x%llx, IRQ=%d\n",
+                vgic->vcpu_base, vgic->maint_irq);
 
        on_each_cpu(vgic_cpu_init_lrs, vgic, 1);
 
        *ops = &vgic_v3_ops;
        *params = vgic;
 
-out:
-       of_node_put(vgic_node);
        return ret;
 }
 
 #include <linux/kvm_host.h>
 #include <linux/interrupt.h>
 #include <linux/io.h>
-#include <linux/of.h>
-#include <linux/of_address.h>
-#include <linux/of_irq.h>
+#include <linux/irq.h>
 #include <linux/rculist.h>
 #include <linux/uaccess.h>
 
 #include <trace/events/kvm.h>
 #include <asm/kvm.h>
 #include <kvm/iodev.h>
+#include <linux/irqchip/arm-gic-common.h>
 
 #define CREATE_TRACE_POINTS
 #include "trace.h"
        .notifier_call = vgic_cpu_notify,
 };
 
-static const struct of_device_id vgic_ids[] = {
-       { .compatible = "arm,cortex-a15-gic",   .data = vgic_v2_probe, },
-       { .compatible = "arm,cortex-a7-gic",    .data = vgic_v2_probe, },
-       { .compatible = "arm,gic-400",          .data = vgic_v2_probe, },
-       { .compatible = "arm,gic-v3",           .data = vgic_v3_probe, },
-       {},
-};
-
-int kvm_vgic_hyp_init(void)
+static int kvm_vgic_probe(void)
 {
-       const struct of_device_id *matched_id;
-       const int (*vgic_probe)(struct device_node *,const struct vgic_ops **,
-                               const struct vgic_params **);
-       struct device_node *vgic_node;
+       const struct gic_kvm_info *gic_kvm_info;
        int ret;
 
-       vgic_node = of_find_matching_node_and_match(NULL,
-                                                   vgic_ids, &matched_id);
-       if (!vgic_node) {
-               kvm_err("error: no compatible GIC node found\n");
+       gic_kvm_info = gic_get_kvm_info();
+       if (!gic_kvm_info)
                return -ENODEV;
+
+       switch (gic_kvm_info->type) {
+       case GIC_V2:
+               ret = vgic_v2_probe(gic_kvm_info, &vgic_ops, &vgic);
+               break;
+       case GIC_V3:
+               ret = vgic_v3_probe(gic_kvm_info, &vgic_ops, &vgic);
+               break;
+       default:
+               ret = -ENODEV;
        }
 
-       vgic_probe = matched_id->data;
-       ret = vgic_probe(vgic_node, &vgic_ops, &vgic);
-       if (ret)
+       return ret;
+}
+
+int kvm_vgic_hyp_init(void)
+{
+       int ret;
+
+       ret = kvm_vgic_probe();
+       if (ret) {
+               kvm_err("error: KVM vGIC probing failed\n");
                return ret;
+       }
 
        ret = request_percpu_irq(vgic->maint_irq, vgic_maintenance_handler,
                                 "vgic", kvm_get_running_vcpus());