Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-28 03:43:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve the test framework for MM syscalls

Browse Source
This commit is contained in:
Ruihan Li
2025-06-21 23:43:46 +08:00
committed by Jianfeng Jiang
parent 93015a6090
commit ee542e043e
4 changed files with 34 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
test/apps/mmap/mmap_and_mremap.c
View File

@ -1,48 +1,35 @@
// SPDX-License-Identifier: MPL-2.0 // SPDX-License-Identifier: MPL-2.0
#define _GNU_SOURCE #define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <sys/mman.h> #include <sys/mman.h>
#include <unistd.h> #include <unistd.h>
#include <string.h> #include <string.h>
#include <errno.h>
#include "../network/test.h" #include "../network/test.h"
#define PAGE_SIZE 4096 #define PAGE_SIZE 4096
const char *content = "kjfkljk*wigo&h"; const char *content = "kjfkljk*wigo&h";
void *x_mmap(void *addr, size_t length, int prot, int flags, int fd, #define CHECK_MM(func) CHECK_WITH(func, _ret != MAP_FAILED)
off_t offset)
{
void *result = mmap(addr, length, prot, flags, fd, offset);
if (result == MAP_FAILED) {
perror("mmap");
exit(EXIT_FAILURE);
}
return result;
}
FN_TEST(mmap_and_mremap) FN_TEST(mmap_and_mremap)
{ {
char *addr = x_mmap(NULL, 3 * PAGE_SIZE, PROT_READ | PROT_WRITE, char *addr = CHECK_MM(mmap(NULL, 3 * PAGE_SIZE, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0); MAP_PRIVATE | MAP_ANONYMOUS, -1, 0));
TEST_SUCC(munmap(addr, 3 * PAGE_SIZE)); TEST_SUCC(munmap(addr, 3 * PAGE_SIZE));
addr = x_mmap(addr, PAGE_SIZE, PROT_READ | PROT_WRITE, addr = CHECK_MM(mmap(addr, PAGE_SIZE, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED, -1, 0); MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED, -1, 0));
strcpy(addr, content); strcpy(addr, content);
char *addr2 = x_mmap(addr + 2 * PAGE_SIZE, PAGE_SIZE, char *addr2 = CHECK_MM(
PROT_READ | PROT_WRITE, mmap(addr + 2 * PAGE_SIZE, PAGE_SIZE, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED, -1, 0); MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED, -1, 0));
char *new_addr = mremap(addr, PAGE_SIZE, 3 * PAGE_SIZE, MREMAP_MAYMOVE); char *new_addr = CHECK_MM(
if (new_addr == MAP_FAILED) { mremap(addr, PAGE_SIZE, 3 * PAGE_SIZE, MREMAP_MAYMOVE));
perror("mremap");
exit(EXIT_FAILURE);
}
// The following operation (if uncommented) would cause a segmentation fault. // The following operation (if uncommented) would cause a segmentation fault.
// strcpy(addr, "Writing to old address"); // strcpy(addr, "Writing to old address");
@ -58,8 +45,8 @@ END_TEST()
FN_TEST(mmap_and_mremap_fixed) FN_TEST(mmap_and_mremap_fixed)
{ {
char *addr1 = x_mmap(NULL, PAGE_SIZE * 2, PROT_READ | PROT_WRITE, char *addr1 = CHECK_MM(mmap(NULL, PAGE_SIZE * 2, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0); MAP_PRIVATE | MAP_ANONYMOUS, -1, 0));
strcpy(addr1, content); strcpy(addr1, content);
// Unmap a target region to ensure we know it's free // Unmap a target region to ensure we know it's free
@ -67,19 +54,15 @@ FN_TEST(mmap_and_mremap_fixed)
TEST_SUCC(munmap(addr2, PAGE_SIZE)); // free it for mremap TEST_SUCC(munmap(addr2, PAGE_SIZE)); // free it for mremap
// Remap from the first address to the second address // Remap from the first address to the second address
if (mremap(addr1, PAGE_SIZE, PAGE_SIZE, MREMAP_MAYMOVE | MREMAP_FIXED, CHECK_WITH(mremap(addr1, PAGE_SIZE, PAGE_SIZE,
addr2) != addr2) { MREMAP_MAYMOVE | MREMAP_FIXED, addr2),
perror("mremap"); _ret == addr2);
exit(EXIT_FAILURE);
}
TEST_RES(strcmp(addr2, content), _ret == 0); TEST_RES(strcmp(addr2, content), _ret == 0);
// Remap from the second address to the first address // Remap from the second address to the first address
if (mremap(addr2, PAGE_SIZE, PAGE_SIZE, MREMAP_MAYMOVE | MREMAP_FIXED, CHECK_WITH(mremap(addr2, PAGE_SIZE, PAGE_SIZE,
addr1) != addr1) { MREMAP_MAYMOVE | MREMAP_FIXED, addr1),
perror("mremap"); _ret == addr1);
exit(EXIT_FAILURE);
}
TEST_RES(strcmp(addr1, content), _ret == 0); TEST_RES(strcmp(addr1, content), _ret == 0);
TEST_SUCC(munmap(addr1, PAGE_SIZE)); TEST_SUCC(munmap(addr1, PAGE_SIZE));

6
test/apps/mmap/mmap_readahead.c
View File

@ -22,9 +22,9 @@ FN_SETUP(mmap_readahead)
CHECK(ftruncate(fd, PAGE_SIZE * NR_PAGES)); CHECK(ftruncate(fd, PAGE_SIZE * NR_PAGES));
addr = mmap(NULL, PAGE_SIZE * NR_PAGES, PROT_READ | PROT_WRITE, addr = CHECK_WITH(mmap(NULL, PAGE_SIZE * NR_PAGES,
MAP_SHARED, fd, 0); PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0),
CHECK(addr == MAP_FAILED ? -1 : 0); _ret != MAP_FAILED);
} }
END_SETUP() END_SETUP()

23
test/apps/mmap/mmap_vmrss.c
View File

@ -47,7 +47,7 @@ long get_vm_rss_kb(rss_type type)
target_field = "VmRSS:"; target_field = "VmRSS:";
break; break;
default: default:
perror("Unknown rss_type\n"); fprintf(stderr, "Unknown rss_type\n");
exit(1); exit(1);
} }
@ -68,14 +68,12 @@ long get_vm_rss_kb(rss_type type)
return rss_kb; return rss_kb;
} }
#define CHECK_MM(func) CHECK_WITH(func, _ret != MAP_FAILED)
FN_TEST(rss_anon) FN_TEST(rss_anon)
{ {
void *mem = mmap(NULL, TOTAL_SIZE, PROT_READ | PROT_WRITE, void *mem = CHECK_MM(mmap(NULL, TOTAL_SIZE, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0); MAP_PRIVATE | MAP_ANONYMOUS, -1, 0));
if (mem == MAP_FAILED) {
perror("mmap");
exit(1);
}
// The first call to `TEST_SUCC` and `get_vm_rss_kb()` may trigger // The first call to `TEST_SUCC` and `get_vm_rss_kb()` may trigger
// lazy mapping of additional pages, such as shared libraries or files. // lazy mapping of additional pages, such as shared libraries or files.
@ -122,11 +120,8 @@ FN_TEST(rss_file)
long rss_file_before = TEST_SUCC(get_vm_rss_kb(file)); long rss_file_before = TEST_SUCC(get_vm_rss_kb(file));
long rss_before = TEST_SUCC(get_vm_rss_kb(total)); long rss_before = TEST_SUCC(get_vm_rss_kb(total));
void *mem = mmap(NULL, TOTAL_SIZE, PROT_READ, MAP_PRIVATE, fd, 0); void *mem =
if (mem == MAP_FAILED) { CHECK_MM(mmap(NULL, TOTAL_SIZE, PROT_READ, MAP_PRIVATE, fd, 0));
perror("mmap");
exit(1);
}
// Trigger page faults // Trigger page faults
for (int i = 0; i < NUM_PAGES; ++i) { for (int i = 0; i < NUM_PAGES; ++i) {
@ -146,7 +141,7 @@ FN_TEST(rss_file)
TEST_RES(get_vm_rss_kb(file), _ret == rss_file_before); TEST_RES(get_vm_rss_kb(file), _ret == rss_file_before);
TEST_RES(get_vm_rss_kb(total), _ret == rss_before); TEST_RES(get_vm_rss_kb(total), _ret == rss_before);
close(fd); TEST_SUCC(close(fd));
unlink(filename); TEST_SUCC(unlink(filename));
} }
END_TEST() END_TEST()

4
test/apps/network/test.h
View File

@ -47,7 +47,7 @@
#define __CHECK(func, cond) \ #define __CHECK(func, cond) \
errno = 0; \ errno = 0; \
long _ret = (func); \ __auto_type _ret = (func); \
if (!(cond)) { \ if (!(cond)) { \
fprintf(stderr, \ fprintf(stderr, \
"fatal error: %s: `" #cond "` is false after `" #func \ "fatal error: %s: `" #cond "` is false after `" #func \
@ -100,7 +100,7 @@ static int __total_failures;
#define __TEST(func, err, cond) \ #define __TEST(func, err, cond) \
errno = 0; \ errno = 0; \
long _ret = (func); \ __auto_type _ret = (func); \
if (errno != (err)) { \ if (errno != (err)) { \
__tests_failed++; \ __tests_failed++; \
fprintf(stderr, \ fprintf(stderr, \