pci: fix memory leaks and needless increment of map address
[dpdk.git] / lib / librte_eal / linuxapp / eal / eal_pci_uio.c
index 2b16fcb..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,34 +92,41 @@ 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
                         */
-                       fd = open(uio_res->path, O_RDWR);
+                       fd = open(uio_res->maps[i].path, O_RDWR);
                        if (fd < 0) {
                                RTE_LOG(ERR, EAL, "Cannot open %s: %s\n",
-                                       uio_res->path, strerror(errno));
+                                       uio_res->maps[i].path, strerror(errno));
                                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)
-                           != uio_res->maps[i].addr) {
-                               RTE_LOG(ERR, EAL,
-                                       "Cannot mmap device resource\n");
+                       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;
                        }
@@ -287,18 +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;
-       uint64_t offset;
-       uint64_t pagesz;
-       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;
@@ -323,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);
@@ -333,19 +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;
        }
 
-       /* update devname for mmap  */
-       snprintf(devname, sizeof(devname),
-               SYSFS_PCI_DEVICES "/" PCI_PRI_FMT "/resource%d",
-               loc->domain, loc->bus, loc->devid, loc->function, 0);
+       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 (rte_eal_process_type() == RTE_PROC_PRIMARY) {
+               /* 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;
                }
        }
 
@@ -354,24 +328,13 @@ 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 */
-       pagesz = sysconf(_SC_PAGESIZE);
-
        maps = uio_res->maps;
        for (i = 0; i != PCI_MAX_RESOURCE; i++) {
                int fd;
@@ -381,55 +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;
-                       offset = j * pagesz;
+               /* 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 devname, 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;
-                       }
+               /* allocate memory to keep path */
+               maps[map_idx].path = rte_malloc(NULL, strlen(devname) + 1, 0);
+               if (maps[map_idx].path == NULL)
+                       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();
+               /*
+                * 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));
+                       rte_free(maps[map_idx].path);
+                       goto error;
+               }
 
-                               mapaddr = pci_map_resource(pci_map_addr, fd, (off_t)offset,
-                                               (size_t)maps[j].size);
-                               if (mapaddr == MAP_FAILED)
-                                       fail = 1;
+               /* try mapping somewhere close to the end of hugepages */
+               if (pci_map_addr == NULL)
+                       pci_map_addr = pci_find_max_end_va();
 
-                               pci_map_addr = RTE_PTR_ADD(mapaddr, (size_t) maps[j].size);
-                       }
+               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;
+               }
 
-                       if (fail) {
-                               rte_free(uio_res);
-                               close(fd);
-                               return -1;
-                       }
-                       close(fd);
+               pci_map_addr = RTE_PTR_ADD(mapaddr,
+                               (size_t)dev->mem_resource[i].len);
 
-                       maps[j].addr = mapaddr;
-                       maps[j].offset = offset;
-                       dev->mem_resource[i].addr = mapaddr;
-               }
+               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;
        }
 
-       TAILQ_INSERT_TAIL(pci_res_list, uio_res, next);
+       uio_res->nb_maps = map_idx;
+
+       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 */