1 /*
2 * mod_ip_tos.c
3 *
4 * Copyright (c) 2001 Dug Song <dugsong@monkey.org>
5 *
6 * $Id: mod_ip_tos.c,v 1.3 2002/04/07 22:55:20 dugsong Exp $
7 */
8
9 #include "config.h"
10
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14
15 #include "argv.h"
16 #include "mod.h"
17 #include "pkt.h"
18
19 struct ip_tos_data {
20 int tos;
21 };
22
23 void *
ip_tos_close(void * d)24 ip_tos_close(void *d)
25 {
26 if (d != NULL)
27 free(d);
28 return (NULL);
29 }
30
31 void *
ip_tos_open(int argc,char * argv[])32 ip_tos_open(int argc, char *argv[])
33 {
34 struct ip_tos_data *data;
35
36 if (argc != 2)
37 return (NULL);
38
39 if ((data = calloc(1, sizeof(*data))) == NULL)
40 return (NULL);
41
42 if (sscanf(argv[1], "%i", &data->tos) != 1 ||
43 data->tos < 0 || data->tos > 255)
44 return (ip_tos_close(data));
45
46 return (data);
47 }
48
49 int
ip_tos_apply(void * d,struct pktq * pktq)50 ip_tos_apply(void *d, struct pktq *pktq)
51 {
52 struct ip_tos_data *data = (struct ip_tos_data *)d;
53 struct pkt *pkt;
54
55 TAILQ_FOREACH(pkt, pktq, pkt_next) {
56 pkt->pkt_ip->ip_tos = data->tos;
57 /* XXX - do incremental checksum */
58 ip_checksum(pkt->pkt_ip, pkt->pkt_ip_data - pkt->pkt_eth_data);
59 }
60 return (0);
61 }
62
63 struct mod mod_ip_tos = {
64 "ip_tos", /* name */
65 "ip_tos <tos>", /* usage */
66 ip_tos_open, /* open */
67 ip_tos_apply, /* apply */
68 ip_tos_close /* close */
69 };
70