^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) #include <sys/mman.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) #include <sys/time.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) #include <sys/resource.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) #ifndef MCL_ONFAULT
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) #define MCL_ONFAULT (MCL_FUTURE << 1)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) #endif
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) static int test_limit(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) int ret = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) struct rlimit lims;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) void *map;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) if (getrlimit(RLIMIT_MEMLOCK, &lims)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) perror("getrlimit");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24) if (mlockall(MCL_ONFAULT | MCL_FUTURE)) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) perror("mlockall");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) map = mmap(NULL, 2 * lims.rlim_max, PROT_READ | PROT_WRITE,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) MAP_PRIVATE | MAP_ANONYMOUS | MAP_POPULATE, -1, 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) if (map != MAP_FAILED)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) printf("mmap should have failed, but didn't\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) else {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) ret = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) munmap(map, 2 * lims.rlim_max);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) munlockall();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) int main(int argc, char **argv)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) int ret = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) ret += test_limit();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) }