tools/testing/selftests/bpf/prog_tests/snprintf.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2021 Google LLC. */
   3
   4 #include <test_progs.h>
   5 #include "test_snprintf.skel.h"
   6 #include "test_snprintf_single.skel.h"
   7
   8 #define EXP_NUM_OUT  "-8 9 96 -424242 1337 DABBAD00"
   9 #define EXP_NUM_RET  sizeof(EXP_NUM_OUT)
  10
  11 #define EXP_IP_OUT   "127.000.000.001 0000:0000:0000:0000:0000:0000:0000:0001"
  12 #define EXP_IP_RET   sizeof(EXP_IP_OUT)
  13
  14 /* The third specifier, %pB, depends on compiler inlining so don't check it */
  15 #define EXP_SYM_OUT  "schedule schedule+0x0/"
  16 #define MIN_SYM_RET  sizeof(EXP_SYM_OUT)
  17
  18 /* The third specifier, %p, is a hashed pointer which changes on every reboot */
  19 #define EXP_ADDR_OUT "0000000000000000 ffff00000add4e55 "
  20 #define EXP_ADDR_RET sizeof(EXP_ADDR_OUT "unknownhashedptr")
  21
  22 #define EXP_STR_OUT  "str1 longstr"
  23 #define EXP_STR_RET  sizeof(EXP_STR_OUT)
  24
  25 #define EXP_OVER_OUT "%over"
  26 #define EXP_OVER_RET 10
  27
  28 #define EXP_PAD_OUT "    4 000"
  29 #define EXP_PAD_RET 900007
  30
  31 #define EXP_NO_ARG_OUT "simple case"
  32 #define EXP_NO_ARG_RET 12
  33
  34 #define EXP_NO_BUF_RET 29
  35
  36 void test_snprintf_positive(void)
  37 {
  38         char exp_addr_out[] = EXP_ADDR_OUT;
  39         char exp_sym_out[]  = EXP_SYM_OUT;
  40         struct test_snprintf *skel;
  41
  42         skel = test_snprintf__open_and_load();
  43         if (!ASSERT_OK_PTR(skel, "skel_open"))
  44                 return;
  45
  46         skel->bss->pid = getpid();
  47
  48         if (!ASSERT_OK(test_snprintf__attach(skel), "skel_attach"))
  49                 goto cleanup;
  50
  51         /* trigger tracepoint */
  52         usleep(1);
  53
  54         ASSERT_STREQ(skel->bss->num_out, EXP_NUM_OUT, "num_out");
  55         ASSERT_EQ(skel->bss->num_ret, EXP_NUM_RET, "num_ret");
  56
  57         ASSERT_STREQ(skel->bss->ip_out, EXP_IP_OUT, "ip_out");
  58         ASSERT_EQ(skel->bss->ip_ret, EXP_IP_RET, "ip_ret");
  59
  60         ASSERT_OK(memcmp(skel->bss->sym_out, exp_sym_out,
  61                          sizeof(exp_sym_out) - 1), "sym_out");
  62         ASSERT_LT(MIN_SYM_RET, skel->bss->sym_ret, "sym_ret");
  63
  64         ASSERT_OK(memcmp(skel->bss->addr_out, exp_addr_out,
  65                          sizeof(exp_addr_out) - 1), "addr_out");
  66         ASSERT_EQ(skel->bss->addr_ret, EXP_ADDR_RET, "addr_ret");
  67
  68         ASSERT_STREQ(skel->bss->str_out, EXP_STR_OUT, "str_out");
  69         ASSERT_EQ(skel->bss->str_ret, EXP_STR_RET, "str_ret");
  70
  71         ASSERT_STREQ(skel->bss->over_out, EXP_OVER_OUT, "over_out");
  72         ASSERT_EQ(skel->bss->over_ret, EXP_OVER_RET, "over_ret");
  73
  74         ASSERT_STREQ(skel->bss->pad_out, EXP_PAD_OUT, "pad_out");
  75         ASSERT_EQ(skel->bss->pad_ret, EXP_PAD_RET, "pad_ret");
  76
  77         ASSERT_STREQ(skel->bss->noarg_out, EXP_NO_ARG_OUT, "no_arg_out");
  78         ASSERT_EQ(skel->bss->noarg_ret, EXP_NO_ARG_RET, "no_arg_ret");
  79
  80         ASSERT_EQ(skel->bss->nobuf_ret, EXP_NO_BUF_RET, "no_buf_ret");
  81
  82 cleanup:
  83         test_snprintf__destroy(skel);
  84 }
  85
  86 #define min(a, b) ((a) < (b) ? (a) : (b))
  87
  88 /* Loads an eBPF object calling bpf_snprintf with up to 10 characters of fmt */
  89 static int load_single_snprintf(char *fmt)
  90 {
  91         struct test_snprintf_single *skel;
  92         int ret;
  93
  94         skel = test_snprintf_single__open();
  95         if (!skel)
  96                 return -EINVAL;
  97
  98         memcpy(skel->rodata->fmt, fmt, min(strlen(fmt) + 1, 10));
  99
 100         ret = test_snprintf_single__load(skel);
 101         test_snprintf_single__destroy(skel);
 102
 103         return ret;
 104 }
 105
 106 void test_snprintf_negative(void)
 107 {
 108         ASSERT_OK(load_single_snprintf("valid %d"), "valid usage");
 109
 110         ASSERT_ERR(load_single_snprintf("0123456789"), "no terminating zero");
 111         ASSERT_ERR(load_single_snprintf("%d %d"), "too many specifiers");
 112         ASSERT_ERR(load_single_snprintf("%pi5"), "invalid specifier 1");
 113         ASSERT_ERR(load_single_snprintf("%a"), "invalid specifier 2");
 114         ASSERT_ERR(load_single_snprintf("%"), "invalid specifier 3");
 115         ASSERT_ERR(load_single_snprintf("%12345678"), "invalid specifier 4");
 116         ASSERT_ERR(load_single_snprintf("%--------"), "invalid specifier 5");
 117         ASSERT_ERR(load_single_snprintf("\x80"), "non ascii character");
 118         ASSERT_ERR(load_single_snprintf("\x1"), "non printable character");
 119 }
 120
 121 void test_snprintf(void)
 122 {
 123         if (test__start_subtest("snprintf_positive"))
 124                 test_snprintf_positive();
 125         if (test__start_subtest("snprintf_negative"))
 126                 test_snprintf_negative();
 127 }