// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2025 Nandakumar Edamana */
#include <linux/bpf.h>
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
#include "bpf_misc.h"

/* Intended to test the abstract multiplication technique(s) used by
 * the verifier. Using assembly to avoid compiler optimizations.
 */
SEC("fentry/bpf_fentry_test1")
void BPF_PROG(mul_precise, int x)
{
        /* First, force the verifier to be uncertain about the value:
         *     unsigned int a = (bpf_get_prandom_u32() & 0x2) | 0x1;
         *
         * Assuming the verifier is using tnum, a must be tnum{.v=0x1, .m=0x2}.
         * Then a * 0x3 would be m0m1 (m for uncertain). Added imprecision
         * would cause the following to fail, because the required return value
         * is 0:
         *     return (a * 0x3) & 0x4);
         */
        asm volatile ("\
        call %[bpf_get_prandom_u32];\
        r0 &= 0x2;\
        r0 |= 0x1;\
        r0 *= 0x3;\
        r0 &= 0x4;\
        if r0 != 0 goto l0_%=;\
        r0 = 0;\
        goto l1_%=;\
l0_%=:\
        r0 = 1;\
l1_%=:\
"       :
        : __imm(bpf_get_prandom_u32)
        : __clobber_all);
}