From nobody Sun Feb 8 10:03:15 2026 Received: from mail-pl1-f194.google.com (mail-pl1-f194.google.com [209.85.214.194]) (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 13B912836B5 for ; Mon, 12 Jan 2026 10:45:45 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.214.194 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768214750; cv=none; b=eGFT/8JLYJaXYfPCc0RsVw1AGhwMWhteYn/fD6vtcHAnnSHfE98mswBvLPqsMRUjaYrYmwTz49WXHE7PhyEAFXdTUkkTguodnV9dH8HFvGp3rAZp3rrTKODE0ZlVNLmjSTplBT/+6JAQ4cIcviiPAglIA6EwCislEzAKgwdXIQs= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768214750; c=relaxed/simple; bh=fHGYT2wKIjsZzvpzHiDp2/v0DzQ4msIuSE/Oy76VwKY=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=JJl/f3EQWQg8pFcwyH5NkY+UykqeWmO0A45eUxZXYRaoKIwvEfG5wiT2OTRiR/x/JDWgPycplpXtrniN4aHqIzR1c7Rakv0eFbO5Zy+WyXa7NwiUAXlQXrpUYY3Z29n8e+UMp9iZAaMWlCv7smybwoH2HluSlmEEqXPmTaYj4qA= 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=ayD2FLn/; arc=none smtp.client-ip=209.85.214.194 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="ayD2FLn/" Received: by mail-pl1-f194.google.com with SMTP id d9443c01a7336-2a081c163b0so41363515ad.0 for ; Mon, 12 Jan 2026 02:45:45 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1768214745; x=1768819545; 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=+OtfIkJz+WopDcxIb9oG5mLIsx9842RNgnwqX2RofzM=; b=ayD2FLn/jzx19QbOI3MjGwLDZAq7zC3mW94Ibe5FC3PDRi5Qvc2o1FKvFHax/f0UTC wjsTPVRhmW7sfDYTyTGgNTmCZ6qfVEGyLMc42wIYTG7lFZ1BdYalkis+X9mZaGn7I+41 +jnm/SE9emPDz2oYqenLrXgHbId+swjV1ZfaMTcNG3HeASYPc+39GFspjAj8pWP5La3B 3WObxC/bGYspC/QTUdh0Sdoy2k+NhV/IuPbKxpmAVBBNToH11q+WutGyGuLBmTl2AIXC O6eAv02baE+PtuYRxiPE2HczebU9+xMkqKum/LZyvNKnC+m18efnFC7d4GoS/4kdzUj1 xaiQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1768214745; x=1768819545; 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=+OtfIkJz+WopDcxIb9oG5mLIsx9842RNgnwqX2RofzM=; b=O76BgY1qgYS7+74FpglDPMoFpV9XaJWtIbIzODIgCjOhTKQV35tZk1Fvay1Yyzljpv d9wLnshlR0FD09j09kFjGmicGF+vlg4iSqZ9DegJqwj5QQf9hi4gSz2BrpAl6qctd7F4 Q65w/Q94djFRnDnpNi+J8ZDc2wMrJesniAtiUaH/MDFV2x+jhMniAbZ8nlYoMTHfhbPu E+qxOx/ClCCXWZdaecvpnNlfRU4BV4pMT8UKkuKPi7QIYPEFzs5HojUgHZDckSeJCZDf mlFt/7L5M70nT56eLO9AbXCHzKapxjm8StDuybwv3MXmia4tIo7gATT16d6Z9oHX2MRk ovCQ== X-Forwarded-Encrypted: i=1; AJvYcCXiGIkKKzWeEujwMRxLpl2wjE1TqAtZ8M20+hqJ6cC+MZGJtySgZzUIM44GN5h+jmPJd3/nZuVJV2r9T3w=@vger.kernel.org X-Gm-Message-State: AOJu0YyxQBBucV37ducpLcgMgm0ODvJvaZkAlgxRTD0up7DFWfY3bmqS KOElvzx2WfHJcxN3JQETXYwwKyrp/mpjdRbilYVgqn5elNl3yHv8c0Pk X-Gm-Gg: AY/fxX5KglWgt/ZT9okHl3snq76BoTaXaq2gRpC40mjpe10B/uz6e7PHP5TA0Ptnh1m 15s6Lg/caTzbMV/uNUNG3tOIXQGrBdUHdLv5IT0yHr1WrCF5C95Nw5aqCPbJ+B2AIh65YAQdJOI YfTxxAuIU5V4ZLrhHWtiWjzGdltwf4oEDbyvQyVQbg+w71tuRblGC3Moc6yFn3fwRMZALDFT0Wf mXecQ9fyNObKg28KIyiQxBqO7/mrIaQb8NmkG4zIVxFK9SZeH1clIEJNj7H3icsIkXJzD7OQOmx AMpSb3pxepRGsJMewo0MBmViqmoLOx0bhjbiRSwqk4NOTrM2S/PGR0pLQLtqSGyRXRPIJJLd09k T/lkcX8R7YxHPVyBF+kGtyz6KHDYLMkHGarCbRXnl91AzYGDn0MdfgUVpOQp8+0UiVS5LxqhElq U1B1qn0Co= X-Google-Smtp-Source: AGHT+IFHYPz6czyuK+LvhbdH8QCXTjjrCywR7dF8EBPHJM5pOUuVPi0LV1ijswiqYj1oDCPUz3rFGg== X-Received: by 2002:a17:902:ce91:b0:297:d4c4:4d99 with SMTP id d9443c01a7336-2a3ee40e4b1mr165119795ad.6.1768214744938; Mon, 12 Jan 2026 02:45:44 -0800 (PST) Received: from 7940hx ([160.187.0.149]) by smtp.gmail.com with ESMTPSA id d9443c01a7336-2a3e3cb2df6sm173551775ad.61.2026.01.12.02.45.39 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Mon, 12 Jan 2026 02:45:44 -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 v4 1/2] bpf, x86: inline bpf_get_current_task() for x86_64 Date: Mon, 12 Jan 2026 18:45:28 +0800 Message-ID: <20260112104529.224645-2-dongml2@chinatelecom.cn> X-Mailer: git-send-email 2.52.0 In-Reply-To: <20260112104529.224645-1-dongml2@chinatelecom.cn> References: <20260112104529.224645-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 --- 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 | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c index 3d44c5d06623..12e99171afd8 100644 --- a/kernel/bpf/verifier.c +++ b/kernel/bpf/verifier.c @@ -17688,6 +17688,8 @@ static bool verifier_inlines_helper_call(struct bpf= _verifier_env *env, s32 imm) switch (imm) { #ifdef CONFIG_X86_64 case BPF_FUNC_get_smp_processor_id: + case BPF_FUNC_get_current_task_btf: + case BPF_FUNC_get_current_task: return env->prog->jit_requested && bpf_jit_supports_percpu_insn(); #endif default: @@ -23273,6 +23275,33 @@ 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)) { +#ifdef CONFIG_SMP + 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); +#else + struct bpf_insn ld_current_addr[2] =3D { + BPF_LD_IMM64(BPF_REG_0, (unsigned long)¤t_task) + }; + insn_buf[0] =3D ld_current_addr[0]; + insn_buf[1] =3D ld_current_addr[1]; + insn_buf[2] =3D BPF_LDX_MEM(BPF_DW, BPF_REG_0, BPF_REG_0, 0); +#endif + 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 10:03:15 2026 Received: from mail-pl1-f196.google.com (mail-pl1-f196.google.com [209.85.214.196]) (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 65E3C28000B for ; Mon, 12 Jan 2026 10:45:51 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.214.196 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768214752; cv=none; b=qoBGvs3eq0jJmkCs/I+BAG/Rn5XuouFAZArgn/A5yL+0jlS5U206aZ4UAX5LbxkG2BaLtdWbd1WXIO2nMPHHqIsh7azIpG1W5jhELYcvwD2ZxZ2qhe3f04+95HP4LVY78MsVVXu5zKv13TdkJkfr4AFGRq5/8RKWSYeRKacccWc= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1768214752; c=relaxed/simple; bh=FctuMLOWdPB2KETKqv0GYttuTULzaHfty9crA5vDYSc=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=PAmp0b1CZSEhiTdoTSdaUaYThMwNhTq9wpEW/eY0rW+VbdgliOHZc234oMgxfy3qxSacgUF+Z5V0OVOkKWG5NqEaQSR/kPPYUdw0L3T5v0us4/fo2kKfgHIC/ID7wDXKhWeytS6RR7ulYb9lwHHyde4BwjNOo66kx/tDSMC/d5A= 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=mm7rC8+2; arc=none smtp.client-ip=209.85.214.196 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="mm7rC8+2" Received: by mail-pl1-f196.google.com with SMTP id d9443c01a7336-2a0d06ffa2aso45815965ad.3 for ; Mon, 12 Jan 2026 02:45:51 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1768214751; x=1768819551; 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=pl/AXVati0RhQqhH4MEINBt0YenTEWtEVHqv/M8mNDk=; b=mm7rC8+23Wa2oeMEWx0GAwY7PX7VXsujPOC+GLvxi9tV8s5WK3tgElNI1zKWKc8+kN BHT7MRgurd+lNIbkfjtYH6OOvnNEH8ofOLZ7vs1yMKh4OdRFgWozzZD/06KAQXYzax1Y ePU0DAVxqjG4j2TOW3n9YtCwi0vyPFH71DUQGglx7NSKKGPaKBQ3BMUicxXoPfsNMiVa XRQ6z6c17GnO/N7+iXA4eTU8uURiRRgOr2EkUBkTMYaK7kR5gIX7zcQNp7qcsT1gvBCk hXfSqzG2WQr9hgSxpSzb8yHoLKH3I5tzlxXOPgI0azOqNMb5lz2h4XO0ukIqVPzr30tR D9SA== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1768214751; x=1768819551; 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=pl/AXVati0RhQqhH4MEINBt0YenTEWtEVHqv/M8mNDk=; b=stpzK1Ow/uZ0gzRu+lsRgwy61WdlFXBB4mAp5n4dbICXvnwFsOgHbAGPjALou3GeXd n2Z35QAChOWPxsM4ojCQm5sKCJHIUNjg8w6mTD0mXw+R2klA1sRjiPdpn+Dblliv3HiY /oHyBRKIEFMMqRtchbmlTodiIDsLhmblbZd0rhImm8OUB/rfBpICzB1hShqLTrymrHD8 /Q+I7bQ7wi45jCXIvvVBV7Bw4h38cIBBr4bTEkemfQ96JohrJJkmSD9+vXhOVnLjotRW A+ujfb8CJyOu/fMkfzMDHodShpcVNywC3kRhtm/5oCAdf+JImW0IJjrI20B1/8dmh0u5 JO6Q== X-Forwarded-Encrypted: i=1; AJvYcCXPCG1mIz9FpgguTtXOK64KkSCSphH1iFMdHkZvxeBUHlaD5RPIfiK6zyrfdDTyRgpf8E61uiIXNDmOH/I=@vger.kernel.org X-Gm-Message-State: AOJu0Yz6Q5vuPwQg/zjgHGVb4Qm1Kn+yHoAzXtfoQmgLD7p9IvhIjN5Z tyllYPEdXLBexKV+2/vjxwQ7z3mLBH7cnJ3wsDex2TbAivhCMU0fEMuz X-Gm-Gg: AY/fxX4NCMMnX5RPOSc6ZP8L6gOSLaSwn+qSR40MhJu0EGYyMnO2T6tefUCe22pSt+1 R10Xhv86LKe/A06BOwykGjZumRcc2/tD9IXgWQyIL68krwt919GEQt3zn+OrzwlKLVsPp7xT3vE nDmQ3yrwWhbLIsDAiAx53A+Y97dX1H/uC+TTEUG4LLH5HQvJFn9bCSzuUdoNMMlIKHnilqzmDqp aGi8Lz3n1X1z/VQyuipkokr5Ogdesh/8dlAO3tnWZlrMMSChAKkx+ao1uc4VHYpoFynqNF8oNac lVePyDBkCW24r84BCiVmEAHgb0+XAl0AmSA6DP4+YbqCaHBWUxECGp93QmK/nYFOLccoCR4CTtj 40FDllbBCH5QcaUS+HNPIcCT47r5qLT2HhHpQ9E9oQfPGprxa2x4w/QxgxMQG69hfBs9HoKpSq4 tUAQtUOv8= X-Google-Smtp-Source: AGHT+IGBW2CRWrtwyacUZsl5YgLmbet1+VCuEiQHvwANEzzhJC/Pwk5wY/VYTlUedfeKRQw8hYzTbQ== X-Received: by 2002:a17:903:1a6b:b0:295:8da5:c634 with SMTP id d9443c01a7336-2a3ee41354amr153710185ad.9.1768214750682; Mon, 12 Jan 2026 02:45:50 -0800 (PST) Received: from 7940hx ([160.187.0.149]) by smtp.gmail.com with ESMTPSA id d9443c01a7336-2a3e3cb2df6sm173551775ad.61.2026.01.12.02.45.45 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Mon, 12 Jan 2026 02:45:50 -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 v4 2/2] selftests/bpf: test the jited inline of bpf_get_current_task Date: Mon, 12 Jan 2026 18:45:29 +0800 Message-ID: <20260112104529.224645-3-dongml2@chinatelecom.cn> X-Mailer: git-send-email 2.52.0 In-Reply-To: <20260112104529.224645-1-dongml2@chinatelecom.cn> References: <20260112104529.224645-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 5829ffd70f8f..47eb78c808c0 100644 --- a/tools/testing/selftests/bpf/prog_tests/verifier.c +++ b/tools/testing/selftests/bpf/prog_tests/verifier.c @@ -110,6 +110,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 @@ -251,6 +252,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