1/*
2 * q_dsmark.c		Differentiated Services field marking.
3 *
4 * Hacked 1998,1999 by Werner Almesberger, EPFL ICA
5 *
6 */
7
8#include <stdio.h>
9#include <stdlib.h>
10#include <unistd.h>
11#include <syslog.h>
12#include <fcntl.h>
13#include <sys/socket.h>
14#include <netinet/in.h>
15#include <arpa/inet.h>
16#include <string.h>
17
18#include "utils.h"
19#include "tc_util.h"
20
21
22static void explain(void)
23{
24	fprintf(stderr,"Usage: dsmark indices INDICES [ default_index "
25	    "DEFAULT_INDEX ] [ set_tc_index ]\n");
26}
27
28
29static int dsmark_parse_opt(struct qdisc_util *qu, int argc, char **argv,
30    struct nlmsghdr *n)
31{
32	struct rtattr *tail;
33	__u16 ind;
34	char *end;
35	int dflt,set_tc_index;
36
37	ind = set_tc_index = 0;
38	dflt = -1;
39	while (argc > 0) {
40		if (!strcmp(*argv,"indices")) {
41			NEXT_ARG();
42			ind = strtoul(*argv,&end,0);
43			if (*end) {
44				explain();
45				return -1;
46			}
47		}
48		else if (!strcmp(*argv,"default_index") || !strcmp(*argv,
49		    "default")) {
50			NEXT_ARG();
51			dflt = strtoul(*argv,&end,0);
52			if (*end) {
53				explain();
54				return -1;
55			}
56		}
57		else if (!strcmp(*argv,"set_tc_index")) {
58			set_tc_index = 1;
59		}
60		else {
61			explain();
62			return -1;
63		}
64		argc--;
65		argv++;
66	}
67	if (!ind) {
68		explain();
69		return -1;
70	}
71	tail = NLMSG_TAIL(n);
72	addattr_l(n,1024,TCA_OPTIONS,NULL,0);
73	addattr_l(n,1024,TCA_DSMARK_INDICES,&ind,sizeof(ind));
74	if (dflt != -1) {
75	    __u16 tmp = dflt;
76
77	    addattr_l(n,1024,TCA_DSMARK_DEFAULT_INDEX,&tmp,sizeof(tmp));
78	}
79	if (set_tc_index) addattr_l(n,1024,TCA_DSMARK_SET_TC_INDEX,NULL,0);
80	tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
81	return 0;
82}
83
84
85static void explain_class(void)
86{
87	fprintf(stderr, "Usage: ... dsmark [ mask MASK ] [ value VALUE ]\n");
88}
89
90
91static int dsmark_parse_class_opt(struct qdisc_util *qu, int argc, char **argv,
92   struct nlmsghdr *n)
93{
94	struct rtattr *tail;
95	__u8 tmp;
96	char *end;
97
98	tail = NLMSG_TAIL(n);
99	addattr_l(n,1024,TCA_OPTIONS,NULL,0);
100	while (argc > 0) {
101		if (!strcmp(*argv,"mask")) {
102			NEXT_ARG();
103			tmp = strtoul(*argv,&end,0);
104			if (*end) {
105				explain_class();
106				return -1;
107			}
108			addattr_l(n,1024,TCA_DSMARK_MASK,&tmp,1);
109		}
110		else if (!strcmp(*argv,"value")) {
111			NEXT_ARG();
112			tmp = strtoul(*argv,&end,0);
113			if (*end) {
114				explain_class();
115				return -1;
116			}
117			addattr_l(n,1024,TCA_DSMARK_VALUE,&tmp,1);
118		}
119		else {
120			explain_class();
121			return -1;
122		}
123		argc--;
124		argv++;
125	}
126	tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
127	return 0;
128}
129
130
131
132static int dsmark_print_opt(struct qdisc_util *qu, FILE *f, struct rtattr *opt)
133{
134	struct rtattr *tb[TCA_DSMARK_MAX+1];
135
136	if (!opt) return 0;
137	memset(tb, 0, sizeof(tb));
138	parse_rtattr(tb, TCA_DSMARK_MAX, RTA_DATA(opt), RTA_PAYLOAD(opt));
139	if (tb[TCA_DSMARK_MASK]) {
140		if (!RTA_PAYLOAD(tb[TCA_DSMARK_MASK]))
141			fprintf(stderr,"dsmark: empty mask\n");
142		else fprintf(f,"mask 0x%02x ",
143			    rta_getattr_u8(tb[TCA_DSMARK_MASK]));
144	}
145	if (tb[TCA_DSMARK_VALUE]) {
146		if (!RTA_PAYLOAD(tb[TCA_DSMARK_VALUE]))
147			fprintf(stderr,"dsmark: empty value\n");
148		else fprintf(f,"value 0x%02x ",
149			    rta_getattr_u8(tb[TCA_DSMARK_VALUE]));
150	}
151	if (tb[TCA_DSMARK_INDICES]) {
152		if (RTA_PAYLOAD(tb[TCA_DSMARK_INDICES]) < sizeof(__u16))
153			fprintf(stderr,"dsmark: indices too short\n");
154		else fprintf(f,"indices 0x%04x ",
155			    rta_getattr_u16(tb[TCA_DSMARK_INDICES]));
156	}
157	if (tb[TCA_DSMARK_DEFAULT_INDEX]) {
158		if (RTA_PAYLOAD(tb[TCA_DSMARK_DEFAULT_INDEX]) < sizeof(__u16))
159			fprintf(stderr,"dsmark: default_index too short\n");
160		else fprintf(f,"default_index 0x%04x ",
161			    rta_getattr_u16(tb[TCA_DSMARK_DEFAULT_INDEX]));
162	}
163	if (tb[TCA_DSMARK_SET_TC_INDEX]) fprintf(f,"set_tc_index ");
164	return 0;
165}
166
167
168struct qdisc_util dsmark_qdisc_util = {
169	.id		= "dsmark",
170	.parse_qopt	= dsmark_parse_opt,
171	.print_qopt	= dsmark_print_opt,
172	.parse_copt	= dsmark_parse_class_opt,
173	.print_copt	= dsmark_print_opt,
174};
175