From patchwork Thu Sep 1 16:15:34 2022 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Alexei Starovoitov X-Patchwork-Id: 12962885 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 1D2B3ECAAD3 for ; Thu, 1 Sep 2022 16:16:04 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id B0E6B6B0098; Thu, 1 Sep 2022 12:16:03 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id A976D940009; Thu, 1 Sep 2022 12:16:03 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 911DE940008; Thu, 1 Sep 2022 12:16:03 -0400 (EDT) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0016.hostedemail.com [216.40.44.16]) by kanga.kvack.org (Postfix) with ESMTP id 7ABF16B0098 for ; Thu, 1 Sep 2022 12:16:03 -0400 (EDT) Received: from smtpin01.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay10.hostedemail.com (Postfix) with ESMTP id A4007C0E28 for ; Thu, 1 Sep 2022 16:16:00 +0000 (UTC) X-FDA: 79864018080.01.C097C30 Received: from mail-pj1-f44.google.com (mail-pj1-f44.google.com [209.85.216.44]) by imf23.hostedemail.com (Postfix) with ESMTP id 4275B14005B for ; Thu, 1 Sep 2022 16:16:00 +0000 (UTC) Received: by mail-pj1-f44.google.com with SMTP id x1-20020a17090ab00100b001fda21bbc90so2877384pjq.3 for ; Thu, 01 Sep 2022 09:15:59 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20210112; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc; bh=Ga0Nwt5Zi5Wm2Kkt3SbRcQrpwa6IAGB0M+wlcmjeY9w=; b=nWj8xtovNxnToI65F02+lMStzi08h3Vfb8Hlqm/P/qDpJUAGiVvgLujS2txPE83BJG OJ6iLWh2nH7xvvIeikFp+AqEJ2N5V7k6VuJBgfVEc7lOy+zPql6nkKkVykOHqm5UZLzG LwTIXuaq8pz60Qg1MCK3VqgtZfekqK/hoZM6GOBlarLzGDy5o4gNAFVNfAICmONL7gd6 OE9Bvyja+wiB/5Sftnhw54lWAKedp9r0ZCWwh9Zsx09b377Gtsz/YM0T2AeLiIAtI32W FU4v9JLYNl8lgTULJPOK/kt1tbmxw0Iw9oHipAn45DTzXWFwV9hHS3atkCbItGkr88J3 VqsA== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20210112; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc; bh=Ga0Nwt5Zi5Wm2Kkt3SbRcQrpwa6IAGB0M+wlcmjeY9w=; b=p+/23xDgIDcz4a0rPikGgoE3BoYbnqZjejAjTxDQq7mBa7USsTpx9ZB2KeRjczIr+U FWlRs3uc/zyW8vpBe0aCIUfeSARg4zHILJAlF2J4VELaw6XlCer4XYtvgUHPrjyznDfV s/wJoqLmV5C1nLd9rQEIBQtRSxuOqTONb85VoChcI61G90hQ/WDydjZRs+reJMPQBcWD AlYULgK5gzXwgXYDQ+C/VpBYTiXMvV2yphXcxjrIjbJATpZ8NoC2kGDQCmGMNb4IblVm W4XrlCwir6cqYTk30C1Gn6OTmmrrMhf6JZoz1/B4RCyDTTtuFkacUlDMyTDSoXqjyR6A k6cA== X-Gm-Message-State: ACgBeo2D7SqpGRh813L6kpAqwsoRXaqCcy58qHyneA+V2VgYT79MhUTj C/qAOncY1MPau8XxTITFWUg= X-Google-Smtp-Source: AA6agR47a1KSz7llECCn+O+oZVaQn5Qfp0qmyRn8bPy/B3mcLMozcBxo0iQsF9ul02mGSEd4HLiAbw== X-Received: by 2002:a17:90a:d585:b0:1f4:f9a5:22a9 with SMTP id v5-20020a17090ad58500b001f4f9a522a9mr9490389pju.49.1662048958991; Thu, 01 Sep 2022 09:15:58 -0700 (PDT) Received: from localhost.localdomain ([2620:10d:c090:500::3:4dc5]) by smtp.gmail.com with ESMTPSA id y15-20020a17090264cf00b0016a058b7547sm13972047pli.294.2022.09.01.09.15.57 (version=TLS1_3 cipher=TLS_CHACHA20_POLY1305_SHA256 bits=256/256); Thu, 01 Sep 2022 09:15:58 -0700 (PDT) From: Alexei Starovoitov To: davem@davemloft.net Cc: daniel@iogearbox.net, andrii@kernel.org, tj@kernel.org, memxor@gmail.com, delyank@fb.com, linux-mm@kvack.org, bpf@vger.kernel.org, kernel-team@fb.com Subject: [PATCH v5 bpf-next 02/15] bpf: Convert hash map to bpf_mem_alloc. Date: Thu, 1 Sep 2022 09:15:34 -0700 Message-Id: <20220901161547.57722-3-alexei.starovoitov@gmail.com> X-Mailer: git-send-email 2.36.1 In-Reply-To: <20220901161547.57722-1-alexei.starovoitov@gmail.com> References: <20220901161547.57722-1-alexei.starovoitov@gmail.com> MIME-Version: 1.0 ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1662048960; 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:dkim-signature; bh=Ga0Nwt5Zi5Wm2Kkt3SbRcQrpwa6IAGB0M+wlcmjeY9w=; b=E5B/J4XYV+yCLKBRF/a1b8SK+PkSqJnJSpgl26CdgoxOSbVVQazoIFVDS4Wn6fetFJD4oN jApCuXDnct+ekx07tgMedptNNasfSKM+frDqGkJ3vXeBGXVzdhLlcP0ngr7J0SN3UuKeAQ QFDc4WPtKjnbxwkQ1UZA9mW/6NsrfYE= ARC-Authentication-Results: i=1; imf23.hostedemail.com; dkim=pass header.d=gmail.com header.s=20210112 header.b=nWj8xtov; spf=pass (imf23.hostedemail.com: domain of alexei.starovoitov@gmail.com designates 209.85.216.44 as permitted sender) smtp.mailfrom=alexei.starovoitov@gmail.com; dmarc=pass (policy=none) header.from=gmail.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1662048960; a=rsa-sha256; cv=none; b=YGbEaQMaJePGEeqzpqtk8kVWtI9b/SSIA3UO9E6eb1e+KfHk05ylCv/wB0K+ql9kQY5/gW vI4YgRYUOuUAr0vKq1xek+dljhCrDUMODz8bbgC1mWqPoMh86ay9+gzP8oWqd8dTxKwzwA YwrXC4mIMFkPIaOiBx/89SaAr4yQCn0= Authentication-Results: imf23.hostedemail.com; dkim=pass header.d=gmail.com header.s=20210112 header.b=nWj8xtov; spf=pass (imf23.hostedemail.com: domain of alexei.starovoitov@gmail.com designates 209.85.216.44 as permitted sender) smtp.mailfrom=alexei.starovoitov@gmail.com; dmarc=pass (policy=none) header.from=gmail.com X-Rspam-User: X-Stat-Signature: 4tb4smqnn5pyhxkub96wade3bjjxsaz9 X-Rspamd-Queue-Id: 4275B14005B X-Rspamd-Server: rspam05 X-HE-Tag: 1662048960-687078 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: Alexei Starovoitov Convert bpf hash map to use bpf memory allocator. Acked-by: Kumar Kartikeya Dwivedi Acked-by: Andrii Nakryiko Signed-off-by: Alexei Starovoitov --- kernel/bpf/hashtab.c | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/kernel/bpf/hashtab.c b/kernel/bpf/hashtab.c index eb1263f03e9b..508e64351f87 100644 --- a/kernel/bpf/hashtab.c +++ b/kernel/bpf/hashtab.c @@ -14,6 +14,7 @@ #include "percpu_freelist.h" #include "bpf_lru_list.h" #include "map_in_map.h" +#include #define HTAB_CREATE_FLAG_MASK \ (BPF_F_NO_PREALLOC | BPF_F_NO_COMMON_LRU | BPF_F_NUMA_NODE | \ @@ -92,6 +93,7 @@ struct bucket { struct bpf_htab { struct bpf_map map; + struct bpf_mem_alloc ma; struct bucket *buckets; void *elems; union { @@ -576,6 +578,10 @@ static struct bpf_map *htab_map_alloc(union bpf_attr *attr) if (err) goto free_prealloc; } + } else { + err = bpf_mem_alloc_init(&htab->ma, htab->elem_size); + if (err) + goto free_map_locked; } return &htab->map; @@ -586,6 +592,7 @@ static struct bpf_map *htab_map_alloc(union bpf_attr *attr) for (i = 0; i < HASHTAB_MAP_LOCK_COUNT; i++) free_percpu(htab->map_locked[i]); bpf_map_area_free(htab->buckets); + bpf_mem_alloc_destroy(&htab->ma); free_htab: lockdep_unregister_key(&htab->lockdep_key); bpf_map_area_free(htab); @@ -862,7 +869,7 @@ static void htab_elem_free(struct bpf_htab *htab, struct htab_elem *l) if (htab->map.map_type == BPF_MAP_TYPE_PERCPU_HASH) free_percpu(htab_elem_get_ptr(l, htab->map.key_size)); check_and_free_fields(htab, l); - kfree(l); + bpf_mem_cache_free(&htab->ma, l); } static void htab_elem_free_rcu(struct rcu_head *head) @@ -986,9 +993,7 @@ static struct htab_elem *alloc_htab_elem(struct bpf_htab *htab, void *key, l_new = ERR_PTR(-E2BIG); goto dec_count; } - l_new = bpf_map_kmalloc_node(&htab->map, htab->elem_size, - GFP_NOWAIT | __GFP_NOWARN, - htab->map.numa_node); + l_new = bpf_mem_cache_alloc(&htab->ma); if (!l_new) { l_new = ERR_PTR(-ENOMEM); goto dec_count; @@ -1007,7 +1012,7 @@ static struct htab_elem *alloc_htab_elem(struct bpf_htab *htab, void *key, pptr = bpf_map_alloc_percpu(&htab->map, size, 8, GFP_NOWAIT | __GFP_NOWARN); if (!pptr) { - kfree(l_new); + bpf_mem_cache_free(&htab->ma, l_new); l_new = ERR_PTR(-ENOMEM); goto dec_count; } @@ -1429,6 +1434,10 @@ static void delete_all_elements(struct bpf_htab *htab) { int i; + /* It's called from a worker thread, so disable migration here, + * since bpf_mem_cache_free() relies on that. + */ + migrate_disable(); for (i = 0; i < htab->n_buckets; i++) { struct hlist_nulls_head *head = select_bucket(htab, i); struct hlist_nulls_node *n; @@ -1439,6 +1448,7 @@ static void delete_all_elements(struct bpf_htab *htab) htab_elem_free(htab, l); } } + migrate_enable(); } static void htab_free_malloced_timers(struct bpf_htab *htab) @@ -1502,6 +1512,7 @@ static void htab_map_free(struct bpf_map *map) bpf_map_free_kptr_off_tab(map); free_percpu(htab->extra_elems); bpf_map_area_free(htab->buckets); + bpf_mem_alloc_destroy(&htab->ma); for (i = 0; i < HASHTAB_MAP_LOCK_COUNT; i++) free_percpu(htab->map_locked[i]); lockdep_unregister_key(&htab->lockdep_key);