This patch enables calculating space left in a hugetlbfs. There are three sources to get the information: 1. from sysfs; 2. from option size specified when mount; 3. use statfs. We should use the minimum one of these three sizes.
Signed-off-by: Jianfeng Tan <jianfeng.tan at intel.com> --- lib/librte_eal/linuxapp/eal/eal_hugepage_info.c | 85 ++++++++++++++++++++++++- 1 file changed, 84 insertions(+), 1 deletion(-) diff --git a/lib/librte_eal/linuxapp/eal/eal_hugepage_info.c b/lib/librte_eal/linuxapp/eal/eal_hugepage_info.c index 18858e2..6db8c33 100644 --- a/lib/librte_eal/linuxapp/eal/eal_hugepage_info.c +++ b/lib/librte_eal/linuxapp/eal/eal_hugepage_info.c @@ -44,6 +44,8 @@ #include <unistd.h> #include <errno.h> #include <sys/queue.h> +#include <sys/vfs.h> +#include <mntent.h> #include <rte_memory.h> #include <rte_memzone.h> @@ -189,6 +191,70 @@ get_hugepage_dir(uint64_t hugepage_sz) return retval; } +/* Caller to make sure this mnt_dir exist + */ +static uint64_t +get_hugetlbfs_mount_size(const char *mnt_dir) +{ + char *start, *end, *opt_size; + struct mntent *ent; + uint64_t size; + FILE *f; + int len; + + f = setmntent("/proc/mounts", "r"); + if (f == NULL) { + RTE_LOG(ERR, EAL, "setmntent() error: %s\n", + strerror(errno)); + return 0; + } + while (NULL != (ent = getmntent(f))) { + if (!strcmp(ent->mnt_dir, mnt_dir)) + break; + } + + start = hasmntopt(ent, "size"); + if (start == NULL) { + RTE_LOG(DEBUG, EAL, "option size not specified for %s\n", + mnt_dir); + size = 0; + goto end; + } + start += strlen("size="); + end = strstr(start, ","); + if (end != NULL) + len = end - start; + else + len = strlen(start); + opt_size = strndup(start, len); + size = rte_str_to_size(opt_size); + free(opt_size); + +end: + endmntent(f); + return size; +} + +/* Caller to make sure this mount has option size + * so that statfs is not zero. + */ +static uint64_t +get_hugetlbfs_free_size(const char *mnt_dir) +{ + int r; + struct statfs stats; + + r = statfs(mnt_dir, &stats); + if (r != 0) { + RTE_LOG(ERR, EAL, "statfs() error: %s\n", + strerror(errno)); + return 0; + } + + return stats.f_bfree * stats.f_bsize; +} + + /* * Clear the hugepage directory of whatever hugepage files * there are. Checks if the file is locked (i.e. @@ -329,9 +395,26 @@ eal_hugepage_info_init(void) if (clear_hugedir(hpi->hugedir) == -1) break; + /* there are three souces of how much space left in a + * hugetlbfs dir. + */ + uint64_t sz_left, sz_sysfs, sz_option, sz_statfs; + + sz_sysfs = get_num_hugepages(dirent->d_name) * + hpi->hugepage_sz; + sz_left = sz_sysfs; + sz_option = get_hugetlbfs_mount_size(hpi->hugedir); + if (sz_option) { + sz_statfs = get_hugetlbfs_free_size(hpi->hugedir); + sz_left = RTE_MIN(sz_sysfs, sz_statfs); + RTE_LOG(INFO, "sz_sysfs: %"PRIu64", sz_option: " + "%"PRIu64", sz_statfs: %"PRIu64"\n", + sz_sysfs, sz_option, sz_statfs); + } + /* for now, put all pages into socket 0, * later they will be sorted */ - hpi->num_pages[0] = get_num_hugepages(dirent->d_name); + hpi->num_pages[0] = sz_left / hpi->hugepage_sz; #ifndef RTE_ARCH_64 /* for 32-bit systems, limit number of hugepages to -- 2.1.4