|
Packit |
d3f73b |
/*
|
|
Packit |
d3f73b |
* m_csum.c checksum updating action
|
|
Packit |
d3f73b |
*
|
|
Packit |
d3f73b |
* This program is free software; you can distribute it and/or
|
|
Packit |
d3f73b |
* modify it under the terms of the GNU General Public License
|
|
Packit |
d3f73b |
* as published by the Free Software Foundation; either version
|
|
Packit |
d3f73b |
* 2 of the License, or (at your option) any later version.
|
|
Packit |
d3f73b |
*
|
|
Packit |
d3f73b |
* Authors: Gregoire Baron <baronchon@n7mm.org>
|
|
Packit |
d3f73b |
*/
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
#include <stdio.h>
|
|
Packit |
d3f73b |
#include <stdlib.h>
|
|
Packit |
d3f73b |
#include <string.h>
|
|
Packit |
d3f73b |
#include <unistd.h>
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
#include <linux/tc_act/tc_csum.h>
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
#include "utils.h"
|
|
Packit |
d3f73b |
#include "tc_util.h"
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
static void
|
|
Packit |
d3f73b |
explain(void)
|
|
Packit |
d3f73b |
{
|
|
Packit |
d3f73b |
fprintf(stderr, "Usage: ... csum <UPDATE>\n"
|
|
Packit |
d3f73b |
"Where: UPDATE := <TARGET> [<UPDATE>]\n"
|
|
Packit |
d3f73b |
" TARGET := { ip4h | icmp | igmp | tcp | udp | udplite | sctp | <SWEETS> }\n"
|
|
Packit |
d3f73b |
" SWEETS := { and | or | \'+\' }\n");
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
static void
|
|
Packit |
d3f73b |
usage(void)
|
|
Packit |
d3f73b |
{
|
|
Packit |
d3f73b |
explain();
|
|
Packit |
d3f73b |
exit(-1);
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
static int
|
|
Packit |
d3f73b |
parse_csum_args(int *argc_p, char ***argv_p, struct tc_csum *sel)
|
|
Packit |
d3f73b |
{
|
|
Packit |
d3f73b |
int argc = *argc_p;
|
|
Packit |
d3f73b |
char **argv = *argv_p;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (argc <= 0)
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
while (argc > 0) {
|
|
Packit |
d3f73b |
if ((matches(*argv, "iph") == 0) ||
|
|
Packit |
d3f73b |
(matches(*argv, "ip4h") == 0) ||
|
|
Packit |
d3f73b |
(matches(*argv, "ipv4h") == 0))
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_IPV4HDR;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if (matches(*argv, "icmp") == 0)
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_ICMP;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if (matches(*argv, "igmp") == 0)
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_IGMP;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if (matches(*argv, "tcp") == 0)
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_TCP;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if (matches(*argv, "udp") == 0)
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_UDP;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if (matches(*argv, "udplite") == 0)
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_UDPLITE;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if (matches(*argv, "sctp") == 0)
|
|
Packit |
d3f73b |
sel->update_flags |= TCA_CSUM_UPDATE_FLAG_SCTP;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
else if ((matches(*argv, "and") == 0) ||
|
|
Packit |
d3f73b |
(matches(*argv, "or") == 0) ||
|
|
Packit |
d3f73b |
(matches(*argv, "+") == 0))
|
|
Packit |
d3f73b |
; /* just ignore: ... csum iph and tcp or udp */
|
|
Packit |
d3f73b |
else
|
|
Packit |
d3f73b |
break;
|
|
Packit |
d3f73b |
argc--;
|
|
Packit |
d3f73b |
argv++;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
*argc_p = argc;
|
|
Packit |
d3f73b |
*argv_p = argv;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
return 0;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
static int
|
|
Packit |
d3f73b |
parse_csum(struct action_util *a, int *argc_p,
|
|
Packit |
d3f73b |
char ***argv_p, int tca_id, struct nlmsghdr *n)
|
|
Packit |
d3f73b |
{
|
|
Packit |
d3f73b |
struct tc_csum sel = {};
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
int argc = *argc_p;
|
|
Packit |
d3f73b |
char **argv = *argv_p;
|
|
Packit |
d3f73b |
int ok = 0;
|
|
Packit |
d3f73b |
struct rtattr *tail;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
while (argc > 0) {
|
|
Packit |
d3f73b |
if (matches(*argv, "csum") == 0) {
|
|
Packit |
d3f73b |
NEXT_ARG();
|
|
Packit |
d3f73b |
if (parse_csum_args(&argc, &argv, &sel)) {
|
|
Packit |
d3f73b |
fprintf(stderr, "Illegal csum construct (%s)\n",
|
|
Packit |
d3f73b |
*argv);
|
|
Packit |
d3f73b |
explain();
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
ok++;
|
|
Packit |
d3f73b |
continue;
|
|
Packit |
d3f73b |
} else if (matches(*argv, "help") == 0) {
|
|
Packit |
d3f73b |
usage();
|
|
Packit |
d3f73b |
} else {
|
|
Packit |
d3f73b |
break;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (!ok) {
|
|
Packit |
d3f73b |
explain();
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (sel.update_flags == 0) {
|
|
Packit |
d3f73b |
fprintf(stderr, "Illegal csum construct, empty <UPDATE> list\n");
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
parse_action_control_dflt(&argc, &argv, &sel.action, false, TC_ACT_OK);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (argc) {
|
|
Packit |
d3f73b |
if (matches(*argv, "index") == 0) {
|
|
Packit |
d3f73b |
NEXT_ARG();
|
|
Packit |
d3f73b |
if (get_u32(&sel.index, *argv, 10)) {
|
|
Packit |
d3f73b |
fprintf(stderr, "Illegal \"index\" (%s) <csum>\n",
|
|
Packit |
d3f73b |
*argv);
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
argc--;
|
|
Packit |
d3f73b |
argv++;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
tail = addattr_nest(n, MAX_MSG, tca_id);
|
|
Packit |
d3f73b |
addattr_l(n, MAX_MSG, TCA_CSUM_PARMS, &sel, sizeof(sel));
|
|
Packit |
d3f73b |
addattr_nest_end(n, tail);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
*argc_p = argc;
|
|
Packit |
d3f73b |
*argv_p = argv;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
return 0;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
static int
|
|
Packit |
d3f73b |
print_csum(struct action_util *au, FILE *f, struct rtattr *arg)
|
|
Packit |
d3f73b |
{
|
|
Packit |
d3f73b |
struct tc_csum *sel;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
struct rtattr *tb[TCA_CSUM_MAX + 1];
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
char *uflag_1 = "";
|
|
Packit |
d3f73b |
char *uflag_2 = "";
|
|
Packit |
d3f73b |
char *uflag_3 = "";
|
|
Packit |
d3f73b |
char *uflag_4 = "";
|
|
Packit |
d3f73b |
char *uflag_5 = "";
|
|
Packit |
d3f73b |
char *uflag_6 = "";
|
|
Packit |
d3f73b |
char *uflag_7 = "";
|
|
Packit |
d3f73b |
SPRINT_BUF(buf);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
int uflag_count = 0;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (arg == NULL)
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
parse_rtattr_nested(tb, TCA_CSUM_MAX, arg);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (tb[TCA_CSUM_PARMS] == NULL) {
|
|
Packit |
d3f73b |
fprintf(stderr, "Missing csum parameters\n");
|
|
Packit |
d3f73b |
return -1;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
sel = RTA_DATA(tb[TCA_CSUM_PARMS]);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (sel->update_flags & TCA_CSUM_UPDATE_FLAG_IPV4HDR) {
|
|
Packit |
d3f73b |
uflag_1 = "iph";
|
|
Packit |
d3f73b |
uflag_count++;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
#define CSUM_UFLAG_BUFFER(flag_buffer, flag_value, flag_string) \
|
|
Packit |
d3f73b |
do { \
|
|
Packit |
d3f73b |
if (sel->update_flags & flag_value) { \
|
|
Packit |
d3f73b |
flag_buffer = uflag_count > 0 ? \
|
|
Packit |
d3f73b |
", " flag_string : flag_string; \
|
|
Packit |
d3f73b |
uflag_count++; \
|
|
Packit |
d3f73b |
} \
|
|
Packit |
d3f73b |
} while (0)
|
|
Packit |
d3f73b |
CSUM_UFLAG_BUFFER(uflag_2, TCA_CSUM_UPDATE_FLAG_ICMP, "icmp");
|
|
Packit |
d3f73b |
CSUM_UFLAG_BUFFER(uflag_3, TCA_CSUM_UPDATE_FLAG_IGMP, "igmp");
|
|
Packit |
d3f73b |
CSUM_UFLAG_BUFFER(uflag_4, TCA_CSUM_UPDATE_FLAG_TCP, "tcp");
|
|
Packit |
d3f73b |
CSUM_UFLAG_BUFFER(uflag_5, TCA_CSUM_UPDATE_FLAG_UDP, "udp");
|
|
Packit |
d3f73b |
CSUM_UFLAG_BUFFER(uflag_6, TCA_CSUM_UPDATE_FLAG_UDPLITE, "udplite");
|
|
Packit |
d3f73b |
CSUM_UFLAG_BUFFER(uflag_7, TCA_CSUM_UPDATE_FLAG_SCTP, "sctp");
|
|
Packit |
d3f73b |
if (!uflag_count) {
|
|
Packit |
d3f73b |
uflag_1 = "?empty";
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
print_string(PRINT_ANY, "kind", "%s ", "csum");
|
|
Packit |
d3f73b |
snprintf(buf, sizeof(buf), "%s%s%s%s%s%s%s",
|
|
Packit |
d3f73b |
uflag_1, uflag_2, uflag_3,
|
|
Packit |
d3f73b |
uflag_4, uflag_5, uflag_6, uflag_7);
|
|
Packit |
d3f73b |
print_string(PRINT_ANY, "csum", "(%s) ", buf);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
print_action_control(f, "action ", sel->action, "\n");
|
|
Packit |
d3f73b |
print_uint(PRINT_ANY, "index", "\tindex %u", sel->index);
|
|
Packit |
d3f73b |
print_int(PRINT_ANY, "ref", " ref %d", sel->refcnt);
|
|
Packit |
d3f73b |
print_int(PRINT_ANY, "bind", " bind %d", sel->bindcnt);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
if (show_stats) {
|
|
Packit |
d3f73b |
if (tb[TCA_CSUM_TM]) {
|
|
Packit |
d3f73b |
struct tcf_t *tm = RTA_DATA(tb[TCA_CSUM_TM]);
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
print_tm(f, tm);
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
print_string(PRINT_FP, NULL, "%s", "\n");
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
return 0;
|
|
Packit |
d3f73b |
}
|
|
Packit |
d3f73b |
|
|
Packit |
d3f73b |
struct action_util csum_action_util = {
|
|
Packit |
d3f73b |
.id = "csum",
|
|
Packit |
d3f73b |
.parse_aopt = parse_csum,
|
|
Packit |
d3f73b |
.print_aopt = print_csum,
|
|
Packit |
d3f73b |
};
|