Add two new helpers to allocate an its itte and an its device.
This will avoid duplication on restore path.

Signed-off-by: Eric Auger <eric.au...@redhat.com>
---
 virt/kvm/arm/vgic/vgic-its.c | 71 ++++++++++++++++++++++++++++++--------------
 1 file changed, 48 insertions(+), 23 deletions(-)

diff --git a/virt/kvm/arm/vgic/vgic-its.c b/virt/kvm/arm/vgic/vgic-its.c
index aeb9d08..07bf180 100644
--- a/virt/kvm/arm/vgic/vgic-its.c
+++ b/virt/kvm/arm/vgic/vgic-its.c
@@ -732,6 +732,26 @@ static void vgic_its_free_collection(struct vgic_its *its, 
u32 coll_id)
        kfree(collection);
 }
 
+static int vgic_its_alloc_itte(struct its_device *device,
+                              struct its_itte **ittep,
+                              struct its_collection *collection,
+                              u32 lpi_id, u32 event_id)
+{
+       struct its_itte *itte;
+
+       itte = kzalloc(sizeof(itte), GFP_KERNEL);
+       if (!itte)
+               return -ENOMEM;
+
+       itte->event_id  = event_id;
+       itte->collection = collection;
+       itte->lpi = lpi_id;
+
+       list_add_tail(&itte->itte_list, &device->itt_head);
+       *ittep = itte;
+       return 0;
+}
+
 /*
  * The MAPTI and MAPI commands map LPIs to ITTEs.
  * Must be called with its_lock mutex held.
@@ -745,7 +765,7 @@ static int vgic_its_cmd_handle_mapi(struct kvm *kvm, struct 
vgic_its *its,
        struct its_itte *itte;
        struct its_device *device;
        struct its_collection *collection, *new_coll = NULL;
-       int lpi_nr;
+       int lpi_nr, ret;
        struct vgic_irq *irq;
 
        device = find_its_device(its, device_id);
@@ -772,19 +792,13 @@ static int vgic_its_cmd_handle_mapi(struct kvm *kvm, 
struct vgic_its *its,
                new_coll = collection;
        }
 
-       itte = kzalloc(sizeof(struct its_itte), GFP_KERNEL);
-       if (!itte) {
+       ret = vgic_its_alloc_itte(device, &itte, collection, lpi_nr, event_id);
+       if (ret) {
                if (new_coll)
                        vgic_its_free_collection(its, coll_id);
-               return -ENOMEM;
+               return ret;
        }
 
-       itte->event_id  = event_id;
-       list_add_tail(&itte->itte_list, &device->itt_head);
-
-       itte->collection = collection;
-       itte->lpi = lpi_nr;
-
        irq = vgic_add_lpi(kvm, lpi_nr);
        if (IS_ERR(irq)) {
                if (new_coll)
@@ -823,6 +837,29 @@ static void vgic_its_unmap_device(struct kvm *kvm, struct 
its_device *device)
        kfree(device);
 }
 
+static int vgic_its_alloc_device(struct vgic_its *its,
+                                struct its_device **devp,
+                                u32 device_id, gpa_t itt_addr_field,
+                                size_t size_field)
+{
+       struct its_device *device;
+
+       device = kzalloc(sizeof(*device), GFP_KERNEL);
+       if (!device)
+               return -ENOMEM;
+
+       device->device_id = device_id;
+       device->itt_addr = itt_addr_field << 8;
+       device->nb_eventid_bits = size_field + 1;
+       device->itt_size = ((2 << (size_field + 1)) - 1) * 16;
+       INIT_LIST_HEAD(&device->itt_head);
+
+       list_add_tail(&device->dev_list, &its->device_list);
+       *devp = device;
+
+       return 0;
+}
+
 /*
  * MAPD maps or unmaps a device ID to Interrupt Translation Tables (ITTs).
  * Must be called with the its_lock mutex held.
@@ -856,19 +893,7 @@ static int vgic_its_cmd_handle_mapd(struct kvm *kvm, 
struct vgic_its *its,
        if (!valid)
                return 0;
 
-       device = kzalloc(sizeof(struct its_device), GFP_KERNEL);
-       if (!device)
-               return -ENOMEM;
-
-       device->device_id = device_id;
-       device->itt_addr = itt_addr << 8;
-       device->nb_eventid_bits = size + 1;
-       device->itt_size = ((2 << (size + 1)) - 1) * 16;
-       INIT_LIST_HEAD(&device->itt_head);
-
-       list_add_tail(&device->dev_list, &its->device_list);
-
-       return 0;
+       return vgic_its_alloc_device(its, &device, device_id, itt_addr, size);
 }
 
 /*
-- 
2.5.5

_______________________________________________
kvmarm mailing list
kvmarm@lists.cs.columbia.edu
https://lists.cs.columbia.edu/mailman/listinfo/kvmarm

Reply via email to