pci: fix memory leaks and needless increment of map address
[dpdk.git] / lib / librte_eal / linuxapp / eal / eal_pci_uio.c
index 7dd9ce1..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)
-static int
-pci_uio_get_mappings(struct rte_pci_device *dev,
-                       struct pci_map maps[], int nb_maps)
-{
-       struct rte_pci_addr *loc = &dev->addr;
-       int i = 0;
-       char filename[PATH_MAX];
-       unsigned long long start_addr, end_addr, flags;
-       FILE *f;
-
-       snprintf(filename, sizeof(filename),
-               SYSFS_PCI_DEVICES "/" PCI_PRI_FMT "/resource",
-               loc->domain, loc->bus, loc->devid, loc->function);
-
-       f = fopen(filename, "r");
-       if (f == NULL) {
-               RTE_LOG(ERR, EAL,
-               "%s(): cannot open sysfs %s\n",
-               __func__, filename);
-               return -1;
-       }
-
-       while (fscanf(f, "%llx %llx %llx", &start_addr,
-                       &end_addr, &flags) == 3 && i < nb_maps) {
-               if (flags & IORESOURCE_MEM) {
-                       maps[i].offset = 0x0;
-                       maps[i].size = end_addr - start_addr + 1;
-                       maps[i].phaddr = start_addr;
-                       i++;
-               }
-       }
-       fclose(f);
-
-       return i;
-}
 
 static int
 pci_uio_set_bus_master(int dev_fd)
@@ -122,18 +92,16 @@ 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 (memcmp(&uio_res->pci_addr, &dev->addr, sizeof(dev->addr)))
+               if (rte_eal_compare_pci_addr(&uio_res->pci_addr, &dev->addr))
                        continue;
 
                for (i = 0; i != uio_res->nb_maps; i++) {
-                       /* ignore mappings unused in primary process */
-                       if (uio_res->maps[i].addr == NULL)
-                               continue;
-
                        /*
                         * open devname, to mmap it
                         */
@@ -144,13 +112,21 @@ pci_uio_map_secondary(struct rte_pci_device *dev)
                                return -1;
                        }
 
-                       if (pci_map_resource(uio_res->maps[i].addr, fd,
-                                            (off_t)uio_res->maps[i].offset,
-                                            (size_t)uio_res->maps[i].size, 0)
-                           != uio_res->maps[i].addr) {
-                               RTE_LOG(ERR, EAL,
-                                       "Cannot mmap device resource file: %s\n",
-                                       uio_res->maps[i].path);
+                       void *mapaddr = pci_map_resource(uio_res->maps[i].addr,
+                                       fd, (off_t)uio_res->maps[i].offset,
+                                       (size_t)uio_res->maps[i].size, 0);
+                       if (mapaddr != uio_res->maps[i].addr) {
+                               if (mapaddr == MAP_FAILED)
+                                       RTE_LOG(ERR, EAL,
+                                                       "Cannot mmap device resource file %s: %s\n",
+                                                       uio_res->maps[i].path,
+                                                       strerror(errno));
+                               else
+                                       RTE_LOG(ERR, EAL,
+                                                       "Cannot mmap device resource file %s to address: %p\n",
+                                                       uio_res->maps[i].path,
+                                                       uio_res->maps[i].addr);
+
                                close(fd);
                                return -1;
                        }
@@ -288,16 +264,17 @@ pci_get_uio_dev(struct rte_pci_device *dev, char *dstbuf,
 int
 pci_uio_map_resource(struct rte_pci_device *dev)
 {
-       int i, j;
+       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;
-       int nb_maps;
        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;
@@ -322,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);
@@ -332,14 +308,18 @@ 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 (rte_eal_process_type() == RTE_PROC_PRIMARY) {
+       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");
-                       return -1;
+                       goto error;
                }
        }
 
@@ -348,21 +328,12 @@ 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);
        memcpy(&uio_res->pci_addr, &dev->addr, sizeof(uio_res->pci_addr));
 
-       /* collect info about device mappings */
-       nb_maps = pci_uio_get_mappings(dev, uio_res->maps,
-                                       RTE_DIM(uio_res->maps));
-       if (nb_maps < 0) {
-               rte_free(uio_res);
-               return nb_maps;
-       }
-       uio_res->nb_maps = nb_maps;
-
        /* Map all BARs */
        maps = uio_res->maps;
        for (i = 0; i != PCI_MAX_RESOURCE; i++) {
@@ -373,65 +344,149 @@ pci_uio_map_resource(struct rte_pci_device *dev)
                if (phaddr == 0)
                        continue;
 
-               for (j = 0; j != nb_maps && (phaddr != maps[j].phaddr ||
-                               dev->mem_resource[i].len != maps[j].size);
-                               j++)
-                       ;
 
-               /* if matching map is found, then use it */
-               if (j != nb_maps) {
-                       int fail = 0;
-
-                       /* update devname for mmap  */
-                       snprintf(devname, sizeof(devname),
+               /* update devname for mmap  */
+               snprintf(devname, sizeof(devname),
                                SYSFS_PCI_DEVICES "/" PCI_PRI_FMT "/resource%d",
                                loc->domain, loc->bus, loc->devid, loc->function,
                                i);
 
-                       /*
-                        * open resource file, to mmap it
-                        */
-                       fd = open(devname, O_RDWR);
-                       if (fd < 0) {
-                               RTE_LOG(ERR, EAL, "Cannot open %s: %s\n",
+               /* 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
+                */
+               fd = open(devname, O_RDWR);
+               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;
+               }
 
-                       if (maps[j].addr != NULL)
-                               fail = 1;
-                       else {
-                               /* try mapping somewhere close to the end of hugepages */
-                               if (pci_map_addr == NULL)
-                                       pci_map_addr = pci_find_max_end_va();
+               /* try mapping somewhere close to the end of hugepages */
+               if (pci_map_addr == NULL)
+                       pci_map_addr = pci_find_max_end_va();
 
-                               mapaddr = pci_map_resource(pci_map_addr, fd, 0,
-                                               (size_t)maps[j].size, 0);
-                               if (mapaddr == MAP_FAILED)
-                                       fail = 1;
+               mapaddr = pci_map_resource(pci_map_addr, fd, 0,
+                               (size_t)dev->mem_resource[i].len, 0);
+               close(fd);
+               if (mapaddr == MAP_FAILED) {
+                       rte_free(maps[map_idx].path);
+                       goto error;
+               }
 
-                               pci_map_addr = RTE_PTR_ADD(mapaddr, (size_t) maps[j].size);
-                       }
+               pci_map_addr = RTE_PTR_ADD(mapaddr,
+                               (size_t)dev->mem_resource[i].len);
+
+               maps[map_idx].phaddr = dev->mem_resource[i].phys_addr;
+               maps[map_idx].size = dev->mem_resource[i].len;
+               maps[map_idx].addr = mapaddr;
+               maps[map_idx].offset = 0;
+               strcpy(maps[map_idx].path, devname);
+               map_idx++;
+               dev->mem_resource[i].addr = mapaddr;
+       }
 
-                       maps[j].path = rte_malloc(NULL, strlen(devname) + 1, 0);
-                       if (maps[j].path == NULL)
-                               fail = 1;
+       uio_res->nb_maps = map_idx;
 
-                       if (fail) {
-                               rte_free(uio_res);
-                               close(fd);
-                               return -1;
-                       }
-                       close(fd);
+       TAILQ_INSERT_TAIL(uio_res_list, uio_res, next);
 
-                       maps[j].addr = mapaddr;
-                       maps[j].offset = 0;
-                       strcpy(maps[j].path, devname);
-                       dev->mem_resource[i].addr = mapaddr;
-               }
+       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;
+}
 
-       TAILQ_INSERT_TAIL(pci_res_list, uio_res, next);
+#ifdef RTE_LIBRTE_EAL_HOTPLUG
+static void
+pci_uio_unmap(struct mapped_pci_resource *uio_res)
+{
+       int i;
 
-       return 0;
+       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 */