Add test programs for ihk_os_getrusage()

2017-09-20 19:48:32 +09:00
parent 47b4bd5aba
commit 4d9a1628f2
63 changed files with 6461 additions and 0 deletions
--- a/test/rusage/rusage011.c
+++ b/test/rusage/rusage011.c
@ -0,0 +1,126 @@
+#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)
+
+int sz_mem[] = {
+	4 * (1ULL<<10),
+	2 * (1ULL<<20),
+	1 * (1ULL<<30),
+	134217728};
+
+#define SZ_INDEX 0
+
+int main(int argc, char** argv) {
+	void* mem;
+	int ret = 0;
+	pid_t pid;
+	int status;
+	key_t key = ftok(argv[0], 0);
+	int shmid;
+	xpmem_segid_t segid;
+
+	shmid = shmget(key, sz_mem[SZ_INDEX], IPC_CREAT | 0660);
+	CHKANDJUMP(shmid == -1, 255, "shmget failed: %s\n", strerror(errno));
+	
+	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]);
+	
+	pid = fork();
+	CHKANDJUMP(pid == -1, 255, "fork failed\n");
+	if(pid == 0) {
+		void *shm = shmat(shmid, NULL, 0);
+		CHKANDJUMP(shm == (void*)-1, 255, "shmat failed: %s\n", strerror(errno));
+
+		while((segid = *(xpmem_segid_t*)shm) == 0) { };
+
+		ret = shmdt(shm);
+		CHKANDJUMP(ret == -1, 255, "shmdt failed\n");
+
+		ret = xpmem_init();
+		CHKANDJUMP(ret != 0, 255, "xpmem_init failed: %s\n", strerror(errno));
+		
+		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 {
+		void *shm = shmat(shmid, NULL, 0);
+		CHKANDJUMP(mem == (void*)-1, 255, "shmat failed: %s\n", strerror(errno));
+
+		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));
+
+		*(xpmem_segid_t*)shm = segid;
+		
+		ret = waitpid(pid, &status, 0);
+		CHKANDJUMP(ret == -1, 255, "waitpid failed\n");
+
+		printf("%lx\n", *((unsigned long*)mem));
+
+		struct shmid_ds buf;
+		ret = shmctl(shmid, IPC_RMID, &buf);
+		CHKANDJUMP(ret == -1, 255, "shmctl failed\n");
+
+		ret = shmdt(shm);
+		CHKANDJUMP(ret == -1, 255, "shmdt failed\n");
+
+		ret = xpmem_remove(segid);
+		CHKANDJUMP(ret == -1, 255, "xpmem_remove failed\n");
+	}
+	
+ fn_exit:
+	return ret;
+ fn_fail:
+	goto fn_exit;
+}