|
Packit Service |
3880ab |
/*
|
|
Packit Service |
3880ab |
* m_sample.c ingress/egress packet sampling module
|
|
Packit Service |
3880ab |
*
|
|
Packit Service |
3880ab |
* This program is free software; you can distribute it and/or
|
|
Packit Service |
3880ab |
* modify it under the terms of the GNU General Public License
|
|
Packit Service |
3880ab |
* as published by the Free Software Foundation; either version
|
|
Packit Service |
3880ab |
* 2 of the License, or (at your option) any later version.
|
|
Packit Service |
3880ab |
*
|
|
Packit Service |
3880ab |
* Authors: Yotam Gigi <yotamg@mellanox.com>
|
|
Packit Service |
3880ab |
*
|
|
Packit Service |
3880ab |
*/
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
#include <stdio.h>
|
|
Packit Service |
3880ab |
#include "utils.h"
|
|
Packit Service |
3880ab |
#include "tc_util.h"
|
|
Packit Service |
3880ab |
#include "tc_common.h"
|
|
Packit Service |
3880ab |
#include <linux/tc_act/tc_sample.h>
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static void explain(void)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
fprintf(stderr,
|
|
Packit Service |
3880ab |
"Usage: sample SAMPLE_CONF\n"
|
|
Packit Service |
3880ab |
"where:\n"
|
|
Packit Service |
3880ab |
"\tSAMPLE_CONF := SAMPLE_PARAMS | SAMPLE_INDEX\n"
|
|
Packit Service |
3880ab |
"\tSAMPLE_PARAMS := rate RATE group GROUP [trunc SIZE] [SAMPLE_INDEX]\n"
|
|
Packit Service |
3880ab |
"\tSAMPLE_INDEX := index INDEX\n"
|
|
Packit Service |
3880ab |
"\tRATE := The ratio of packets observed at the data source to the samples generated.\n"
|
|
Packit Service |
3880ab |
"\tGROUP := the psample sampling group\n"
|
|
Packit Service |
3880ab |
"\tSIZE := the truncation size\n"
|
|
Packit Service |
3880ab |
"\tINDEX := integer index of the sample action\n");
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static void usage(void)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
explain();
|
|
Packit Service |
3880ab |
exit(-1);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static int parse_sample(struct action_util *a, int *argc_p, char ***argv_p,
|
|
Packit Service |
3880ab |
int tca_id, struct nlmsghdr *n)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
struct tc_sample p = { 0 };
|
|
Packit Service |
3880ab |
bool trunc_set = false;
|
|
Packit Service |
3880ab |
bool group_set = false;
|
|
Packit Service |
3880ab |
bool rate_set = false;
|
|
Packit Service |
3880ab |
char **argv = *argv_p;
|
|
Packit Service |
3880ab |
struct rtattr *tail;
|
|
Packit Service |
3880ab |
int argc = *argc_p;
|
|
Packit Service |
3880ab |
__u32 trunc;
|
|
Packit Service |
3880ab |
__u32 group;
|
|
Packit Service |
3880ab |
__u32 rate;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (argc <= 1) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "sample bad argument count %d\n", argc);
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (matches(*argv, "sample") == 0) {
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
fprintf(stderr, "sample bad argument %s\n", *argv);
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
while (argc > 0) {
|
|
Packit Service |
3880ab |
if (matches(*argv, "rate") == 0) {
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
if (get_u32(&rate, *argv, 10) != 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Illegal rate %s\n", *argv);
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
rate_set = true;
|
|
Packit Service |
3880ab |
} else if (matches(*argv, "group") == 0) {
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
if (get_u32(&group, *argv, 10) != 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Illegal group num %s\n",
|
|
Packit Service |
3880ab |
*argv);
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
group_set = true;
|
|
Packit Service |
3880ab |
} else if (matches(*argv, "trunc") == 0) {
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
if (get_u32(&trunc, *argv, 10) != 0) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Illegal truncation size %s\n",
|
|
Packit Service |
3880ab |
*argv);
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
trunc_set = true;
|
|
Packit Service |
3880ab |
} else if (matches(*argv, "help") == 0) {
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
} else {
|
|
Packit Service |
3880ab |
break;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
NEXT_ARG_FWD();
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
parse_action_control_dflt(&argc, &argv, &p.action, false, TC_ACT_PIPE);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (argc) {
|
|
Packit Service |
3880ab |
if (matches(*argv, "index") == 0) {
|
|
Packit Service |
3880ab |
NEXT_ARG();
|
|
Packit Service |
3880ab |
if (get_u32(&p.index, *argv, 10)) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "sample: Illegal \"index\"\n");
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
NEXT_ARG_FWD();
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (!p.index && !group_set) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "param \"group\" not set\n");
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (!p.index && !rate_set) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "param \"rate\" not set\n");
|
|
Packit Service |
3880ab |
usage();
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
tail = addattr_nest(n, MAX_MSG, tca_id);
|
|
Packit Service |
3880ab |
addattr_l(n, MAX_MSG, TCA_SAMPLE_PARMS, &p, sizeof(p));
|
|
Packit Service |
3880ab |
if (rate_set)
|
|
Packit Service |
3880ab |
addattr32(n, MAX_MSG, TCA_SAMPLE_RATE, rate);
|
|
Packit Service |
3880ab |
if (group_set)
|
|
Packit Service |
3880ab |
addattr32(n, MAX_MSG, TCA_SAMPLE_PSAMPLE_GROUP, group);
|
|
Packit Service |
3880ab |
if (trunc_set)
|
|
Packit Service |
3880ab |
addattr32(n, MAX_MSG, TCA_SAMPLE_TRUNC_SIZE, trunc);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
addattr_nest_end(n, tail);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
*argc_p = argc;
|
|
Packit Service |
3880ab |
*argv_p = argv;
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
static int print_sample(struct action_util *au, FILE *f, struct rtattr *arg)
|
|
Packit Service |
3880ab |
{
|
|
Packit Service |
3880ab |
struct rtattr *tb[TCA_SAMPLE_MAX + 1];
|
|
Packit Service |
3880ab |
struct tc_sample *p;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (arg == NULL)
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
parse_rtattr_nested(tb, TCA_SAMPLE_MAX, arg);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (!tb[TCA_SAMPLE_PARMS] || !tb[TCA_SAMPLE_RATE] ||
|
|
Packit Service |
3880ab |
!tb[TCA_SAMPLE_PSAMPLE_GROUP]) {
|
|
Packit Service |
3880ab |
fprintf(stderr, "Missing sample parameters\n");
|
|
Packit Service |
3880ab |
return -1;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
p = RTA_DATA(tb[TCA_SAMPLE_PARMS]);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_string(PRINT_ANY, "kind", "%s ", "sample");
|
|
Packit Service |
3880ab |
print_uint(PRINT_ANY, "rate", "rate 1/%u ",
|
|
Packit Service |
3880ab |
rta_getattr_u32(tb[TCA_SAMPLE_RATE]));
|
|
Packit Service |
3880ab |
print_uint(PRINT_ANY, "group", "group %u",
|
|
Packit Service |
3880ab |
rta_getattr_u32(tb[TCA_SAMPLE_PSAMPLE_GROUP]));
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (tb[TCA_SAMPLE_TRUNC_SIZE])
|
|
Packit Service |
3880ab |
print_uint(PRINT_ANY, "trunc_size", " trunc_size %u",
|
|
Packit Service |
3880ab |
rta_getattr_u32(tb[TCA_SAMPLE_TRUNC_SIZE]));
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_action_control(f, " ", p->action, "");
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_nl();
|
|
Packit Service |
3880ab |
print_uint(PRINT_ANY, "index", "\t index %u", p->index);
|
|
Packit Service |
3880ab |
print_int(PRINT_ANY, "ref", " ref %d", p->refcnt);
|
|
Packit Service |
3880ab |
print_int(PRINT_ANY, "bind", " bind %d", p->bindcnt);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
if (show_stats) {
|
|
Packit Service |
3880ab |
if (tb[TCA_SAMPLE_TM]) {
|
|
Packit Service |
3880ab |
struct tcf_t *tm = RTA_DATA(tb[TCA_SAMPLE_TM]);
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
print_tm(f, tm);
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
print_nl();
|
|
Packit Service |
3880ab |
return 0;
|
|
Packit Service |
3880ab |
}
|
|
Packit Service |
3880ab |
|
|
Packit Service |
3880ab |
struct action_util sample_action_util = {
|
|
Packit Service |
3880ab |
.id = "sample",
|
|
Packit Service |
3880ab |
.parse_aopt = parse_sample,
|
|
Packit Service |
3880ab |
.print_aopt = print_sample,
|
|
Packit Service |
3880ab |
};
|