From mboxrd@z Thu Jan 1 00:00:00 1970 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 AD492E95A63 for ; Sat, 7 Oct 2023 13:50:11 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 57E138D0014; Sat, 7 Oct 2023 09:50:09 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 52C6D8D0005; Sat, 7 Oct 2023 09:50:09 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 356768D0014; Sat, 7 Oct 2023 09:50:09 -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 26F9E8D0005 for ; Sat, 7 Oct 2023 09:50:09 -0400 (EDT) Received: from smtpin24.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay07.hostedemail.com (Postfix) with ESMTP id 00CA9160277 for ; Sat, 7 Oct 2023 13:50:08 +0000 (UTC) X-FDA: 81318799338.24.B406B91 Received: from dggsgout12.his.huawei.com (unknown [45.249.212.56]) by imf30.hostedemail.com (Postfix) with ESMTP id 1C71C80004 for ; Sat, 7 Oct 2023 13:50:04 +0000 (UTC) Authentication-Results: imf30.hostedemail.com; dkim=none; dmarc=none; spf=none (imf30.hostedemail.com: domain of houtao@huaweicloud.com has no SPF policy when checking 45.249.212.56) smtp.mailfrom=houtao@huaweicloud.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1696686607; 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=qMFR7Mf6xY+9t5IHDOy3Nj6hkXbCJvR9n26C7dJQRmQ=; b=fBtqdNr3DSerSKXNhWCZGojQCP5IoIg6u/UgyOCI8Jcaz+hhO5kWtsllRsEmgGuaJ15C/1 yADRxEDEUht42TlAnLpacjSDTCbFTUhBob1MaI9rFloIqwnVxY82dFwlZT1GmXFegIIkwy uGQCHd2amNRRf8vgx6HMDT3bAQWkUKo= ARC-Authentication-Results: i=1; imf30.hostedemail.com; dkim=none; dmarc=none; spf=none (imf30.hostedemail.com: domain of houtao@huaweicloud.com has no SPF policy when checking 45.249.212.56) smtp.mailfrom=houtao@huaweicloud.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1696686607; a=rsa-sha256; cv=none; b=nSfI5WVyz3dMHpZomMwLHzW6fPOGq4yEigNkFk5A5uPHU8NpROnIIaGH4zdj8zCIm2VxxW LsXTlD4oYxGdPRJEySzIhSJaz9rpqmJXtilHM7AVveptNuu+AWEVSOFIuTqgqvy4d1qDwZ Q+ikUqBkVpnMUZIfCVCz1AIymOxwIt4= Received: from mail02.huawei.com (unknown [172.30.67.143]) by dggsgout12.his.huawei.com (SkyGuard) with ESMTP id 4S2mrF5mVsz4f3kF9 for ; Sat, 7 Oct 2023 21:49:49 +0800 (CST) Received: from huaweicloud.com (unknown [10.175.124.27]) by APP4 (Coremail) with SMTP id gCh0CgAnvdz9YSFl4YF2CQ--.30763S5; Sat, 07 Oct 2023 21:49:54 +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 1/6] mm/percpu.c: introduce alloc_size_percpu() Date: Sat, 7 Oct 2023 21:51:01 +0800 Message-Id: <20231007135106.3031284-2-houtao@huaweicloud.com> X-Mailer: git-send-email 2.29.2 In-Reply-To: <20231007135106.3031284-1-houtao@huaweicloud.com> References: <20231007135106.3031284-1-houtao@huaweicloud.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-CM-TRANSID:gCh0CgAnvdz9YSFl4YF2CQ--.30763S5 X-Coremail-Antispam: 1UD129KBjvJXoW7Aw17ZFykAw4rCFWDJrWUJwb_yoW8tw4UpF Wku3s3tr1rXFn7Ww1ftw1UZw4rWw4kWayxJ343WFy5ZryavFyagF1vvrW5uFyrGFn2vry2 qa90qF4fCFW3X3DanT9S1TB71UUUUUUqnTZGkaVYY2UrUUUUjbIjqfuFe4nvWSU5nxnvy2 9KBjDU0xBIdaVrnRJUUUB0b4IE77IF4wAFF20E14v26rWj6s0DM7CY07I20VC2zVCF04k2 6cxKx2IYs7xG6rWj6s0DM7CIcVAFz4kK6r1j6r18M28IrcIa0xkI8VA2jI8067AKxVWUGw A2048vs2IY020Ec7CjxVAFwI0_Gr0_Xr1l8cAvFVAK0II2c7xJM28CjxkF64kEwVA0rcxS w2x7M28EF7xvwVC0I7IYx2IY67AKxVWDJVCq3wA2z4x0Y4vE2Ix0cI8IcVCY1x0267AKxV W8Jr0_Cr1UM28EF7xvwVC2z280aVAFwI0_GcCE3s1l84ACjcxK6I8E87Iv6xkF7I0E14v2 6rxl6s0DM2AIxVAIcxkEcVAq07x20xvEncxIr21l5I8CrVACY4xI64kE6c02F40Ex7xfMc Ij6xIIjxv20xvE14v26r1j6r18McIj6I8E87Iv67AKxVWUJVW8JwAm72CE4IkC6x0Yz7v_ Jr0_Gr1lF7xvr2IYc2Ij64vIr41lFIxGxcIEc7CjxVA2Y2ka0xkIwI1l42xK82IYc2Ij64 vIr41l4I8I3I0E4IkC6x0Yz7v_Jr0_Gr1lx2IqxVAqx4xG67AKxVWUJVWUGwC20s026x8G jcxK67AKxVWUGVWUWwC2zVAF1VAY17CE14v26r4a6rW5MIIYrxkI7VAKI48JMIIF0xvE2I x0cI8IcVAFwI0_Jr0_JF4lIxAIcVC0I7IYx2IY6xkF7I0E14v26r4j6F4UMIIF0xvE42xK 8VAvwI8IcIk0rVWUJVWUCwCI42IY6I8E87Iv67AKxVWUJVW8JwCI42IY6I8E87Iv6xkF7I 0E14v26r4j6r4UJbIYCTnIWIevJa73UjIFyTuYvjxU2mL9UUUUU X-CM-SenderInfo: xkrx3t3r6k3tpzhluzxrxghudrp/ X-CFilter-Loop: Reflected X-Rspamd-Queue-Id: 1C71C80004 X-Rspam-User: X-Rspamd-Server: rspam04 X-Stat-Signature: fwn1qgery3prrq8w1t94pbzjw47a6q8m X-HE-Tag: 1696686604-286238 X-HE-Meta: U2FsdGVkX19WtbQuDJVV39tH/qZtovFy+KT9gdIk1ycuwVPi37RQTwROdCwgniWJJg1GLNJ0LDsImV/zLB/sh2ovQMcZYgmysgFTbdwkF58Vri73/z/lmDNet/an+I/2oz4ev22YV4oFGeS8cQ6mbWwEqHW4SVqijFxQdNeMlsMs9rkgIwaNrEoZhj9sIFQSB2oP58xDz7MlcXnJn5wAqX2wFXYXpGil1VPtP2g2u7TgRAyGzrJEywEszbF7EjhIvoMBEt7dugKRq4hCRHxQXaEixhjr87hcv6L1pssFt/AMAH3X4cerndqmnhjKgEeL9tiyTpa6mrt4tKkiRmLus6HNtXsU77ABQLDZ7nxa0Xx78J2M47VgVXiV2JCEZiazAv9jXn8auWEh53IrynPd0zWa04ysCNKHeBley0L9++TDsIS/cBmNlJhYVPdHrgonK+2TvMMJYx6v/YEIguzpOyhOMQn0GYDHEWTwQ/P7uIFFNgckkb26X+zWYmXyardVgSrs628dDzPq9i8guFVWy/s59tTCFfPTGzxy4yyGclpzCmoww+7o0b+fTnLda1CPbCoAaMicPQsqiPiBP1WZ7U9So5fbSW2KPq5wPh5wE+58bu6vzIEftEABnok4c8bBC/Jj285yep9Lk1EEpmCqSNQijp1ckN0F/3xqwKv+7UpHPX7lHmG9NrYZHlXL9Qw5n0L4E/zWmiX1tuTjTZp2pUqQXU/8WtEFl9afhEuCq84r+sNQ/jVd1Rb0SSwEfHgRFeup+XppTlC4bFsS23dICimRHrxXCoxzyBDi8jBcMj9UAM+ZV/D2UuIY6OTLL0jeHKPSuyMua1gbm3QU/LJNw4lEGztWonhsDrwGM3oUgZrAMCm4j34RrdCYGkVkH0L2Pv0bz2Alra6xKjnG9CuBc2H/2WsAWbDWcJRkWFXVv/uEYIKGQsm2mRhlwxkKhZqvFsrUlQ21ZOv5Nqrnn0U cGwjYTG+ xpvt/fTgCZlEZtPe0lHabn6KSHUoHRhGAAHWKVpeUBSoQgYW+tQmWLlrMjpQTLk8YnC2PFreKn5YQ9c2LdxZmAiCi4LwFstNhHnd6zeYAmtpwDG7F6Csik+SrxITihT9fDfKC0/u6SX44HUDjOFhJLQdcXroR73wMab3rx2q1RX2odosqj511fk6NfEiOr0BZ1wmQGWc8r0O7e2zsP9rlPhm29n1bsFrCo4f5Ev5t06+Tq8xNPgYXYUe0qHhqjqoQMh6WFF2vqURI0k9GQGxN06NXXP++YulXIx1O5o+ExYc9nggYL7kBzvyCCw== 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 alloc_size_percpu() 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 multiple per-cpu area caches for multiple area sizes and it needs the size of dynamic per-cpu area to select the corresponding cache when bpf program frees the dynamic per-cpu area. Signed-off-by: Hou Tao --- include/linux/percpu.h | 1 + mm/percpu.c | 29 +++++++++++++++++++++++++++++ 2 files changed, 30 insertions(+) diff --git a/include/linux/percpu.h b/include/linux/percpu.h index 68fac2e7cbe6..d140d9d79567 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 alloc_size_percpu(void __percpu *__pdata); DEFINE_FREE(free_percpu, void __percpu *, free_percpu(_T)) diff --git a/mm/percpu.c b/mm/percpu.c index 7b40b3963f10..f541cfc3cb2d 100644 --- a/mm/percpu.c +++ b/mm/percpu.c @@ -2244,6 +2244,35 @@ static void pcpu_balance_workfn(struct work_struct *work) mutex_unlock(&pcpu_alloc_mutex); } +/** + * alloc_size_percpu - the size of the dynamic percpu area + * @ptr: pointer to the dynamic percpu area + * + * Return the size of the dynamic percpu area @ptr. + * + * RETURNS: + * The size of the dynamic percpu area. + * + * CONTEXT: + * Can be called from atomic context. + */ +size_t alloc_size_percpu(void __percpu *ptr) +{ + struct pcpu_chunk *chunk; + int 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 -- 2.29.2