113 lines
2.4 KiB
C
113 lines
2.4 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>
|
|
#include <libgen.h>
|
|
#include "util.h"
|
|
|
|
#define BUFF_SIZE 1024
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
void *mem, *attach;
|
|
int rc = 0;
|
|
int status;
|
|
pid_t pid;
|
|
xpmem_segid_t segid;
|
|
xpmem_apid_t apid;
|
|
struct xpmem_addr addr;
|
|
key_t key = ftok(argv[0], 0);
|
|
int shmid;
|
|
|
|
printf("*** %s start ***\n", basename(argv[0]));
|
|
|
|
shmid = shmget(key, SZ_MEM, IPC_CREAT | 0660);
|
|
CHKANDJUMP(shmid == -1, "shmget");
|
|
|
|
mem = mmap(0, SZ_MEM, PROT_READ | PROT_WRITE,
|
|
MAP_ANONYMOUS | MAP_PRIVATE |
|
|
MAP_HUGETLB | (LARGE_PAGE_SHIFT << MAP_HUGE_SHIFT), -1, 0);
|
|
CHKANDJUMP(mem == NULL, "mmap");
|
|
memset(mem, 0, SZ_MEM);
|
|
|
|
fflush(0);
|
|
pid = fork();
|
|
CHKANDJUMP(pid == -1, "fork failed\n");
|
|
|
|
if (pid == 0) {
|
|
/* Child process */
|
|
void *shm = shmat(shmid, NULL, 0);
|
|
|
|
CHKANDJUMP(shm == (void *)-1, "shmat in child");
|
|
|
|
while ((segid = *(xpmem_segid_t *)shm) == 0) {
|
|
};
|
|
|
|
rc = shmdt(shm);
|
|
CHKANDJUMP(rc == -1, "shmdt");
|
|
|
|
rc = xpmem_init();
|
|
CHKANDJUMP(rc != 0, "xpmem_init in child");
|
|
|
|
apid = xpmem_get(segid, XPMEM_RDWR, XPMEM_PERMIT_MODE, NULL);
|
|
OKNG(apid == -1, "xpmem_get in child");
|
|
|
|
addr.apid = apid;
|
|
addr.offset = 0;
|
|
attach = xpmem_attach(addr, SZ_MEM, NULL);
|
|
OKNG(attach == (void *)-1, "xpmem_attach in child");
|
|
|
|
*((unsigned long *)attach) = TEST_VAL;
|
|
|
|
rc = xpmem_detach(attach);
|
|
OKNG(rc == -1, "xpmem_detach in child");
|
|
|
|
fflush(0);
|
|
_exit(0);
|
|
} else {
|
|
/* Parent process */
|
|
void *shm = shmat(shmid, NULL, 0);
|
|
struct shmid_ds buf;
|
|
|
|
CHKANDJUMP(shm == (void *)-1, "shmat in parent");
|
|
rc = xpmem_init();
|
|
CHKANDJUMP(rc != 0, "xpmem_init");
|
|
|
|
segid = xpmem_make(mem, SZ_MEM, XPMEM_PERMIT_MODE,
|
|
(void *)0666);
|
|
OKNG(segid == -1, "xpmem_make");
|
|
|
|
*(xpmem_segid_t *)shm = segid;
|
|
|
|
rc = waitpid(pid, &status, 0);
|
|
CHKANDJUMP(rc == -1, "waitpid failed\n");
|
|
|
|
OKNG(*((unsigned long *)mem) != TEST_VAL, "validate TEST_VAL");
|
|
|
|
rc = shmctl(shmid, IPC_RMID, &buf);
|
|
CHKANDJUMP(rc == -1, "shmctl");
|
|
|
|
rc = shmdt(shm);
|
|
CHKANDJUMP(rc == -1, "shmdt");
|
|
|
|
rc = xpmem_remove(segid);
|
|
OKNG(rc == -1, "xpmem_remove");
|
|
}
|
|
|
|
printf("*** %s PASSED\n\n", basename(argv[0]));
|
|
return 0;
|
|
|
|
fn_fail:
|
|
printf("*** %s FAILED\n\n", basename(argv[0]));
|
|
|
|
return -1;
|
|
}
|