pci: fix memory leaks and needless increment of map address
[dpdk.git] / lib / librte_eal / linuxapp / eal / eal_pci_uio.c
index c5e0cf3..d6a14b1 100644 (file)
@@ -32,6 +32,7 @@
  */
 
 #include <string.h>
+#include <unistd.h>
 #include <fcntl.h>
 #include <dirent.h>
 #include <sys/stat.h>
@@ -40,9 +41,9 @@
 
 #include <rte_log.h>
 #include <rte_pci.h>
+#include <rte_eal_memconfig.h>
 #include <rte_common.h>
 #include <rte_malloc.h>
-#include <rte_tailq.h>
 
 #include "rte_pci_dev_ids.h"
 #include "eal_filesystem.h"
 
 void *pci_map_addr = NULL;
 
+static struct rte_tailq_elem rte_uio_tailq = {
+       .name = "UIO_RESOURCE_LIST",
+};
+EAL_REGISTER_TAILQ(rte_uio_tailq)
 
 #define OFF_MAX              ((uint64_t)(off_t)-1)
 
@@ -87,8 +92,10 @@ pci_uio_map_secondary(struct rte_pci_device *dev)
 {
        int fd, i;
        struct mapped_pci_resource *uio_res;
+       struct mapped_pci_res_list *uio_res_list =
+                       RTE_TAILQ_CAST(rte_uio_tailq.head, mapped_pci_res_list);
 
-       TAILQ_FOREACH(uio_res, pci_res_list, next) {
+       TAILQ_FOREACH(uio_res, uio_res_list, next) {
 
                /* skip this element if it doesn't match our PCI address */
                if (rte_eal_compare_pci_addr(&uio_res->pci_addr, &dev->addr))
@@ -257,7 +264,7 @@ pci_get_uio_dev(struct rte_pci_device *dev, char *dstbuf,
 int
 pci_uio_map_resource(struct rte_pci_device *dev)
 {
-       int i, map_idx;
+       int i, map_idx = 0;
        char dirname[PATH_MAX];
        char cfgname[PATH_MAX];
        char devname[PATH_MAX]; /* contains the /dev/uioX */
@@ -265,7 +272,9 @@ pci_uio_map_resource(struct rte_pci_device *dev)
        int uio_num;
        uint64_t phaddr;
        struct rte_pci_addr *loc = &dev->addr;
-       struct mapped_pci_resource *uio_res;
+       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;
@@ -290,9 +299,8 @@ pci_uio_map_resource(struct rte_pci_device *dev)
        if (dev->intr_handle.fd < 0) {
                RTE_LOG(ERR, EAL, "Cannot open %s: %s\n",
                        devname, strerror(errno));
-               return -1;
+               goto error;
        }
-       dev->intr_handle.type = RTE_INTR_HANDLE_UIO;
 
        snprintf(cfgname, sizeof(cfgname),
                        "/sys/class/uio/uio%u/device/config", uio_num);
@@ -300,13 +308,19 @@ pci_uio_map_resource(struct rte_pci_device *dev)
        if (dev->intr_handle.uio_cfg_fd < 0) {
                RTE_LOG(ERR, EAL, "Cannot open %s: %s\n",
                        cfgname, strerror(errno));
-               return -1;
+               goto error;
        }
 
-       /* set bus master that is not done by uio_pci_generic */
-       if (pci_uio_set_bus_master(dev->intr_handle.uio_cfg_fd)) {
-               RTE_LOG(ERR, EAL, "Cannot set up bus mastering!\n");
-               return -1;
+       if (dev->kdrv == RTE_KDRV_IGB_UIO)
+               dev->intr_handle.type = RTE_INTR_HANDLE_UIO;
+       else {
+               dev->intr_handle.type = RTE_INTR_HANDLE_UIO_INTX;
+
+               /* set bus master that is not done by uio_pci_generic */
+               if (pci_uio_set_bus_master(dev->intr_handle.uio_cfg_fd)) {
+                       RTE_LOG(ERR, EAL, "Cannot set up bus mastering!\n");
+                       goto error;
+               }
        }
 
        /* allocate the mapping details for secondary processes*/
@@ -314,7 +328,7 @@ pci_uio_map_resource(struct rte_pci_device *dev)
        if (uio_res == NULL) {
                RTE_LOG(ERR, EAL,
                        "%s(): cannot store uio mmap details\n", __func__);
-               return -1;
+               goto error;
        }
 
        snprintf(uio_res->path, sizeof(uio_res->path), "%s", devname);
@@ -322,9 +336,8 @@ pci_uio_map_resource(struct rte_pci_device *dev)
 
        /* Map all BARs */
        maps = uio_res->maps;
-       for (i = 0, map_idx = 0; i != PCI_MAX_RESOURCE; i++) {
+       for (i = 0; i != PCI_MAX_RESOURCE; i++) {
                int fd;
-               int fail = 0;
 
                /* skip empty BAR */
                phaddr = dev->mem_resource[i].phys_addr;
@@ -338,6 +351,11 @@ pci_uio_map_resource(struct rte_pci_device *dev)
                                loc->domain, loc->bus, loc->devid, loc->function,
                                i);
 
+               /* allocate memory to keep path */
+               maps[map_idx].path = rte_malloc(NULL, strlen(devname) + 1, 0);
+               if (maps[map_idx].path == NULL)
+                       goto error;
+
                /*
                 * open resource file, to mmap it
                 */
@@ -345,7 +363,8 @@ pci_uio_map_resource(struct rte_pci_device *dev)
                if (fd < 0) {
                        RTE_LOG(ERR, EAL, "Cannot open %s: %s\n",
                                        devname, strerror(errno));
-                       return -1;
+                       rte_free(maps[map_idx].path);
+                       goto error;
                }
 
                /* try mapping somewhere close to the end of hugepages */
@@ -354,23 +373,15 @@ pci_uio_map_resource(struct rte_pci_device *dev)
 
                mapaddr = pci_map_resource(pci_map_addr, fd, 0,
                                (size_t)dev->mem_resource[i].len, 0);
-               if (mapaddr == MAP_FAILED)
-                       fail = 1;
+               close(fd);
+               if (mapaddr == MAP_FAILED) {
+                       rte_free(maps[map_idx].path);
+                       goto error;
+               }
 
                pci_map_addr = RTE_PTR_ADD(mapaddr,
                                (size_t)dev->mem_resource[i].len);
 
-               maps[map_idx].path = rte_malloc(NULL, strlen(devname) + 1, 0);
-               if (maps[map_idx].path == NULL)
-                       fail = 1;
-
-               if (fail) {
-                       rte_free(uio_res);
-                       close(fd);
-                       return -1;
-               }
-               close(fd);
-
                maps[map_idx].phaddr = dev->mem_resource[i].phys_addr;
                maps[map_idx].size = dev->mem_resource[i].len;
                maps[map_idx].addr = mapaddr;
@@ -382,7 +393,100 @@ pci_uio_map_resource(struct rte_pci_device *dev)
 
        uio_res->nb_maps = map_idx;
 
-       TAILQ_INSERT_TAIL(pci_res_list, uio_res, next);
+       TAILQ_INSERT_TAIL(uio_res_list, uio_res, next);
 
        return 0;
+
+error:
+       for (i = 0; i < map_idx; i++) {
+               pci_unmap_resource(uio_res->maps[i].addr,
+                               (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;
+       }
+       return -1;
+}
+
+#ifdef RTE_LIBRTE_EAL_HOTPLUG
+static void
+pci_uio_unmap(struct mapped_pci_resource *uio_res)
+{
+       int i;
+
+       if (uio_res == NULL)
+               return;
+
+       for (i = 0; i != uio_res->nb_maps; i++) {
+               pci_unmap_resource(uio_res->maps[i].addr,
+                               (size_t)uio_res->maps[i].size);
+               rte_free(uio_res->maps[i].path);
+       }
+}
+
+static struct mapped_pci_resource *
+pci_uio_find_resource(struct rte_pci_device *dev)
+{
+       struct mapped_pci_resource *uio_res;
+       struct mapped_pci_res_list *uio_res_list =
+                       RTE_TAILQ_CAST(rte_uio_tailq.head, mapped_pci_res_list);
+
+       if (dev == NULL)
+               return NULL;
+
+       TAILQ_FOREACH(uio_res, uio_res_list, next) {
+
+               /* skip this element if it doesn't match our PCI address */
+               if (!rte_eal_compare_pci_addr(&uio_res->pci_addr, &dev->addr))
+                       return uio_res;
+       }
+       return NULL;
+}
+
+/* unmap the PCI resource of a PCI device in virtual memory */
+void
+pci_uio_unmap_resource(struct rte_pci_device *dev)
+{
+       struct mapped_pci_resource *uio_res;
+       struct mapped_pci_res_list *uio_res_list =
+                       RTE_TAILQ_CAST(rte_uio_tailq.head, mapped_pci_res_list);
+
+       if (dev == NULL)
+               return;
+
+       /* find an entry for the device */
+       uio_res = pci_uio_find_resource(dev);
+       if (uio_res == NULL)
+               return;
+
+       /* secondary processes - just free maps */
+       if (rte_eal_process_type() != RTE_PROC_PRIMARY)
+               return pci_uio_unmap(uio_res);
+
+       TAILQ_REMOVE(uio_res_list, uio_res, next);
+
+       /* unmap all resources */
+       pci_uio_unmap(uio_res);
+
+       /* free uio resource */
+       rte_free(uio_res);
+
+       /* close fd if in primary process */
+       close(dev->intr_handle.fd);
+       dev->intr_handle.fd = -1;
+
+       /* close cfg_fd if in primary process */
+       close(dev->intr_handle.uio_cfg_fd);
+       dev->intr_handle.uio_cfg_fd = -1;
+
+       dev->intr_handle.type = RTE_INTR_HANDLE_UNKNOWN;
 }
+#endif /* RTE_LIBRTE_EAL_HOTPLUG */