From patchwork Wed Feb 21 08:18:01 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: "GONG, Ruiqi" X-Patchwork-Id: 13565091 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 2940EC48BF6 for ; Wed, 21 Feb 2024 08:11:42 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 3FB086B009C; Wed, 21 Feb 2024 03:11:42 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id 3AAFE6B009D; Wed, 21 Feb 2024 03:11:42 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 29A986B009E; Wed, 21 Feb 2024 03:11:42 -0500 (EST) 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 19A666B009C for ; Wed, 21 Feb 2024 03:11:42 -0500 (EST) Received: from smtpin03.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay08.hostedemail.com (Postfix) with ESMTP id E35AE140986 for ; Wed, 21 Feb 2024 08:11:41 +0000 (UTC) X-FDA: 81815092002.03.AF455D6 Received: from szxga03-in.huawei.com (szxga03-in.huawei.com [45.249.212.189]) by imf22.hostedemail.com (Postfix) with ESMTP id B0C6AC000A for ; Wed, 21 Feb 2024 08:11:38 +0000 (UTC) Authentication-Results: imf22.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf22.hostedemail.com: domain of gongruiqi1@huawei.com designates 45.249.212.189 as permitted sender) smtp.mailfrom=gongruiqi1@huawei.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1708503100; 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: references; bh=QswBzDOL/bQpw/1cySEjF/CQyPWFC1XkxLNwNJXPJPc=; b=LZpL8rslOj6T86rPDsGZ0nJljy0Uzw4XRmS+yM65vjAJ1VkOyA54FJ/f44DIPQTmnqNv0j Zw8cYVUy80FSNPfq1n+BEi6Uh0EwTicPdNipLkZ/uwn1DcIvtRB4inchUjDH8tEdLvTNix doo66NpLxOD0qqTicJT4/EnqoI1L9sY= ARC-Authentication-Results: i=1; imf22.hostedemail.com; dkim=none; dmarc=pass (policy=quarantine) header.from=huawei.com; spf=pass (imf22.hostedemail.com: domain of gongruiqi1@huawei.com designates 45.249.212.189 as permitted sender) smtp.mailfrom=gongruiqi1@huawei.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1708503100; a=rsa-sha256; cv=none; b=8J3BYKuLK2Mu6ycRziytTGzumuntbUnN9EEUOgLV7zML5FKda1khk0Tr6bfPLnotQrTuRA biZcQJBQAQiy3lorc3S7KhL5v3XnXD3dFfQN6sD8Heh28qFnwBlWwRJ7B3LLDHAHF3Qvbt H3mM9t3Vn03a+slj6kOHSv6ybSz3OBw= Received: from mail.maildlp.com (unknown [172.19.88.105]) by szxga03-in.huawei.com (SkyGuard) with ESMTP id 4Tfpq72rMSzNlm7; Wed, 21 Feb 2024 16:10:11 +0800 (CST) Received: from dggpemd100002.china.huawei.com (unknown [7.185.36.164]) by mail.maildlp.com (Postfix) with ESMTPS id 55334140114; Wed, 21 Feb 2024 16:11:34 +0800 (CST) Received: from huawei.com (10.67.174.33) by dggpemd100002.china.huawei.com (7.185.36.164) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.2.1258.28; Wed, 21 Feb 2024 16:11:34 +0800 From: "GONG, Ruiqi" To: , , Johannes Weiner , Michal Hocko , Roman Gushchin , Shakeel Butt , Muchun Song CC: , , Wang Weiyang , Xiu Jianfeng Subject: [PATCH stable] memcg: add refcnt for pcpu stock to avoid UAF problem in drain_all_stock() Date: Wed, 21 Feb 2024 16:18:01 +0800 Message-ID: <20240221081801.69764-1-gongruiqi1@huawei.com> X-Mailer: git-send-email 2.25.1 MIME-Version: 1.0 X-Originating-IP: [10.67.174.33] X-ClientProxiedBy: dggems703-chm.china.huawei.com (10.3.19.180) To dggpemd100002.china.huawei.com (7.185.36.164) X-Rspam-User: X-Stat-Signature: ojbat41gh1i41do5gkq6pfktjxn6cm98 X-Rspamd-Server: rspam07 X-Rspamd-Queue-Id: B0C6AC000A X-HE-Tag: 1708503098-852679 X-HE-Meta: U2FsdGVkX1/es1QzUKilMFICtdCkaRh264hf6nL5mMBfYx1wMTv8sEG4ooex/fsSrQ1SL8qqq7lrqmpHw3hFPZMjVNFh7sT0kMGnLj0d7Gb2Bjwtqwardxj5SRGKxVAO45GAvjoREYvgGLmcFT/58YMKmECCqkzCl9tX/Ap4EfxnzHeFa9hWOYNmm40Axmne1tPmteA0I1PZ63qQSi+K0zvIe9NYXGUh8N8oYvmb0WexEriw/HGjmjnmogPjSs4TXgpcdQgJvz6OjcgPZVH/JfZrpUiUQBX/IzwVUCADjqC9a7bUArPXLDa/P9GYQ8vMTp2SXSpmsiEhQ66OmkUt8AyIOfAPdMv76rdilfTk26TW4L3he7fu81v49Aw+jRXMQxg9UpblUzWIg9gNwV0UtYPBxqp3lzPAXeZYeQaVRzoZutt5HVJsDdQZJsWflKbRO+GVUsOQ72mK7Ds7AXirqazLTSOMAKjt89komrwd9ZXl2CAssqZBr6tN3AsAwIhYte2V72Jc3tDEARCzg0wimu6BH/UJ8H7ycgmIJWoSg3TWJYsbPjkfH0dbbtykkstBCf9tprxpe1NfudA0UGOx/As1myIh1oTA7Ed902Dz2b/6BJECoPXaB3YjdT6mhGodpbXLmsL/kew41mte7ik38dTuZlgQLmdUg9Xa04DpWQIrLZ7hNluyTkC3tyY8SBqe/7/d1SWRGZNqzYvgDdMJMOTpBCUa+4bD8b/dsOrVZYtRzFkoTzVvXGgcHXktDeqzuIezSM0puaosf25Pe8gku1+ZV7R2cWvgCIYoBdGwnwZXUPEr6iGtVrWF/gH3m4F5KWidyCXKzaXdXp/D/N+QZduUOkpai4gsmMbv41aokIxnJHvrQKN1NcSt4L9hEJzGrbMAlCBYOrc= 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 1a3e1f40962c445b997151a542314f3c6097f8c3 upstream. There was a kernel panic happened on an in-house environment running 3.10, and the same problem was reproduced on 4.19: general protection fault: 0000 [#1] SMP PTI CPU: 1 PID: 2085 Comm: bash Kdump: loaded Tainted: G L 4.19.90+ #7 Hardware name: QEMU Standard PC (i440FX + PIIX, 1996), BIOS rel-1.15.0-0-g2dd4b9b3f840-prebuilt.qemu.org 04/01/2014 RIP: 0010 drain_all_stock+0xad/0x140 Code: 00 00 4d 85 ff 74 2c 45 85 c9 74 27 4d 39 fc 74 42 41 80 bc 24 28 04 00 00 00 74 17 49 8b 04 24 49 8b 17 48 8b 88 90 02 00 00 <48> 39 8a 90 02 00 00 74 02 eb 86 48 63 88 3c 01 00 00 39 8a 3c 01 RSP: 0018:ffffa7efc5813d70 EFLAGS: 00010202 RAX: ffff8cb185548800 RBX: ffff8cb89f420160 RCX: ffff8cb1867b6000 RDX: babababababababa RSI: 0000000000000001 RDI: 0000000000231876 RBP: 0000000000000000 R08: 0000000000000415 R09: 0000000000000002 R10: 0000000000000000 R11: 0000000000000001 R12: ffff8cb186f89040 R13: 0000000000020160 R14: 0000000000000001 R15: ffff8cb186b27040 FS: 00007f4a308d3740(0000) GS:ffff8cb89f440000(0000) knlGS:0000000000000000 CS: 0010 DS: 0000 ES: 0000 CR0: 0000000080050033 CR2: 00007ffe4d634a68 CR3: 000000010b022000 CR4: 00000000000006e0 DR0: 0000000000000000 DR1: 0000000000000000 DR2: 0000000000000000 DR3: 0000000000000000 DR6: 00000000fffe0ff0 DR7: 0000000000000400 Call Trace: mem_cgroup_force_empty_write+0x31/0xb0 cgroup_file_write+0x60/0x140 ? __check_object_size+0x136/0x147 kernfs_fop_write+0x10e/0x190 __vfs_write+0x37/0x1b0 ? selinux_file_permission+0xe8/0x130 ? security_file_permission+0x2e/0xb0 vfs_write+0xb6/0x1a0 ksys_write+0x57/0xd0 do_syscall_64+0x63/0x250 ? async_page_fault+0x8/0x30 entry_SYSCALL_64_after_hwframe+0x5c/0xc1 Modules linked in: ... It is found that in case of stock->nr_pages == 0, the memcg on stock->cached could be freed due to its refcnt decreased to 0, which made stock->cached become a dangling pointer. It could cause a UAF problem in drain_all_stock() in the following concurrent scenario. Note that drain_all_stock() doesn't disable irq but only preemption. CPU1 CPU2 ============================================================================== stock->cached = memcgA (freed) drain_all_stock(memcgB) rcu_read_lock() memcg = CPU1's stock->cached (memcgA) (interrupted) refill_stock(memcgC) drain_stock(memcgA) stock->cached = memcgC stock->nr_pages += xxx (> 0) stock->nr_pages > 0 mem_cgroup_is_descendant(memcgA, memcgB) [UAF] rcu_read_unlock() This problem is, unintenionally, fixed at 5.9, where commit 1a3e1f40962c ("mm: memcontrol: decouple reference counting from page accounting") adds memcg refcnt for stock. Therefore affected LTS versions include 4.19 and 5.4. For 4.19, memcg's css offline process doesn't call drain_all_stock(). so it's easier for the released memcg to be left on the stock. For 5.4, although mem_cgroup_css_offline() does call drain_all_stock(), but the flushing could be skipped when stock->nr_pages happens to be 0, and besides the async draining could be delayed and take place after the UAF problem has happened. Fix this problem by adding (and decreasing) memcg's refcnt when memcg is put onto (and removed from) stock, just like how commit 1a3e1f40962c ("mm: memcontrol: decouple reference counting from page accounting") does. After all, "being on the stock" is a kind of reference with regards to memcg. As such, it's guaranteed that a css on stock would not be freed. Cc: stable@vger.kernel.org # 4.19 5.4 Fixes: cdec2e4265df ("memcg: coalesce charging via percpu storage") Signed-off-by: GONG, Ruiqi Acked-by: Michal Hocko --- mm/memcontrol.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mm/memcontrol.c b/mm/memcontrol.c index 5a366cf79821..8c04296df1c7 100644 --- a/mm/memcontrol.c +++ b/mm/memcontrol.c @@ -2015,6 +2015,9 @@ static void drain_stock(struct memcg_stock_pcp *stock) { struct mem_cgroup *old = stock->cached; + if (!old) + return; + if (stock->nr_pages) { page_counter_uncharge(&old->memory, stock->nr_pages); if (do_memsw_account()) @@ -2022,6 +2025,8 @@ static void drain_stock(struct memcg_stock_pcp *stock) css_put_many(&old->css, stock->nr_pages); stock->nr_pages = 0; } + + css_put(&old->css); stock->cached = NULL; } @@ -2057,6 +2062,7 @@ static void refill_stock(struct mem_cgroup *memcg, unsigned int nr_pages) stock = this_cpu_ptr(&memcg_stock); if (stock->cached != memcg) { /* reset if necessary */ drain_stock(stock); + css_get(&memcg->css); stock->cached = memcg; } stock->nr_pages += nr_pages;