-
Notifications
You must be signed in to change notification settings - Fork 52
Replace ashmem backend with memfd_create (runtime fallback) #21
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
base: master
Are you sure you want to change the base?
Changes from 9 commits
31e393d
48e2dc4
345ad9c
a408bf9
3819687
26196e6
4089679
fdaef26
2872df7
a7e23ad
aae4e89
6f02e69
4cb1de1
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 |
|---|---|---|
| @@ -1,24 +1,35 @@ | ||
| #define __ANDROID_UNAVAILABLE_SYMBOLS_ARE_WEAK__ 1 | ||
| #include <android/log.h> | ||
| #if __ANDROID_API__ >= 26 | ||
| #include <android/sharedmem.h> | ||
| #endif | ||
| #include <errno.h> | ||
| #include <pthread.h> | ||
| #include <stdbool.h> | ||
| #include <stdint.h> | ||
| #include <stdio.h> | ||
| #include <stdlib.h> | ||
| #include <string.h> | ||
| #define _GNU_SOURCE | ||
| #include <sys/mman.h> | ||
| #ifndef MFD_CLOEXEC | ||
| #define MFD_CLOEXEC 0x0001U | ||
| #endif | ||
| #ifndef MFD_ALLOW_SEALING | ||
| #define MFD_ALLOW_SEALING 0x0002U | ||
| #endif | ||
| #include <sys/socket.h> | ||
| #include <sys/stat.h> | ||
| #include <sys/syscall.h> | ||
| #include <linux/memfd.h> | ||
| #include <sys/un.h> | ||
| #include <unistd.h> | ||
| #include <paths.h> | ||
|
|
||
| #if __ANDROID_API__ < 26 | ||
| #ifndef __NR_memfd_create | ||
| #error "__NR_memfd_create not defined" | ||
| #endif | ||
|
|
||
| #define __u32 uint32_t | ||
| #include <linux/ashmem.h> | ||
| #endif | ||
|
|
||
| #include "shm.h" | ||
|
|
||
|
|
@@ -111,46 +122,72 @@ static int ancil_recv_fd(int sock) | |
| return ((int*) CMSG_DATA(cmsg))[0]; | ||
| } | ||
|
|
||
| static int ashmem_get_size_region(int fd) | ||
| { | ||
| #if __ANDROID_API__ >= 26 | ||
| return ASharedMemory_getSize(fd); | ||
| #else | ||
| return TEMP_FAILURE_RETRY(ioctl(fd, ASHMEM_GET_SIZE, NULL)); | ||
| #endif | ||
| } | ||
|
|
||
| /* | ||
| * ashmem_create_region - creates a new named ashmem region and returns the file | ||
| * descriptor, or <0 on error. | ||
| * Probe backends at runtime from newest/preferred to oldest: | ||
| * 1. memfd_create (Linux 3.17+, anonymous, no FS dependency) | ||
| * 2. ASharedMemory (Android API 26+, NDK ashmem wrapper) | ||
| * 3. /dev/ashmem (legacy, deprecated on some kernels) | ||
| * | ||
| * `name' is the label to give the region (visible in /proc/pid/maps) | ||
| * `size' is the size of the region, in page-aligned bytes | ||
| * On newer devices (Honor kernel 5.10.226), ashmem SET_SIZE returns | ||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. No need to specify your own device here.
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Done. |
||
| * ENOTTY even though /dev/ashmem exists. Runtime fallback handles this. | ||
| */ | ||
| static int ashmem_create_region(char const* name, size_t size) | ||
| static int shmem_create_region(char const* name, size_t size) | ||
| { | ||
| #if __ANDROID_API__ >= 26 | ||
| return ASharedMemory_create(name, size); | ||
| #else | ||
| // From https://android.googlesource.com/platform/system/core/+/master/libcutils/ashmem-dev.c | ||
| int fd = open("/dev/ashmem", O_RDWR); | ||
| int fd = -1; | ||
|
|
||
| // 1) memfd_create — Linux 3.17+, available on most Android 10+ devices | ||
| // Use direct syscall for NDK compatibility. | ||
| fd = syscall(__NR_memfd_create, name, MFD_CLOEXEC); | ||
| if (fd >= 0) { | ||
| if (ftruncate(fd, (off_t)size) == 0) | ||
| return fd; | ||
| close(fd); | ||
| } | ||
|
|
||
| // 2) ASharedMemory — Android 8+ (API 26+), weak symbol | ||
| if (ASharedMemory_create) { | ||
| fd = ASharedMemory_create(name, size); | ||
| if (fd >= 0) return fd; | ||
| } | ||
|
|
||
| // 3) /dev/ashmem — legacy, may return ENOTTY on newer kernels | ||
| fd = open("/dev/ashmem", O_RDWR); | ||
| if (fd < 0) return fd; | ||
|
|
||
| char name_buffer[ASHMEM_NAME_LEN] = {0}; | ||
| strncpy(name_buffer, name, sizeof(name_buffer)); | ||
| name_buffer[sizeof(name_buffer)-1] = 0; | ||
|
|
||
| int ret = ioctl(fd, ASHMEM_SET_NAME, name_buffer); | ||
| if (ret < 0) goto error; | ||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Still no ion/dma_heap fallback here.
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Done. Added dma-buf heap and ion. |
||
|
|
||
| ret = ioctl(fd, ASHMEM_SET_SIZE, size); | ||
| if (ret < 0) goto error; | ||
| if (ioctl(fd, ASHMEM_SET_NAME, name_buffer) < 0) | ||
| goto error; | ||
| if (ioctl(fd, ASHMEM_SET_SIZE, size) < 0) | ||
| goto error; | ||
|
|
||
| return fd; | ||
| error: | ||
| close(fd); | ||
| return ret; | ||
| #endif | ||
| { | ||
| int save_errno = errno; | ||
| close(fd); | ||
| errno = save_errno; | ||
| return -1; | ||
| } | ||
| } | ||
|
|
||
| static int shmem_get_size_region(int fd) | ||
| { | ||
| // Try fstat first — works for memfd and regular fds. | ||
| // ashmem fds also succeed fstat but st_size is always 0 | ||
| // (ashmem doesn't maintain i_size, only accessible via ioctl). | ||
| struct stat st; | ||
| if (fstat(fd, &st) == 0 && st.st_size > 0) | ||
| return (int)st.st_size; | ||
|
|
||
| // Fall back to backend-specific size query | ||
| if (ASharedMemory_getSize) { | ||
| int ret = ASharedMemory_getSize(fd); | ||
| if (ret > 0) return ret; | ||
| } | ||
| return TEMP_FAILURE_RETRY(ioctl(fd, ASHMEM_GET_SIZE, NULL)); | ||
| } | ||
|
|
||
| static void ashv_check_pid() | ||
|
|
@@ -275,9 +312,9 @@ static int ashv_read_remote_segment(int shmid) | |
| } | ||
| close(recvsock); | ||
|
|
||
| int size = ashmem_get_size_region(descriptor); | ||
| int size = shmem_get_size_region(descriptor); | ||
| if (size == 0 || size == -1) { | ||
| DBG ("%s: ERROR: ashmem_get_size_region() returned %d on socket %s: %s", __PRETTY_FUNCTION__, size, addr.sun_path + 1, strerror(errno)); | ||
| DBG ("%s: ERROR: shmem_get_size_region() returned %d on socket %s: %s", __PRETTY_FUNCTION__, size, addr.sun_path + 1, strerror(errno)); | ||
| return -1; | ||
| } | ||
|
|
||
|
|
@@ -399,14 +436,14 @@ int shmget(key_t key, size_t size, int flags) | |
| shmem = realloc(shmem, shmem_amount * sizeof(shmem_t)); | ||
| size = ROUND_UP(size, getpagesize()); | ||
| shmem[idx].size = size; | ||
| shmem[idx].descriptor = ashmem_create_region(buf, size); | ||
| shmem[idx].descriptor = shmem_create_region(buf, size); | ||
| shmem[idx].addr = NULL; | ||
| shmem[idx].id = shmid; | ||
| shmem[idx].markedForDeletion = false; | ||
| shmem[idx].key = key; | ||
|
|
||
| if (shmem[idx].descriptor < 0) { | ||
| DBG("%s: ashmem_create_region() failed for size %zu: %s", __PRETTY_FUNCTION__, size, strerror(errno)); | ||
| DBG("%s: shmem_create_region() failed for size %zu: %s", __PRETTY_FUNCTION__, size, strerror(errno)); | ||
| shmem_amount --; | ||
| shmem = realloc(shmem, shmem_amount * sizeof(shmem_t)); | ||
| pthread_mutex_unlock (&mutex); | ||
|
|
||
Uh oh!
There was an error while loading. Please reload this page.
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.
__NR_memfd_create is not restricted by ANDROID_API guard. It is not needed here. Probably
should be enough.
Which means there should not be custom
__NR_memfd_createdefinition here and ndk builtin is fine.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.
Done.
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.
MFD_*defines are not needed as well.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.
Done.