pci: introduce functions to alloc and free uio resource

[dpdk.git] / lib / librte_eal / linuxapp / eal / eal_pci_uio.c

diff --git a/lib/librte_eal/linuxapp/eal/eal_pci_uio.c b/lib/librte_eal/linuxapp/eal/eal_pci_uio.c
index d8f61b5..de92ce5 100644 (file)
--- a/lib/librte_eal/linuxapp/eal/eal_pci_uio.c
+++ b/lib/librte_eal/linuxapp/eal/eal_pci_uio.c
@@ -252,30 +252,34 @@ pci_get_uio_dev(struct rte_pci_device *dev, char *dstbuf,
        return uio_num;
 }
 
-/* map the PCI resource of a PCI device in virtual memory */
-int
-pci_uio_map_resource(struct rte_pci_device *dev)
+static void
+pci_uio_free_resource(struct rte_pci_device *dev,
+               struct mapped_pci_resource *uio_res)
+{
+       rte_free(uio_res);
+
+       if (dev->intr_handle.uio_cfg_fd >= 0) {
+               close(dev->intr_handle.uio_cfg_fd);
+               dev->intr_handle.uio_cfg_fd = -1;
+       }
+       if (dev->intr_handle.fd) {
+               close(dev->intr_handle.fd);
+               dev->intr_handle.fd = -1;
+               dev->intr_handle.type = RTE_INTR_HANDLE_UNKNOWN;
+       }
+}
+
+static int
+pci_uio_alloc_resource(struct rte_pci_device *dev,
+               struct mapped_pci_resource **uio_res)
 {
-       int i, map_idx = 0;
        char dirname[PATH_MAX];
        char cfgname[PATH_MAX];
        char devname[PATH_MAX]; /* contains the /dev/uioX */
-       void *mapaddr;
        int uio_num;
-       uint64_t phaddr;
-       struct rte_pci_addr *loc = &dev->addr;
-       struct mapped_pci_resource *uio_res = NULL;
-       struct mapped_pci_res_list *uio_res_list =
-                       RTE_TAILQ_CAST(rte_uio_tailq.head, mapped_pci_res_list);
-       struct pci_map *maps;
+       struct rte_pci_addr *loc;
 
-       dev->intr_handle.fd = -1;
-       dev->intr_handle.uio_cfg_fd = -1;
-       dev->intr_handle.type = RTE_INTR_HANDLE_UNKNOWN;
-
-       /* secondary processes - use already recorded details */
-       if (rte_eal_process_type() != RTE_PROC_PRIMARY)
-               return pci_uio_map_secondary(dev);
+       loc = &dev->addr;
 
        /* find uio resource */
        uio_num = pci_get_uio_dev(dev, dirname, sizeof(dirname));
@@ -316,15 +320,49 @@ pci_uio_map_resource(struct rte_pci_device *dev)
        }
 
        /* allocate the mapping details for secondary processes*/
-       uio_res = rte_zmalloc("UIO_RES", sizeof(*uio_res), 0);
-       if (uio_res == NULL) {
+       *uio_res = rte_zmalloc("UIO_RES", sizeof(**uio_res), 0);
+       if (*uio_res == NULL) {
                RTE_LOG(ERR, EAL,
                        "%s(): cannot store uio mmap details\n", __func__);
                goto error;
        }
 
-       snprintf(uio_res->path, sizeof(uio_res->path), "%s", devname);
-       memcpy(&uio_res->pci_addr, &dev->addr, sizeof(uio_res->pci_addr));
+       snprintf((*uio_res)->path, sizeof((*uio_res)->path), "%s", devname);
+       memcpy(&(*uio_res)->pci_addr, &dev->addr, sizeof((*uio_res)->pci_addr));
+
+       return 0;
+
+error:
+       pci_uio_free_resource(dev, *uio_res);
+       return -1;
+}
+
+/* map the PCI resource of a PCI device in virtual memory */
+int
+pci_uio_map_resource(struct rte_pci_device *dev)
+{
+       int i, map_idx = 0, ret;
+       char devname[PATH_MAX]; /* contains the /dev/uioX */
+       void *mapaddr;
+       uint64_t phaddr;
+       struct rte_pci_addr *loc = &dev->addr;
+       struct mapped_pci_resource *uio_res = NULL;
+       struct mapped_pci_res_list *uio_res_list =
+               RTE_TAILQ_CAST(rte_uio_tailq.head, mapped_pci_res_list);
+       struct pci_map *maps;
+
+       dev->intr_handle.fd = -1;
+       dev->intr_handle.uio_cfg_fd = -1;
+       dev->intr_handle.type = RTE_INTR_HANDLE_UNKNOWN;
+
+       /* secondary processes - use already recorded details */
+       if (rte_eal_process_type() != RTE_PROC_PRIMARY)
+               return pci_uio_map_secondary(dev);
+
+       /* allocate uio resource */
+       ret = pci_uio_alloc_resource(dev, &uio_res);
+       if (ret)
+               return ret;
 
        /* Map all BARs */
        maps = uio_res->maps;
@@ -398,16 +436,7 @@ error:
                                (size_t)uio_res->maps[i].size);
                rte_free(maps[i].path);
        }
-       rte_free(uio_res);
-       if (dev->intr_handle.uio_cfg_fd >= 0) {
-               close(dev->intr_handle.uio_cfg_fd);
-               dev->intr_handle.uio_cfg_fd = -1;
-       }
-       if (dev->intr_handle.fd >= 0) {
-               close(dev->intr_handle.fd);
-               dev->intr_handle.fd = -1;
-               dev->intr_handle.type = RTE_INTR_HANDLE_UNKNOWN;
-       }
+       pci_uio_free_resource(dev, uio_res);
        return -1;
 }