mir/criu
2
0
Fork 0
mirror of https://github.com/checkpoint-restore/criu synced 2025-09-03 07:45:17 +00:00
Code Issues Releases Wiki Activity

parasite: add a function to execute a parasite code in threads

Browse Source 
Signed-off-by: Andrey Vagin <avagin@openvz.org>
Acked-by: Pavel Emelyanov <xemul@parallels.com>
Signed-off-by: Cyrill Gorcunov <gorcunov@openvz.org>
This commit is contained in:
Andrey Vagin
2012-02-21 09:25:16 +03:00
committed by Cyrill Gorcunov
parent 385d7e0369
commit fd5520f3c3
1 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
parasite-syscall.c
View File

@@ -77,9 +77,8 @@ static void parasite_setup_regs(unsigned long new_ip, user_regs_struct_t *regs)
} }
/* we run at @regs->ip */ /* we run at @regs->ip */
static int __parasite_execute(struct parasite_ctl *ctl, user_regs_struct_t *regs) static int __parasite_execute(struct parasite_ctl *ctl, pid_t pid, user_regs_struct_t *regs)
{ {
pid_t pid = ctl->pid;
siginfo_t siginfo; siginfo_t siginfo;
int status; int status;
int ret = -1; int ret = -1;
@@ -226,19 +225,29 @@ err:
return ret; return ret;
} }
static int parasite_execute(unsigned long cmd, struct parasite_ctl *ctl, static int parasite_execute_by_pid(unsigned long cmd, struct parasite_ctl *ctl,
pid_t pid,
parasite_status_t *args, int args_size) parasite_status_t *args, int args_size)
{ {
int ret; int ret;
user_regs_struct_t regs_orig, regs;
user_regs_struct_t regs = ctl->regs_orig; if (ctl->pid == pid)
regs = ctl->regs_orig;
else {
if (ptrace(PTRACE_GETREGS, pid, NULL, &regs_orig)) {
pr_err("Can't obtain registers (pid: %d)\n", pid);
return -1;
}
regs = regs_orig;
}
memcpy(ctl->addr_cmd, &cmd, sizeof(cmd)); memcpy(ctl->addr_cmd, &cmd, sizeof(cmd));
memcpy(ctl->addr_args, args, args_size); memcpy(ctl->addr_args, args, args_size);
parasite_setup_regs(ctl->parasite_ip, &regs); parasite_setup_regs(ctl->parasite_ip, &regs);
ret = __parasite_execute(ctl, &regs); ret = __parasite_execute(ctl, pid, &regs);
memcpy(args, ctl->addr_args, args_size); memcpy(args, ctl->addr_args, args_size);
if (!ret) if (!ret)
@@ -248,9 +257,21 @@ static int parasite_execute(unsigned long cmd, struct parasite_ctl *ctl,
pr_err("Parasite exited with %d ret (%li at %li)\n", pr_err("Parasite exited with %d ret (%li at %li)\n",
ret, args->sys_ret, args->line); ret, args->sys_ret, args->line);
if (ctl->pid != pid)
if (ptrace(PTRACE_SETREGS, pid, NULL, &regs_orig)) {
pr_panic("Can't restore registers (pid: %d)\n", ctl->pid);
return -1;
}
return ret; return ret;
} }
static int parasite_execute(unsigned long cmd, struct parasite_ctl *ctl,
parasite_status_t *args, int args_size)
{
return parasite_execute_by_pid(cmd, ctl, ctl->pid, args, args_size);
}
static void *mmap_seized(struct parasite_ctl *ctl, static void *mmap_seized(struct parasite_ctl *ctl,
void *addr, size_t length, int prot, void *addr, size_t length, int prot,
int flags, int fd, off_t offset) int flags, int fd, off_t offset)
@@ -269,7 +290,7 @@ static void *mmap_seized(struct parasite_ctl *ctl,
parasite_setup_regs(ctl->syscall_ip, &regs); parasite_setup_regs(ctl->syscall_ip, &regs);
ret = __parasite_execute(ctl, &regs); ret = __parasite_execute(ctl, ctl->pid, &regs);
if (ret) if (ret)
goto err; goto err;
@@ -290,7 +311,7 @@ static int munmap_seized(struct parasite_ctl *ctl, void *addr, size_t length)
parasite_setup_regs(ctl->syscall_ip, &regs); parasite_setup_regs(ctl->syscall_ip, &regs);
ret = __parasite_execute(ctl, &regs); ret = __parasite_execute(ctl, ctl->pid, &regs);
if (!ret) if (!ret)
ret = (int)regs.ax; ret = (int)regs.ax;