LXR linux/tools/testing/selftests/bpf/test_btf

   1#include <stdio.h>
   2#include <stdlib.h>
   3#include <string.h>
   4#include <unistd.h>
   5#include <errno.h>
   6#include <linux/err.h>
   7#include <btf.h>
   8
   9#define CHECK(condition, format...) ({                                  \
  10        int __ret = !!(condition);                                      \
  11        if (__ret) {                                                    \
  12                fprintf(stderr, "%s:%d:FAIL ", __func__, __LINE__);     \
  13                fprintf(stderr, format);                                \
  14        }                                                               \
  15        __ret;                                                          \
  16})
  17
  18void btf_dump_printf(void *ctx, const char *fmt, va_list args)
  19{
  20        vfprintf(ctx, fmt, args);
  21}
  22
  23struct btf_dump_test_case {
  24        const char *name;
  25        struct btf_dump_opts opts;
  26} btf_dump_test_cases[] = {
  27        {.name = "btf_dump_test_case_syntax", .opts = {}},
  28        {.name = "btf_dump_test_case_ordering", .opts = {}},
  29        {.name = "btf_dump_test_case_padding", .opts = {}},
  30        {.name = "btf_dump_test_case_packing", .opts = {}},
  31        {.name = "btf_dump_test_case_bitfields", .opts = {}},
  32        {.name = "btf_dump_test_case_multidim", .opts = {}},
  33        {.name = "btf_dump_test_case_namespacing", .opts = {}},
  34};
  35
  36static int btf_dump_all_types(const struct btf *btf,
  37                              const struct btf_dump_opts *opts)
  38{
  39        size_t type_cnt = btf__get_nr_types(btf);
  40        struct btf_dump *d;
  41        int err = 0, id;
  42
  43        d = btf_dump__new(btf, NULL, opts, btf_dump_printf);
  44        if (IS_ERR(d))
  45                return PTR_ERR(d);
  46
  47        for (id = 1; id <= type_cnt; id++) {
  48                err = btf_dump__dump_type(d, id);
  49                if (err)
  50                        goto done;
  51        }
  52
  53done:
  54        btf_dump__free(d);
  55        return err;
  56}
  57
  58int test_btf_dump_case(int n, struct btf_dump_test_case *test_case)
  59{
  60        char test_file[256], out_file[256], diff_cmd[1024];
  61        struct btf *btf = NULL;
  62        int err = 0, fd = -1;
  63        FILE *f = NULL;
  64
  65        fprintf(stderr, "Test case #%d (%s): ", n, test_case->name);
  66
  67        snprintf(test_file, sizeof(test_file), "%s.o", test_case->name);
  68
  69        btf = btf__parse_elf(test_file, NULL);
  70        if (CHECK(IS_ERR(btf),
  71            "failed to load test BTF: %ld\n", PTR_ERR(btf))) {
  72                err = -PTR_ERR(btf);
  73                btf = NULL;
  74                goto done;
  75        }
  76
  77        snprintf(out_file, sizeof(out_file),
  78                 "/tmp/%s.output.XXXXXX", test_case->name);
  79        fd = mkstemp(out_file);
  80        if (CHECK(fd < 0, "failed to create temp output file: %d\n", fd)) {
  81                err = fd;
  82                goto done;
  83        }
  84        f = fdopen(fd, "w");
  85        if (CHECK(f == NULL, "failed to open temp output file: %s(%d)\n",
  86                  strerror(errno), errno)) {
  87                close(fd);
  88                goto done;
  89        }
  90
  91        test_case->opts.ctx = f;
  92        err = btf_dump_all_types(btf, &test_case->opts);
  93        fclose(f);
  94        close(fd);
  95        if (CHECK(err, "failure during C dumping: %d\n", err)) {
  96                goto done;
  97        }
  98
  99        snprintf(test_file, sizeof(test_file), "progs/%s.c", test_case->name);
 100        if (access(test_file, R_OK) == -1)
 101                /*
 102                 * When the test is run with O=, kselftest copies TEST_FILES
 103                 * without preserving the directory structure.
 104                 */
 105                snprintf(test_file, sizeof(test_file), "%s.c",
 106                        test_case->name);
 107        /*
 108         * Diff test output and expected test output, contained between
 109         * START-EXPECTED-OUTPUT and END-EXPECTED-OUTPUT lines in test case.
 110         * For expected output lines, everything before '*' is stripped out.
 111         * Also lines containing comment start and comment end markers are
 112         * ignored. 
 113         */
 114        snprintf(diff_cmd, sizeof(diff_cmd),
 115                 "awk '/START-EXPECTED-OUTPUT/{out=1;next} "
 116                 "/END-EXPECTED-OUTPUT/{out=0} "
 117                 "/\\/\\*|\\*\\//{next} " /* ignore comment start/end lines */
 118                 "out {sub(/^[ \\t]*\\*/, \"\"); print}' '%s' | diff -u - '%s'",
 119                 test_file, out_file);
 120        err = system(diff_cmd);
 121        if (CHECK(err,
 122                  "differing test output, output=%s, err=%d, diff cmd:\n%s\n",
 123                  out_file, err, diff_cmd))
 124                goto done;
 125
 126        remove(out_file);
 127        fprintf(stderr, "OK\n");
 128
 129done:
 130        btf__free(btf);
 131        return err;
 132}
 133
 134int main() {
 135        int test_case_cnt, i, err, failed = 0;
 136
 137        test_case_cnt = sizeof(btf_dump_test_cases) /
 138                        sizeof(btf_dump_test_cases[0]);
 139
 140        for (i = 0; i < test_case_cnt; i++) {
 141                err = test_btf_dump_case(i, &btf_dump_test_cases[i]);
 142                if (err)
 143                        failed++;
 144        }
 145
 146        fprintf(stderr, "%d tests succeeded, %d tests failed.\n",
 147                test_case_cnt - failed, failed);
 148
 149        return failed;
 150}
 151