From patchwork Thu Mar 28 11:10:08 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Jinjiang Tu X-Patchwork-Id: 13608404 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id 022DEC54E64 for ; Thu, 28 Mar 2024 11:10:41 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 7EA786B008C; Thu, 28 Mar 2024 07:10:40 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 7996F6B0092; Thu, 28 Mar 2024 07:10:40 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 660AF6B0096; Thu, 28 Mar 2024 07:10:40 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0011.hostedemail.com [216.40.44.11]) by kanga.kvack.org (Postfix) with ESMTP id 4B27C6B008C for ; Thu, 28 Mar 2024 07:10:40 -0400 (EDT) Received: from smtpin27.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay05.hostedemail.com (Postfix) with ESMTP id 0C78441133 for ; Thu, 28 Mar 2024 11:10:40 +0000 (UTC) X-FDA: 81946179840.27.73141AB Received: from szxga01-in.huawei.com (szxga01-in.huawei.com [45.249.212.187]) by imf14.hostedemail.com (Postfix) with ESMTP id 3AED6100005 for ; Thu, 28 Mar 2024 11:10:36 +0000 (UTC) Authentication-Results: imf14.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf14.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.187 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1711624238; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=cTwV8t7oMvp0yB+otbVu8cFnsneSqJ/QAllYDKvgrF4=; b=CQJFVbq2Kfwln0r2b2xLNiHAjMgm+cOoIzqKr2/GYHZFbyqO32l9IMHO1bj+a3/yvUQox9 v37lBz/KunlTKdf5Af9Iear5U15MB16oWctcvNj4D+pj0kYSIBF4vaEYuedSEZufmhdyUX H65OREIKqQsMvHHPnmf2Pe62tHWSZGI= ARC-Authentication-Results: i=1; imf14.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf14.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.187 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1711624238; a=rsa-sha256; cv=none; b=qpxfSgK2f5x4vkMObOVKBJhSgaPjWOHsdURGx5JcrZiAWj/N/vr/SHhXwLex5U7o9AT5Gf uKF+gXgDT4KaRuZENdDQGDBTkmzdKQXYkg8TqztuOpZnrZGg5HTAav1eEtqba+x/rWgTXv nTnrHBH4uOIQ/5fxLBlcm2OilgBC82A= Received: from mail.maildlp.com (unknown [172.19.163.174]) by szxga01-in.huawei.com (SkyGuard) with ESMTP id 4V513p5vZ1ztQbm; Thu, 28 Mar 2024 19:08:06 +0800 (CST) Received: from dggpeml500011.china.huawei.com (unknown [7.185.36.84]) by mail.maildlp.com (Postfix) with ESMTPS id 69A0E1401E0; Thu, 28 Mar 2024 19:10:33 +0800 (CST) Received: from huawei.com (10.175.113.32) by dggpeml500011.china.huawei.com (7.185.36.84) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.1.2507.35; Thu, 28 Mar 2024 19:10:32 +0800 From: Jinjiang Tu To: , , , , , , , CC: Subject: [PATCH v4 1/3] mm/ksm: fix ksm exec support for prctl Date: Thu, 28 Mar 2024 19:10:08 +0800 Message-ID: <20240328111010.1502191-2-tujinjiang@huawei.com> X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240328111010.1502191-1-tujinjiang@huawei.com> References: <20240328111010.1502191-1-tujinjiang@huawei.com> MIME-Version: 1.0 X-Originating-IP: [10.175.113.32] X-ClientProxiedBy: dggems704-chm.china.huawei.com (10.3.19.181) To dggpeml500011.china.huawei.com (7.185.36.84) X-Rspam-User: X-Rspamd-Server: rspam12 X-Rspamd-Queue-Id: 3AED6100005 X-Stat-Signature: wbb83sksi9wjdqxrafkduoyqszzm67f1 X-HE-Tag: 1711624236-561661 X-HE-Meta: U2FsdGVkX1+CmrE/VlUOLrSM/TYY1EYorSSuvthplXqK+T+J3bLFsluDt4WzfdCFL96OidF9Le62h68J/CNe8cGEdKw+R7T6dZxtA58OKdQOD75FCCHhhymW9Op476n/cL7R8AQrmO8YXNIWZWkjHcUco2k8iOwq+inRhnruDAP3PAatc8tUZn9gUkfUZUhtmxQOX+RaNOpRZs31HVHqGBhwqjgYsp+VfnyfyleAm5An8sEuwU29x7uFVRpaTtB0MHZjjFODzhEKvu33GYyUXYWEhtCcbG9HkFQmxB8PdfvYYlSwVUsOyRuJz3LE/XLZLr0/tGo1h9p1FHx7LIAysPlMcIduu+xKVUi+/asGBKudnEMnfyy+7WtnQOEfYunU2MzUb6NquMCpdyLDGlopn3BA666tYeAIMZQs32C9qB3FrJ/GT/rvAVuf9eILVIeHJ/D45afqhQT2Sc42u0rMYqVjic+Qq4aA/XMg9c7RQz/xeKP1kEZMARGzSRLoQLYA0lSyAXZifeEAQVBx93IPc++kYLbRz3ugD9bii2qfAotnCtf0gLxqwACaHEvWWQE1NIza+X8Q5NapeYCIPVjF0s1aS3alLEqPQPyPQxW3cSzU7+OP65mN/vKGitBQZRNgNKSmydHtK9NK8VMM5fXgxB+57AtOWzkjPdbCqgB+6Y2vRW8hILJAeXsD2Jhj6n0WboG1BksPYyLG8U/YOKZckHeZ6yZTvCOOK5S+YJX5yTgv1nMH8anTWANwhQ8BYoO2ux7f6bf6BhZkNO0fuqne4eEpjlB3OxQgSFuBCk72OCYiHXHK5+2QmHRM1OhxLflFkdmltEpvYVfesm5Rr4EaRCviUMLIwuwsSon4UW8T4q8dEx+LcWY1h4cR4VLPr+ArYumOGd2z09hjraITDHLHwMYBLtkTm+5PHecqQsYzEgxus+EBdbJUEtZ8ukfreSERiL55hr9xhAqt9ebVyuM lbrkibie PlzFsOq2jHTEhx+uIZ00dKKB32jYPSkli1BmVAqXRfU5KXlZOf8r76lerBUIiJ6i0rDGp8Ijkkbb+uv2L6V80cAzUxt+sMBWHj+22qiYIv5UBfvaS5StrhDbNHwTo8ZbpZ+6VoIN634IwuM5BQRP3AB1Nr3L3Cda3fwqDOYMb9mFX/PY= X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: commit 3c6f33b7273a ("mm/ksm: support fork/exec for prctl") inherits MMF_VM_MERGE_ANY flag when a task calls execve(). Howerver, it doesn't create the mm_slot, so ksmd will not try to scan this task. To fix it, allocate and add the mm_slot to ksm_mm_head in __bprm_mm_init() when the mm has MMF_VM_MERGE_ANY flag. Fixes: 3c6f33b7273a ("mm/ksm: support fork/exec for prctl") Signed-off-by: Jinjiang Tu Reviewed-by: David Hildenbrand --- fs/exec.c | 11 +++++++++++ include/linux/ksm.h | 13 +++++++++++++ 2 files changed, 24 insertions(+) diff --git a/fs/exec.c b/fs/exec.c index ff6f26671cfc..c2890d32232e 100644 --- a/fs/exec.c +++ b/fs/exec.c @@ -67,6 +67,7 @@ #include #include #include +#include #include #include @@ -267,6 +268,14 @@ static int __bprm_mm_init(struct linux_binprm *bprm) goto err_free; } + /* + * Need to be called with mmap write lock + * held, to avoid race with ksmd. + */ + err = ksm_execve(mm); + if (err) + goto err_ksm; + /* * Place the stack at the largest stack address the architecture * supports. Later, we'll move this to an appropriate place. We don't @@ -288,6 +297,8 @@ static int __bprm_mm_init(struct linux_binprm *bprm) bprm->p = vma->vm_end - sizeof(void *); return 0; err: + ksm_exit(mm); +err_ksm: mmap_write_unlock(mm); err_free: bprm->vma = NULL; diff --git a/include/linux/ksm.h b/include/linux/ksm.h index 401348e9f92b..7e2b1de3996a 100644 --- a/include/linux/ksm.h +++ b/include/linux/ksm.h @@ -59,6 +59,14 @@ static inline int ksm_fork(struct mm_struct *mm, struct mm_struct *oldmm) return 0; } +static inline int ksm_execve(struct mm_struct *mm) +{ + if (test_bit(MMF_VM_MERGE_ANY, &mm->flags)) + return __ksm_enter(mm); + + return 0; +} + static inline void ksm_exit(struct mm_struct *mm) { if (test_bit(MMF_VM_MERGEABLE, &mm->flags)) @@ -107,6 +115,11 @@ static inline int ksm_fork(struct mm_struct *mm, struct mm_struct *oldmm) return 0; } +static inline int ksm_execve(struct mm_struct *mm) +{ + return 0; +} + static inline void ksm_exit(struct mm_struct *mm) { } From patchwork Thu Mar 28 11:10:09 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Jinjiang Tu X-Patchwork-Id: 13608405 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id E1027C54E64 for ; Thu, 28 Mar 2024 11:10:44 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 810A96B0092; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 7C1106B0096; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 639C86B0098; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0010.hostedemail.com [216.40.44.10]) by kanga.kvack.org (Postfix) with ESMTP id 46E276B0092 for ; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) Received: from smtpin04.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay07.hostedemail.com (Postfix) with ESMTP id C98BC160F10 for ; Thu, 28 Mar 2024 11:10:40 +0000 (UTC) X-FDA: 81946179840.04.03813C7 Received: from szxga04-in.huawei.com (szxga04-in.huawei.com [45.249.212.190]) by imf21.hostedemail.com (Postfix) with ESMTP id 5A1811C001C for ; Thu, 28 Mar 2024 11:10:37 +0000 (UTC) Authentication-Results: imf21.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf21.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.190 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1711624239; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=q4TG5FLVhpdjvrCxKGfy/GgQApZKPsrbW7gZGUGHI1Q=; b=piCrRtnFEHukrMwXAD8Lxw61i9juLq5hc3S1JaoDOYOYvEoEey6xiPfYTv1gjp3yVCmDlC Cs1MXIYqc4LhpXZPyRVT/vw4QUaMjVXYph2I18TGtDYZRpdpdZ8kJIMX74+krlmWaiEEKv v1Fr1DYLCvy9JUh0nwMGrXTHDdokYqY= ARC-Authentication-Results: i=1; imf21.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf21.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.190 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1711624239; a=rsa-sha256; cv=none; b=qmb/bu5pRwAl5EWXCmbhl2COtdctN6yA/HQYhpaqQlhtPIWNmZ+2ZBv/DH4m02VtNohSsZ ra7AM6a65K5fZ2Ry0ockczDMg2941ShEG2477TB5TJT0nTTDKDUl7KA8OWiUEhXBAspc+b quPc0x5pnAqoLWjVnv38WTYtKW/ogVo= Received: from mail.maildlp.com (unknown [172.19.88.214]) by szxga04-in.huawei.com (SkyGuard) with ESMTP id 4V514J2JJwz1xsTZ; Thu, 28 Mar 2024 19:08:32 +0800 (CST) Received: from dggpeml500011.china.huawei.com (unknown [7.185.36.84]) by mail.maildlp.com (Postfix) with ESMTPS id DD29E1A016C; Thu, 28 Mar 2024 19:10:33 +0800 (CST) Received: from huawei.com (10.175.113.32) by dggpeml500011.china.huawei.com (7.185.36.84) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.1.2507.35; Thu, 28 Mar 2024 19:10:33 +0800 From: Jinjiang Tu To: , , , , , , , CC: Subject: [PATCH v4 2/3] selftest/mm: ksm_functional_tests: refactor mmap_and_merge_range() Date: Thu, 28 Mar 2024 19:10:09 +0800 Message-ID: <20240328111010.1502191-3-tujinjiang@huawei.com> X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240328111010.1502191-1-tujinjiang@huawei.com> References: <20240328111010.1502191-1-tujinjiang@huawei.com> MIME-Version: 1.0 X-Originating-IP: [10.175.113.32] X-ClientProxiedBy: dggems704-chm.china.huawei.com (10.3.19.181) To dggpeml500011.china.huawei.com (7.185.36.84) X-Rspamd-Server: rspam09 X-Rspamd-Queue-Id: 5A1811C001C X-Stat-Signature: jfd8855jnexjxymwx46a183nakco9raf X-Rspam-User: X-HE-Tag: 1711624237-493267 X-HE-Meta: U2FsdGVkX18pM3Zn1kQ0cI6ulvzKcxBCcnBnp4dDxg1V9UppBvPq6HVLYLRQiw/1WVCd/foxwpagZnGzkuwxPu+L170DQ6uW1BKT4bxe+DY122JrQfRaezCR9uEE2/CIKE6u+BBPTbzdmfFbA+7jd09fX9aYf3+LlGLkF1yGPEB0Qz4I28kzVRUlgwOOR1ns4OHCkLXXyQ5fYBwNcRtUFkAo06I7TiE+c2sh1IHl4e+VpehiyJQnSC8208pU1VUDiZNgdf1JlK6m/BPlyweRA5sNxn1AkW8wNS1X0QHUkSZpPYJnKSv+NZU8WwjzJIrSeJKyHTVPXu7OdD39MRRIMUP22zqk/b/ZcFGXJ5X2r/q4EUq99+lGSePYIJ8P25QFs4qqnj5SL2H79fGf/xQQydTVGBCkxHqGaAdU6dEcSmnf1uMBayRVERVQrYANa/2nsK+1+ErVXYmZmE9djBM9fQEB90XObZTJLBRQpwbUB81LOp7Y0ASew7s0GRvv6d+NtPOQL3RyD1jSr6JPc3ufiSBN6pDoCQO5AmaIyzfXEuVB/ivNQFwiiU+IXrz3IJt39thgE21a4OaV/OegWjU5KgcjZNi5RgeYXcIzKWH+DJRvBrJltTsQz/Jb+ZRM5XDPKrgHqhts7wl1hpezR3W5gWhFTIXLMd97fe9ODHqubP2kEy6nd1zJS5GsG8ZXf2RzcOLBaLEeouSkpiEAcfSA1QI9uqII+ntm5Oel87quEM18Abb2cb6Lmw7SqCRUWEJrP6zymQ304BtfMkZt3wtXTnVnDU0loEbX35Yo/pj+ejBHLUKMQHS3IHBSXSPooyYMXyjs3NQfUbiN5mAE7C+CZgyQchd3h9/fnjLOvSeqYn6MnWhbsapuIV55EiFwmvhbiR8mlh0UDw8= X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: In order to extend test_prctl_fork() and test_prctl_fork_exec() to make sure that deduplication really happens, mmap_and_merge_range() needs to be refactored. Firstly, mmap_and_merge_range() will be called with no need to call enable KSM by madvise or prctl. So, switch the 'bool use_prctl' parameter to enum ksm_merge_mode. Secondly, mmap_and_merge_range() will be called in child process in the two testcases, it isn't appropriate to call ksft_test_result_{fail, skip}, because the global variables ksft_{fail, skip} aren't consistent with the parent process. Thus, convert calls of ksft_test_result_{fail, skip} to ksft_print_msg(), return differrent error according to the two cases, and rename mmap_and_merge_range() to __mmap_and_merge_range(). For existing callers, introduce new mmap_and_merge_range() to handle different return values of __mmap_and_merge_range(). Suggested-by: David Hildenbrand Signed-off-by: Jinjiang Tu Reviewed-by: David Hildenbrand --- .../selftests/mm/ksm_functional_tests.c | 86 +++++++++++++------ 1 file changed, 61 insertions(+), 25 deletions(-) diff --git a/tools/testing/selftests/mm/ksm_functional_tests.c b/tools/testing/selftests/mm/ksm_functional_tests.c index d615767e396b..5462873df84c 100644 --- a/tools/testing/selftests/mm/ksm_functional_tests.c +++ b/tools/testing/selftests/mm/ksm_functional_tests.c @@ -28,6 +28,15 @@ #define MiB (1024 * KiB) #define FORK_EXEC_CHILD_PRG_NAME "ksm_fork_exec_child" +#define MAP_MERGE_FAIL ((void *)-1) +#define MAP_MERGE_SKIP ((void *)-2) + +enum ksm_merge_mode { + KSM_MERGE_PRCTL, + KSM_MERGE_MADVISE, + KSM_MERGE_NONE, /* PRCTL already set */ +}; + static int mem_fd; static int ksm_fd; static int ksm_full_scans_fd; @@ -146,33 +155,34 @@ static int ksm_unmerge(void) return 0; } -static char *mmap_and_merge_range(char val, unsigned long size, int prot, - bool use_prctl) +static char *__mmap_and_merge_range(char val, unsigned long size, int prot, + enum ksm_merge_mode mode) { char *map; + char *err_map = MAP_MERGE_FAIL; int ret; /* Stabilize accounting by disabling KSM completely. */ if (ksm_unmerge()) { - ksft_test_result_fail("Disabling (unmerging) KSM failed\n"); - return MAP_FAILED; + ksft_print_msg("Disabling (unmerging) KSM failed\n"); + return err_map; } if (get_my_merging_pages() > 0) { - ksft_test_result_fail("Still pages merged\n"); - return MAP_FAILED; + ksft_print_msg("Still pages merged\n"); + return err_map; } map = mmap(NULL, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); if (map == MAP_FAILED) { - ksft_test_result_fail("mmap() failed\n"); - return MAP_FAILED; + ksft_print_msg("mmap() failed\n"); + return err_map; } /* Don't use THP. Ignore if THP are not around on a kernel. */ if (madvise(map, size, MADV_NOHUGEPAGE) && errno != EINVAL) { - ksft_test_result_fail("MADV_NOHUGEPAGE failed\n"); + ksft_print_msg("MADV_NOHUGEPAGE failed\n"); goto unmap; } @@ -180,27 +190,36 @@ static char *mmap_and_merge_range(char val, unsigned long size, int prot, memset(map, val, size); if (mprotect(map, size, prot)) { - ksft_test_result_skip("mprotect() failed\n"); + ksft_print_msg("mprotect() failed\n"); + err_map = MAP_MERGE_SKIP; goto unmap; } - if (use_prctl) { + switch (mode) { + case KSM_MERGE_PRCTL: ret = prctl(PR_SET_MEMORY_MERGE, 1, 0, 0, 0); if (ret < 0 && errno == EINVAL) { - ksft_test_result_skip("PR_SET_MEMORY_MERGE not supported\n"); + ksft_print_msg("PR_SET_MEMORY_MERGE not supported\n"); + err_map = MAP_MERGE_SKIP; goto unmap; } else if (ret) { - ksft_test_result_fail("PR_SET_MEMORY_MERGE=1 failed\n"); + ksft_print_msg("PR_SET_MEMORY_MERGE=1 failed\n"); goto unmap; } - } else if (madvise(map, size, MADV_MERGEABLE)) { - ksft_test_result_fail("MADV_MERGEABLE failed\n"); - goto unmap; + break; + case KSM_MERGE_MADVISE: + if (madvise(map, size, MADV_MERGEABLE)) { + ksft_print_msg("MADV_MERGEABLE failed\n"); + goto unmap; + } + break; + case KSM_MERGE_NONE: + break; } /* Run KSM to trigger merging and wait. */ if (ksm_merge()) { - ksft_test_result_fail("Running KSM failed\n"); + ksft_print_msg("Running KSM failed\n"); goto unmap; } @@ -209,14 +228,31 @@ static char *mmap_and_merge_range(char val, unsigned long size, int prot, * accounted differently (depending on kernel support). */ if (val && !get_my_merging_pages()) { - ksft_test_result_fail("No pages got merged\n"); + ksft_print_msg("No pages got merged\n"); goto unmap; } return map; unmap: munmap(map, size); - return MAP_FAILED; + return err_map; +} + +static char *mmap_and_merge_range(char val, unsigned long size, int prot, + enum ksm_merge_mode mode) +{ + char *map; + char *ret = MAP_FAILED; + + map = __mmap_and_merge_range(val, size, prot, mode); + if (map == MAP_MERGE_FAIL) + ksft_test_result_fail("Merging memory failed"); + else if (map == MAP_MERGE_SKIP) + ksft_test_result_skip("Merging memory skipped"); + else + ret = map; + + return ret; } static void test_unmerge(void) @@ -226,7 +262,7 @@ static void test_unmerge(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, false); + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); if (map == MAP_FAILED) return; @@ -264,7 +300,7 @@ static void test_unmerge_zero_pages(void) } /* Let KSM deduplicate zero pages. */ - map = mmap_and_merge_range(0x00, size, PROT_READ | PROT_WRITE, false); + map = mmap_and_merge_range(0x00, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); if (map == MAP_FAILED) return; @@ -312,7 +348,7 @@ static void test_unmerge_discarded(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, false); + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); if (map == MAP_FAILED) return; @@ -344,7 +380,7 @@ static void test_unmerge_uffd_wp(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, false); + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); if (map == MAP_FAILED) return; @@ -545,7 +581,7 @@ static void test_prctl_unmerge(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, true); + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_PRCTL); if (map == MAP_FAILED) return; @@ -568,7 +604,7 @@ static void test_prot_none(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0x11, size, PROT_NONE, false); + map = mmap_and_merge_range(0x11, size, PROT_NONE, KSM_MERGE_MADVISE); if (map == MAP_FAILED) goto unmap; From patchwork Thu Mar 28 11:10:10 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Jinjiang Tu X-Patchwork-Id: 13608406 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id AE869C54E64 for ; Thu, 28 Mar 2024 11:10:47 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id C7EA36B0096; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id C2DEC6B0098; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id AF60B6B0099; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0015.hostedemail.com [216.40.44.15]) by kanga.kvack.org (Postfix) with ESMTP id 8F4A46B0098 for ; Thu, 28 Mar 2024 07:10:41 -0400 (EDT) Received: from smtpin04.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay09.hostedemail.com (Postfix) with ESMTP id 51C9F810EB for ; Thu, 28 Mar 2024 11:10:41 +0000 (UTC) X-FDA: 81946179882.04.DB8BADC Received: from szxga05-in.huawei.com (szxga05-in.huawei.com [45.249.212.191]) by imf11.hostedemail.com (Postfix) with ESMTP id A989840014 for ; Thu, 28 Mar 2024 11:10:38 +0000 (UTC) Authentication-Results: imf11.hostedemail.com; dkim=none; spf=pass (imf11.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.191 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com; dmarc=pass (policy=quarantine) header.from=huawei.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1711624239; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=S5ooruH/cHPfkDWNWhBMEtiFounIviZyb4zr5uPuOTM=; b=J23eZ7qT2AHVf4EvbtAOzXv5EtRcy2ILBDPdJEKyA6q4gppjItM1qelXUscb5aw46Js6gy NBq4SIXDj/r6+OHMUC5vH053N/FNyUPHsneQR7PO7hYQotsjU9xALxve2MjNWevOBLNuE2 RZSv4ofDUsokdgPxsBKosQ0ddLIk3b0= ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1711624239; a=rsa-sha256; cv=none; b=jgC4rvlgenhmY9OASnKHR9LYQBxpbjSCwUpsIjQZ9wgmOv9AJCpBLxBXdnh6miJVE8soWX os5j4ydG/DM3hJuPRhKxp+iqCnzRm37KBn1NSjAh6Z7/zjTM2V8NsQaMTESxkKP5c8Km4X mbgfNVhcCaskbOiZhHEGLTZ7dnUzZyY= ARC-Authentication-Results: i=1; imf11.hostedemail.com; dkim=none; spf=pass (imf11.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.191 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com; dmarc=pass (policy=quarantine) header.from=huawei.com Received: from mail.maildlp.com (unknown [172.19.88.163]) by szxga05-in.huawei.com (SkyGuard) with ESMTP id 4V513Z0wJcz1h4Fr; Thu, 28 Mar 2024 19:07:54 +0800 (CST) Received: from dggpeml500011.china.huawei.com (unknown [7.185.36.84]) by mail.maildlp.com (Postfix) with ESMTPS id 5EFCB180063; Thu, 28 Mar 2024 19:10:34 +0800 (CST) Received: from huawei.com (10.175.113.32) by dggpeml500011.china.huawei.com (7.185.36.84) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.1.2507.35; Thu, 28 Mar 2024 19:10:33 +0800 From: Jinjiang Tu To: , , , , , , , CC: Subject: [PATCH v4 3/3] selftest/mm: ksm_functional_tests: extend test case for ksm fork/exec Date: Thu, 28 Mar 2024 19:10:10 +0800 Message-ID: <20240328111010.1502191-4-tujinjiang@huawei.com> X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240328111010.1502191-1-tujinjiang@huawei.com> References: <20240328111010.1502191-1-tujinjiang@huawei.com> MIME-Version: 1.0 X-Originating-IP: [10.175.113.32] X-ClientProxiedBy: dggems704-chm.china.huawei.com (10.3.19.181) To dggpeml500011.china.huawei.com (7.185.36.84) X-Rspamd-Queue-Id: A989840014 X-Rspam-User: X-Rspamd-Server: rspam11 X-Stat-Signature: rwoidiybhojux4xxbo9j778ke1pkaqw3 X-HE-Tag: 1711624238-229451 X-HE-Meta: U2FsdGVkX1+6KYIiKq4VCNnPHONN3hgC2C0gppDRkw+QOTFXj2E4XddEqTxtdqTpCl0HN27L0IFbMOPKvRULqOPH0gm/1x0w9oYumNoEorpdbeGpchyopuPRgnFxoqG+YpuKKuJnzOt1SiDS+ZotdohTizVLa7F7nehmPzF22HjqsmUO76x03f5pi711HX3JvBgClJtenyZ8KikppsLJmWGG6BkbAqedhl6M3lHbIzy99sVamDECvls1JI3cxqN3rQq1FVp5zIRA1hroX86XmQTfOzkiP2ijknclwQLTocJbc37eLW9Ramljc+9YfFZHNk4D8aGAinEgHdVzbdY7dVsl3wEZGFuGs/GgE6BBOG+gPKJIvUJStfiFQY3X1rJgDFLzodB3FW7oz7TF+qLhSiREcDLTRHP6iuF4ULHi+zaHOak8ZUrGhebGFLNecbEJwez0pLJaWg4TvY66gUSok+BAGY7t2+gwWT38Rh7rKCbnk+l8pzbTkiErli0iCGNlXbCB9hINERIpACl23OaR6AeI1qpCpBTNnLlIo+42ezLjgN5UvvrB+qBnZ3DBiDh3+shml8B6JXH6kSYnVbXBKnmxNQNBNwZyygxACXl2KuJVrgN+g9VHutbQxn+JsSzNZSHYNxoytco80i/I78ltGk9/tuc4kZbyz91sJ3U4uoid2k30HVU3NlHW1C9jSWmRnU3pQP4gP3VBB3nTSnfsYZRr5BV0TFjLlQOaInH+Ei1boALy0Ceb86y/3x0Q8BpGyW3+7DOrwbqLokIXcOjcBHS/z8xFwyBmZo2e4j2/HgPg1QzanGGgUA9Hz84cH3zodVLIFNFk4+Zmb3IvadwCL79PI5tOMnR5Z2YcCFMtoi7ND4D0o/3V1G9BWfisWWG4zpWr7NvGuWBGr7cMlQ9IVvtl6hg0IodNauaeevbho5i61L3ENlZXaZdpeIQETejNZMs+JIA1F+15WWO8Yx9 xh9DteKt WBJwjeI0bhPyaaikS+wfnod5u7yo1mLDqKytU3+RVHGRrlOk+ns1tXUHx0md8jIkH/VSgqr3TUk8FeqZ1h8ER4hBPTQ/KpJYB+L28vvyebAbXpS1EWp/cW9io7Lw7yNOu6HOwCmdYCA5Rh0yR22VPpcUi1+IKEgsRagJ/sDkZ8CCEtuApU6Ss+xQQ2tgBGDb6NPWNMtJ/9iCf6zVG79Kacpphsg== X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: This extends test_prctl_fork() and test_prctl_fork_exec() to make sure that deduplication really happens, instead of only testing the MMF_VM_MERGE_ANY flag is set. Suggested-by: David Hildenbrand Signed-off-by: Jinjiang Tu Reviewed-by: David Hildenbrand --- .../selftests/mm/ksm_functional_tests.c | 49 ++++++++++++++----- 1 file changed, 38 insertions(+), 11 deletions(-) diff --git a/tools/testing/selftests/mm/ksm_functional_tests.c b/tools/testing/selftests/mm/ksm_functional_tests.c index 5462873df84c..2d277620fad2 100644 --- a/tools/testing/selftests/mm/ksm_functional_tests.c +++ b/tools/testing/selftests/mm/ksm_functional_tests.c @@ -475,6 +475,36 @@ static void test_prctl(void) ksft_test_result_pass("Setting/clearing PR_SET_MEMORY_MERGE works\n"); } +static int test_child_ksm(void) +{ + const unsigned int size = 2 * MiB; + char *map; + + /* Test if KSM is enabled for the process. */ + if (prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0) != 1) + return -1; + + /* Test if merge could really happen. */ + map = __mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_NONE); + if (map == MAP_MERGE_FAIL) + return -2; + else if (map == MAP_MERGE_SKIP) + return -3; + + munmap(map, size); + return 0; +} + +static void test_child_ksm_err(int status) +{ + if (status == -1) + ksft_test_result_fail("unexpected PR_GET_MEMORY_MERGE result in child\n"); + else if (status == -2) + ksft_test_result_fail("Merge in child failed\n"); + else if (status == -3) + ksft_test_result_skip("Merge in child skiped\n"); +} + /* Verify that prctl ksm flag is inherited. */ static void test_prctl_fork(void) { @@ -494,7 +524,7 @@ static void test_prctl_fork(void) child_pid = fork(); if (!child_pid) { - exit(prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0)); + exit(test_child_ksm()); } else if (child_pid < 0) { ksft_test_result_fail("fork() failed\n"); return; @@ -503,8 +533,11 @@ static void test_prctl_fork(void) if (waitpid(child_pid, &status, 0) < 0) { ksft_test_result_fail("waitpid() failed\n"); return; - } else if (WEXITSTATUS(status) != 1) { - ksft_test_result_fail("unexpected PR_GET_MEMORY_MERGE result in child\n"); + } + + status = WEXITSTATUS(status); + if (status) { + test_child_ksm_err(status); return; } @@ -516,12 +549,6 @@ static void test_prctl_fork(void) ksft_test_result_pass("PR_SET_MEMORY_MERGE value is inherited\n"); } -static int ksm_fork_exec_child(void) -{ - /* Test if KSM is enabled for the process. */ - return prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0) == 1; -} - static void test_prctl_fork_exec(void) { int ret, status; @@ -554,7 +581,7 @@ static void test_prctl_fork_exec(void) if (WIFEXITED(status)) { status = WEXITSTATUS(status); if (status) { - ksft_test_result_fail("KSM not enabled\n"); + test_child_ksm_err(status); return; } } else { @@ -635,7 +662,7 @@ int main(int argc, char **argv) int err; if (argc > 1 && !strcmp(argv[1], FORK_EXEC_CHILD_PRG_NAME)) { - exit(ksm_fork_exec_child() == 1 ? 0 : 1); + exit(test_child_ksm()); } #ifdef __NR_userfaultfd