@@ -123,12 +123,17 @@ test_misc_flags(void)
#define vdev "--vdev"
#define file_prefix "--file-prefix"
+#define FS_HUGETLB "hugetlbfs"
+
#define memtest "memtest"
#define memtest1 "memtest1"
#define memtest2 "memtest2"
#define SOCKET_MEM_STRLEN (RTE_MAX_NUMA_NODES * 20)
#define launch_proc(ARGV) process_dup(ARGV, RTE_DIM(ARGV), __func__)
+#ifdef RTE_EXEC_ENV_LINUX
+#include <mntent.h>
+
enum hugepage_action {
HUGEPAGE_CHECK_EXISTS = 0,
HUGEPAGE_CHECK_LOCKED,
@@ -136,24 +141,6 @@ enum hugepage_action {
HUGEPAGE_INVALID
};
-/* if string contains a hugepage path */
-static int
-get_hugepage_path(char * src, int src_len, char * dst, int dst_len)
-{
-#define NUM_TOKENS 4
- char *tokens[NUM_TOKENS];
-
- /* if we couldn't properly split the string */
- if (rte_strsplit(src, src_len, tokens, NUM_TOKENS, ' ') < NUM_TOKENS)
- return 0;
-
- if (strncmp(tokens[2], "hugetlbfs", sizeof("hugetlbfs")) == 0) {
- strlcpy(dst, tokens[1], dst_len);
- return 1;
- }
- return 0;
-}
-
/*
* Cycles through hugepage directories and looks for hugepage
* files associated with a given prefix. Depending on value of
@@ -165,45 +152,39 @@ get_hugepage_path(char * src, int src_len, char * dst, int dst_len)
* Returns -1 if it encounters an error
*/
static int
-process_hugefiles(const char * prefix, enum hugepage_action action)
+process_hugefiles(const char *prefix, enum hugepage_action action)
{
- FILE * hugedir_handle = NULL;
- DIR * hugepage_dir = NULL;
- struct dirent *dirent = NULL;
-
- char hugefile_prefix[PATH_MAX] = {0};
- char hugedir[PATH_MAX] = {0};
- char line[PATH_MAX] = {0};
-
- int fd, lck_result, result = 0;
+ const struct mntent *entry;
+ char hugefile_prefix[PATH_MAX];
+ int result = 0;
- const int prefix_len = snprintf(hugefile_prefix,
- sizeof(hugefile_prefix), "%smap_", prefix);
- if (prefix_len <= 0 || prefix_len >= (int)sizeof(hugefile_prefix)
- || prefix_len >= (int)sizeof(dirent->d_name)) {
+ const int prefix_len = snprintf(hugefile_prefix, sizeof(hugefile_prefix), "%smap_", prefix);
+ if (prefix_len <= 0 || prefix_len >= NAME_MAX) {
printf("Error creating hugefile filename prefix\n");
return -1;
}
/* get hugetlbfs mountpoints from /proc/mounts */
- hugedir_handle = fopen("/proc/mounts", "r");
-
- if (hugedir_handle == NULL) {
+ FILE *mounts = setmntent("/proc/mounts", "r");
+ if (mounts == NULL) {
printf("Error parsing /proc/mounts!\n");
return -1;
}
- /* read and parse script output */
- while (fgets(line, sizeof(line), hugedir_handle) != NULL) {
+ /* foreach mountpoint */
+ while ((entry = getmntent(mounts)) != NULL) {
+ DIR *hugepage_dir;
+ struct dirent *dirent;
- /* check if we have a hugepage filesystem path */
- if (!get_hugepage_path(line, sizeof(line), hugedir, sizeof(hugedir)))
+ /* only want hugetlbfs filesystems */
+ if (strcmp(entry->mnt_type, FS_HUGETLB) != 0)
continue;
/* check if directory exists */
- if ((hugepage_dir = opendir(hugedir)) == NULL) {
- fclose(hugedir_handle);
- printf("Error reading %s: %s\n", hugedir, strerror(errno));
+ hugepage_dir = opendir(entry->mnt_dir);
+ if (hugepage_dir == NULL) {
+ endmntent(mounts);
+ printf("Error reading %s: %s\n", entry->mnt_dir, strerror(errno));
return -1;
}
@@ -222,10 +203,10 @@ process_hugefiles(const char * prefix, enum hugepage_action action)
break;
case HUGEPAGE_DELETE:
{
- char file_path[PATH_MAX] = {0};
+ char file_path[PATH_MAX];
snprintf(file_path, sizeof(file_path),
- "%s/%s", hugedir, dirent->d_name);
+ "%s/%s", entry->mnt_dir, dirent->d_name);
/* remove file */
if (remove(file_path) < 0) {
@@ -240,6 +221,8 @@ process_hugefiles(const char * prefix, enum hugepage_action action)
break;
case HUGEPAGE_CHECK_LOCKED:
{
+ int fd;
+
/* try and lock the file */
fd = openat(dirfd(hugepage_dir), dirent->d_name, O_RDONLY);
@@ -253,10 +236,7 @@ process_hugefiles(const char * prefix, enum hugepage_action action)
}
/* non-blocking lock */
- lck_result = flock(fd, LOCK_EX | LOCK_NB);
-
- /* if lock succeeds, there's something wrong */
- if (lck_result != -1) {
+ if (flock(fd, LOCK_EX | LOCK_NB) != -1) {
result = 0;
/* unlock the resulting lock */
@@ -278,11 +258,10 @@ process_hugefiles(const char * prefix, enum hugepage_action action)
closedir(hugepage_dir);
} /* read /proc/mounts */
end:
- fclose(hugedir_handle);
+ endmntent(mounts);
return result;
}
-#ifdef RTE_EXEC_ENV_LINUX
/*
* count the number of "node*" files in /sys/devices/system/node/
*/
@@ -854,10 +833,10 @@ test_no_huge_flag(void)
static int
test_misc_flags(void)
{
- char hugepath[PATH_MAX] = {0};
- char hugepath_dir[PATH_MAX] = {0};
- char hugepath_dir2[PATH_MAX] = {0};
- char hugepath_dir3[PATH_MAX] = {0};
+ const char *hugepath = "";
+ char hugepath_dir[PATH_MAX];
+ char hugepath_dir2[PATH_MAX];
+ char hugepath_dir3[PATH_MAX];
#ifdef RTE_EXEC_ENV_FREEBSD
/* BSD target doesn't support prefixes at this point */
const char * prefix = "";
@@ -865,9 +844,7 @@ test_misc_flags(void)
#else
const char *prefix = file_prefix_arg();
const char * nosh_prefix = "--file-prefix=noshconf";
- FILE * hugedir_handle = NULL;
- char line[PATH_MAX] = {0};
- unsigned i, isempty = 1;
+ struct mntent *entry;
if (prefix == NULL)
return -1;
@@ -877,29 +854,24 @@ test_misc_flags(void)
*/
/* get hugetlbfs mountpoints from /proc/mounts */
- hugedir_handle = fopen("/proc/mounts", "r");
-
- if (hugedir_handle == NULL) {
+ FILE *mounts = setmntent("/proc/mounts", "r");
+ if (mounts == NULL) {
printf("Error opening /proc/mounts!\n");
return -1;
}
- /* read /proc/mounts */
- while (fgets(line, sizeof(line), hugedir_handle) != NULL) {
-
- /* find first valid hugepath */
- if (get_hugepage_path(line, sizeof(line), hugepath, sizeof(hugepath)))
+ /* foreach mount point */
+ hugepath = NULL;
+ while ((entry = getmntent(mounts)) != NULL) {
+ /* only want hugetlbfs filesystems */
+ if (strcmp(entry->mnt_type, FS_HUGETLB) == 0) {
+ hugepath = strdupa(entry->mnt_dir);
break;
+ }
}
+ endmntent(mounts);
- fclose(hugedir_handle);
-
- /* check if path is not empty */
- for (i = 0; i < sizeof(hugepath); i++)
- if (hugepath[i] != '\0')
- isempty = 0;
-
- if (isempty) {
+ if (hugepath == NULL) {
printf("No mounted hugepage dir found!\n");
return -1;
}
@@ -1168,6 +1140,17 @@ test_misc_flags(void)
return -1;
}
+#ifdef RTE_EXEC_ENV_FREEBSD
+
+static int
+test_file_prefix(void)
+{
+ printf("file_prefix not supported on FreeBSD, skipping test\n");
+ return TEST_SKIPPED;
+}
+
+#else
+
static int
test_file_prefix(void)
{
@@ -1186,16 +1169,12 @@ test_file_prefix(void)
* run in legacy mode, and not present at all after run in default
* mem mode
*/
- char prefix[PATH_MAX] = "";
+ char prefix[PATH_MAX];
-#ifdef RTE_EXEC_ENV_FREEBSD
- return 0;
-#else
if (get_current_prefix(prefix, sizeof(prefix)) == NULL) {
printf("Error - unable to get current prefix!\n");
return -1;
}
-#endif
/* this should fail unless the test itself is run with "memtest" prefix */
const char *argv0[] = {prgname, mp_flag, "-m",
@@ -1431,6 +1410,7 @@ test_file_prefix(void)
return 0;
}
+#endif
/* This function writes in passed buf pointer a valid --socket-mem= option
* for num_sockets then concatenates the provided suffix string.