root/tools/testing/selftests/bpf/prog_tests/test_csum_diff.c
// SPDX-License-Identifier: GPL-2.0
/* Copyright Amazon.com Inc. or its affiliates */
#include <test_progs.h>
#include "csum_diff_test.skel.h"

#define BUFF_SZ 512

struct testcase {
        unsigned long long to_buff[BUFF_SZ / 8];
        unsigned int to_buff_len;
        unsigned long long from_buff[BUFF_SZ / 8];
        unsigned int from_buff_len;
        unsigned short seed;
        unsigned short result;
};

#define NUM_PUSH_TESTS 4

struct testcase push_tests[NUM_PUSH_TESTS] = {
        {
                .to_buff = {
                        0xdeadbeefdeadbeef,
                },
                .to_buff_len = 8,
                .from_buff = {},
                .from_buff_len = 0,
                .seed = 0,
                .result = 0x3b3b
        },
        {
                .to_buff = {
                        0xdeadbeefdeadbeef,
                        0xbeefdeadbeefdead,
                },
                .to_buff_len = 16,
                .from_buff = {},
                .from_buff_len = 0,
                .seed = 0x1234,
                .result = 0x88aa
        },
        {
                .to_buff = {
                        0xdeadbeefdeadbeef,
                        0xbeefdeadbeefdead,
                },
                .to_buff_len = 15,
                .from_buff = {},
                .from_buff_len = 0,
                .seed = 0x1234,
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
                .result = 0xcaa9
#else
                .result = 0x87fd
#endif
        },
        {
                .to_buff = {
                        0x327b23c66b8b4567,
                        0x66334873643c9869,
                        0x19495cff74b0dc51,
                        0x625558ec2ae8944a,
                        0x46e87ccd238e1f29,
                        0x507ed7ab3d1b58ba,
                        0x41b71efb2eb141f2,
                        0x7545e14679e2a9e3,
                        0x5bd062c2515f007c,
                        0x4db127f812200854,
                        0x1f16e9e80216231b,
                        0x66ef438d1190cde7,
                        0x3352255a140e0f76,
                        0x0ded7263109cf92e,
                        0x1befd79f7fdcc233,
                        0x6b68079a41a7c4c9,
                        0x25e45d324e6afb66,
                        0x431bd7b7519b500d,
                        0x7c83e4583f2dba31,
                        0x62bbd95a257130a3,
                        0x628c895d436c6125,
                        0x721da317333ab105,
                        0x2d1d5ae92443a858,
                        0x75a2a8d46763845e,
                        0x79838cb208edbdab,
                        0x0b03e0c64353d0cd,
                        0x54e49eb4189a769b,
                        0x2ca8861171f32454,
                        0x02901d820836c40e,
                        0x081386413a95f874,
                        0x7c3dbd3d1e7ff521,
                        0x6ceaf087737b8ddc,
                        0x4516dde922221a70,
                        0x614fd4a13006c83e,
                        0x5577f8e1419ac241,
                        0x05072367440badfc,
                        0x77465f013804823e,
                        0x5c482a977724c67e,
                        0x5e884adc2463b9ea,
                        0x2d51779651ead36b,
                        0x153ea438580bd78f,
                        0x70a64e2a3855585c,
                        0x2a487cb06a2342ec,
                        0x725a06fb1d4ed43b,
                        0x57e4ccaf2cd89a32,
                        0x4b588f547a6d8d3c,
                        0x6de91b18542289ec,
                        0x7644a45c38437fdb,
                        0x684a481a32fff902,
                        0x749abb43579478fe,
                        0x1ba026fa3dc240fb,
                        0x75c6c33a79a1deaa,
                        0x70c6a52912e685fb,
                        0x374a3fe6520eedd1,
                        0x23f9c13c4f4ef005,
                        0x275ac794649bb77c,
                        0x1cf10fd839386575,
                        0x235ba861180115be,
                        0x354fe9f947398c89,
                        0x741226bb15b5af5c,
                        0x10233c990d34b6a8,
                        0x615740953f6ab60f,
                        0x77ae35eb7e0c57b1,
                        0x310c50b3579be4f1,
                },
                .to_buff_len = 512,
                .from_buff = {},
                .from_buff_len = 0,
                .seed = 0xffff,
                .result = 0xca45
        },
};

#define NUM_PULL_TESTS 4

struct testcase pull_tests[NUM_PULL_TESTS] = {
        {
                .from_buff = {
                        0xdeadbeefdeadbeef,
                },
                .from_buff_len = 8,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0,
                .result = 0xc4c4
        },
        {
                .from_buff = {
                        0xdeadbeefdeadbeef,
                        0xbeefdeadbeefdead,
                },
                .from_buff_len = 16,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0x1234,
                .result = 0x9bbd
        },
        {
                .from_buff = {
                        0xdeadbeefdeadbeef,
                        0xbeefdeadbeefdead,
                },
                .from_buff_len = 15,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0x1234,
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
                .result = 0x59be
#else
                .result = 0x9c6a
#endif
        },
        {
                .from_buff = {
                        0x327b23c66b8b4567,
                        0x66334873643c9869,
                        0x19495cff74b0dc51,
                        0x625558ec2ae8944a,
                        0x46e87ccd238e1f29,
                        0x507ed7ab3d1b58ba,
                        0x41b71efb2eb141f2,
                        0x7545e14679e2a9e3,
                        0x5bd062c2515f007c,
                        0x4db127f812200854,
                        0x1f16e9e80216231b,
                        0x66ef438d1190cde7,
                        0x3352255a140e0f76,
                        0x0ded7263109cf92e,
                        0x1befd79f7fdcc233,
                        0x6b68079a41a7c4c9,
                        0x25e45d324e6afb66,
                        0x431bd7b7519b500d,
                        0x7c83e4583f2dba31,
                        0x62bbd95a257130a3,
                        0x628c895d436c6125,
                        0x721da317333ab105,
                        0x2d1d5ae92443a858,
                        0x75a2a8d46763845e,
                        0x79838cb208edbdab,
                        0x0b03e0c64353d0cd,
                        0x54e49eb4189a769b,
                        0x2ca8861171f32454,
                        0x02901d820836c40e,
                        0x081386413a95f874,
                        0x7c3dbd3d1e7ff521,
                        0x6ceaf087737b8ddc,
                        0x4516dde922221a70,
                        0x614fd4a13006c83e,
                        0x5577f8e1419ac241,
                        0x05072367440badfc,
                        0x77465f013804823e,
                        0x5c482a977724c67e,
                        0x5e884adc2463b9ea,
                        0x2d51779651ead36b,
                        0x153ea438580bd78f,
                        0x70a64e2a3855585c,
                        0x2a487cb06a2342ec,
                        0x725a06fb1d4ed43b,
                        0x57e4ccaf2cd89a32,
                        0x4b588f547a6d8d3c,
                        0x6de91b18542289ec,
                        0x7644a45c38437fdb,
                        0x684a481a32fff902,
                        0x749abb43579478fe,
                        0x1ba026fa3dc240fb,
                        0x75c6c33a79a1deaa,
                        0x70c6a52912e685fb,
                        0x374a3fe6520eedd1,
                        0x23f9c13c4f4ef005,
                        0x275ac794649bb77c,
                        0x1cf10fd839386575,
                        0x235ba861180115be,
                        0x354fe9f947398c89,
                        0x741226bb15b5af5c,
                        0x10233c990d34b6a8,
                        0x615740953f6ab60f,
                        0x77ae35eb7e0c57b1,
                        0x310c50b3579be4f1,
                },
                .from_buff_len = 512,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0xffff,
                .result = 0x35ba
        },
};

#define NUM_DIFF_TESTS 4

struct testcase diff_tests[NUM_DIFF_TESTS] = {
        {
                .from_buff = {
                        0xdeadbeefdeadbeef,
                },
                .from_buff_len = 8,
                .to_buff = {
                        0xabababababababab,
                },
                .to_buff_len = 8,
                .seed = 0,
                .result = 0x7373
        },
        {
                .from_buff = {
                        0xdeadbeefdeadbeef,
                },
                .from_buff_len = 7,
                .to_buff = {
                        0xabababababababab,
                },
                .to_buff_len = 7,
                .seed = 0,
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
                .result = 0xa673
#else
                .result = 0x73b7
#endif
        },
        {
                .from_buff = {
                        0,
                },
                .from_buff_len = 8,
                .to_buff = {
                        0xabababababababab,
                },
                .to_buff_len = 8,
                .seed = 0,
                .result = 0xaeae
        },
        {
                .from_buff = {
                        0xdeadbeefdeadbeef
                },
                .from_buff_len = 8,
                .to_buff = {
                        0,
                },
                .to_buff_len = 8,
                .seed = 0xffff,
                .result = 0xc4c4
        },
};

#define NUM_EDGE_TESTS 4

struct testcase edge_tests[NUM_EDGE_TESTS] = {
        {
                .from_buff = {},
                .from_buff_len = 0,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0,
                .result = 0
        },
        {
                .from_buff = {
                        0x1234
                },
                .from_buff_len = 0,
                .to_buff = {
                        0x1234
                },
                .to_buff_len = 0,
                .seed = 0,
                .result = 0
        },
        {
                .from_buff = {},
                .from_buff_len = 0,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0x1234,
                .result = 0x1234
        },
        {
                .from_buff = {},
                .from_buff_len = 512,
                .to_buff = {},
                .to_buff_len = 0,
                .seed = 0xffff,
                .result = 0xffff
        },
};

static unsigned short trigger_csum_diff(const struct csum_diff_test *skel)
{
        u8 tmp_out[64 << 2] = {};
        u8 tmp_in[64] = {};
        int err;
        int pfd;

        LIBBPF_OPTS(bpf_test_run_opts, topts,
                .data_in = tmp_in,
                .data_size_in = sizeof(tmp_in),
                .data_out = tmp_out,
                .data_size_out = sizeof(tmp_out),
                .repeat = 1,
        );
        pfd = bpf_program__fd(skel->progs.compute_checksum);
        err = bpf_prog_test_run_opts(pfd, &topts);
        if (err)
                return -1;

        return skel->bss->result;
}

static void test_csum_diff(struct testcase *tests, int num_tests)
{
        struct csum_diff_test *skel;
        unsigned short got;
        int err;

        for (int i = 0; i < num_tests; i++) {
                skel = csum_diff_test__open();
                if (!ASSERT_OK_PTR(skel, "csum_diff_test open"))
                        return;

                skel->rodata->to_buff_len = tests[i].to_buff_len;
                skel->rodata->from_buff_len = tests[i].from_buff_len;

                err = csum_diff_test__load(skel);
                if (!ASSERT_EQ(err, 0, "csum_diff_test load"))
                        goto out;

                memcpy(skel->bss->to_buff, tests[i].to_buff, tests[i].to_buff_len);
                memcpy(skel->bss->from_buff, tests[i].from_buff, tests[i].from_buff_len);
                skel->bss->seed = tests[i].seed;

                got = trigger_csum_diff(skel);
                ASSERT_EQ(got, tests[i].result, "csum_diff result");

                csum_diff_test__destroy(skel);
        }

        return;
out:
        csum_diff_test__destroy(skel);
}

void test_test_csum_diff(void)
{
        if (test__start_subtest("csum_diff_push"))
                test_csum_diff(push_tests, NUM_PUSH_TESTS);
        if (test__start_subtest("csum_diff_pull"))
                test_csum_diff(pull_tests, NUM_PULL_TESTS);
        if (test__start_subtest("csum_diff_diff"))
                test_csum_diff(diff_tests, NUM_DIFF_TESTS);
        if (test__start_subtest("csum_diff_edge"))
                test_csum_diff(edge_tests, NUM_EDGE_TESTS);
}