Commit cabc30da authored by Christophe Leroy's avatar Christophe Leroy Committed by Linus Torvalds

selftests/vm: fix map_hugetlb length used for testing read and write

Commit fa7b9a80 ("tools/selftest/vm: allow choosing mem size and page
size in map_hugetlb") added the possibility to change the size of memory
mapped for the test, but left the read and write test using the default
value.  This is unnoticed when mapping a length greater than the default
one, but segfaults otherwise.

Fix read_bytes() and write_bytes() by giving them the real length.

Also fix the call to munmap().

Fixes: fa7b9a80 ("tools/selftest/vm: allow choosing mem size and page size in map_hugetlb")
Signed-off-by: default avatarChristophe Leroy <christophe.leroy@c-s.fr>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
Reviewed-by: default avatarLeonardo Bras <leonardo@linux.ibm.com>
Cc: Michael Ellerman <mpe@ellerman.id.au>
Cc: Shuah Khan <shuah@kernel.org>
Cc: <stable@vger.kernel.org>
Link: http://lkml.kernel.org/r/9a404a13c871c4bd0ba9ede68f69a1225180dd7e.1580978385.git.christophe.leroy@c-s.frSigned-off-by: default avatarLinus Torvalds <torvalds@linux-foundation.org>
parent d4af73e3
...@@ -45,20 +45,20 @@ static void check_bytes(char *addr) ...@@ -45,20 +45,20 @@ static void check_bytes(char *addr)
printf("First hex is %x\n", *((unsigned int *)addr)); printf("First hex is %x\n", *((unsigned int *)addr));
} }
static void write_bytes(char *addr) static void write_bytes(char *addr, size_t length)
{ {
unsigned long i; unsigned long i;
for (i = 0; i < LENGTH; i++) for (i = 0; i < length; i++)
*(addr + i) = (char)i; *(addr + i) = (char)i;
} }
static int read_bytes(char *addr) static int read_bytes(char *addr, size_t length)
{ {
unsigned long i; unsigned long i;
check_bytes(addr); check_bytes(addr);
for (i = 0; i < LENGTH; i++) for (i = 0; i < length; i++)
if (*(addr + i) != (char)i) { if (*(addr + i) != (char)i) {
printf("Mismatch at %lu\n", i); printf("Mismatch at %lu\n", i);
return 1; return 1;
...@@ -96,11 +96,11 @@ int main(int argc, char **argv) ...@@ -96,11 +96,11 @@ int main(int argc, char **argv)
printf("Returned address is %p\n", addr); printf("Returned address is %p\n", addr);
check_bytes(addr); check_bytes(addr);
write_bytes(addr); write_bytes(addr, length);
ret = read_bytes(addr); ret = read_bytes(addr, length);
/* munmap() length of MAP_HUGETLB memory must be hugepage aligned */ /* munmap() length of MAP_HUGETLB memory must be hugepage aligned */
if (munmap(addr, LENGTH)) { if (munmap(addr, length)) {
perror("munmap"); perror("munmap");
exit(1); exit(1);
} }
......
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