Esempio n. 1
0
int opal_crs_criu_checkpoint(pid_t pid, opal_crs_base_snapshot_t *base_snapshot,
                             opal_crs_base_ckpt_options_t *options,
                             opal_crs_state_type_t *state)
{
    int ret;
    int fd = 0;
    int oh = mca_crs_criu_component.super.output_handle;
    opal_crs_criu_snapshot_t *snapshot = NULL;
    char *dest = NULL;

    opal_output_verbose(10, oh, "crs:criu: checkpoint(%d, ---)", pid);

    snapshot = (opal_crs_criu_snapshot_t *)base_snapshot;
    snapshot->super.component_name = strdup(mca_crs_criu_component.super.base_version.mca_component_name);

    if (NULL == snapshot->super.metadata) {
        if (NULL == (snapshot->super.metadata = fopen(snapshot->super.metadata_filename, "a"))) {
            opal_output(oh, "crs:criu: checkpoint(): Error: Unable to open the file (%s)",
                        snapshot->super.metadata_filename);
            *state = OPAL_CRS_ERROR;
            goto cleanup;
        }
    }
    fprintf(snapshot->super.metadata, "%s%s\n", CRS_METADATA_COMP, snapshot->super.component_name);

    fclose(snapshot->super.metadata);
    snapshot->super.metadata = NULL;

    ret = criu_init_opts();

    if (ret < 0) {
        criu_error(ret, pid);
        *state = OPAL_CRS_ERROR;
        goto cleanup;
    }

    opal_output_verbose(10, oh, "crs:criu: criu_init_opts() returned %d", ret);

    dest = snapshot->super.snapshot_directory;
    opal_output_verbose(10, oh, "crs:criu: opening snapshot directory %s", dest);
    fd = open(dest, O_DIRECTORY);

    if (fd < 0) {
        *state = OPAL_CRS_ERROR;
        opal_output(oh, "crs:criu: checkpoint(): Error: Unable to open checkpoint "
                    "directory (%s) for pid (%d)", dest, pid);
        goto cleanup;
    }

    /* http://criu.org/C_API */
    criu_set_images_dir_fd(fd);
    criu_set_pid(pid);

    criu_set_log_file(mca_crs_criu_component.log_file);
    criu_set_log_level(mca_crs_criu_component.log_level);
    criu_set_tcp_established(mca_crs_criu_component.tcp_established);
    criu_set_shell_job(mca_crs_criu_component.shell_job);
    criu_set_ext_unix_sk(mca_crs_criu_component.ext_unix_sk);
    criu_set_leave_running(mca_crs_criu_component.leave_running);
    ret = criu_dump();

    if (ret < 0) {
        criu_error(ret, pid);
        *state = OPAL_CRS_ERROR;
        goto cleanup;
    }

    *state = OPAL_CRS_CONTINUE;

 cleanup:

    if (fd > 0) {
        close(fd);
    }

    if (OPAL_CRS_ERROR == *state) {
        return OPAL_ERROR;
    }
    return OPAL_SUCCESS;
}
Esempio n. 2
0
int main(int argc, char **argv)
{
    int pid, ret, fd, p[2];

    printf("--- Start loop ---\n");
    pipe(p);
    pid = fork();
    if (pid < 0) {
        perror("Can't");
        return -1;
    }

    if (!pid) {
        printf("   `- loop: initializing\n");
        if (setsid() < 0)
            exit(1);
        if (signal(SIGUSR1, sh) == SIG_ERR)
            exit(1);

        close(0);
        close(1);
        close(2);
        close(p[0]);

        ret = SUCC_ECODE;
        write(p[1], &ret, sizeof(ret));
        close(p[1]);

        while (!stop)
            sleep(1);
        exit(SUCC_ECODE);
    }

    close(p[1]);

    /* Wait for kid to start */
    ret = -1;
    read(p[0], &ret, sizeof(ret));
    if (ret != SUCC_ECODE) {
        printf("Error starting loop\n");
        goto err;
    }

    /* Wait for pipe to get closed, then dump */
    read(p[0], &ret, 1);
    close(p[0]);

    printf("--- Dump loop ---\n");
    criu_init_opts();
    criu_set_service_address(argv[1]);
    criu_set_pid(pid);
    criu_set_log_file("dump.log");
    criu_set_log_level(4);
    fd = open(argv[2], O_DIRECTORY);
    criu_set_images_dir_fd(fd);

    ret = criu_dump();
    if (ret < 0) {
        what_err_ret_mean(ret);
        kill(pid, SIGKILL);
        goto err;
    }

    printf("   `- Dump succeeded\n");
    waitpid(pid, NULL, 0);

    printf("--- Restore loop ---\n");
    criu_init_opts();
    criu_set_log_level(4);
    criu_set_log_file("restore.log");
    criu_set_images_dir_fd(fd);

    pid = criu_restore_child();
    if (pid <= 0) {
        what_err_ret_mean(pid);
        return -1;
    }

    printf("   `- Restore returned pid %d\n", pid);
    kill(pid, SIGUSR1);
err:
    if (waitpid(pid, &ret, 0) < 0) {
        perror("   Can't wait kid");
        return -1;
    }

    return chk_exit(ret, SUCC_ECODE);
}
Esempio n. 3
0
int main(int argc, char *argv[])
{
	int ret, fd;

	criu_set_service_address("criu_service.socket");

	puts("--- Check ---");
	ret = criu_check();
	if (ret < 0) {
		what_err_ret_mean(ret);
		return -1;
	} else
		puts("Success");

	puts("--- Dump loop ---");
	criu_init_opts();
	criu_set_pid(atoi(argv[1]));
	criu_set_log_file("dump.log");
	criu_set_log_level(4);
	fd = open("imgs_loop", O_DIRECTORY);
	criu_set_images_dir_fd(fd);

	ret = criu_dump();
	if (ret < 0) {
		what_err_ret_mean(ret);
		return -1;
	} else if (ret == 0)
		puts("Success");

	puts("--- Restore loop ---");
	criu_init_opts();
	criu_set_log_level(4);
	criu_set_log_file("restore.log");
	criu_set_images_dir_fd(fd);

	ret = criu_restore();
	if (ret < 0) {
		what_err_ret_mean(ret);
		return -1;
	} else if (ret > 0) {
		puts("Success");
		printf("pid %d\n", ret);
	}

	puts("--- Dump myself ---");
	criu_init_opts();
	criu_set_leave_running(true);
	criu_set_shell_job(true);
	criu_set_images_dir_fd(open("imgs_test", O_DIRECTORY));

	ret = criu_dump();
	if (ret < 0) {
		what_err_ret_mean(ret);
		return -1;
	} else {
		puts("Success");
		if (ret == 1)
			puts("Restored");
	}

	return 0;
}