1 /* $NetBSD: qop_wfq.c,v 1.6 2008/09/11 17:58:59 joerg Exp $ */
2 /* $KAME: qop_wfq.c,v 1.6 2001/12/03 08:20:56 kjc Exp $ */
3 /*
4 * Copyright (C) 1999-2000
5 * Sony Computer Science Laboratories, Inc. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY SONY CSL AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL SONY CSL OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <sys/param.h>
30 #include <sys/socket.h>
31 #include <sys/sockio.h>
32 #include <sys/ioctl.h>
33 #include <sys/fcntl.h>
34 #include <net/if.h>
35 #include <netinet/in.h>
36 #include <arpa/inet.h>
37
38 #include <stdio.h>
39 #include <stdlib.h>
40 #include <unistd.h>
41 #include <stddef.h>
42 #include <string.h>
43 #include <ctype.h>
44 #include <errno.h>
45 #include <syslog.h>
46 #include <netdb.h>
47
48 #include <altq/altq.h>
49 #include <altq/altq_wfq.h>
50 #include "altq_qop.h"
51 #include "qop_wfq.h"
52
53 static int wfq_attach(struct ifinfo *);
54 static int wfq_detach(struct ifinfo *);
55 static int wfq_enable(struct ifinfo *);
56 static int wfq_disable(struct ifinfo *);
57
58 #define WFQ_DEVICE "/dev/altq/wfq"
59
60 static int wfq_fd = -1;
61 static int wfq_refcount = 0;
62
63 static struct qdisc_ops wfq_qdisc = {
64 ALTQT_WFQ,
65 "wfq",
66 wfq_attach,
67 wfq_detach,
68 NULL, /* clear */
69 wfq_enable,
70 wfq_disable,
71 NULL, /* add class */
72 NULL, /* modify class */
73 NULL, /* delete class */
74 NULL, /* add filter */
75 NULL /* delete filter */
76 };
77
78 /*
79 * parser interface
80 */
81 #define EQUAL(s1, s2) (strcmp((s1), (s2)) == 0)
82
83 int
wfq_interface_parser(const char * ifname,int argc,char ** argv)84 wfq_interface_parser(const char *ifname, int argc, char **argv)
85 {
86 u_int bandwidth = 100000000; /* 100Mbps */
87 u_int tbrsize = 0;
88 int hash_policy = 0; /* 0: use default */
89 int nqueues = 0; /* 0: use default */
90 int qsize = 0; /* 0: use default */
91
92 /*
93 * process options
94 */
95 while (argc > 0) {
96 if (EQUAL(*argv, "bandwidth")) {
97 argc--; argv++;
98 if (argc > 0)
99 bandwidth = atobps(*argv);
100 } else if (EQUAL(*argv, "tbrsize")) {
101 argc--; argv++;
102 if (argc > 0)
103 tbrsize = atobytes(*argv);
104 } else if (EQUAL(*argv, "nqueues")) {
105 argc--; argv++;
106 if (argc > 0)
107 nqueues = (int)strtol(*argv, NULL, 0);
108 } else if (EQUAL(*argv, "qsize")) {
109 argc--; argv++;
110 if (argc > 0)
111 qsize = atobytes(*argv);
112 } else if (EQUAL(*argv, "hash")) {
113 argc--; argv++;
114 if (argc > 0) {
115 if (EQUAL(*argv, "dstaddr"))
116 hash_policy = WFQ_HASH_DSTADDR;
117 else if (EQUAL(*argv, "full"))
118 hash_policy = WFQ_HASH_FULL;
119 else if (EQUAL(*argv, "srcport"))
120 hash_policy = WFQ_HASH_SRCPORT;
121 else if (EQUAL(*argv, "srcaddr"))
122 hash_policy = WFQ_HASH_SRCADDR;
123 else {
124 LOG(LOG_ERR, 0,
125 "Unknown hash policy '%s'", *argv);
126 return (0);
127 }
128 }
129 } else if (EQUAL(*argv, "wfq")) {
130 /* just skip */
131 } else {
132 LOG(LOG_ERR, 0, "Unknown keyword '%s'", *argv);
133 return (0);
134 }
135 argc--; argv++;
136 }
137
138 if (qcmd_tbr_register(ifname, bandwidth, tbrsize) != 0)
139 return (0);
140
141 if (qsize != 0 && qsize < 1500) {
142 LOG(LOG_ERR, 0, "qsize too small: %d bytes", qsize);
143 return (0);
144 }
145
146 if (qcmd_wfq_add_if(ifname, bandwidth,
147 hash_policy, nqueues, qsize) != 0)
148 return (0);
149 return (1);
150 }
151
152 /*
153 * qcmd api
154 */
155 int
qcmd_wfq_add_if(const char * ifname,u_int bandwidth,int hash_policy,int nqueues,int qsize)156 qcmd_wfq_add_if(const char *ifname, u_int bandwidth, int hash_policy,
157 int nqueues, int qsize)
158 {
159 int error;
160
161 error = qop_wfq_add_if(NULL, ifname, bandwidth,
162 hash_policy, nqueues, qsize);
163 if (error != 0)
164 LOG(LOG_ERR, errno, "%s: can't add wfq on interface '%s'",
165 qoperror(error), ifname);
166 return (error);
167 }
168
169 /*
170 * qop api
171 */
172 int
qop_wfq_add_if(struct ifinfo ** rp,const char * ifname,u_int bandwidth,int hash_policy,int nqueues,int qsize)173 qop_wfq_add_if(struct ifinfo **rp, const char *ifname, u_int bandwidth,
174 int hash_policy, int nqueues, int qsize)
175 {
176 struct ifinfo *ifinfo = NULL;
177 struct wfq_ifinfo *wfq_ifinfo;
178 int error;
179
180 if ((wfq_ifinfo = calloc(1, sizeof(*wfq_ifinfo))) == NULL)
181 return (QOPERR_NOMEM);
182 wfq_ifinfo->hash_policy = hash_policy;
183 wfq_ifinfo->nqueues = nqueues;
184 wfq_ifinfo->qsize = qsize;
185
186 error = qop_add_if(&ifinfo, ifname, bandwidth,
187 &wfq_qdisc, wfq_ifinfo);
188 if (error != 0) {
189 free(wfq_ifinfo);
190 return (error);
191 }
192
193 if (rp != NULL)
194 *rp = ifinfo;
195 return (0);
196 }
197
198 /*
199 * system call interfaces for qdisc_ops
200 */
201 static int
wfq_attach(struct ifinfo * ifinfo)202 wfq_attach(struct ifinfo *ifinfo)
203 {
204 struct wfq_interface iface;
205 struct wfq_ifinfo *wfq_ifinfo;
206 struct wfq_conf conf;
207
208 if (wfq_fd < 0 &&
209 (wfq_fd = open(WFQ_DEVICE, O_RDWR)) < 0 &&
210 (wfq_fd = open_module(WFQ_DEVICE, O_RDWR)) < 0) {
211 LOG(LOG_ERR, errno, "WFQ open");
212 return (QOPERR_SYSCALL);
213 }
214
215 wfq_refcount++;
216 memset(&iface, 0, sizeof(iface));
217 strncpy(iface.wfq_ifacename, ifinfo->ifname, IFNAMSIZ);
218
219 if (ioctl(wfq_fd, WFQ_IF_ATTACH, &iface) < 0)
220 return (QOPERR_SYSCALL);
221
222 /* set wfq parameters */
223 wfq_ifinfo = (struct wfq_ifinfo *)ifinfo->private;
224 if (wfq_ifinfo->hash_policy != 0 || wfq_ifinfo->nqueues != 0 ||
225 wfq_ifinfo->qsize != 0) {
226 memset(&conf, 0, sizeof(conf));
227 strncpy(conf.iface.wfq_ifacename, ifinfo->ifname, IFNAMSIZ);
228 conf.hash_policy = wfq_ifinfo->hash_policy;
229 conf.nqueues = wfq_ifinfo->nqueues;
230 conf.qlimit = wfq_ifinfo->qsize;
231 if (ioctl(wfq_fd, WFQ_CONFIG, &conf) < 0) {
232 LOG(LOG_ERR, errno, "WFQ_CONFIG");
233 return (QOPERR_SYSCALL);
234 }
235 }
236 #if 1
237 LOG(LOG_INFO, 0, "wfq attached to %s", iface.wfq_ifacename);
238 #endif
239 return (0);
240 }
241
242 static int
wfq_detach(struct ifinfo * ifinfo)243 wfq_detach(struct ifinfo *ifinfo)
244 {
245 struct wfq_interface iface;
246
247 memset(&iface, 0, sizeof(iface));
248 strncpy(iface.wfq_ifacename, ifinfo->ifname, IFNAMSIZ);
249
250 if (ioctl(wfq_fd, WFQ_IF_DETACH, &iface) < 0)
251 return (QOPERR_SYSCALL);
252
253 if (--wfq_refcount == 0) {
254 close(wfq_fd);
255 wfq_fd = -1;
256 }
257 return (0);
258 }
259
260 static int
wfq_enable(struct ifinfo * ifinfo)261 wfq_enable(struct ifinfo *ifinfo)
262 {
263 struct wfq_interface iface;
264
265 memset(&iface, 0, sizeof(iface));
266 strncpy(iface.wfq_ifacename, ifinfo->ifname, IFNAMSIZ);
267
268 if (ioctl(wfq_fd, WFQ_ENABLE, &iface) < 0)
269 return (QOPERR_SYSCALL);
270 return (0);
271 }
272
273 static int
wfq_disable(struct ifinfo * ifinfo)274 wfq_disable(struct ifinfo *ifinfo)
275 {
276 struct wfq_interface iface;
277
278 memset(&iface, 0, sizeof(iface));
279 strncpy(iface.wfq_ifacename, ifinfo->ifname, IFNAMSIZ);
280
281 if (ioctl(wfq_fd, WFQ_DISABLE, &iface) < 0)
282 return (QOPERR_SYSCALL);
283 return (0);
284 }
285