Files
mckernel/test/mng_mod/issues/925/arm64/rusage012.c
Shiratori, Takehiro 739472bd86 Test "xpmem: support for fork()" on arm64
Change-Id: I12c628312157f35e239d3c5e67fa38adf156406b
Refs #925
2018-11-22 04:50:58 +00:00

108 lines
3.2 KiB
C

#include <unistd.h>
#include <stdio.h>
#include <sys/mman.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <xpmem.h>
#define DEBUG
#ifdef DEBUG
#define dprintf(...) \
do { \
char msg[1024]; \
sprintf(msg, __VA_ARGS__); \
fprintf(stderr, "%s,%s", __FUNCTION__, msg); \
} while (0);
#define eprintf(...) \
do { \
char msg[1024]; \
sprintf(msg, __VA_ARGS__); \
fprintf(stderr, "%s,%s", __FUNCTION__, msg); \
} while (0);
#else
#define dprintf(...) do { } while (0)
#define eprintf(...) do { } while (0)
#endif
#define CHKANDJUMP(cond, err, ...) \
do { \
if(cond) { \
eprintf(__VA_ARGS__); \
ret = err; \
goto fn_fail; \
} \
} while(0)
#define CONFIG_64K_PAGE
#ifdef CONFIG_64K_PAGE
int sz_mem[] = {
(1ULL << 16), /* 64KiB */
(1ULL << 25), /* 32MiB */
(1ULL << 29), /* 512MiB */
(1ULL << 27)}; /* 128MiB */
#else /* CONFIG_4K_PAGE */
int sz_mem[] = {
4 * (1ULL<<10), /* 4096(4KiB) */
2 * (1ULL<<20), /* 2097152(2MiB) */
1 * (1ULL<<30), /* 1073741824(1GiB) */
134217728}; /* 128MiB */
#endif
#define SZ_INDEX 0
int main(int argc, char** argv) {
void* mem;
int ret = 0;
pid_t pid;
int status;
xpmem_segid_t segid;
mem = mmap(0, sz_mem[SZ_INDEX], PROT_READ|PROT_WRITE, MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
CHKANDJUMP(mem == MAP_FAILED, 255, "mmap failed\n");
memset(mem, 0, sz_mem[SZ_INDEX]);
ret = xpmem_init();
CHKANDJUMP(ret != 0, 255, "xpmem_init failed: %s\n", strerror(errno));
segid = xpmem_make(mem, sz_mem[SZ_INDEX], XPMEM_PERMIT_MODE, (void*)0666);
CHKANDJUMP(segid == -1, 255, "xpmem_ioctl failed: %s\n", strerror(errno));
pid = fork();
CHKANDJUMP(pid == -1, 255, "fork failed\n");
if(pid == 0) {
xpmem_apid_t apid = xpmem_get(segid, XPMEM_RDWR, XPMEM_PERMIT_MODE, NULL);
CHKANDJUMP(apid == -1, 255, "xpmem_get failed: %s\n", strerror(errno));
struct xpmem_addr addr = { .apid = apid, .offset = 0 };
void* attach = xpmem_attach(addr, sz_mem[SZ_INDEX], NULL);
CHKANDJUMP(attach == (void*)-1, 255, "xpmem_attach failed: %s\n", strerror(errno));
*((unsigned long*)attach) = 0x1234;
ret = xpmem_detach(attach);
CHKANDJUMP(ret == -1, 255, "xpmem_detach failed\n");
_exit(123);
} else {
ret = waitpid(pid, &status, 0);
CHKANDJUMP(ret == -1, 255, "waitpid failed\n");
printf("%lx\n", *((unsigned long*)mem));
ret = xpmem_remove(segid);
CHKANDJUMP(ret == -1, 255, "xpmem_remove failed\n");
}
fn_exit:
return ret;
fn_fail:
goto fn_exit;
}