Commit 57bcb57d authored by Sandipan Das's avatar Sandipan Das Committed by Linus Torvalds

selftests: vm: pkeys: use the correct huge page size

The huge page size can vary across architectures.  This will ensure that
the correct huge page size is used when accessing the hugetlb controls
under sysfs.  Instead of using a hardcoded page size (i.e.  2MB), this now
uses the HPAGE_SIZE macro which is arch-specific.
Signed-off-by: default avatarSandipan Das <sandipan@linux.ibm.com>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
Acked-by: default avatarDave Hansen <dave.hansen@intel.com>
Cc: "Desnes A. Nunes do Rosario" <desnesn@linux.vnet.ibm.com>
Cc: Florian Weimer <fweimer@redhat.com>
Cc: Ingo Molnar <mingo@kernel.org>
Cc: Ram Pai <linuxram@us.ibm.com>
Cc: Thiago Jung Bauermann <bauerman@linux.ibm.com>
Cc: "Aneesh Kumar K.V" <aneesh.kumar@linux.ibm.com>
Cc: Michael Ellerman <mpe@ellerman.id.au>
Cc: Michal Hocko <mhocko@kernel.org>
Cc: Michal Suchanek <msuchanek@suse.de>
Cc: Shuah Khan <shuah@kernel.org>
Link: http://lkml.kernel.org/r/66882a5d6e45c73c3a52bc4aef9754e48afa4f88.1585646528.git.sandipan@linux.ibm.comSigned-off-by: default avatarLinus Torvalds <torvalds@linux-foundation.org>
parent 6e373263
...@@ -739,12 +739,15 @@ void *malloc_pkey_anon_huge(long size, int prot, u16 pkey) ...@@ -739,12 +739,15 @@ void *malloc_pkey_anon_huge(long size, int prot, u16 pkey)
} }
int hugetlb_setup_ok; int hugetlb_setup_ok;
#define SYSFS_FMT_NR_HUGE_PAGES "/sys/kernel/mm/hugepages/hugepages-%ldkB/nr_hugepages"
#define GET_NR_HUGE_PAGES 10 #define GET_NR_HUGE_PAGES 10
void setup_hugetlbfs(void) void setup_hugetlbfs(void)
{ {
int err; int err;
int fd; int fd;
char buf[] = "123"; char buf[256];
long hpagesz_kb;
long hpagesz_mb;
if (geteuid() != 0) { if (geteuid() != 0) {
fprintf(stderr, "WARNING: not run as root, can not do hugetlb test\n"); fprintf(stderr, "WARNING: not run as root, can not do hugetlb test\n");
...@@ -755,11 +758,16 @@ void setup_hugetlbfs(void) ...@@ -755,11 +758,16 @@ void setup_hugetlbfs(void)
/* /*
* Now go make sure that we got the pages and that they * Now go make sure that we got the pages and that they
* are 2M pages. Someone might have made 1G the default. * are PMD-level pages. Someone might have made PUD-level
* pages the default.
*/ */
fd = open("/sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages", O_RDONLY); hpagesz_kb = HPAGE_SIZE / 1024;
hpagesz_mb = hpagesz_kb / 1024;
sprintf(buf, SYSFS_FMT_NR_HUGE_PAGES, hpagesz_kb);
fd = open(buf, O_RDONLY);
if (fd < 0) { if (fd < 0) {
perror("opening sysfs 2M hugetlb config"); fprintf(stderr, "opening sysfs %ldM hugetlb config: %s\n",
hpagesz_mb, strerror(errno));
return; return;
} }
...@@ -767,13 +775,14 @@ void setup_hugetlbfs(void) ...@@ -767,13 +775,14 @@ void setup_hugetlbfs(void)
err = read(fd, buf, sizeof(buf)-1); err = read(fd, buf, sizeof(buf)-1);
close(fd); close(fd);
if (err <= 0) { if (err <= 0) {
perror("reading sysfs 2M hugetlb config"); fprintf(stderr, "reading sysfs %ldM hugetlb config: %s\n",
hpagesz_mb, strerror(errno));
return; return;
} }
if (atoi(buf) != GET_NR_HUGE_PAGES) { if (atoi(buf) != GET_NR_HUGE_PAGES) {
fprintf(stderr, "could not confirm 2M pages, got: '%s' expected %d\n", fprintf(stderr, "could not confirm %ldM pages, got: '%s' expected %d\n",
buf, GET_NR_HUGE_PAGES); hpagesz_mb, buf, GET_NR_HUGE_PAGES);
return; return;
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment