mirror of
https://github.com/checkpoint-restore/criu.git
synced 2026-07-28 12:36:47 +00:00
104 lines
1.7 KiB
C
104 lines
1.7 KiB
C
#include "criu.h"
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <signal.h>
|
|
#include "lib.h"
|
|
|
|
static int stop = 0;
|
|
static void sh(int sig)
|
|
{
|
|
stop = 1;
|
|
}
|
|
|
|
#define SUCC_ECODE 42
|
|
|
|
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");
|
|
wait(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);
|
|
}
|