Test "Wait for LWK to run at shutdown." on arm64

Change-Id: I96785dda7a1a7eb36ceeb31401d71b4e40efb185
Refs: #898
Refs: #928

test/mng_mod/issues/898_928/arm64/CT_005.c

@@ -0,0 +1,170 @@
+/* CT_005.c COPYRIGHT FUJITSU LIMITED 2018 */
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <ihklib.h>
+#include <sys/types.h>
+#include "mck_bps_conflict.h"
+#include "ct_okng.h"
+
+static char prefix[256] = MCK_DIR;
+
+static char test_name[64] = "CT_005";
+
+int main(int argc, char** argv) {
+    int ret = 0, status, ret_ihklib, pid;
+	FILE *fp;
+	char buf[65536];
+	size_t nread;
+
+	char cmd[1024];
+	char fn[256];
+	char kargs[256];
+
+	int cpus[4] = {6, 7, 8, 9};
+	int num_cpus = 4;
+
+	struct ihk_mem_chunk mem_chunks[4];
+	int num_mem_chunks;
+
+	printf("*** %s start *************************\n", test_name);
+	fflush(stdout);
+	/*--------------------------------------------
+	 * Preparing                                  
+	 *--------------------------------------------*/
+	sprintf(cmd, "%s/sbin/mcstop+release.sh", prefix);
+	status = system(cmd);
+
+	// ihk_os_destroy_pseudofs
+	ret_ihklib = ihk_os_destroy_pseudofs(0, 0, 0);
+	fp = popen("cat /proc/mounts | grep /tmp/mcos/mcos0_sys", "r");
+	nread = fread(buf, 1, sizeof(buf), fp);
+	buf[nread] = 0;
+
+	sprintf(cmd, "insmod %s/kmod/ihk.ko", prefix);
+	status = system(cmd);
+
+	sprintf(cmd, "insmod %s/kmod/%s %s", prefix, PART_MOD_NAME, PART_MOD_NAME);
+	status = system(cmd);
+
+	sprintf(cmd, "insmod %s/kmod/mcctrl.ko", prefix);
+	status = system(cmd);
+
+	/*--------------------------------------------
+	 * Test                                  
+	 *--------------------------------------------*/
+	// create 0
+    ret_ihklib = ihk_create_os(0);
+
+	// reserve cpus
+	ret_ihklib = ihk_reserve_cpu(0, cpus, num_cpus);
+	//OKNG(ret_ihklib == 0, "ihk_reserve_cpu\n");
+
+	// assign cpus
+    ret_ihklib = ihk_os_assign_cpu(0, cpus, num_cpus);
+    //OKNG(ret_ihklib == 0, "ihk_os_assign_cpu\n");
+
+	// reserve mem 128m@0,128m@1
+	num_mem_chunks = 2;
+	mem_chunks[0].size = 128*1024*1024ULL;
+	mem_chunks[0].numa_node_number = 0;
+	mem_chunks[1].size = 128*1024*1024ULL;
+	mem_chunks[1].numa_node_number = 1;
+    ret_ihklib = ihk_reserve_mem(0, mem_chunks, num_mem_chunks);
+    //OKNG(ret_ihklib == 0, "ihk_reserve_mem (2)\n");
+
+	// assign mem 128m@0,128m@1
+	num_mem_chunks = 2;
+	mem_chunks[0].size = 128*1024*1024ULL;
+	mem_chunks[0].numa_node_number = 0;
+	mem_chunks[1].size = 128*1024*1024ULL;
+	mem_chunks[1].numa_node_number = 1;
+    ret_ihklib = ihk_os_assign_mem(0, mem_chunks, num_mem_chunks);
+    //OKNG(ret_ihklib == 0, "ihk_os_assign_mem (2)\n");
+
+	// load
+	sprintf(fn, "%s/%s/kernel/mckernel.img", prefix, TARGET);
+    ret_ihklib = ihk_os_load(0, fn);
+	//OKNG(ret_ihklib == 0, "ihk_os_load\n");
+
+	// kargs
+	sprintf(kargs, "hidos ksyslogd=0");
+    ret_ihklib = ihk_os_kargs(0, kargs);
+	//OKNG(ret_ihklib == 0, "ihk_os_kargs\n");
+
+	// boot
+    ret_ihklib = ihk_os_boot(0);
+	OKNG(ret_ihklib == 0, "ihk_os_boot\n");
+
+	/* Make sure that all initialization related transactions between McKernel and IHK finish
+	   sysfs_init(void) (in mckernel/kernel/sysfs.c)
+	   packet.msg = SCD_MSG_SYSFS_REQ_SETUP;
+	   sysfsm_work_main() in (mckernel/executer/kernel/mcctrl/sysfs.c)
+       sysfsm_req_setup
+       sysfsm_setup */
+	usleep(100*1000);
+
+	// create pseudofs
+	ret_ihklib = ihk_os_create_pseudofs(0, 0, 0);
+	fp = popen("cat /proc/mounts | grep /tmp/mcos/mcos0_sys", "r");
+	nread = fread(buf, 1, sizeof(buf), fp);
+	buf[nread] = 0;
+	//OKNG(ret_ihklib == 0 &&
+	//	 strstr(buf, "/tmp/mcos/mcos0_sys") != NULL, "ihk_os_create_pseudofs()\n");
+
+	// mcexec
+	pid = fork();
+	if (pid == 0) {
+		printf("  start long mcexec...\n");
+		fflush(stdout);
+		sprintf(cmd, "%s/bin/mcexec sleep 5", prefix);
+    	fp = popen(cmd, "r");
+		nread = fread(buf, 1, sizeof(buf), fp);
+		return 0;
+	}
+	sleep(2);
+
+	// shutdown
+shutdown:
+    ret_ihklib = ihk_os_shutdown(0);
+	OKNG(ret_ihklib == 0, "shutdown during mcexec returned 0\n");
+	printf(" (But, mcexec process remain due to #846)\n");
+	fflush(stdout);
+goto done_test;
+
+	// get status. Note that the smp_ihk_os_shutdown() transitions 
+	// smp-x86 status to BUILTIN_OS_STATUS_SHUTDOWN
+	// and smp_ihk_os_query_status() transitions os status to IHK_OS_STATUS_NOT_BOOTED.
+	ret_ihklib = ihk_os_get_status(0);
+	//OKNG(ret_ihklib == IHK_STATUS_SHUTDOWN ||
+	//	 ret_ihklib == IHK_STATUS_INACTIVE, "ihk_os_get_status (5) returned %d\n", ret_ihklib);
+
+destroy:
+    ret_ihklib = ihk_destroy_os(0, 0);
+	//OKNG(ret_ihklib == 0, "destroy immediately after boot\n");
+
+	sprintf(cmd, "rmmod %s/kmod/mcctrl.ko", prefix);
+	status = system(cmd);
+
+	// destroy pseudofs
+	ret_ihklib = ihk_os_destroy_pseudofs(0, 0, 0);
+	fp = popen("cat /proc/mounts | grep /tmp/mcos/mcos0_sys", "r");
+	nread = fread(buf, 1, sizeof(buf), fp);
+	buf[nread] = 0;
+
+	sprintf(cmd, "rmmod %s/kmod/%s", prefix, PART_MOD_NAME);
+	status = system(cmd);
+
+	sprintf(cmd, "rmmod %s/kmod/ihk.ko", prefix);
+	status = system(cmd);
+
+ done_test:
+	printf("*** All tests finished\n\n");
+	fflush(stdout);
+
+ fn_exit:
+    return ret;
+ fn_fail:
+    goto fn_exit;
+}