bpf/prog_tests/snprintf.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /* Copyright (c) 2021 Google LLC. */
0003
0004 #include <test_progs.h>
0005 #include "test_snprintf.skel.h"
0006 #include "test_snprintf_single.skel.h"
0007
0008 #define EXP_NUM_OUT  "-8 9 96 -424242 1337 DABBAD00"
0009 #define EXP_NUM_RET  sizeof(EXP_NUM_OUT)
0010
0011 #define EXP_IP_OUT   "127.000.000.001 0000:0000:0000:0000:0000:0000:0000:0001"
0012 #define EXP_IP_RET   sizeof(EXP_IP_OUT)
0013
0014 /* The third specifier, %pB, depends on compiler inlining so don't check it */
0015 #define EXP_SYM_OUT  "schedule schedule+0x0/"
0016 #define MIN_SYM_RET  sizeof(EXP_SYM_OUT)
0017
0018 /* The third specifier, %p, is a hashed pointer which changes on every reboot */
0019 #define EXP_ADDR_OUT "0000000000000000 ffff00000add4e55 "
0020 #define EXP_ADDR_RET sizeof(EXP_ADDR_OUT "unknownhashedptr")
0021
0022 #define EXP_STR_OUT  "str1         a  b c      d e longstr"
0023 #define EXP_STR_RET  sizeof(EXP_STR_OUT)
0024
0025 #define EXP_OVER_OUT "%over"
0026 #define EXP_OVER_RET 10
0027
0028 #define EXP_PAD_OUT "    4 000"
0029 #define EXP_PAD_RET 900007
0030
0031 #define EXP_NO_ARG_OUT "simple case"
0032 #define EXP_NO_ARG_RET 12
0033
0034 #define EXP_NO_BUF_RET 29
0035
0036 static void test_snprintf_positive(void)
0037 {
0038     char exp_addr_out[] = EXP_ADDR_OUT;
0039     char exp_sym_out[]  = EXP_SYM_OUT;
0040     struct test_snprintf *skel;
0041
0042     skel = test_snprintf__open_and_load();
0043     if (!ASSERT_OK_PTR(skel, "skel_open"))
0044         return;
0045
0046     skel->bss->pid = getpid();
0047
0048     if (!ASSERT_OK(test_snprintf__attach(skel), "skel_attach"))
0049         goto cleanup;
0050
0051     /* trigger tracepoint */
0052     usleep(1);
0053
0054     ASSERT_STREQ(skel->bss->num_out, EXP_NUM_OUT, "num_out");
0055     ASSERT_EQ(skel->bss->num_ret, EXP_NUM_RET, "num_ret");
0056
0057     ASSERT_STREQ(skel->bss->ip_out, EXP_IP_OUT, "ip_out");
0058     ASSERT_EQ(skel->bss->ip_ret, EXP_IP_RET, "ip_ret");
0059
0060     ASSERT_OK(memcmp(skel->bss->sym_out, exp_sym_out,
0061              sizeof(exp_sym_out) - 1), "sym_out");
0062     ASSERT_LT(MIN_SYM_RET, skel->bss->sym_ret, "sym_ret");
0063
0064     ASSERT_OK(memcmp(skel->bss->addr_out, exp_addr_out,
0065              sizeof(exp_addr_out) - 1), "addr_out");
0066     ASSERT_EQ(skel->bss->addr_ret, EXP_ADDR_RET, "addr_ret");
0067
0068     ASSERT_STREQ(skel->bss->str_out, EXP_STR_OUT, "str_out");
0069     ASSERT_EQ(skel->bss->str_ret, EXP_STR_RET, "str_ret");
0070
0071     ASSERT_STREQ(skel->bss->over_out, EXP_OVER_OUT, "over_out");
0072     ASSERT_EQ(skel->bss->over_ret, EXP_OVER_RET, "over_ret");
0073
0074     ASSERT_STREQ(skel->bss->pad_out, EXP_PAD_OUT, "pad_out");
0075     ASSERT_EQ(skel->bss->pad_ret, EXP_PAD_RET, "pad_ret");
0076
0077     ASSERT_STREQ(skel->bss->noarg_out, EXP_NO_ARG_OUT, "no_arg_out");
0078     ASSERT_EQ(skel->bss->noarg_ret, EXP_NO_ARG_RET, "no_arg_ret");
0079
0080     ASSERT_EQ(skel->bss->nobuf_ret, EXP_NO_BUF_RET, "no_buf_ret");
0081
0082 cleanup:
0083     test_snprintf__destroy(skel);
0084 }
0085
0086 /* Loads an eBPF object calling bpf_snprintf with up to 10 characters of fmt */
0087 static int load_single_snprintf(char *fmt)
0088 {
0089     struct test_snprintf_single *skel;
0090     int ret;
0091
0092     skel = test_snprintf_single__open();
0093     if (!skel)
0094         return -EINVAL;
0095
0096     memcpy(skel->rodata->fmt, fmt, MIN(strlen(fmt) + 1, 10));
0097
0098     ret = test_snprintf_single__load(skel);
0099     test_snprintf_single__destroy(skel);
0100
0101     return ret;
0102 }
0103
0104 static void test_snprintf_negative(void)
0105 {
0106     ASSERT_OK(load_single_snprintf("valid %d"), "valid usage");
0107
0108     ASSERT_ERR(load_single_snprintf("0123456789"), "no terminating zero");
0109     ASSERT_ERR(load_single_snprintf("%d %d"), "too many specifiers");
0110     ASSERT_ERR(load_single_snprintf("%pi5"), "invalid specifier 1");
0111     ASSERT_ERR(load_single_snprintf("%a"), "invalid specifier 2");
0112     ASSERT_ERR(load_single_snprintf("%"), "invalid specifier 3");
0113     ASSERT_ERR(load_single_snprintf("%12345678"), "invalid specifier 4");
0114     ASSERT_ERR(load_single_snprintf("%--------"), "invalid specifier 5");
0115     ASSERT_ERR(load_single_snprintf("%lc"), "invalid specifier 6");
0116     ASSERT_ERR(load_single_snprintf("%llc"), "invalid specifier 7");
0117     ASSERT_ERR(load_single_snprintf("\x80"), "non ascii character");
0118     ASSERT_ERR(load_single_snprintf("\x1"), "non printable character");
0119 }
0120
0121 void test_snprintf(void)
0122 {
0123     if (test__start_subtest("snprintf_positive"))
0124         test_snprintf_positive();
0125     if (test__start_subtest("snprintf_negative"))
0126         test_snprintf_negative();
0127 }