-
Notifications
You must be signed in to change notification settings - Fork 15.2k
[libc] Add Linux mman extension remap_file_pages. #110307
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 6 commits
50bf5e9
5a8891a
17d7626
cbc1c33
0caed2f
6cdfe53
db96b76
6a181ef
09f8671
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| //===------- Linux implementation of the remap_file_pages function --------===// | ||
| // | ||
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| // See https://llvm.org/LICENSE.txt for license information. | ||
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| // | ||
| //===----------------------------------------------------------------------===// | ||
|
|
||
| #include "src/sys/mman/remap_file_pages.h" | ||
|
|
||
| #include "src/__support/OSUtil/syscall.h" // For internal syscall function. | ||
| #include "src/__support/common.h" | ||
| #include "src/__support/macros/config.h" | ||
| #include "src/errno/libc_errno.h" | ||
| #include <sys/syscall.h> // For syscall numbers. | ||
|
|
||
| namespace LIBC_NAMESPACE_DECL { | ||
|
|
||
| LLVM_LIBC_FUNCTION(int, remap_file_pages, | ||
| (void *addr, size_t size, int prot, size_t pgoff, | ||
| int flags)) { | ||
| #ifdef SYS_remap_file_pages | ||
| int ret = LIBC_NAMESPACE::syscall_impl<int>(SYS_remap_file_pages, | ||
| reinterpret_cast<long>(addr), | ||
| size, prot, pgoff, flags); | ||
| #else | ||
| #error "remap_file_pages syscall is not available." | ||
| #endif | ||
|
|
||
| // A negative return value indicates an error with the magnitude of the | ||
| // value being the error code. | ||
| if (ret < 0) { | ||
| libc_errno = -ret; | ||
| return -1; | ||
| } | ||
|
|
||
| return 0; | ||
| } | ||
|
|
||
| } // namespace LIBC_NAMESPACE_DECL |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,22 @@ | ||
| //===-- Implementation header for remap_file_pages function -----*- C++ -*-===// | ||
| // | ||
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| // See https://llvm.org/LICENSE.txt for license information. | ||
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| // | ||
| //===----------------------------------------------------------------------===// | ||
|
|
||
| #ifndef LLVM_LIBC_SRC_SYS_MMAN_REMAP_FILE_PAGES_H | ||
| #define LLVM_LIBC_SRC_SYS_MMAN_REMAP_FILE_PAGES_H | ||
|
|
||
| #include "src/__support/macros/config.h" | ||
| #include <stddef.h> | ||
|
|
||
| namespace LIBC_NAMESPACE_DECL { | ||
|
|
||
| int remap_file_pages(void *addr, size_t size, int prot, size_t pgoff, | ||
| int flags); | ||
|
|
||
| } // namespace LIBC_NAMESPACE_DECL | ||
|
|
||
| #endif // LLVM_LIBC_SRC_SYS_MMAN_REMAP_FILE_PAGES_H |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,88 @@ | ||
| //===-- Unittests for remap_file_pages ------------------------------------===// | ||
| // | ||
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| // See https://llvm.org/LICENSE.txt for license information. | ||
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| // | ||
| //===----------------------------------------------------------------------===// | ||
|
|
||
| #include "src/errno/libc_errno.h" | ||
| #include "src/sys/mman/mmap.h" | ||
| #include "src/sys/mman/munmap.h" | ||
| #include "src/sys/mman/remap_file_pages.h" | ||
| #include "src/unistd/sysconf.h" | ||
| #include "test/UnitTest/ErrnoSetterMatcher.h" | ||
| #include "test/UnitTest/Test.h" | ||
| #include <fcntl.h> | ||
|
|
||
| #include <sys/mman.h> | ||
|
|
||
| using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Fails; | ||
| using LIBC_NAMESPACE::testing::ErrnoSetterMatcher::Succeeds; | ||
|
|
||
| TEST(LlvmLibcRemapFilePagesTest, NoError) { | ||
| size_t page_size = sysconf(_SC_PAGE_SIZE); | ||
| ASSERT_GT(page_size, size_t(0)); | ||
|
|
||
| // Create a file-backed mapping | ||
| int fd = open("/dev/zero", O_RDWR); | ||
|
||
| ASSERT_GT(fd, 0); | ||
|
|
||
| // First, allocate some memory using mmap | ||
| size_t alloc_size = 2 * page_size; | ||
| LIBC_NAMESPACE::libc_errno = 0; | ||
| void *addr = LIBC_NAMESPACE::mmap(nullptr, alloc_size, PROT_READ | PROT_WRITE, | ||
| MAP_SHARED, fd, 0); | ||
|
||
| ASSERT_ERRNO_SUCCESS(); | ||
| EXPECT_NE(addr, MAP_FAILED); | ||
|
|
||
| // Reset error number for the new function | ||
| LIBC_NAMESPACE::libc_errno = 0; | ||
|
||
|
|
||
| // Now try to remap the pages | ||
| EXPECT_THAT(LIBC_NAMESPACE::remap_file_pages(addr, page_size, 0, 1, 0), | ||
| Succeeds()); | ||
|
|
||
| // Reset error number for the new function | ||
| LIBC_NAMESPACE::libc_errno = 0; | ||
|
|
||
| // Clean up | ||
| EXPECT_THAT(LIBC_NAMESPACE::munmap(addr, alloc_size), Succeeds()); | ||
| } | ||
|
|
||
| TEST(LlvmLibcRemapFilePagesTest, ErrorInvalidFlags) { | ||
| size_t page_size = sysconf(_SC_PAGE_SIZE); | ||
| ASSERT_GT(page_size, size_t(0)); | ||
|
|
||
| // Create a file-backed mapping | ||
| int fd = open("/dev/zero", O_RDWR); | ||
| ASSERT_GT(fd, 0); | ||
|
||
|
|
||
| // First, allocate some memory using mmap | ||
| size_t alloc_size = 2 * page_size; | ||
| LIBC_NAMESPACE::libc_errno = 0; | ||
| void *addr = LIBC_NAMESPACE::mmap(nullptr, alloc_size, PROT_READ | PROT_WRITE, | ||
| MAP_SHARED, fd, 0); | ||
| ASSERT_ERRNO_SUCCESS(); | ||
| EXPECT_NE(addr, MAP_FAILED); | ||
|
|
||
| // Try to remap pages with an invalid flag MAP_PRIVATE | ||
| EXPECT_THAT(LIBC_NAMESPACE::remap_file_pages(addr, page_size, PROT_READ, 0, | ||
| MAP_PRIVATE), | ||
| Fails(EINVAL)); | ||
|
|
||
| // Clean up | ||
| EXPECT_THAT(LIBC_NAMESPACE::munmap(addr, page_size), Succeeds()); | ||
| } | ||
|
|
||
| TEST(LlvmLibcRemapFilePagesTest, ErrorInvalidAddress) { | ||
| size_t page_size = sysconf(_SC_PAGESIZE); | ||
| ASSERT_GT(page_size, size_t(0)); | ||
|
|
||
| // Use an address that we haven't mapped | ||
| void *invalid_addr = reinterpret_cast<void *>(0x12345000); | ||
|
|
||
| EXPECT_THAT(LIBC_NAMESPACE::remap_file_pages(invalid_addr, page_size, | ||
| PROT_READ, 0, 0), | ||
| Fails(EINVAL)); | ||
| } | ||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.