| // SPDX-License-Identifier: GPL-2.0 |
| /* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */ |
| |
| #include <linux/bpf.h> |
| #include <bpf/bpf_helpers.h> |
| #include "bpf_misc.h" |
| |
| struct value_t { |
| long long a[32]; |
| }; |
| |
| struct { |
| __uint(type, BPF_MAP_TYPE_HASH); |
| __uint(max_entries, 1); |
| __type(key, long long); |
| __type(value, struct value_t); |
| } map_hash SEC(".maps"); |
| |
| SEC("socket") |
| __description("bpf_jit_convergence je <-> jmp") |
| __success __retval(0) |
| __arch_x86_64 |
| __jited(" pushq %rbp") |
| __naked void btf_jit_convergence_je_jmp(void) |
| { |
| asm volatile ( |
| "call %[bpf_get_prandom_u32];" |
| "if r0 == 0 goto l20_%=;" |
| "if r0 == 1 goto l21_%=;" |
| "if r0 == 2 goto l22_%=;" |
| "if r0 == 3 goto l23_%=;" |
| "if r0 == 4 goto l24_%=;" |
| "call %[bpf_get_prandom_u32];" |
| "call %[bpf_get_prandom_u32];" |
| "l20_%=:" |
| "l21_%=:" |
| "l22_%=:" |
| "l23_%=:" |
| "l24_%=:" |
| "r1 = 0;" |
| "*(u64 *)(r10 - 8) = r1;" |
| "r2 = r10;" |
| "r2 += -8;" |
| "r1 = %[map_hash] ll;" |
| "call %[bpf_map_lookup_elem];" |
| "if r0 == 0 goto l1_%=;" |
| "r6 = r0;" |
| "call %[bpf_get_prandom_u32];" |
| "r7 = r0;" |
| "r5 = r6;" |
| "if r0 != 0x0 goto l12_%=;" |
| "call %[bpf_get_prandom_u32];" |
| "r1 = r0;" |
| "r2 = r6;" |
| "if r1 == 0x0 goto l0_%=;" |
| "l9_%=:" |
| "r2 = *(u64 *)(r6 + 0x0);" |
| "r2 += 0x1;" |
| "*(u64 *)(r6 + 0x0) = r2;" |
| "goto l1_%=;" |
| "l12_%=:" |
| "r1 = r7;" |
| "r1 += 0x98;" |
| "r2 = r5;" |
| "r2 += 0x90;" |
| "r2 = *(u32 *)(r2 + 0x0);" |
| "r3 = r7;" |
| "r3 &= 0x1;" |
| "r2 *= 0xa8;" |
| "if r3 == 0x0 goto l2_%=;" |
| "r1 += r2;" |
| "r1 -= r7;" |
| "r1 += 0x8;" |
| "if r1 <= 0xb20 goto l3_%=;" |
| "r1 = 0x0;" |
| "goto l4_%=;" |
| "l3_%=:" |
| "r1 += r7;" |
| "l4_%=:" |
| "if r1 == 0x0 goto l8_%=;" |
| "goto l9_%=;" |
| "l2_%=:" |
| "r1 += r2;" |
| "r1 -= r7;" |
| "r1 += 0x10;" |
| "if r1 <= 0xb20 goto l6_%=;" |
| "r1 = 0x0;" |
| "goto l7_%=;" |
| "l6_%=:" |
| "r1 += r7;" |
| "l7_%=:" |
| "if r1 == 0x0 goto l8_%=;" |
| "goto l9_%=;" |
| "l0_%=:" |
| "r1 = 0x3;" |
| "*(u64 *)(r10 - 0x10) = r1;" |
| "r2 = r1;" |
| "goto l1_%=;" |
| "l8_%=:" |
| "r1 = r5;" |
| "r1 += 0x4;" |
| "r1 = *(u32 *)(r1 + 0x0);" |
| "*(u64 *)(r10 - 0x8) = r1;" |
| "l1_%=:" |
| "r0 = 0;" |
| "exit;" |
| : |
| : __imm(bpf_get_prandom_u32), |
| __imm(bpf_map_lookup_elem), |
| __imm_addr(map_hash) |
| : __clobber_all); |
| } |
| |
| char _license[] SEC("license") = "GPL"; |
