From nobody Fri Dec 19 17:33:21 2025 Received: from mail-pj1-f42.google.com (mail-pj1-f42.google.com [209.85.216.42]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 770113328FB for ; Thu, 6 Nov 2025 13:20:35 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.216.42 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1762435237; cv=none; b=Z9S4SmHd8TC03ABaCmyOB31h/SKRP5+5sgGH4xMU3OslIbFRa9D2uWZZoa/uofJwYHMp33+btHidB4UDORh8NVxlg3W3CryUl61q/f7PBqiHimAQLHBwlFiynu80XoYK9Z3sIQc288BwZFnRAJEcuqlLO4MkZHxbQGEntJhRHOs= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1762435237; c=relaxed/simple; bh=F5X5Y5A4ts8MLXZLKzfKP+tr0DZu+o/90/KYpaX5Qb0=; h=From:To:Cc:Subject:Date:Message-Id:In-Reply-To:References: MIME-Version; b=OAJNOQ0JDbaYqMtTKZkJ6+JwpBr10Lwkg+3YYrajOrphCJwcKej4Bl7mqqVXKkfyE4awyQ0lcGCM7P7WMpgFE/sYZeb/7UvK1yXKAQGyg/FT5yHDAAEeqPL2VMxEqbYe1vOeYadMuPoxw/xIb63FdflUf4hkIj6GOkI3FzDUkG8= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=gmail.com; spf=pass smtp.mailfrom=gmail.com; dkim=pass (2048-bit key) header.d=gmail.com header.i=@gmail.com header.b=GyQscZaf; arc=none smtp.client-ip=209.85.216.42 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=gmail.com Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=gmail.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=gmail.com header.i=@gmail.com header.b="GyQscZaf" Received: by mail-pj1-f42.google.com with SMTP id 98e67ed59e1d1-340c39ee02dso996835a91.1 for ; Thu, 06 Nov 2025 05:20:35 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1762435235; x=1763040035; darn=vger.kernel.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=3oe9IB7XRuftK/ORLSLXvtfYwT74Xn4ffToyKR5CyQA=; b=GyQscZaf8nREVM6RKPr779zcbhPzppTWxVAttqaAS1kdmiugEXA3yHDHadnl6n/SuU XLGLOjHGAMcUKSzHMVkScwc7CmAo+vPZDK4/DJYNekd4MMbH+pWGl/GKs9YxXzPSPpnS V1k16MxcePyUe5clRTArIJUiIbg0mow7Yo/6AI+PQl7tVYuWZLyka44KMLdQD02zyyNm p/+5Vx8B/spM98NSwALTreTLaUAdZCTfrvoxkUCWj7Hj8tsFcsUozHtVR5PRe635I6xy 7iX8oyalFwbb0+Wrv5k8O+yKtQHf9UVua3kem3IQ2Rf1VqajAyuSqTBrt2TRB+gfFpdC eg3g== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1762435235; x=1763040035; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=3oe9IB7XRuftK/ORLSLXvtfYwT74Xn4ffToyKR5CyQA=; b=jln+Ev+QFjYw82wc7aFNP6J3bb4nUtHf0fed4ela/fEZe9pHWbVzP473mdUBDOl44Y 7gUL7oa5GTn++0PaXIq0l81vK1cNW3xcFHJaQz9bt9ekMlpKgSyv0T7QJdVIvqREU9g4 1v+MG8FCvyPPmD2SnzFkxPKw8SseaLXV4fJrX+7+phvUj7sCl+1GZyug1EucJ4UjeZXP qnqpCUqzlozZk5KRgvMB+6Mbc07duz8TnODEuCiS3NPuTm12vGEeFIQUkHz5i1Q3JRoD NpVBs8LPIHFMlbxgEFjbd2zBZhAQefBLVYtPuZ90bEkJseHESkszOuMOT2Pj+5CjJf60 xeiQ== X-Forwarded-Encrypted: i=1; AJvYcCWZHbNSb8hd3ZBEt418dI8qTtNK1bZVBawp8VDLOA+yasMZnUYpWSHcRYaKxH8PwN5s5Km8Vek5fChr/VY=@vger.kernel.org X-Gm-Message-State: AOJu0YyFOLEHxbImTRQtdpNPDj6D9WwLI58tXXPQOvGuuDgJEI/pwPK5 Puo37tloYV3jihJcH22qph+nxtyn9kREcnaWrMQO4WVowxnEQaENuQaa X-Gm-Gg: ASbGncvertzPMXopdJI+ymIcZnx2bfRPIlnlkyvwLE2tKJeqN58ZHkEzd/PrylVr2Gx 0xgdQtAX1abQFYyWgnc7q9TeTTQiFC3q7shCZ7UegxE7mnwR8VaASKka79eO1Qx6evYuuqhklXB Ib19lYwvHZ1OfEeLiQqDZxvyqFDSPkti1gEh1jLcKWYw0qCy8yGoQsHCKGkMdAP2okhaa7vltmk gAniN7zkYCY4F3U1VE1hEy/BbW5IjE9HeahTCnL9yIJsBwXcXn+qoUPFv90N7CxFj4+CZnB4K/l PSTdHiF1oRKvtJgkyUPBDp5F8sJK+Gpyh5ipAYVUl7XuOlqGCCGcs1vF/GPgrisBojggvcml8Lm ouQTKbW7yq6LomavRM2YmXv9Dx/4v44YagO2YOWP3LW0vqLLjl//kE1yPGBaiWLfm9UBEsIKfKv PEX/FmL+0rZ+V+N0UZ X-Google-Smtp-Source: AGHT+IHdCvg4WM6MDWt7CjISOISNrqfsnmPepoxmBDS9wWIOhAC432xRASHA8HBRdlSBjrUGDiQajg== X-Received: by 2002:a17:90a:1508:b0:340:d578:f299 with SMTP id 98e67ed59e1d1-341a6c00d45mr5789547a91.3.1762435234764; Thu, 06 Nov 2025 05:20:34 -0800 (PST) Received: from pengdl-pc.mioffice.cn ([43.224.245.249]) by smtp.gmail.com with ESMTPSA id 98e67ed59e1d1-341d3e0b0b2sm1914869a91.21.2025.11.06.05.20.31 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Thu, 06 Nov 2025 05:20:33 -0800 (PST) From: Donglin Peng To: ast@kernel.org Cc: eddyz87@gmail.com, andrii.nakryiko@gmail.com, zhangxiaoqin@xiaomi.com, linux-kernel@vger.kernel.org, bpf@vger.kernel.org, Donglin Peng , Alan Maguire , Song Liu , Donglin Peng Subject: [PATCH v5 5/7] btf: Optimize type lookup with binary search Date: Thu, 6 Nov 2025 21:19:54 +0800 Message-Id: <20251106131956.1222864-6-dolinux.peng@gmail.com> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20251106131956.1222864-1-dolinux.peng@gmail.com> References: <20251106131956.1222864-1-dolinux.peng@gmail.com> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Content-Type: text/plain; charset="utf-8" From: Donglin Peng Improve btf_find_by_name_kind() performance by adding binary search support for sorted types. Falls back to linear search for compatibility. Cc: Eduard Zingerman Cc: Alexei Starovoitov Cc: Andrii Nakryiko Cc: Alan Maguire Cc: Song Liu Cc: Xiaoqin Zhang Signed-off-by: Donglin Peng Signed-off-by: Donglin Peng --- kernel/bpf/btf.c | 117 +++++++++++++++++++++++++++++++++++++++++++---- 1 file changed, 107 insertions(+), 10 deletions(-) diff --git a/kernel/bpf/btf.c b/kernel/bpf/btf.c index 0de8fc8a0e0b..66cb739a0598 100644 --- a/kernel/bpf/btf.c +++ b/kernel/bpf/btf.c @@ -192,6 +192,8 @@ */ #define BTF_MAX_SIZE (16 * 1024 * 1024) =20 +#define BTF_NEED_SORT_CHECK ((u32)-1) + #define for_each_member_from(i, from, struct_type, member) \ for (i =3D from, member =3D btf_type_member(struct_type) + from; \ i < btf_type_vlen(struct_type); \ @@ -259,6 +261,7 @@ struct btf { void *nohdr_data; struct btf_header hdr; u32 nr_types; /* includes VOID for base BTF */ + u32 nr_sorted_types; /* exclude VOID for base BTF */ u32 types_size; u32 data_size; refcount_t refcnt; @@ -494,6 +497,11 @@ static bool btf_type_is_modifier(const struct btf_type= *t) return false; } =20 +static int btf_start_id(const struct btf *btf) +{ + return btf->start_id + (btf->base_btf ? 0 : 1); +} + bool btf_type_is_void(const struct btf_type *t) { return t =3D=3D &btf_void; @@ -544,24 +552,110 @@ u32 btf_nr_types(const struct btf *btf) return total; } =20 +/* Performs binary search within specified type ID range to find the leftm= ost + * BTF type matching the given name. The search assumes types are sorted by + * name in lexicographical order within the specified range. + * + * Return: Type ID of leftmost matching type, or -ENOENT if not found + */ +static s32 btf_find_by_name_kind_bsearch(const struct btf *btf, const char= *name, + s32 start_id, s32 end_id) +{ + const struct btf_type *t; + const char *tname; + s32 l, r, m, lmost =3D -ENOENT; + int ret; + + l =3D start_id; + r =3D end_id; + while (l <=3D r) { + m =3D l + (r - l) / 2; + t =3D btf_type_by_id(btf, m); + tname =3D btf_name_by_offset(btf, t->name_off); + ret =3D strcmp(tname, name); + if (ret < 0) { + l =3D m + 1; + } else { + if (ret =3D=3D 0) + lmost =3D m; + r =3D m - 1; + } + } + + return lmost; +} + +/* Searches for a BTF type with the specified name and kind. The function + * first recursively searches in the base BTF (if present), then searches + * in the current BTF using either binary search (if types are sorted) + * or linear search. + * + * Binary search is used when types are name-sorted (nr_sorted_types > 0). + * After finding a name match, it scans forward to find the first type + * that also matches the specified kind. Linear search is used for unsorted + * types, checking each type sequentially. + * + * Return: Type ID of matching type on success, -ENOENT if not found + */ s32 btf_find_by_name_kind(const struct btf *btf, const char *name, u8 kind) { + const struct btf *base_btf =3D btf_base_btf(btf);; const struct btf_type *t; const char *tname; - u32 i, total; + int err =3D -ENOENT; =20 - total =3D btf_nr_types(btf); - for (i =3D 1; i < total; i++) { - t =3D btf_type_by_id(btf, i); - if (BTF_INFO_KIND(t->info) !=3D kind) - continue; + if (base_btf) { + err =3D btf_find_by_name_kind(base_btf, name, kind); + if (err > 0) + goto out; + } =20 - tname =3D btf_name_by_offset(btf, t->name_off); - if (!strcmp(tname, name)) - return i; + if (btf->nr_sorted_types !=3D BTF_NEED_SORT_CHECK) { + /* binary search */ + bool skip_first; + s32 start_id, end_id;; + int ret; + + start_id =3D btf_start_id(btf); + end_id =3D start_id + btf->nr_sorted_types - 1; + ret =3D btf_find_by_name_kind_bsearch(btf, name, start_id, end_id); + if (ret < 0) + goto out; + skip_first =3D true; + do { + t =3D btf_type_by_id(btf, ret); + if (BTF_INFO_KIND(t->info) !=3D kind) { + if (skip_first) { + skip_first =3D false; + continue; + } + } else if (skip_first) { + return ret; + } + tname =3D btf_name_by_offset(btf, t->name_off); + if (!strcmp(tname, name)) + return ret; + else + break; + } while (++ret <=3D end_id); + } else { + /* linear search */ + u32 i, total; + + total =3D btf_nr_types(btf); + for (i =3D btf_start_id(btf); i < total; i++) { + t =3D btf_type_by_id(btf, i); + if (BTF_INFO_KIND(t->info) !=3D kind) + continue; + + tname =3D btf_name_by_offset(btf, t->name_off); + if (!strcmp(tname, name)) + return i; + } } =20 - return -ENOENT; +out: + return err; } =20 s32 bpf_find_btf_id(const char *name, u32 kind, struct btf **btf_p) @@ -5791,6 +5885,7 @@ static struct btf *btf_parse(const union bpf_attr *at= tr, bpfptr_t uattr, u32 uat goto errout; } env->btf =3D btf; + btf->nr_sorted_types =3D BTF_NEED_SORT_CHECK; =20 data =3D kvmalloc(attr->btf_size, GFP_KERNEL | __GFP_NOWARN); if (!data) { @@ -6210,6 +6305,7 @@ static struct btf *btf_parse_base(struct btf_verifier= _env *env, const char *name btf->data =3D data; btf->data_size =3D data_size; btf->kernel_btf =3D true; + btf->nr_sorted_types =3D BTF_NEED_SORT_CHECK; snprintf(btf->name, sizeof(btf->name), "%s", name); =20 err =3D btf_parse_hdr(env); @@ -6327,6 +6423,7 @@ static struct btf *btf_parse_module(const char *modul= e_name, const void *data, btf->start_id =3D base_btf->nr_types; btf->start_str_off =3D base_btf->hdr.str_len; btf->kernel_btf =3D true; + btf->nr_sorted_types =3D BTF_NEED_SORT_CHECK; snprintf(btf->name, sizeof(btf->name), "%s", module_name); =20 btf->data =3D kvmemdup(data, data_size, GFP_KERNEL | __GFP_NOWARN); --=20 2.34.1