From patchwork Wed Mar 27 06:09:20 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Jinjiang Tu X-Patchwork-Id: 13605606 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 0B10AC47DD9 for ; Wed, 27 Mar 2024 06:09:58 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id D37E76B0095; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id CEE356B0096; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id BBEB96B0095; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0017.hostedemail.com [216.40.44.17]) by kanga.kvack.org (Postfix) with ESMTP id ABBB76B0095 for ; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) Received: from smtpin21.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay08.hostedemail.com (Postfix) with ESMTP id 4EC7614062E for ; Wed, 27 Mar 2024 06:09:54 +0000 (UTC) X-FDA: 81941793108.21.D1B0D02 Received: from szxga04-in.huawei.com (szxga04-in.huawei.com [45.249.212.190]) by imf15.hostedemail.com (Postfix) with ESMTP id 1E490A0005 for ; Wed, 27 Mar 2024 06:09:51 +0000 (UTC) Authentication-Results: imf15.hostedemail.com; dkim=none; spf=pass (imf15.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.190 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com; dmarc=pass (policy=quarantine) header.from=huawei.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1711519792; a=rsa-sha256; cv=none; b=zVtKtQTaYjwDjpgb4k6/IPRdyh2Cjb94r2rfmg4tb1X54hKxCga0Pa4nkYUG0nMpKQZfwn rK6GfQH4JdJa1G/96Y8cKf+kmqYNS7H0ubqrtVt2OdM1gNw4hkhGrilZ6Yp8bKxr6Ih2vh hGkKpRIkWF0weerRb+ykPlM9X4vP43Y= ARC-Authentication-Results: i=1; imf15.hostedemail.com; dkim=none; spf=pass (imf15.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.190 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=1711519792; 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=VSYd4pTxe45tA5Bx2we1UFSyrEfDYpNqJnIAcY3hM+s=; b=mPZ//A0m3ZHINZLzhmEQIxsTZ9ds8HpLMN1vNuhZB1w6wvica2aOkyzKBZV8xIGFqY1dMd NcIkaD1+v6YCb1+RlpF/YP39BHOvrGILwr9Y1CX1R7KfqTBl+FM4iZuOaLRvAPGC0fbbmk rlyaPRLi1QUoK+2/rVMe5uRoPbtmWJw= Received: from mail.maildlp.com (unknown [172.19.163.44]) by szxga04-in.huawei.com (SkyGuard) with ESMTP id 4V4GRd73Rhz1xsc4; Wed, 27 Mar 2024 14:07:41 +0800 (CST) Received: from dggpeml500011.china.huawei.com (unknown [7.185.36.84]) by mail.maildlp.com (Postfix) with ESMTPS id 40A751402CA; Wed, 27 Mar 2024 14:09:42 +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; Wed, 27 Mar 2024 14:09:41 +0800 From: Jinjiang Tu To: , , , , , , , CC: Subject: [PATCH v3 1/3] mm/ksm: fix ksm exec support for prctl Date: Wed, 27 Mar 2024 14:09:20 +0800 Message-ID: <20240327060922.1484395-2-tujinjiang@huawei.com> X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240327060922.1484395-1-tujinjiang@huawei.com> References: <20240327060922.1484395-1-tujinjiang@huawei.com> MIME-Version: 1.0 X-Originating-IP: [10.175.113.32] X-ClientProxiedBy: dggems703-chm.china.huawei.com (10.3.19.180) To dggpeml500011.china.huawei.com (7.185.36.84) X-Rspam-User: X-Rspamd-Server: rspam06 X-Rspamd-Queue-Id: 1E490A0005 X-Stat-Signature: i1hjmodmmjairpe7t6r41o6hpbkjobic X-HE-Tag: 1711519791-581099 X-HE-Meta: U2FsdGVkX1+m6ANCjDswk5TL71o104kXdjf0kplQwcEz0lJK6HvXZkMQvTnJ4IjIpm7Q9Q16/5jkm0R3hS1R2OovaZvz9pN39zjI8KbUB45FT2NRjL2q9pZ68oN7UforgcAH7zBVwBgSiIIB+viYUsphhbhjGkcFA+/s+m8SOVb6cuRf7R75jiepRbt4kvJz/PRzzKMGU3TXdcr9AHF4CfwN1cOfT2fimh8MnLfoW/ANqUgmLfRGDm5FHQbkpChaSSwBfuIVOVWIMF/MtY5torlmdCdSNwRIWydZnBqqKsm2w04kAQRRKGtNyeZm/WR1ji6K4AX1TuScj7URIig3e6Epdz7bL2nbUYY+EvdX6qd6zjSlf0zrREYm0YSkfVKwQXLSvoXxCcHQsjbPeGYuqlHBTzHIcTvqrSYpZ31KDU6rMX+bIVkataaq3ctSfuFgAxHRjePCRh4BRrTV5G794YFpTxdnWIQ2T1D/JPv5mgw327beeOjyfv8/8RK6R52vAnrcyox06F3n3xNHpH9Mneu8ktGx8QlJMMDy6paHomDQryju9ZmdbwiT1hQaIQENe4nuD2vtilIo7I4bg5HuV7hLMeTXMQOw3Mnixx2z4sA3ZU86yglmF9vFXHxQobd+lvIJniP3zkXX38s6FEttjBC2FVPbRYI/JM5cgJ/hnX5k5s4wdy+FIydTRjcp73vehqj7dMrhHAF9BR1PPrMWbG+1F5too+ee141KFCG+czvmWpHrwIpEior120y63pVjGFR1aVaIiJHr1Xz3WGOb6bHsWiB9X6dEu9adXDnThF9NoYk+IOirhQdEwk+0uuJiqloNHO5vFlrI1WLcGWi1vJooxnV2v31nMUZlBO15cjKzO97ycWr70GtMh+J/skdXB2nrb7VYLug+x6zmzXDTegXRETpDedyibnMWLWyHW6DbQ8I6oE9Yv4Zl5dcOFx1x9JPYjKQwMnIvmHWJqcD B3OP/TEW Ypcl8kLjQti+/7rPWrMTg3Q8jPRyD9QIEsVQvB5AtigRcXsW8ylRLcIxx21CozJGIqyN0mPYGDNfOBbSHrVGdafpyqUNbb6GpR8M11jkr9Pfuw3YKRv38LlWJuPg+eucL6iejkYBdwrxOgdPrt4lMIaK/0QTb0K2AusXUVxECUDuVzPJ6yjJsMpONnQ== 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 Wed Mar 27 06:09:21 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Jinjiang Tu X-Patchwork-Id: 13605605 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 CCDD6C54E67 for ; Wed, 27 Mar 2024 06:09:55 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 87E3F6B008A; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 82BE36B0095; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 726376B0096; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0012.hostedemail.com [216.40.44.12]) by kanga.kvack.org (Postfix) with ESMTP id 61EB26B008A for ; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) Received: from smtpin01.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay05.hostedemail.com (Postfix) with ESMTP id 2F15E406D6 for ; Wed, 27 Mar 2024 06:09:54 +0000 (UTC) X-FDA: 81941793108.01.4DD1CD6 Received: from szxga04-in.huawei.com (szxga04-in.huawei.com [45.249.212.190]) by imf10.hostedemail.com (Postfix) with ESMTP id B8AD2C0004 for ; Wed, 27 Mar 2024 06:09:51 +0000 (UTC) Authentication-Results: imf10.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf10.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=1711519792; 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=NhPLJM3Cd3azoJDiiiyuK4BfrsuZ+cy36ol4s5YcX0k=; b=ITH+B3UvJuhgCH8Sjw3pzWaPMjSH/qRc0YgnjWfqj1jKdgf0rwwHh5iGSGwlIodE8DzVgg g5myBclLqO+HB2SVcq5Jn2L/dN4M6iNSvLVZ70Nuz6npLj5gvFkN9fN/SLskhd6Pcr70BZ P1+JetoNbi7B7mPWZsmjfPgoYRXStik= ARC-Authentication-Results: i=1; imf10.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf10.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=1711519792; a=rsa-sha256; cv=none; b=AWljXH3lSPT1oQuXk/fQ79V95+7rP95FdrEyTv5GTXuaBj2GU7PipiLxAi4O8jT42XdeWA h48QFDykbySDauFTaAWlWjNSeSLaNn/nXREqBvBTJJUiAwgvB3mHawW9tLnuIBGlHh++zF u1R+F9pTX7kOAEddoGn7OkqG8KJDpDM= Received: from mail.maildlp.com (unknown [172.19.162.112]) by szxga04-in.huawei.com (SkyGuard) with ESMTP id 4V4GQv2PZtz2Bj1M; Wed, 27 Mar 2024 14:07:03 +0800 (CST) Received: from dggpeml500011.china.huawei.com (unknown [7.185.36.84]) by mail.maildlp.com (Postfix) with ESMTPS id B44921400DA; Wed, 27 Mar 2024 14:09:42 +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; Wed, 27 Mar 2024 14:09:42 +0800 From: Jinjiang Tu To: , , , , , , , CC: Subject: [PATCH v3 2/3] selftest/mm: ksm_functional_tests: refactor mmap_and_merge_range() Date: Wed, 27 Mar 2024 14:09:21 +0800 Message-ID: <20240327060922.1484395-3-tujinjiang@huawei.com> X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240327060922.1484395-1-tujinjiang@huawei.com> References: <20240327060922.1484395-1-tujinjiang@huawei.com> MIME-Version: 1.0 X-Originating-IP: [10.175.113.32] X-ClientProxiedBy: dggems703-chm.china.huawei.com (10.3.19.180) To dggpeml500011.china.huawei.com (7.185.36.84) X-Rspam-User: X-Stat-Signature: 6kbmubeijby836s3kop6kufuqixy1odc X-Rspamd-Server: rspam07 X-Rspamd-Queue-Id: B8AD2C0004 X-HE-Tag: 1711519791-561478 X-HE-Meta: U2FsdGVkX1/czfJ/XzgXYHLu8yD8QqMi44Ydj/+HvVgSmssN7etHUgzzxVHqUblfXwsauQ7ILEaaURKCrIf5cKm49CFHk2e+XfIzR+ydPysyHg71q4bNGbHvohmQuo55oLOSAMtgN7Y0TVOqNqWmNrkX1kDZmBivrSgKOZ7u/rqQUmAR2VTG0Pd2Krp7zC2wNZYnAaV+RovovxecIbjTp9wTjfN9IqkT/P5ukmFxIKlh5sTMG2CV69VMA/BMQ8QGUEk6R29olykn4WT153Faj0olJAaukehlb+Q8lVJgktsKWhrT6AmZBNXZlXf3MCDxgs8XMpJV6pPJRxiOV1oBl3qStTcwO1WwjVaeU2iE8x/OyIVgYQMiXZ/gv9G4HR2uOm78XAZupzy09fYmBrOmJUqWzLWyzVpTxK80TWRuT47n961Pf9YjoMqpqXZ9MLiQ3ezzVyGzCmFYVaDm370E8NMIDQ1M+UBwYJtn1+fmCedM4SI3bEaJMFQJk7EnfpG6UQGIdDqFlAt5UTUbxGgq+ehB8WWhX9+6e5W75zLMnDZUDlZk6kJlzcDrbyprmd6398FbWIelD+iQ6QKo6FqKkVJwpbcxqhdLzX2HWIRSkTv2yCFjgbkG5cuwqdWOtwsr4s1ygLGeRHiu9Jdd6s4P3+PI5zXaBMBgG7+FUZw3F8gw+TDFgo4YOOJMsyN87Ud4GGu6uWn05xKlKPOsxatk0tUDih7Dg2eQT1BePkHbXv22Gs0xN/cw4o4n8U/KuOI7LJZ9T31oHm2iq9sDg9RKypoe68pgMDA22rPtjXVNIPc9wL5LvkSop2eCHAcBjjoSU0I0b2bQa0uPFSt7GVWucVmJ10Kn+Lmw/twB8Lty2MeyhFAKPmMFuT1+eHEEfbOMl+WZJSVROZcTUUHrx/CrwRMpYW+3kzzCuF3NrcxPVloN6Ootv+XPBz6kwHr/N+dPmcaPtZyjS8fjvcBxHEZ BxfP5cKJ UjxoRpkJaoQfNng+rXd9g6KBtqOsp24gPkPhY7FmmGB6HvLXM0FJcZSaOk0f6KCsLL2ZpYJHZOGX/8d8H8jEsF/767WQQ3yymiUWgeVe5ja8kg6Vc+SkI7R7OylaMyNLjlZchF6TViGAOE0TR76n5iH/4/4wFr6wfa5yt/r7kodV2xhbVa05yc+6jhsBUznFYDHKooz+hnxmNoZFpW6hwYNlS3Q== 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() and return differrent error according to the two cases. The callers of mmap_and_merge_range() then call ksft_test_result_{fail, skip} according to the return value. Introduce mmap_and_merge_err_print() helper to make it easier. Suggested-by: David Hildenbrand Signed-off-by: Jinjiang Tu --- .../selftests/mm/ksm_functional_tests.c | 93 +++++++++++++------ 1 file changed, 63 insertions(+), 30 deletions(-) diff --git a/tools/testing/selftests/mm/ksm_functional_tests.c b/tools/testing/selftests/mm/ksm_functional_tests.c index d615767e396b..da58545e58e1 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; @@ -147,32 +156,33 @@ static int ksm_unmerge(void) } static char *mmap_and_merge_range(char val, unsigned long size, int prot, - bool use_prctl) + 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,28 @@ 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 inline int mmap_and_merge_err_print(char *map) +{ + int ret = -1; + + if (map == MAP_MERGE_FAIL) + ksft_test_result_fail("Merging memory failed"); + else if (map == MAP_MERGE_FAIL) + ksft_test_result_skip("Merging memory skiped"); + else + ret = 0; + + return ret; } static void test_unmerge(void) @@ -226,8 +259,8 @@ static void test_unmerge(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, false); - if (map == MAP_FAILED) + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); + if (mmap_and_merge_err_print(map)) return; if (madvise(map, size, MADV_UNMERGEABLE)) { @@ -264,8 +297,8 @@ static void test_unmerge_zero_pages(void) } /* Let KSM deduplicate zero pages. */ - map = mmap_and_merge_range(0x00, size, PROT_READ | PROT_WRITE, false); - if (map == MAP_FAILED) + map = mmap_and_merge_range(0x00, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); + if (mmap_and_merge_err_print(map)) return; /* Check if ksm_zero_pages is updated correctly after KSM merging */ @@ -312,8 +345,8 @@ 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); - if (map == MAP_FAILED) + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); + if (mmap_and_merge_err_print(map)) return; /* Discard half of all mapped pages so we have pte_none() entries. */ @@ -344,8 +377,8 @@ 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); - if (map == MAP_FAILED) + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_MADVISE); + if (mmap_and_merge_err_print(map)) return; /* See if UFFD is around. */ @@ -545,8 +578,8 @@ 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); - if (map == MAP_FAILED) + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_PRCTL); + if (mmap_and_merge_err_print(map)) return; if (prctl(PR_SET_MEMORY_MERGE, 0, 0, 0, 0)) { @@ -568,8 +601,8 @@ static void test_prot_none(void) ksft_print_msg("[RUN] %s\n", __func__); - map = mmap_and_merge_range(0x11, size, PROT_NONE, false); - if (map == MAP_FAILED) + map = mmap_and_merge_range(0x11, size, PROT_NONE, KSM_MERGE_MADVISE); + if (mmap_and_merge_err_print(map)) goto unmap; /* Store a unique value in each page on one half using ptrace */ From patchwork Wed Mar 27 06:09:22 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Jinjiang Tu X-Patchwork-Id: 13605607 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 50055CD1280 for ; Wed, 27 Mar 2024 06:10:00 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 01BBF6B0096; Wed, 27 Mar 2024 02:09:55 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id EC4626B009A; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id C72206B0099; Wed, 27 Mar 2024 02:09:54 -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 B64C26B0098 for ; Wed, 27 Mar 2024 02:09:54 -0400 (EDT) Received: from smtpin06.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay03.hostedemail.com (Postfix) with ESMTP id 7CBACA0994 for ; Wed, 27 Mar 2024 06:09:54 +0000 (UTC) X-FDA: 81941793108.06.AA709BD Received: from szxga05-in.huawei.com (szxga05-in.huawei.com [45.249.212.191]) by imf20.hostedemail.com (Postfix) with ESMTP id DE0101C000D for ; Wed, 27 Mar 2024 06:09:51 +0000 (UTC) Authentication-Results: imf20.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf20.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.191 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=1711519792; 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=3wm5s2+OnW88nMQQZAuHTrskP4Ow1S/ft1oHx7m30Ww=; b=Pa1VrofXAyPhFm7NNjUtbKBxjWJ0Xwd+cUK7dmIGzqSTAAFI31cXafpH+X8Nwkyfm+xcl6 5wJhzoWxIm605A7E4mJtS2BChLd9CpW/dZEdvHM/Q4owdXEqhEJEKdxt//6qnbyEIppy56 zlnw2Zibi509OSvU9eoyNLwwSRbrcs4= ARC-Authentication-Results: i=1; imf20.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf20.hostedemail.com: domain of tujinjiang@huawei.com designates 45.249.212.191 as permitted sender) smtp.mailfrom=tujinjiang@huawei.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1711519792; a=rsa-sha256; cv=none; b=RYAGckQNVgUHe+/JYLZvNMqi3wFc/db7JVVD9jWhRBsBLb74iWpxxsN5LtyG0xTi1a6NUh k0E4qe/CHEvPQ8qTf1Sk8lwiJ2lkrC8gjykQvQHwtXWiVGoB7NQWZVM0PaNjRMAgI4p51u SG+uV5+tX87LfjDpf26f/d7DgqqKAyo= Received: from mail.maildlp.com (unknown [172.19.163.44]) by szxga05-in.huawei.com (SkyGuard) with ESMTP id 4V4GQv736Bz1h45m; Wed, 27 Mar 2024 14:07:03 +0800 (CST) Received: from dggpeml500011.china.huawei.com (unknown [7.185.36.84]) by mail.maildlp.com (Postfix) with ESMTPS id 307561402CA; Wed, 27 Mar 2024 14:09:43 +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; Wed, 27 Mar 2024 14:09:42 +0800 From: Jinjiang Tu To: , , , , , , , CC: Subject: [PATCH v3 3/3] selftest/mm: ksm_functional_tests: extend test case for ksm fork/exec Date: Wed, 27 Mar 2024 14:09:22 +0800 Message-ID: <20240327060922.1484395-4-tujinjiang@huawei.com> X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240327060922.1484395-1-tujinjiang@huawei.com> References: <20240327060922.1484395-1-tujinjiang@huawei.com> MIME-Version: 1.0 X-Originating-IP: [10.175.113.32] X-ClientProxiedBy: dggems703-chm.china.huawei.com (10.3.19.180) To dggpeml500011.china.huawei.com (7.185.36.84) X-Rspam-User: X-Stat-Signature: bhmewitrg5fwt8szojtzh48ks6gfktzk X-Rspamd-Server: rspam07 X-Rspamd-Queue-Id: DE0101C000D X-HE-Tag: 1711519791-461204 X-HE-Meta: U2FsdGVkX1/doTEYtnQS+Efi7rTFpMAkaeM4cif8N//74pVgyM/lXz89Wp8aebzXzJzhRF+BEbtGw+1/WloBhca0DGSlsHn2FvuoT9E1O8D43eT4HPn9Kwzwjj6d4aNG/HF0Far2c/ULEAMfERD8BL736cZlD95kRH/rMRlnUHgQ1KeUbwO1rdDXFWa1s7NHsD9cQL4SOEBXUdBfv2X7wyfkIegn/1yQP/oNQ/rBc3lZ0TxlyUh2sYIESBWxtPck07C17PR7kpdJFvwptNzNYrvpozFMdd+u7cHaXSbQoYMCWv0jVvn7AlHb7pPOeJk0PHRWR0++QwD5X5MpezSw2H6LismazWGFelJRmewAz2m0iSkbQ7tq9tbrlnt5YdnmwpYVwXtedPTAHdam8RFvbnRq1+ySZkcCZHphsYS+SspAQ/0P2CSF3Kw7FVaSwuwrZiPdGZu0nGobcQj+TBwYWXovNEiCRqLYzfCEO1AriLFyxCzpm8HfAaRq6TAF8caO2O9uZ7sQzgqok3QxK8QM3J3r8N4yBo+R4oj6m1h/Xn+QoxPFJwpHnEr8dT0fEpF3ebluWoj209Et7RubtHKWhpI8RiDwdFXNzwD8YValFvV2IoP7B/EFSFCAPGC9opPFhwFVNaTZUuFRnBUbWM/WfJvm5z53Dn7VTUeVR/pifjfdbkVUl0+ZZi/u56I+EK88C+m8dbq2+Q5NOFHsR68ItceKxXlFgKLZGGMXXIyLVGR/MLHFwCfsxmE9GY/auPStDaOJIoKuMcXECIDKNcwtKSILL53hDQT/oUWz0eZGUbDGTdkS3A9NrB37OmMaZM7OkEDGZ+n+m2eqiRfN69Q+HMj9wmehV6luRqTgVNY1XTGZuPFwTxKHNsNcffhcTDBjc1h/owzXsGHSFaM/P9ilZZpKNhHIYWEkQp9MejI6nsfOr0Ue0qaWWmVtuGQGiTkvcGkHqqsswIFk8sMSBXI Jg973gJE KwJrX8hgWRHKsC1AhDiPykSJsEEB2W1LpmxBc8KbKz1+QZxI6ZJJus7yHxRkbYahp9qsovS/SAUjwXcBrhfq+F+yG0BDHuTSzrDe6OCaW4UaigRocddonteOwzr8bk3Mk8x27I3ER6yrFJ9Yv81Q2WwzLNwTuzRKhmyPSWiXVXXJA5IvYUaHEx9CjCsLsGPdjjzzr+7rQFjreS4yXTGRmOWKK8g== 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 test the MMF_VM_MERGE_ANY flag is set. Suggested-by: David Hildenbrand Signed-off-by: Jinjiang Tu --- .../selftests/mm/ksm_functional_tests.c | 52 ++++++++++++++++--- 1 file changed, 46 insertions(+), 6 deletions(-) diff --git a/tools/testing/selftests/mm/ksm_functional_tests.c b/tools/testing/selftests/mm/ksm_functional_tests.c index da58545e58e1..c56e34946096 100644 --- a/tools/testing/selftests/mm/ksm_functional_tests.c +++ b/tools/testing/selftests/mm/ksm_functional_tests.c @@ -491,7 +491,20 @@ static void test_prctl_fork(void) child_pid = fork(); if (!child_pid) { - exit(prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0)); + const unsigned int size = 2 * MiB; + char *map; + + if (prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0) != 1) + exit(-1); + + map = mmap_and_merge_range(0xcf, size, PROT_READ | PROT_WRITE, KSM_MERGE_NONE); + if (map == MAP_MERGE_FAIL) + exit(-2); + else if (map == MAP_MERGE_SKIP) + exit(-3); + + munmap(map, size); + exit(0); } else if (child_pid < 0) { ksft_test_result_fail("fork() failed\n"); return; @@ -500,8 +513,16 @@ 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 != 0) { + 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"); return; } @@ -515,8 +536,21 @@ static void test_prctl_fork(void) static int ksm_fork_exec_child(void) { + const unsigned int size = 2 * MiB; + char *map; + /* Test if KSM is enabled for the process. */ - return prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0) == 1; + if (prctl(PR_GET_MEMORY_MERGE, 0, 0, 0, 0) != 1) + return -1; + + 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_prctl_fork_exec(void) @@ -550,9 +584,15 @@ static void test_prctl_fork_exec(void) if (waitpid(child_pid, &status, 0) > 0) { if (WIFEXITED(status)) { status = WEXITSTATUS(status); - if (status) { + if (status == -1) { ksft_test_result_fail("KSM not enabled\n"); return; + } else if (status == -2) { + ksft_test_result_fail("Merge in child failed\n"); + return; + } else if (status == -3) { + ksft_test_result_skip("Merge in child skiped\n"); + return; } } else { ksft_test_result_fail("program didn't terminate normally\n"); @@ -632,7 +672,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(ksm_fork_exec_child()); } #ifdef __NR_userfaultfd