[CRIU] [PATCH 3/7] compel/criu: Add ARCH_HAS_LONG_PAGES to PIE binaries
Dmitry Safonov
dima at arista.com
Mon Mar 26 16:06:58 MSK 2018
For architectures like aarch64/ppc64 it's needed to propagate the size
of page inside PIEs. For the parasite page size will be defined during
seizing, and for restorer during early initialization.
Afterward we can use PAGE_SIZE in PIEs like we did before.
Signed-off-by: Dmitry Safonov <dima at arista.com>
---
compel/include/rpc-pie-priv.h | 3 +++
compel/plugins/std/infect.c | 13 +++++++++++++
compel/src/lib/infect.c | 3 +++
criu/cr-restore.c | 3 +++
criu/include/restorer.h | 3 +++
criu/pie/restorer.c | 12 ++++++++++++
include/common/arch/aarch64/asm/page.h | 9 +++++++++
include/common/arch/ppc64/asm/page.h | 9 +++++++++
8 files changed, 55 insertions(+)
diff --git a/compel/include/rpc-pie-priv.h b/compel/include/rpc-pie-priv.h
index f25ca89eb344..15f5b14cab00 100644
--- a/compel/include/rpc-pie-priv.h
+++ b/compel/include/rpc-pie-priv.h
@@ -38,6 +38,9 @@ struct parasite_init_args {
uint64_t sigreturn_addr;
uint64_t sigframe; /* pointer to sigframe */
futex_t daemon_connected;
+#ifdef ARCH_HAS_LONG_PAGES
+ uint32_t page_size;
+#endif
};
struct parasite_unmap_args {
diff --git a/compel/plugins/std/infect.c b/compel/plugins/std/infect.c
index 2d3aa3df80d1..17beefd34744 100644
--- a/compel/plugins/std/infect.c
+++ b/compel/plugins/std/infect.c
@@ -3,6 +3,7 @@
#include "common/scm.h"
#include "common/compiler.h"
#include "common/lock.h"
+#include "common/page.h"
#define pr_err(fmt, ...) print_on_level(1, fmt, ##__VA_ARGS__)
#define pr_info(fmt, ...) print_on_level(3, fmt, ##__VA_ARGS__)
@@ -19,6 +20,15 @@ static int tsock = -1;
static struct rt_sigframe *sigframe;
+#ifdef ARCH_HAS_LONG_PAGES
+static unsigned __page_size;
+
+unsigned __attribute((weak)) page_size(void)
+{
+ return __page_size;
+}
+#endif
+
int parasite_get_rpc_sock(void)
{
return tsock;
@@ -142,6 +152,9 @@ static noinline __used int parasite_init_daemon(void *data)
args->sigreturn_addr = (uint64_t)(uintptr_t)fini_sigreturn;
sigframe = (void*)(uintptr_t)args->sigframe;
+#ifdef ARCH_HAS_LONG_PAGES
+ __page_size = args->page_size;
+#endif
ret = tsock = sys_socket(PF_UNIX, SOCK_SEQPACKET, 0);
if (tsock < 0) {
diff --git a/compel/src/lib/infect.c b/compel/src/lib/infect.c
index ecc6599b12d0..d51b8aa88a3d 100644
--- a/compel/src/lib/infect.c
+++ b/compel/src/lib/infect.c
@@ -613,6 +613,9 @@ static int parasite_init_daemon(struct parasite_ctl *ctl)
args->sigframe = (uintptr_t)ctl->rsigframe;
args->log_level = compel_log_get_loglevel();
+#ifdef ARCH_HAS_LONG_PAGES
+ args->page_size = PAGE_SIZE;
+#endif
futex_set(&args->daemon_connected, 0);
diff --git a/criu/cr-restore.c b/criu/cr-restore.c
index db913b2dae2e..f39a768b01cd 100644
--- a/criu/cr-restore.c
+++ b/criu/cr-restore.c
@@ -3644,6 +3644,9 @@ static int sigreturn_restore(pid_t pid, struct task_restore_args *task_args, uns
task_args->premmapped_len = rsti(current)->premmapped_len;
task_args->task_size = kdat.task_size;
+#ifdef ARCH_HAS_LONG_PAGES
+ task_args->page_size = PAGE_SIZE;
+#endif
RST_MEM_FIXUP_PPTR(task_args->vmas);
RST_MEM_FIXUP_PPTR(task_args->rings);
diff --git a/criu/include/restorer.h b/criu/include/restorer.h
index 15307d9c0701..70223be55410 100644
--- a/criu/include/restorer.h
+++ b/criu/include/restorer.h
@@ -209,6 +209,9 @@ struct task_restore_args {
void **breakpoint;
enum faults fault_strategy;
+#ifdef ARCH_HAS_LONG_PAGES
+ unsigned page_size;
+#endif
} __aligned(64);
/*
diff --git a/criu/pie/restorer.c b/criu/pie/restorer.c
index 091026103805..ee384df1fbf1 100644
--- a/criu/pie/restorer.c
+++ b/criu/pie/restorer.c
@@ -36,6 +36,7 @@
#include "uffd.h"
#include "common/lock.h"
+#include "common/page.h"
#include "restorer.h"
#include "aio.h"
#include "seccomp.h"
@@ -76,6 +77,14 @@ bool fault_injected(enum faults f)
return __fault_injected(f, fi_strategy);
}
+#ifdef ARCH_HAS_LONG_PAGES
+static unsigned __page_size;
+unsigned page_size(void)
+{
+ return __page_size;
+}
+#endif
+
/*
* These are stubs for std compel plugin.
*/
@@ -1230,6 +1239,9 @@ long __export_restore_task(struct task_restore_args *args)
zombies = args->zombies;
n_zombies = args->zombies_n;
*args->breakpoint = rst_sigreturn;
+#ifdef ARCH_HAS_LONG_PAGES
+ __page_size = args->page_size;
+#endif
ksigfillset(&act.rt_sa_mask);
act.rt_sa_handler = sigchld_handler;
diff --git a/include/common/arch/aarch64/asm/page.h b/include/common/arch/aarch64/asm/page.h
index de1fe5428c50..4126c8474a7d 100644
--- a/include/common/arch/aarch64/asm/page.h
+++ b/include/common/arch/aarch64/asm/page.h
@@ -1,6 +1,9 @@
#ifndef __CR_ASM_PAGE_H__
#define __CR_ASM_PAGE_H__
+#define ARCH_HAS_LONG_PAGES
+
+#ifndef CR_NOGLIBC
#include <unistd.h>
#ifndef PAGE_SHIFT
@@ -18,4 +21,10 @@
#define PAGE_PFN(addr) ((addr) / PAGE_SIZE)
#define page_size() sysconf(_SC_PAGESIZE)
+#else /* CR_NOGLIBC */
+
+extern unsigned page_size(void);
+#define PAGE_SIZE page_size()
+
+#endif /* CR_NOGLIBC */
#endif /* __CR_ASM_PAGE_H__ */
diff --git a/include/common/arch/ppc64/asm/page.h b/include/common/arch/ppc64/asm/page.h
index 9d10455f1c47..a95af55ef8a0 100644
--- a/include/common/arch/ppc64/asm/page.h
+++ b/include/common/arch/ppc64/asm/page.h
@@ -1,6 +1,9 @@
#ifndef __CR_ASM_PAGE_H__
#define __CR_ASM_PAGE_H__
+#define ARCH_HAS_LONG_PAGES
+
+#ifndef CR_NOGLIBC
#include <unistd.h>
/*
@@ -22,4 +25,10 @@
#define PAGE_PFN(addr) ((addr) / PAGE_SIZE)
#define page_size() sysconf(_SC_PAGESIZE)
+#else /* CR_NOGLIBC */
+
+extern unsigned page_size(void);
+#define PAGE_SIZE page_size()
+
+#endif /* CR_NOGLIBC */
#endif /* __CR_ASM_PAGE_H__ */
--
2.13.6
More information about the CRIU
mailing list