From patchwork Fri Oct 20 13:31:57 2023 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Hou Tao X-Patchwork-Id: 13430743 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 6E050CDB474 for ; Fri, 20 Oct 2023 13:31:07 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id DE0838D00D0; Fri, 20 Oct 2023 09:31:06 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id D9B578D00BC; Fri, 20 Oct 2023 09:31:06 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id C801A8D00D0; Fri, 20 Oct 2023 09:31:06 -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 B540C8D0003 for ; Fri, 20 Oct 2023 09:31:06 -0400 (EDT) Received: from smtpin09.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay07.hostedemail.com (Postfix) with ESMTP id 83B1C160F2C for ; Fri, 20 Oct 2023 13:31:06 +0000 (UTC) X-FDA: 81365925732.09.D813C74 Received: from dggsgout11.his.huawei.com (dggsgout11.his.huawei.com [45.249.212.51]) by imf18.hostedemail.com (Postfix) with ESMTP id 25CEE1C0019 for ; Fri, 20 Oct 2023 13:31:00 +0000 (UTC) Authentication-Results: imf18.hostedemail.com; dkim=none; spf=none (imf18.hostedemail.com: domain of houtao@huaweicloud.com has no SPF policy when checking 45.249.212.51) smtp.mailfrom=houtao@huaweicloud.com; dmarc=none ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1697808663; 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-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=81mFlfTc1r40XBjlg2DB+PfcJDKb+iScqkXbFv6ZLMw=; b=bYEUsvteawGZ2ydqfXQV0gnNHsd/bg5dM+qb246p6Oy2QSeCtAnMEI+I6vovPaz2tNYsdQ Vk8m695BSwtvlP3CmGqwYm4d+nF0e3cAygy8D2LVXN4GgmstXy9HfXw58pQcphJYs6ONUL otvhbomBQ19LPgxtCD795+eZ8rIoRnM= ARC-Authentication-Results: i=1; imf18.hostedemail.com; dkim=none; spf=none (imf18.hostedemail.com: domain of houtao@huaweicloud.com has no SPF policy when checking 45.249.212.51) smtp.mailfrom=houtao@huaweicloud.com; dmarc=none ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1697808663; a=rsa-sha256; cv=none; b=qdc2wgep526rb/bA9dR5j5C84kBF2R9JGCdLWB8LvoamptfT7UhYoyykVQRo1RTSyf64vB Kj0uNTbbM7FN18eCIvVEymqrYBLNgj5RnrmVPmZUjySfuEnI1Sw3laFZupbhKCmk57IgS5 V5jU8jErmwOHC42tXNOo26zzS7eHYW8= Received: from mail02.huawei.com (unknown [172.30.67.153]) by dggsgout11.his.huawei.com (SkyGuard) with ESMTP id 4SBlpN4mjbz4f3mHg for ; Fri, 20 Oct 2023 21:30:52 +0800 (CST) Received: from huaweicloud.com (unknown [10.175.124.27]) by APP4 (Coremail) with SMTP id gCh0CgBn+dgIgTJlmYjjDQ--.7231S6; Fri, 20 Oct 2023 21:30:55 +0800 (CST) From: Hou Tao To: bpf@vger.kernel.org, linux-mm@kvack.org Cc: Martin KaFai Lau , Alexei Starovoitov , Andrii Nakryiko , Song Liu , Hao Luo , Yonghong Song , Daniel Borkmann , KP Singh , Stanislav Fomichev , Jiri Olsa , John Fastabend , houtao1@huawei.com, Dennis Zhou , Tejun Heo , Christoph Lameter , Andrew Morton Subject: [PATCH bpf-next v3 2/7] mm/percpu.c: introduce pcpu_alloc_size() Date: Fri, 20 Oct 2023 21:31:57 +0800 Message-Id: <20231020133202.4043247-3-houtao@huaweicloud.com> X-Mailer: git-send-email 2.29.2 In-Reply-To: <20231020133202.4043247-1-houtao@huaweicloud.com> References: <20231020133202.4043247-1-houtao@huaweicloud.com> MIME-Version: 1.0 X-CM-TRANSID: gCh0CgBn+dgIgTJlmYjjDQ--.7231S6 X-Coremail-Antispam: 1UD129KBjvJXoW7Aw13Xr4kXw1kZFyUur15urg_yoW5JFy3pF Wkuryrtr48Xrn7Ww1ft3WUZw4rWw4kWF4xJay3WFy3Zr1aqFyYgr1qyrW5uFyrGFn2vr12 qrZ0qFWxCFW5J3DanT9S1TB71UUUUUUqnTZGkaVYY2UrUUUUjbIjqfuFe4nvWSU5nxnvy2 9KBjDU0xBIdaVrnRJUUUBFb4IE77IF4wAFF20E14v26rWj6s0DM7CY07I20VC2zVCF04k2 6cxKx2IYs7xG6rWj6s0DM7CIcVAFz4kK6r1j6r18M28IrcIa0xkI8VA2jI8067AKxVWUXw A2048vs2IY020Ec7CjxVAFwI0_Xr0E3s1l8cAvFVAK0II2c7xJM28CjxkF64kEwVA0rcxS w2x7M28EF7xvwVC0I7IYx2IY67AKxVWDJVCq3wA2z4x0Y4vE2Ix0cI8IcVCY1x0267AKxV W8Jr0_Cr1UM28EF7xvwVC2z280aVAFwI0_GcCE3s1l84ACjcxK6I8E87Iv6xkF7I0E14v2 6rxl6s0DM2AIxVAIcxkEcVAq07x20xvEncxIr21l5I8CrVACY4xI64kE6c02F40Ex7xfMc Ij6xIIjxv20xvE14v26r1j6r18McIj6I8E87Iv67AKxVWUJVW8JwAm72CE4IkC6x0Yz7v_ Jr0_Gr1lF7xvr2IYc2Ij64vIr41lFIxGxcIEc7CjxVA2Y2ka0xkIwI1l42xK82IYc2Ij64 vIr41l4I8I3I0E4IkC6x0Yz7v_Jr0_Gr1lx2IqxVAqx4xG67AKxVWUJVWUGwC20s026x8G jcxK67AKxVWUGVWUWwC2zVAF1VAY17CE14v26r4a6rW5MIIYrxkI7VAKI48JMIIF0xvE2I x0cI8IcVAFwI0_Jr0_JF4lIxAIcVC0I7IYx2IY6xkF7I0E14v26F4j6r4UJwCI42IY6xAI w20EY4v20xvaj40_Jr0_JF4lIxAIcVC2z280aVAFwI0_Jr0_Gr1lIxAIcVC2z280aVCY1x 0267AKxVW8Jr0_Cr1UYxBIdaVFxhVjvjDU0xZFpf9x07UC9aPUUUUU= X-CM-SenderInfo: xkrx3t3r6k3tpzhluzxrxghudrp/ X-CFilter-Loop: Reflected X-Rspamd-Queue-Id: 25CEE1C0019 X-Rspam-User: X-Stat-Signature: zeikjys7xmr7g15ytt94jx941n1qht1j X-Rspamd-Server: rspam01 X-HE-Tag: 1697808660-13370 X-HE-Meta: U2FsdGVkX19BY+Q2j9nkuc+Rf3WB1bgBYkc5aaLMOE42o6TmCdVQkFdVaRUIm1y7c8JA3VHh9IYMiPr6JzNlhr17XMwF1jcmzCxcG3cAmpDTCrqQLv1450B6RGfg08fwkOSV7B7i4vZNT77KopMXrJMyGWgBWtNpE7c+yxezqqSW3k6MCSw7Aj9IMUsJyY4OCjwEKdzf/IQoCXk9DZ58NCZygXW0ggOlC5qyGlWHONvblRw7A+2dbYxdWZvw4h2Z649nWlBFauAbucJCL2LYvPliv1le9SIv3NhNrAgR600oZxkoaRj8/+E8Kltrm3gF313tn18HLwNlSIffEKygK7nQRvFInMPp9VAnWJeJQhnkLkAIzDMtY9aGHBz4d+tY1nD5i/vznV2qDhRvL7mDBl/WzxQ9awNibzR2sWE/EvqJl+0uHXcLENtb/P5apdZmi9TYgzJfwuLdJy+4T2mL8sH8cLjWbP10fV63LdKc2yAHJnOnitg2uwSpHWWnN9r1qfmub30GoCmRqVU3tQIVx3aKn8RsQKcpI7I2OMxfGG7uDU21BzqatZwnPrODsQI4iDVsxJZPKtNW6ckrU6W5iHh3BhIVV4AhNAeq+UitC7utRmNM+kBzOLMuonvMSOuX5gM96AuPmNhiMgXg1WM4JkxXC22LbGz6Q4+kpntXeWIbkje1PKsaZZjKViiotLGH9PtEdKqTxphAIxdhD7xxsBk3iKSXn0P/0DxpDnKr7is9Xo2xUiGyYogISDHe4E38sHUTf6+9u+dLNAg2wASFpVXGIUwkgHzlEMSFGWWF8kpsapvOh/9BgBS4nxv0zejTldDH0E+b6+YKHRAiuyaCLs9VaZHubXcB/eBQQ+tQSacyQf8r0/zhh06laEWJheHXATZyjy831GM09UHTXhh6uWSNHTVTUf+FFBvyXo6ifHNudz9SJr9UDQbfLVNAMVZDwy8S8ovwzO7KYJzenD9 83SSWQvW MSXzJL5TUw4mW7w9ENe0EksOoHRYkpR59SRIT9bemFnzAFngZMBpJAVLziYomWWfpnrgMrQIao2BwFTiGqr44h9z/EGWRvcjWZJ1RflQHl+yGHeTVgZ37EPTH+c64OoqpN/NjFEIKZk+E/gK7GC+kNRUhlNMUJfM8FQGKi6flWDVxFYKQcXC3cdz8lRd3sAQAkieJPmK1ba79JmXnmirFBoTKFQtYt3aegx9zVAwa5fyghWZdATR6xI8ozaMpIcSVytBy7osrFht4DfAkONYYgOhw4s5NAE90vS+Uw1q1KOGxBE/UfihG4Tvou+EqyBf2ULbX8LwmnTO16RU= 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: From: Hou Tao Introduce pcpu_alloc_size() to get the size of the dynamic per-cpu area. It will be used by bpf memory allocator in the following patches. BPF memory allocator maintains per-cpu area caches for multiple area sizes and its free API only has the to-be-freed per-cpu pointer, so it needs the size of dynamic per-cpu area to select the corresponding cache when bpf program frees the dynamic per-cpu pointer. Acked-by: Dennis Zhou Signed-off-by: Hou Tao --- include/linux/percpu.h | 1 + mm/percpu.c | 31 +++++++++++++++++++++++++++++++ 2 files changed, 32 insertions(+) diff --git a/include/linux/percpu.h b/include/linux/percpu.h index 68fac2e7cbe67..8c677f185901b 100644 --- a/include/linux/percpu.h +++ b/include/linux/percpu.h @@ -132,6 +132,7 @@ extern void __init setup_per_cpu_areas(void); extern void __percpu *__alloc_percpu_gfp(size_t size, size_t align, gfp_t gfp) __alloc_size(1); extern void __percpu *__alloc_percpu(size_t size, size_t align) __alloc_size(1); extern void free_percpu(void __percpu *__pdata); +extern size_t pcpu_alloc_size(void __percpu *__pdata); DEFINE_FREE(free_percpu, void __percpu *, free_percpu(_T)) diff --git a/mm/percpu.c b/mm/percpu.c index 76b9c5e63c562..1759b91c8944a 100644 --- a/mm/percpu.c +++ b/mm/percpu.c @@ -2244,6 +2244,37 @@ static void pcpu_balance_workfn(struct work_struct *work) mutex_unlock(&pcpu_alloc_mutex); } +/** + * pcpu_alloc_size - the size of the dynamic percpu area + * @ptr: pointer to the dynamic percpu area + * + * Returns the size of the @ptr allocation. This is undefined for statically + * defined percpu variables as there is no corresponding chunk->bound_map. + * + * RETURNS: + * The size of the dynamic percpu area. + * + * CONTEXT: + * Can be called from atomic context. + */ +size_t pcpu_alloc_size(void __percpu *ptr) +{ + struct pcpu_chunk *chunk; + unsigned long bit_off, end; + void *addr; + + if (!ptr) + return 0; + + addr = __pcpu_ptr_to_addr(ptr); + /* No pcpu_lock here: ptr has not been freed, so chunk is still alive */ + chunk = pcpu_chunk_addr_search(addr); + bit_off = (addr - chunk->base_addr) / PCPU_MIN_ALLOC_SIZE; + end = find_next_bit(chunk->bound_map, pcpu_chunk_map_bits(chunk), + bit_off + 1); + return (end - bit_off) * PCPU_MIN_ALLOC_SIZE; +} + /** * free_percpu - free percpu area * @ptr: pointer to area to free