]>
Commit | Line | Data |
---|---|---|
0b1abd84 YG |
1 | /* |
2 | * m_sample.c ingress/egress packet sampling module | |
3 | * | |
4 | * This program is free software; you can distribute it and/or | |
5 | * modify it under the terms of the GNU General Public License | |
6 | * as published by the Free Software Foundation; either version | |
7 | * 2 of the License, or (at your option) any later version. | |
8 | * | |
9 | * Authors: Yotam Gigi <yotamg@mellanox.com> | |
10 | * | |
11 | */ | |
12 | ||
13 | #include <stdio.h> | |
14 | #include "utils.h" | |
15 | #include "tc_util.h" | |
16 | #include "tc_common.h" | |
17 | #include <linux/tc_act/tc_sample.h> | |
18 | ||
19 | static void explain(void) | |
20 | { | |
21 | fprintf(stderr, "Usage: sample SAMPLE_CONF\n"); | |
22 | fprintf(stderr, "where:\n"); | |
23 | fprintf(stderr, "\tSAMPLE_CONF := SAMPLE_PARAMS | SAMPLE_INDEX\n"); | |
24 | fprintf(stderr, "\tSAMPLE_PARAMS := rate RATE group GROUP [trunc SIZE] [SAMPLE_INDEX]\n"); | |
25 | fprintf(stderr, "\tSAMPLE_INDEX := index INDEX\n"); | |
26 | fprintf(stderr, "\tRATE := The ratio of packets observed at the data source to the samples generated.\n"); | |
27 | fprintf(stderr, "\tGROUP := the psample sampling group\n"); | |
28 | fprintf(stderr, "\tSIZE := the truncation size\n"); | |
29 | fprintf(stderr, "\tINDEX := integer index of the sample action\n"); | |
30 | } | |
31 | ||
32 | static void usage(void) | |
33 | { | |
34 | explain(); | |
35 | exit(-1); | |
36 | } | |
37 | ||
38 | static int parse_sample(struct action_util *a, int *argc_p, char ***argv_p, | |
39 | int tca_id, struct nlmsghdr *n) | |
40 | { | |
41 | struct tc_sample p = { 0 }; | |
42 | bool trunc_set = false; | |
43 | bool group_set = false; | |
44 | bool rate_set = false; | |
45 | char **argv = *argv_p; | |
46 | struct rtattr *tail; | |
47 | int argc = *argc_p; | |
48 | __u32 trunc; | |
49 | __u32 group; | |
50 | __u32 rate; | |
51 | ||
52 | if (argc <= 1) { | |
53 | fprintf(stderr, "sample bad argument count %d\n", argc); | |
54 | usage(); | |
55 | return -1; | |
56 | } | |
57 | ||
58 | if (matches(*argv, "sample") == 0) { | |
59 | NEXT_ARG(); | |
60 | } else { | |
61 | fprintf(stderr, "sample bad argument %s\n", *argv); | |
62 | return -1; | |
63 | } | |
64 | ||
65 | while (argc > 0) { | |
66 | if (matches(*argv, "rate") == 0) { | |
67 | NEXT_ARG(); | |
bf7d1488 | 68 | if (get_u32(&rate, *argv, 10) != 0) { |
0b1abd84 YG |
69 | fprintf(stderr, "Illegal rate %s\n", *argv); |
70 | usage(); | |
71 | return -1; | |
72 | } | |
73 | rate_set = true; | |
74 | } else if (matches(*argv, "group") == 0) { | |
75 | NEXT_ARG(); | |
bf7d1488 | 76 | if (get_u32(&group, *argv, 10) != 0) { |
0b1abd84 YG |
77 | fprintf(stderr, "Illegal group num %s\n", |
78 | *argv); | |
79 | usage(); | |
80 | return -1; | |
81 | } | |
82 | group_set = true; | |
83 | } else if (matches(*argv, "trunc") == 0) { | |
84 | NEXT_ARG(); | |
bf7d1488 | 85 | if (get_u32(&trunc, *argv, 10) != 0) { |
0b1abd84 YG |
86 | fprintf(stderr, "Illegal truncation size %s\n", |
87 | *argv); | |
88 | usage(); | |
89 | return -1; | |
90 | } | |
91 | trunc_set = true; | |
92 | } else if (matches(*argv, "help") == 0) { | |
93 | usage(); | |
94 | } else { | |
95 | break; | |
96 | } | |
97 | ||
98 | NEXT_ARG_FWD(); | |
99 | } | |
100 | ||
e67aba55 | 101 | parse_action_control_dflt(&argc, &argv, &p.action, false, TC_ACT_PIPE); |
0b1abd84 YG |
102 | |
103 | if (argc) { | |
104 | if (matches(*argv, "index") == 0) { | |
105 | NEXT_ARG(); | |
106 | if (get_u32(&p.index, *argv, 10)) { | |
107 | fprintf(stderr, "sample: Illegal \"index\"\n"); | |
108 | return -1; | |
109 | } | |
110 | NEXT_ARG_FWD(); | |
111 | } | |
112 | } | |
113 | ||
114 | if (!p.index && !group_set) { | |
115 | fprintf(stderr, "param \"group\" not set\n"); | |
116 | usage(); | |
117 | } | |
118 | ||
119 | if (!p.index && !rate_set) { | |
120 | fprintf(stderr, "param \"rate\" not set\n"); | |
121 | usage(); | |
122 | } | |
123 | ||
c14f9d92 | 124 | tail = addattr_nest(n, MAX_MSG, tca_id); |
0b1abd84 YG |
125 | addattr_l(n, MAX_MSG, TCA_SAMPLE_PARMS, &p, sizeof(p)); |
126 | if (rate_set) | |
127 | addattr32(n, MAX_MSG, TCA_SAMPLE_RATE, rate); | |
128 | if (group_set) | |
129 | addattr32(n, MAX_MSG, TCA_SAMPLE_PSAMPLE_GROUP, group); | |
130 | if (trunc_set) | |
131 | addattr32(n, MAX_MSG, TCA_SAMPLE_TRUNC_SIZE, trunc); | |
132 | ||
c14f9d92 | 133 | addattr_nest_end(n, tail); |
0b1abd84 YG |
134 | |
135 | *argc_p = argc; | |
136 | *argv_p = argv; | |
137 | return 0; | |
138 | } | |
139 | ||
140 | static int print_sample(struct action_util *au, FILE *f, struct rtattr *arg) | |
141 | { | |
142 | struct rtattr *tb[TCA_SAMPLE_MAX + 1]; | |
143 | struct tc_sample *p; | |
144 | ||
145 | if (arg == NULL) | |
146 | return -1; | |
147 | ||
148 | parse_rtattr_nested(tb, TCA_SAMPLE_MAX, arg); | |
149 | ||
150 | if (!tb[TCA_SAMPLE_PARMS] || !tb[TCA_SAMPLE_RATE] || | |
151 | !tb[TCA_SAMPLE_PSAMPLE_GROUP]) { | |
7ada016a | 152 | print_string(PRINT_FP, NULL, "%s", "[NULL sample parameters]"); |
0b1abd84 YG |
153 | return -1; |
154 | } | |
155 | p = RTA_DATA(tb[TCA_SAMPLE_PARMS]); | |
156 | ||
7ada016a RM |
157 | print_string(PRINT_ANY, "kind", "%s ", "sample"); |
158 | print_uint(PRINT_ANY, "rate", "rate 1/%u ", | |
159 | rta_getattr_u32(tb[TCA_SAMPLE_RATE])); | |
160 | print_uint(PRINT_ANY, "group", "group %u", | |
161 | rta_getattr_u32(tb[TCA_SAMPLE_PSAMPLE_GROUP])); | |
0b1abd84 YG |
162 | |
163 | if (tb[TCA_SAMPLE_TRUNC_SIZE]) | |
7ada016a RM |
164 | print_uint(PRINT_ANY, "trunc_size", " trunc_size %u", |
165 | rta_getattr_u32(tb[TCA_SAMPLE_TRUNC_SIZE])); | |
0b1abd84 | 166 | |
e9fa1658 RM |
167 | print_action_control(f, " ", p->action, ""); |
168 | ||
7ada016a RM |
169 | print_string(PRINT_FP, NULL, "%s", _SL_); |
170 | print_uint(PRINT_ANY, "index", "\t index %u", p->index); | |
171 | print_int(PRINT_ANY, "ref", " ref %d", p->refcnt); | |
172 | print_int(PRINT_ANY, "bind", " bind %d", p->bindcnt); | |
0b1abd84 YG |
173 | |
174 | if (show_stats) { | |
175 | if (tb[TCA_SAMPLE_TM]) { | |
176 | struct tcf_t *tm = RTA_DATA(tb[TCA_SAMPLE_TM]); | |
177 | ||
178 | print_tm(f, tm); | |
179 | } | |
180 | } | |
7ada016a | 181 | print_string(PRINT_FP, NULL, "%s", _SL_); |
0b1abd84 YG |
182 | return 0; |
183 | } | |
184 | ||
185 | struct action_util sample_action_util = { | |
186 | .id = "sample", | |
187 | .parse_aopt = parse_sample, | |
188 | .print_aopt = print_sample, | |
189 | }; |