[CRIU] [RFC PATCH v2 22/23] lazy-pages: request userfaultfd events during API handshake

[Mike Rapoport]

Signed-off-by: Mike Rapoport <rppt at linux.vnet.ibm.com>
---
 criu/uffd.c | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/criu/uffd.c b/criu/uffd.c
index b357d08..3262c7d 100644
--- a/criu/uffd.c
+++ b/criu/uffd.c
@@ -49,6 +49,12 @@
 #define lp_err(lpi, fmt, arg...) pr_err("%d-%d: " fmt, lpi->pid, lpi->lpfd.fd, ##arg)
 #define lp_perror(lpi, fmt, arg...) pr_perror("%d-%d: " fmt, lpi->pid, lpi->lpfd.fd, ##arg)
 
+#define NEED_UFFD_API_FEATURES (UFFD_FEATURE_EVENT_FORK |	    \
+				UFFD_FEATURE_EVENT_EXIT |	    \
+				UFFD_FEATURE_EVENT_REMAP |	    \
+				UFFD_FEATURE_EVENT_UNMAP |	    \
+				UFFD_FEATURE_EVENT_REMOVE)
+
 #define LAZY_PAGES_SOCK_NAME	"lazy-pages.socket"
 
 static mutex_t *lazy_sock_mutex;
@@ -229,7 +235,7 @@ int setup_uffd(int pid, struct task_restore_args *task_args)
 	 * Check if the UFFD_API is the one which is expected
 	 */
 	uffdio_api.api = UFFD_API;
-	uffdio_api.features = 0;
+	uffdio_api.features = kdat.uffd_features & NEED_UFFD_API_FEATURES;
 	if (ioctl(task_args->uffd, UFFDIO_API, &uffdio_api)) {
 		pr_err("Checking for UFFDIO_API failed.\n");
 		goto err;
-- 
1.9.1