mem: make use of memory hotplug for init
[dpdk.git] / lib / librte_eal / linuxapp / eal / eal_hugepage_info.c
index 7a21e8f..2e0819f 100644 (file)
@@ -1,34 +1,5 @@
-/*-
- *   BSD LICENSE
- *
- *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
- *   All rights reserved.
- *
- *   Redistribution and use in source and binary forms, with or without
- *   modification, are permitted provided that the following conditions
- *   are met:
- *
- *     * Redistributions of source code must retain the above copyright
- *       notice, this list of conditions and the following disclaimer.
- *     * Redistributions in binary form must reproduce the above copyright
- *       notice, this list of conditions and the following disclaimer in
- *       the documentation and/or other materials provided with the
- *       distribution.
- *     * Neither the name of Intel Corporation nor the names of its
- *       contributors may be used to endorse or promote products derived
- *       from this software without specific prior written permission.
- *
- *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2010-2014 Intel Corporation
  */
 
 #include <string.h>
@@ -44,9 +15,9 @@
 #include <unistd.h>
 #include <errno.h>
 #include <sys/queue.h>
+#include <sys/stat.h>
 
 #include <rte_memory.h>
-#include <rte_memzone.h>
 #include <rte_eal.h>
 #include <rte_launch.h>
 #include <rte_per_lcore.h>
@@ -60,6 +31,7 @@
 #include "eal_filesystem.h"
 
 static const char sys_dir_path[] = "/sys/kernel/mm/hugepages";
+static const char sys_pages_numa_dir_path[] = "/sys/devices/system/node";
 
 /* this function is only called from eal_hugepage_info_init which itself
  * is only called from a primary process */
@@ -100,6 +72,45 @@ get_num_hugepages(const char *subdir)
        return num_pages;
 }
 
+static uint32_t
+get_num_hugepages_on_node(const char *subdir, unsigned int socket)
+{
+       char path[PATH_MAX], socketpath[PATH_MAX];
+       DIR *socketdir;
+       unsigned long num_pages = 0;
+       const char *nr_hp_file = "free_hugepages";
+
+       snprintf(socketpath, sizeof(socketpath), "%s/node%u/hugepages",
+               sys_pages_numa_dir_path, socket);
+
+       socketdir = opendir(socketpath);
+       if (socketdir) {
+               /* Keep calm and carry on */
+               closedir(socketdir);
+       } else {
+               /* Can't find socket dir, so ignore it */
+               return 0;
+       }
+
+       snprintf(path, sizeof(path), "%s/%s/%s",
+                       socketpath, subdir, nr_hp_file);
+       if (eal_parse_sysfs_value(path, &num_pages) < 0)
+               return 0;
+
+       if (num_pages == 0)
+               RTE_LOG(WARNING, EAL, "No free hugepages reported in %s\n",
+                               subdir);
+
+       /*
+        * we want to return a uint32_t and more than this looks suspicious
+        * anyway ...
+        */
+       if (num_pages > UINT32_MAX)
+               num_pages = UINT32_MAX;
+
+       return num_pages;
+}
+
 static uint64_t
 get_default_hp_size(void)
 {
@@ -189,6 +200,18 @@ get_hugepage_dir(uint64_t hugepage_sz)
        return retval;
 }
 
+/*
+ * uses fstat to report the size of a file on disk
+ */
+static off_t
+get_file_size(int fd)
+{
+       struct stat st;
+       if (fstat(fd, &st) < 0)
+               return 0;
+       return st.st_size;
+}
+
 /*
  * Clear the hugepage directory of whatever hugepage files
  * there are. Checks if the file is locked (i.e.
@@ -219,6 +242,8 @@ clear_hugedir(const char * hugedir)
        }
 
        while(dirent != NULL){
+               struct flock lck = {0};
+
                /* skip files that don't match the hugepage pattern */
                if (fnmatch(filter, dirent->d_name, 0) > 0) {
                        dirent = readdir(dir);
@@ -235,11 +260,17 @@ clear_hugedir(const char * hugedir)
                }
 
                /* non-blocking lock */
-               lck_result = flock(fd, LOCK_EX | LOCK_NB);
+               lck.l_type = F_RDLCK;
+               lck.l_whence = SEEK_SET;
+               lck.l_start = 0;
+               lck.l_len = get_file_size(fd);
+
+               lck_result = fcntl(fd, F_SETLK, &lck);
 
                /* if lock succeeds, unlock and remove the file */
                if (lck_result != -1) {
-                       flock(fd, LOCK_UN);
+                       lck.l_type = F_UNLCK;
+                       fcntl(fd, F_SETLK, &lck);
                        unlinkat(dir_fd, dirent->d_name, 0);
                }
                close (fd);
@@ -278,7 +309,7 @@ eal_hugepage_info_init(void)
 {
        const char dirent_start_text[] = "hugepages-";
        const size_t dirent_start_len = sizeof(dirent_start_text) - 1;
-       unsigned i, num_sizes = 0;
+       unsigned int i, total_pages, num_sizes = 0;
        DIR *dir;
        struct dirent *dirent;
 
@@ -332,9 +363,28 @@ eal_hugepage_info_init(void)
                if (clear_hugedir(hpi->hugedir) == -1)
                        break;
 
-               /* for now, put all pages into socket 0,
-                * later they will be sorted */
-               hpi->num_pages[0] = get_num_hugepages(dirent->d_name);
+               /*
+                * first, try to put all hugepages into relevant sockets, but
+                * if first attempts fails, fall back to collecting all pages
+                * in one socket and sorting them later
+                */
+               total_pages = 0;
+               /* we also don't want to do this for legacy init */
+               if (!internal_config.legacy_mem)
+                       for (i = 0; i < rte_socket_count(); i++) {
+                               int socket = rte_socket_id_by_idx(i);
+                               unsigned int num_pages =
+                                               get_num_hugepages_on_node(
+                                                       dirent->d_name, socket);
+                               hpi->num_pages[socket] = num_pages;
+                               total_pages += num_pages;
+                       }
+               /*
+                * we failed to sort memory from the get go, so fall
+                * back to old way
+                */
+               if (total_pages == 0)
+                       hpi->num_pages[0] = get_num_hugepages(dirent->d_name);
 
 #ifndef RTE_ARCH_64
                /* for 32-bit systems, limit number of hugepages to
@@ -358,10 +408,19 @@ eal_hugepage_info_init(void)
              sizeof(internal_config.hugepage_info[0]), compare_hpi);
 
        /* now we have all info, check we have at least one valid size */
-       for (i = 0; i < num_sizes; i++)
+       for (i = 0; i < num_sizes; i++) {
+               /* pages may no longer all be on socket 0, so check all */
+               unsigned int j, num_pages = 0;
+
+               for (j = 0; j < RTE_MAX_NUMA_NODES; j++) {
+                       struct hugepage_info *hpi =
+                                       &internal_config.hugepage_info[i];
+                       num_pages += hpi->num_pages[j];
+               }
                if (internal_config.hugepage_info[i].hugedir != NULL &&
-                   internal_config.hugepage_info[i].num_pages[0] > 0)
+                               num_pages > 0)
                        return 0;
+       }
 
        /* no valid hugepage mounts available, return error */
        return -1;