From nobody Sun Feb 8 12:20:09 2026 Received: from mail-pg1-f195.google.com (mail-pg1-f195.google.com [209.85.215.195]) (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 23CDD1E0B86 for ; Mon, 19 Jan 2026 07:03:08 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.215.195 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768806189; cv=none; b=YkM/PkYfK9BzauOg4+I9Z1AyyD7xCNzQlnFTacxQAQvpfZStjhOBhAI7CYLjQF687WFMg8OfaD28cmYIG/UkTcnxz9F6z+srkQKyWpK6R11ZPUy16OpU5oMYS8ZYdmxfrkv+BGt/xrKLWOC2PczN+j0uCr+B+elQINQnWwPeML0= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768806189; c=relaxed/simple; bh=oYZy79n5OSjjTlrj54yqdATsU4yMrFm/rl00CpbxXBY=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=U3G+S7NDwyFm0bV+ivlnm1UkzR0SfEeEabRZuOIkpO1MVkxS0KwQVhI/PqZOIkuPJjZHOdEwAHv88oxho5pGyZ2Op87dBqlnLkRvJG9QHfhbL36nzLnZBTbMGhki53VL8Axb1BK/HunjgigtN7qsvj4q4zGLKt6l8v1/6EmjYtw= 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=VpMmlL+3; arc=none smtp.client-ip=209.85.215.195 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="VpMmlL+3" Received: by mail-pg1-f195.google.com with SMTP id 41be03b00d2f7-c2dc870e194so1667648a12.2 for ; Sun, 18 Jan 2026 23:03:08 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1768806187; x=1769410987; 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=AbzRpqckj4NQsnl95AKvf3X1E1Gz7eIXNEvtt8X/tco=; b=VpMmlL+3mMxGoX5bWrcaneTHcuw4TuJEiOdsIgELLbVEbGL4YDwKYTZNstR9AJemDG b68T/R2GkVjO2P7RjJTu6Lq9sNzTsrbBHtatox1ye/0EB4y811ndD3ok/Vimguq0/DDo bsr/t9DyEIvYfIeZvan7B0oMe5GpflThy7B++O6OIL/zpCRfXTSb+NoVnU5LIkcgIfAT FfrnPft9XyzGLxCKP93wNlXrjwfR6q5xkqQYsTDpgrDCvvbTYnKqJePVcrAvqSS5YXep fnzEfbXYAHyXbVGzlnoN3luBxGdzXtKkF6DHYdThjRiXL4qGuaEAhJD/LxezIXrt95vB QEOQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1768806187; x=1769410987; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-gg:x-gm-message-state:from :to:cc:subject:date:message-id:reply-to; bh=AbzRpqckj4NQsnl95AKvf3X1E1Gz7eIXNEvtt8X/tco=; b=xLjG/Y3X4M231dsnkCTg929o9H+5jawdUAKN7yoS92CkY4feNWXBNgkqA0U4MqZvLE nM1EDHHS2wouCq1zMIOy0uR/WW/YZhWq+wq3BgM7AXnGztBamZRBbAWvQYflFdSQ2R60 L5XuhIojSImTGBoduAsyPehXIfLVagGDK9Ns781TQgJ99KLPUnLzw/lYI43G+wqNsfEi hdu7xA6oFL1aY66CGwTSTezhWg2u69BKRJylDr2qFV7ts0QvbxWApqmz8LlorTZmTWHF DwgaEhFRK/pZmEU0g03xLQ8SZELA8GJUCxJRRhQMvSEo8tn2mGhCKeLxBpoNzWlsLkQx L5rw== X-Forwarded-Encrypted: i=1; AJvYcCU1nuYtsLJBvdFLB8xe7HZtqNNK8fYGGx1i5I0isOZZIiJmIi0Xn64f40Z81IK8AZNcUMsEfBfH4h3dLkE=@vger.kernel.org X-Gm-Message-State: AOJu0Yxgl3vwJHm3uqRlAUV/5qpagZaNK+HarC4aQJa0KB/vk78vHv55 5bjVd9c1np4u//TPF6xRqX5X7s/X2t8KSUAwDA4cLwgnIuaOF8mw/7JH X-Gm-Gg: AZuq6aL4BcsKInH6Q20Zb2nz+/4q1kxgvNLY5WrYoZzuyfYqfH+SFCcEFnd7+wMIXGh qR3Boid2KBx4g17hTb3tT0unA6vBUFY6O+vBTCTcrh5x14r4i0r+ATjV80C5b/i0HDDd57x/L2E Fx46eIWUAYdXSq+zN/ySS6YxqtWYonbx4J6PH8LPkt2b1bnYSgoJiLV6yj8lpH6rSbJPTIF7V9Z htoQ73+QdUHO/T+ox0VkU6l/lZeEeJAWvYm/MCK4AEMyazVfMwU4JswHSf6fYTlkqLPppR6G2K8 Vhwvw/zsrJs9CX7raBk/IyTPJCwsTKhxuZaIEIg+CZpz31ZvdEbRTqw8I40kWYyKtzCCDonMyPk 1qtqKEHjJgS7eU0Jfal6LmNnZQSVlztbr9UQneDSEme43/w0p9lkDv3k/CH/q0CagObKJIE1Gf0 Jh+yP86it4 X-Received: by 2002:a17:90a:ec86:b0:343:87b1:285 with SMTP id 98e67ed59e1d1-35272f6c412mr7433738a91.18.1768806187446; Sun, 18 Jan 2026 23:03:07 -0800 (PST) Received: from 7940hx ([103.173.155.241]) by smtp.gmail.com with ESMTPSA id 98e67ed59e1d1-352677ec7dasm10772027a91.8.2026.01.18.23.03.01 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Sun, 18 Jan 2026 23:03:07 -0800 (PST) From: Menglong Dong X-Google-Original-From: Menglong Dong To: ast@kernel.org, eddyz87@gmail.com Cc: daniel@iogearbox.net, john.fastabend@gmail.com, andrii@kernel.org, martin.lau@linux.dev, song@kernel.org, yonghong.song@linux.dev, kpsingh@kernel.org, sdf@fomichev.me, haoluo@google.com, jolsa@kernel.org, bpf@vger.kernel.org, linux-kernel@vger.kernel.org Subject: [PATCH bpf-next v5 1/2] bpf, x86: inline bpf_get_current_task() for x86_64 Date: Mon, 19 Jan 2026 15:02:45 +0800 Message-ID: <20260119070246.249499-2-dongml2@chinatelecom.cn> X-Mailer: git-send-email 2.52.0 In-Reply-To: <20260119070246.249499-1-dongml2@chinatelecom.cn> References: <20260119070246.249499-1-dongml2@chinatelecom.cn> 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" Inline bpf_get_current_task() and bpf_get_current_task_btf() for x86_64 to obtain better performance. In !CONFIG_SMP case, the percpu variable is just a normal variable, and we can read the current_task directly. Signed-off-by: Menglong Dong Acked-by: Eduard Zingerman --- v5: - don't support the !CONFIG_SMP case v4: - handle the !CONFIG_SMP case v3: - implement it in the verifier with BPF_MOV64_PERCPU_REG() instead of in x86_64 JIT. --- kernel/bpf/verifier.c | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c index 9de0ec0c3ed9..c4e2ffadfb1f 100644 --- a/kernel/bpf/verifier.c +++ b/kernel/bpf/verifier.c @@ -17739,6 +17739,10 @@ static bool verifier_inlines_helper_call(struct bp= f_verifier_env *env, s32 imm) switch (imm) { #ifdef CONFIG_X86_64 case BPF_FUNC_get_smp_processor_id: +#ifdef CONFIG_SMP + case BPF_FUNC_get_current_task_btf: + case BPF_FUNC_get_current_task: +#endif return env->prog->jit_requested && bpf_jit_supports_percpu_insn(); #endif default: @@ -23319,6 +23323,24 @@ static int do_misc_fixups(struct bpf_verifier_env = *env) insn =3D new_prog->insnsi + i + delta; goto next_insn; } + + /* Implement bpf_get_current_task() and bpf_get_current_task_btf() inlin= e. */ + if ((insn->imm =3D=3D BPF_FUNC_get_current_task || insn->imm =3D=3D BPF_= FUNC_get_current_task_btf) && + verifier_inlines_helper_call(env, insn->imm)) { + insn_buf[0] =3D BPF_MOV64_IMM(BPF_REG_0, (u32)(unsigned long)¤t_t= ask); + insn_buf[1] =3D BPF_MOV64_PERCPU_REG(BPF_REG_0, BPF_REG_0); + insn_buf[2] =3D BPF_LDX_MEM(BPF_DW, BPF_REG_0, BPF_REG_0, 0); + cnt =3D 3; + + new_prog =3D bpf_patch_insn_data(env, i + delta, insn_buf, cnt); + if (!new_prog) + return -ENOMEM; + + delta +=3D cnt - 1; + env->prog =3D prog =3D new_prog; + insn =3D new_prog->insnsi + i + delta; + goto next_insn; + } #endif /* Implement bpf_get_func_arg inline. */ if (prog_type =3D=3D BPF_PROG_TYPE_TRACING && --=20 2.52.0 From nobody Sun Feb 8 12:20:09 2026 Received: from mail-pf1-f195.google.com (mail-pf1-f195.google.com [209.85.210.195]) (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 95CF8283FF9 for ; Mon, 19 Jan 2026 07:03:13 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.210.195 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768806194; cv=none; b=LBE48ww1qT33MlpVStdVybsSn8XfjK+VnjKJ5C4KelCjiORu6Tniub9kb6ad3bAigEnA2CXrNNR+3UnnYp22lrdwxkWn0PBBOeE/W1zC1a1H3bdz3kn8VkFmUF4tqh27S7xYoLpHNYuH3t+txvxEbMIIQCT+rJdG1NmJ0nCUTGM= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768806194; c=relaxed/simple; bh=E/tkA7w9WWDmbfkNKYISzjMSp/U2pRPxGq4HARC6vZ4=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=t1pHAFGX6C2yAWfG5cjD3QhzD23DiFOQcy/bBo4vmJVGHgysawuE6pRvXjLHoOm0JKlXKeuu/BtgXK0zGDXe4Z86Ied9KCJ5QtbDZPOyucpYw0oKpATPGFhqek9/kyNZYGsw0NfCWcoQ+JEt83X9c1qInY++e914UCUGJbNGxs8= 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=BbbyfEJ4; arc=none smtp.client-ip=209.85.210.195 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="BbbyfEJ4" Received: by mail-pf1-f195.google.com with SMTP id d2e1a72fcca58-81ef4b87291so2008462b3a.0 for ; Sun, 18 Jan 2026 23:03:13 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1768806193; x=1769410993; 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=Jm4jeH98Vgcx0inG/SSLZR49gdamhbduwf0TgJLnrGI=; b=BbbyfEJ4/UJAfcMLmvggcAaGfxKKL5RN/0/9lJCXTALPbjHOkBrSiCY3bPY42X/ODK oNHgo3tOG8+9OVPfH9oRK47Tk7xqIE0NIPbfRFopRI2OmN7NN2SheGDEFEYzD/QrZSQk RG++HlJqGmOsVjMBy0dn0MVbMqs8siaUADrxNszCk58ctg7VvuQ9zrcdO51UPr48cY1S 3Gp+qt+ehPhxvtv2a90L0RoF456j+ZRI/vmcziXvsOa1dI7cUYl9kUoTtz4G4loyHrME Ho0Uigpy/qCln0eXf7EwZDek7d1QipcuQsMB48ZAjLhp9Syqms4RUu6mJjnHuM0WSosL CWmQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1768806193; x=1769410993; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-gg:x-gm-message-state:from :to:cc:subject:date:message-id:reply-to; bh=Jm4jeH98Vgcx0inG/SSLZR49gdamhbduwf0TgJLnrGI=; b=OUTw7jE+7foo8cHkd9/4hdQ34AbBSF99FV4UmSrCdEsm3wcEDw6G3n3BrvkzKV3JZw TZhwcZnGWZrDZQj/Beo1sMeuy4A0hHqo82d+meY136xCIhsdfIrlzWyUIA6+i92+Nasa YB5deApTKeSHfuVdBPIdL9pJpm+CQPkxyb3DjBlPgMY6tRG0ijS2XO8S4HvhOYifMAal EwgYzFf5QJyqpzMISqi98SIlvghR2SiZ7VPD0MiOipBGUdNWKHcLdanCbvDUGNmV7F84 c31KWyeLziQWv9Pn4+7PqH8q0Z3Hk/G16zkASbiGRxd7uEiXbBVvvxFXc1Q1ijcyD8pM hwWA== X-Forwarded-Encrypted: i=1; AJvYcCX+Kgd/HlqUxWksCQhEoNxUp+7yrzsgBzre7xuBBj0Iw+4M1RvIGd5tAVHIoecOsnM/og99xgr00m7l6PQ=@vger.kernel.org X-Gm-Message-State: AOJu0YxHWVEjtNY+mS/qIKsKF3Cw+irrxxhMuE4p7tM1sosrs0IwW1xA x7LFUJPY//qKroDr6bjJ9OcNgbETIZWev2oq+vTGDUZKhofsrkr2qF2z X-Gm-Gg: AY/fxX7c17GEvDQaZC9+1fehfDF+G9w34t8SQ1fk4VCj9sZDy0oGvQfVIcG573ZTQ6p yfTvVKAIg14IURrcbBnOdzl3XlG0NXDIEj5Y7sSXXHrntcXzwsOB9WAyxVebu4s6GQxFoekAYwa yrq04HBvSyEl8hGsHTNLlUQQ4ni8Hek+yqwNt934AEDJO2EWs5IIYOsS6MXVyQBm9eLrhyFvd+1 /xDWfNy0pvquECyLiyI6eLNFPTUW+9FlOW/Cb2W2GcENs7CMe7TywOhoDGZunZ9fq+BVlyt4SxL Ioa/kMD97TPkO4XP4/TA3buHDuu9hw51fu5TJtLGhHxWqFr4ht7RvkOV9Xz26YtN2JB6FZo2Vtx 0Ppmgbli1J8wpi2oauxW3p9CHq/1XO0GhOAA8MYb2Dpy9dR4UmuXV8iUgBGg5jhY5TjouMmZITs gv9BDJxnRk X-Received: by 2002:a05:6a21:339c:b0:35f:aa1b:bbff with SMTP id adf61e73a8af0-38dfe590741mr9912368637.11.1768806192917; Sun, 18 Jan 2026 23:03:12 -0800 (PST) Received: from 7940hx ([103.173.155.241]) by smtp.gmail.com with ESMTPSA id 98e67ed59e1d1-352677ec7dasm10772027a91.8.2026.01.18.23.03.07 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Sun, 18 Jan 2026 23:03:12 -0800 (PST) From: Menglong Dong X-Google-Original-From: Menglong Dong To: ast@kernel.org, eddyz87@gmail.com Cc: daniel@iogearbox.net, john.fastabend@gmail.com, andrii@kernel.org, martin.lau@linux.dev, song@kernel.org, yonghong.song@linux.dev, kpsingh@kernel.org, sdf@fomichev.me, haoluo@google.com, jolsa@kernel.org, bpf@vger.kernel.org, linux-kernel@vger.kernel.org Subject: [PATCH bpf-next v5 2/2] selftests/bpf: test the jited inline of bpf_get_current_task Date: Mon, 19 Jan 2026 15:02:46 +0800 Message-ID: <20260119070246.249499-3-dongml2@chinatelecom.cn> X-Mailer: git-send-email 2.52.0 In-Reply-To: <20260119070246.249499-1-dongml2@chinatelecom.cn> References: <20260119070246.249499-1-dongml2@chinatelecom.cn> 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" Add the testcase for the jited inline of bpf_get_current_task(). Signed-off-by: Menglong Dong --- .../selftests/bpf/prog_tests/verifier.c | 2 ++ .../selftests/bpf/progs/verifier_jit_inline.c | 35 +++++++++++++++++++ 2 files changed, 37 insertions(+) create mode 100644 tools/testing/selftests/bpf/progs/verifier_jit_inline.c diff --git a/tools/testing/selftests/bpf/prog_tests/verifier.c b/tools/test= ing/selftests/bpf/prog_tests/verifier.c index 38c5ba70100c..2ae7b096bd64 100644 --- a/tools/testing/selftests/bpf/prog_tests/verifier.c +++ b/tools/testing/selftests/bpf/prog_tests/verifier.c @@ -111,6 +111,7 @@ #include "verifier_xdp_direct_packet_access.skel.h" #include "verifier_bits_iter.skel.h" #include "verifier_lsm.skel.h" +#include "verifier_jit_inline.skel.h" #include "irq.skel.h" =20 #define MAX_ENTRIES 11 @@ -253,6 +254,7 @@ void test_verifier_bits_iter(void) { RUN(verifier_bits_= iter); } void test_verifier_lsm(void) { RUN(verifier_lsm); } void test_irq(void) { RUN(irq); } void test_verifier_mtu(void) { RUN(verifier_mtu); } +void test_verifier_jit_inline(void) { RUN(verifier_jit_inlin= e); } =20 static int init_test_val_map(struct bpf_object *obj, char *map_name) { diff --git a/tools/testing/selftests/bpf/progs/verifier_jit_inline.c b/tool= s/testing/selftests/bpf/progs/verifier_jit_inline.c new file mode 100644 index 000000000000..0938ca1dac87 --- /dev/null +++ b/tools/testing/selftests/bpf/progs/verifier_jit_inline.c @@ -0,0 +1,35 @@ +// SPDX-License-Identifier: GPL-2.0 + +#include +#include +#include "bpf_misc.h" + +#if defined(__TARGET_ARCH_x86) || defined(__TARGET_ARCH_arm64) + +SEC("fentry/bpf_fentry_test1") +__description("Jit inline, bpf_get_current_task") +__success __retval(0) +__arch_x86_64 +__jited(" addq %gs:{{.*}}, %rax") +__arch_arm64 +__jited(" mrs x7, SP_EL0") +int inline_bpf_get_current_task(void) +{ + bpf_get_current_task(); + + return 0; +} + +#else + +SEC("kprobe") +__description("Jit inline is not supported, use a dummy test") +__success +int dummy_test(void) +{ + return 0; +} + +#endif + +char _license[] SEC("license") =3D "GPL"; --=20 2.52.0