85 lines
2.3 KiB
C
85 lines
2.3 KiB
C
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
#define _GNU_SOURCE
|
|
|
|
#include <sys/fcntl.h>
|
|
#include <sys/mman.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#include "../test.h"
|
|
|
|
#define PAGE_SIZE 4096
|
|
const char *filename = "testfile";
|
|
|
|
FN_TEST(mprotect_shared_writable_mapping_on_read_only_file)
|
|
{
|
|
int fd = TEST_SUCC(open(filename, O_RDWR | O_CREAT | O_TRUNC, 0600));
|
|
TEST_SUCC(ftruncate(fd, PAGE_SIZE));
|
|
TEST_SUCC(write(fd, "AAAA", 5));
|
|
|
|
TEST_SUCC(close(fd));
|
|
fd = TEST_SUCC(open(filename, O_RDONLY));
|
|
|
|
char *addr =
|
|
TEST_SUCC(mmap(NULL, PAGE_SIZE, PROT_READ, MAP_SHARED, fd, 0));
|
|
TEST_ERRNO(mprotect(addr, PAGE_SIZE, PROT_READ | PROT_WRITE), EACCES);
|
|
|
|
TEST_SUCC(munmap(addr, PAGE_SIZE));
|
|
TEST_SUCC(close(fd));
|
|
TEST_SUCC(unlink(filename));
|
|
}
|
|
END_TEST()
|
|
|
|
FN_TEST(mprotect_private_writable_mapping_copy_on_write)
|
|
{
|
|
int fd = TEST_SUCC(open(filename, O_RDWR | O_CREAT | O_TRUNC, 0600));
|
|
TEST_SUCC(ftruncate(fd, PAGE_SIZE));
|
|
TEST_SUCC(write(fd, "AAAA", 5));
|
|
|
|
char *addr1 =
|
|
TEST_SUCC(mmap(NULL, PAGE_SIZE, PROT_READ, MAP_PRIVATE, fd, 0));
|
|
char *addr2 =
|
|
TEST_SUCC(mmap(NULL, PAGE_SIZE, PROT_READ, MAP_PRIVATE, fd, 0));
|
|
TEST_RES(strcmp(addr1, "AAAA"), _ret == 0);
|
|
TEST_RES(strcmp(addr2, "AAAA"), _ret == 0);
|
|
TEST_SUCC(mprotect(addr1, PAGE_SIZE, PROT_READ | PROT_WRITE));
|
|
memcpy(addr1, "BBBB", 5);
|
|
TEST_RES(strcmp(addr1, "BBBB"), _ret == 0);
|
|
TEST_RES(strcmp(addr2, "AAAA"), _ret == 0);
|
|
|
|
TEST_SUCC(munmap(addr1, PAGE_SIZE));
|
|
TEST_SUCC(munmap(addr2, PAGE_SIZE));
|
|
TEST_SUCC(close(fd));
|
|
TEST_SUCC(unlink(filename));
|
|
}
|
|
END_TEST()
|
|
|
|
FN_TEST(mprotect_invalid_permission)
|
|
{
|
|
#define PROT_INVALID 0x20
|
|
|
|
void *addr = TEST_SUCC(mmap(NULL, PAGE_SIZE,
|
|
PROT_READ | PROT_WRITE | PROT_INVALID,
|
|
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0));
|
|
|
|
TEST_SUCC(mprotect(addr, PAGE_SIZE, PROT_READ | PROT_EXEC));
|
|
TEST_ERRNO(mprotect(addr, PAGE_SIZE, PROT_READ | PROT_INVALID), EINVAL);
|
|
|
|
TEST_SUCC(munmap(addr, PAGE_SIZE));
|
|
}
|
|
END_TEST()
|
|
|
|
FN_TEST(mprotect_merge_after_split_mapping)
|
|
{
|
|
void *addr = TEST_SUCC(mmap(NULL, 2 * PAGE_SIZE, PROT_READ | PROT_WRITE,
|
|
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0));
|
|
|
|
// Split the mapping into two.
|
|
TEST_SUCC(mprotect(addr, PAGE_SIZE, PROT_READ));
|
|
// After the first mapping is protected, the two mappings will merge.
|
|
TEST_SUCC(mprotect(addr, 2 * PAGE_SIZE, PROT_READ | PROT_WRITE));
|
|
|
|
TEST_SUCC(munmap(addr, 2 * PAGE_SIZE));
|
|
}
|
|
END_TEST() |