1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (C) 2021. Huawei Technologies Co., Ltd */ 3 #include <argp.h> 4 #include "bench.h" 5 #include "strncmp_bench.skel.h" 6 7 static struct strncmp_ctx { 8 struct strncmp_bench *skel; 9 } ctx; 10 11 static struct strncmp_args { 12 u32 cmp_str_len; 13 } args = { 14 .cmp_str_len = 32, 15 }; 16 17 enum { 18 ARG_CMP_STR_LEN = 5000, 19 }; 20 21 static const struct argp_option opts[] = { 22 { "cmp-str-len", ARG_CMP_STR_LEN, "CMP_STR_LEN", 0, 23 "Set the length of compared string" }, 24 {}, 25 }; 26 27 static error_t strncmp_parse_arg(int key, char *arg, struct argp_state *state) 28 { 29 switch (key) { 30 case ARG_CMP_STR_LEN: 31 args.cmp_str_len = strtoul(arg, NULL, 10); 32 if (!args.cmp_str_len || 33 args.cmp_str_len >= sizeof(ctx.skel->bss->str)) { 34 fprintf(stderr, "Invalid cmp str len (limit %zu)\n", 35 sizeof(ctx.skel->bss->str)); 36 argp_usage(state); 37 } 38 break; 39 default: 40 return ARGP_ERR_UNKNOWN; 41 } 42 43 return 0; 44 } 45 46 const struct argp bench_strncmp_argp = { 47 .options = opts, 48 .parser = strncmp_parse_arg, 49 }; 50 51 static void strncmp_validate(void) 52 { 53 if (env.consumer_cnt != 0) { 54 fprintf(stderr, "strncmp benchmark doesn't support consumer!\n"); 55 exit(1); 56 } 57 } 58 59 static void strncmp_setup(void) 60 { 61 int err; 62 char *target; 63 size_t i, sz; 64 65 sz = sizeof(ctx.skel->rodata->target); 66 if (!sz || sz < sizeof(ctx.skel->bss->str)) { 67 fprintf(stderr, "invalid string size (target %zu, src %zu)\n", 68 sz, sizeof(ctx.skel->bss->str)); 69 exit(1); 70 } 71 72 setup_libbpf(); 73 74 ctx.skel = strncmp_bench__open(); 75 if (!ctx.skel) { 76 fprintf(stderr, "failed to open skeleton\n"); 77 exit(1); 78 } 79 80 srandom(time(NULL)); 81 target = ctx.skel->rodata->target; 82 for (i = 0; i < sz - 1; i++) 83 target[i] = '1' + random() % 9; 84 target[sz - 1] = '\0'; 85 86 ctx.skel->rodata->cmp_str_len = args.cmp_str_len; 87 88 memcpy(ctx.skel->bss->str, target, args.cmp_str_len); 89 ctx.skel->bss->str[args.cmp_str_len] = '\0'; 90 /* Make bss->str < rodata->target */ 91 ctx.skel->bss->str[args.cmp_str_len - 1] -= 1; 92 93 err = strncmp_bench__load(ctx.skel); 94 if (err) { 95 fprintf(stderr, "failed to load skeleton\n"); 96 strncmp_bench__destroy(ctx.skel); 97 exit(1); 98 } 99 } 100 101 static void strncmp_attach_prog(struct bpf_program *prog) 102 { 103 struct bpf_link *link; 104 105 link = bpf_program__attach(prog); 106 if (!link) { 107 fprintf(stderr, "failed to attach program!\n"); 108 exit(1); 109 } 110 } 111 112 static void strncmp_no_helper_setup(void) 113 { 114 strncmp_setup(); 115 strncmp_attach_prog(ctx.skel->progs.strncmp_no_helper); 116 } 117 118 static void strncmp_helper_setup(void) 119 { 120 strncmp_setup(); 121 strncmp_attach_prog(ctx.skel->progs.strncmp_helper); 122 } 123 124 static void *strncmp_producer(void *ctx) 125 { 126 while (true) 127 (void)syscall(__NR_getpgid); 128 return NULL; 129 } 130 131 static void strncmp_measure(struct bench_res *res) 132 { 133 res->hits = atomic_swap(&ctx.skel->bss->hits, 0); 134 } 135 136 const struct bench bench_strncmp_no_helper = { 137 .name = "strncmp-no-helper", 138 .argp = &bench_strncmp_argp, 139 .validate = strncmp_validate, 140 .setup = strncmp_no_helper_setup, 141 .producer_thread = strncmp_producer, 142 .measure = strncmp_measure, 143 .report_progress = hits_drops_report_progress, 144 .report_final = hits_drops_report_final, 145 }; 146 147 const struct bench bench_strncmp_helper = { 148 .name = "strncmp-helper", 149 .argp = &bench_strncmp_argp, 150 .validate = strncmp_validate, 151 .setup = strncmp_helper_setup, 152 .producer_thread = strncmp_producer, 153 .measure = strncmp_measure, 154 .report_progress = hits_drops_report_progress, 155 .report_final = hits_drops_report_final, 156 }; 157