Home | History | Annotate | Line # | Download | only in altq
altq_conf.c revision 1.17
      1 /*	$NetBSD: altq_conf.c,v 1.17 2006/10/12 19:59:08 peter Exp $	*/
      2 /*	$KAME: altq_conf.c,v 1.24 2005/04/13 03:44:24 suz Exp $	*/
      3 
      4 /*
      5  * Copyright (C) 1997-2003
      6  *	Sony Computer Science Laboratories Inc.  All rights reserved.
      7  *
      8  * Redistribution and use in source and binary forms, with or without
      9  * modification, are permitted provided that the following conditions
     10  * are met:
     11  * 1. Redistributions of source code must retain the above copyright
     12  *    notice, this list of conditions and the following disclaimer.
     13  * 2. Redistributions in binary form must reproduce the above copyright
     14  *    notice, this list of conditions and the following disclaimer in the
     15  *    documentation and/or other materials provided with the distribution.
     16  *
     17  * THIS SOFTWARE IS PROVIDED BY SONY CSL AND CONTRIBUTORS ``AS IS'' AND
     18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     20  * ARE DISCLAIMED.  IN NO EVENT SHALL SONY CSL OR CONTRIBUTORS BE LIABLE
     21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     27  * SUCH DAMAGE.
     28  */
     29 
     30 #include <sys/cdefs.h>
     31 __KERNEL_RCSID(0, "$NetBSD: altq_conf.c,v 1.17 2006/10/12 19:59:08 peter Exp $");
     32 
     33 #ifdef _KERNEL_OPT
     34 #include "opt_altq.h"
     35 #include "opt_inet.h"
     36 #endif
     37 
     38 /*
     39  * altq device interface.
     40  */
     41 #include <sys/param.h>
     42 #include <sys/systm.h>
     43 #include <sys/socket.h>
     44 #include <sys/kernel.h>
     45 #include <sys/proc.h>
     46 #include <sys/errno.h>
     47 #include <sys/kauth.h>
     48 
     49 #include <net/if.h>
     50 
     51 #include <altq/altq.h>
     52 #include <altq/altqconf.h>
     53 #include <altq/altq_conf.h>
     54 
     55 #ifdef ALTQ3_COMPAT
     56 
     57 #ifdef ALTQ_CBQ
     58 altqdev_decl(cbq);
     59 #endif
     60 #ifdef ALTQ_WFQ
     61 altqdev_decl(wfq);
     62 #endif
     63 #ifdef ALTQ_AFMAP
     64 altqdev_decl(afm);
     65 #endif
     66 #ifdef ALTQ_FIFOQ
     67 altqdev_decl(fifoq);
     68 #endif
     69 #ifdef ALTQ_RED
     70 altqdev_decl(red);
     71 #endif
     72 #ifdef ALTQ_RIO
     73 altqdev_decl(rio);
     74 #endif
     75 #ifdef ALTQ_LOCALQ
     76 altqdev_decl(localq);
     77 #endif
     78 #ifdef ALTQ_HFSC
     79 altqdev_decl(hfsc);
     80 #endif
     81 #ifdef ALTQ_CDNR
     82 altqdev_decl(cdnr);
     83 #endif
     84 #ifdef ALTQ_BLUE
     85 altqdev_decl(blue);
     86 #endif
     87 #ifdef ALTQ_PRIQ
     88 altqdev_decl(priq);
     89 #endif
     90 #ifdef ALTQ_JOBS
     91 altqdev_decl(jobs);
     92 #endif
     93 
     94 /*
     95  * altq minor device (discipline) table
     96  */
     97 static struct altqsw altqsw[] = {				/* minor */
     98 	{"altq", noopen,	noclose,	noioctl},  /* 0 (reserved) */
     99 #ifdef ALTQ_CBQ
    100 	{"cbq",	cbqopen,	cbqclose,	cbqioctl},	/* 1 */
    101 #else
    102 	{"noq",	noopen,		noclose,	noioctl},	/* 1 */
    103 #endif
    104 #ifdef ALTQ_WFQ
    105 	{"wfq",	wfqopen,	wfqclose,	wfqioctl},	/* 2 */
    106 #else
    107 	{"noq",	noopen,		noclose,	noioctl},	/* 2 */
    108 #endif
    109 #ifdef ALTQ_AFMAP
    110 	{"afm",	afmopen,	afmclose,	afmioctl},	/* 3 */
    111 #else
    112 	{"noq",	noopen,		noclose,	noioctl},	/* 3 */
    113 #endif
    114 #ifdef ALTQ_FIFOQ
    115 	{"fifoq", fifoqopen,	fifoqclose,	fifoqioctl},	/* 4 */
    116 #else
    117 	{"noq",	noopen,		noclose,	noioctl},	/* 4 */
    118 #endif
    119 #ifdef ALTQ_RED
    120 	{"red", redopen,	redclose,	redioctl},	/* 5 */
    121 #else
    122 	{"noq",	noopen,		noclose,	noioctl},	/* 5 */
    123 #endif
    124 #ifdef ALTQ_RIO
    125 	{"rio", rioopen,	rioclose,	rioioctl},	/* 6 */
    126 #else
    127 	{"noq",	noopen,		noclose,	noioctl},	/* 6 */
    128 #endif
    129 #ifdef ALTQ_LOCALQ
    130 	{"localq",localqopen,	localqclose,	localqioctl}, /* 7 (local use) */
    131 #else
    132 	{"noq",	noopen,		noclose,	noioctl},  /* 7 (local use) */
    133 #endif
    134 #ifdef ALTQ_HFSC
    135 	{"hfsc",hfscopen,	hfscclose,	hfscioctl},	/* 8 */
    136 #else
    137 	{"noq",	noopen,		noclose,	noioctl},	/* 8 */
    138 #endif
    139 #ifdef ALTQ_CDNR
    140 	{"cdnr",cdnropen,	cdnrclose,	cdnrioctl},	/* 9 */
    141 #else
    142 	{"noq",	noopen,		noclose,	noioctl},	/* 9 */
    143 #endif
    144 #ifdef ALTQ_BLUE
    145 	{"blue",blueopen,	blueclose,	blueioctl},	/* 10 */
    146 #else
    147 	{"noq",	noopen,		noclose,	noioctl},	/* 10 */
    148 #endif
    149 #ifdef ALTQ_PRIQ
    150 	{"priq",priqopen,	priqclose,	priqioctl},	/* 11 */
    151 #else
    152 	{"noq",	noopen,		noclose,	noioctl},	/* 11 */
    153 #endif
    154 #ifdef ALTQ_JOBS
    155 	{"jobs",jobsopen,	jobsclose,	jobsioctl},	/* 12 */
    156 #else
    157 	{"noq", noopen,		noclose,	noioctl},	/* 12 */
    158 #endif
    159 };
    160 
    161 /*
    162  * altq major device support
    163  */
    164 int	naltqsw = sizeof (altqsw) / sizeof (altqsw[0]);
    165 
    166 dev_type_open(altqopen);
    167 dev_type_close(altqclose);
    168 dev_type_ioctl(altqioctl);
    169 
    170 const struct cdevsw altq_cdevsw = {
    171 	altqopen, altqclose, noread, nowrite, altqioctl,
    172 	nostop, notty, nopoll, nommap, nokqfilter, D_OTHER,
    173 };
    174 
    175 int
    176 altqopen(dev_t dev, int flag, int fmt, struct lwp *l)
    177 {
    178 	int unit = minor(dev);
    179 
    180 	if (unit == 0)
    181 		return (0);
    182 	if (unit < naltqsw)
    183 		return (*altqsw[unit].d_open)(dev, flag, fmt, l);
    184 
    185 	return ENXIO;
    186 }
    187 
    188 int
    189 altqclose(dev_t dev, int flag, int fmt, struct lwp *l)
    190 {
    191 	int unit = minor(dev);
    192 
    193 	if (unit == 0)
    194 		return (0);
    195 	if (unit < naltqsw)
    196 		return (*altqsw[unit].d_close)(dev, flag, fmt, l);
    197 
    198 	return ENXIO;
    199 }
    200 
    201 int
    202 altqioctl(dev_t dev, ioctlcmd_t cmd, caddr_t addr, int flag, struct lwp *l)
    203 {
    204 	int unit = minor(dev);
    205 
    206 	if (unit == 0) {
    207 		struct ifnet *ifp;
    208 		struct altqreq *typereq;
    209 		struct tbrreq *tbrreq;
    210 		int error;
    211 
    212 		switch (cmd) {
    213 		case ALTQGTYPE:
    214 		case ALTQTBRGET:
    215 			break;
    216 		default:
    217 #if (__FreeBSD_version > 400000)
    218 			if ((error = suser(p)) != 0)
    219 				return (error);
    220 #else
    221 			if ((error = kauth_authorize_generic(l->l_cred,
    222 			    KAUTH_GENERIC_ISSUSER, &l->l_acflag)) != 0)
    223 				return (error);
    224 #endif
    225 			break;
    226 		}
    227 
    228 		switch (cmd) {
    229 		case ALTQGTYPE:
    230 			typereq = (struct altqreq *)addr;
    231 			if ((ifp = ifunit(typereq->ifname)) == NULL)
    232 				return (EINVAL);
    233 			typereq->arg = (u_long)ifp->if_snd.altq_type;
    234 			return (0);
    235 		case ALTQTBRSET:
    236 			tbrreq = (struct tbrreq *)addr;
    237 			if ((ifp = ifunit(tbrreq->ifname)) == NULL)
    238 				return (EINVAL);
    239 			return tbr_set(&ifp->if_snd, &tbrreq->tb_prof);
    240 		case ALTQTBRGET:
    241 			tbrreq = (struct tbrreq *)addr;
    242 			if ((ifp = ifunit(tbrreq->ifname)) == NULL)
    243 				return (EINVAL);
    244 			return tbr_get(&ifp->if_snd, &tbrreq->tb_prof);
    245 		default:
    246 			return (EINVAL);
    247 		}
    248 	}
    249 	if (unit < naltqsw)
    250 		return (*altqsw[unit].d_ioctl)(dev, cmd, addr, flag, l);
    251 
    252 	return ENXIO;
    253 }
    254 
    255 #ifdef __FreeBSD__
    256 static int altq_devsw_installed = 0;
    257 #endif
    258 
    259 #ifdef __FreeBSD__
    260 static void
    261 altq_drvinit(void *unused)
    262 {
    263 	int unit;
    264 
    265 #if 0
    266 	mtx_init(&altq_mtx, "altq global lock", MTX_DEF);
    267 #endif
    268 	altq_devsw_installed = 1;
    269 	printf("altq: attached. Major number assigned automatically.\n");
    270 
    271 	/* create minor devices */
    272 	for (unit = 0; unit < naltqsw; unit++) {
    273 		if (unit == 0 || altqsw[unit].d_open != NULL)
    274 			altqsw[unit].dev = make_dev(&altq_cdevsw, unit,
    275 			    UID_ROOT, GID_WHEEL, 0644, "altq/%s",
    276 			    altqsw[unit].d_name);
    277 	}
    278 }
    279 
    280 SYSINIT(altqdev,SI_SUB_DRIVERS,SI_ORDER_MIDDLE+CDEV_MAJOR,altq_drvinit,NULL)
    281 
    282 #endif /* FreeBSD */
    283 
    284 #ifdef ALTQ_KLD
    285 /*
    286  * KLD support
    287  */
    288 static int altq_module_register(struct altq_module_data *);
    289 static int altq_module_deregister(struct altq_module_data *);
    290 
    291 static struct altq_module_data *altq_modules[ALTQT_MAX];
    292 #if __FreeBSD_version < 502103
    293 static struct altqsw noqdisc = {"noq", noopen, noclose, noioctl};
    294 #else
    295 static struct altqsw noqdisc = {"noq"};
    296 #endif
    297 
    298 void altq_module_incref(int type)
    299 {
    300 	if (type < 0 || type >= ALTQT_MAX || altq_modules[type] == NULL)
    301 		return;
    302 
    303 	altq_modules[type]->ref++;
    304 }
    305 
    306 void altq_module_declref(int type)
    307 {
    308 	if (type < 0 || type >= ALTQT_MAX || altq_modules[type] == NULL)
    309 		return;
    310 
    311 	altq_modules[type]->ref--;
    312 }
    313 
    314 static int
    315 altq_module_register(struct altq_module_data *mdata)
    316 {
    317 	int type = mdata->type;
    318 
    319 	if (type < 0 || type >= ALTQT_MAX)
    320 		return (EINVAL);
    321 #if (__FreeBSD_version < 502103)
    322 	if (altqsw[type].d_open != noopen)
    323 #else
    324 	if (altqsw[type].d_open != NULL)
    325 #endif
    326 		return (EBUSY);
    327 	altqsw[type] = *mdata->altqsw;	/* set discipline functions */
    328 	altq_modules[type] = mdata;	/* save module data pointer */
    329 #if (__FreeBSD_version < 502103)
    330 	make_dev(&altq_cdevsw, type, UID_ROOT, GID_WHEEL, 0644,
    331 		 "altq/%s", altqsw[type].d_name);
    332 #else
    333 	altqsw[type].dev = make_dev(&altq_cdevsw, type, UID_ROOT, GID_WHEEL,
    334 	    0644, "altq/%s", altqsw[type].d_name);
    335 #endif
    336 	return (0);
    337 }
    338 
    339 static int
    340 altq_module_deregister(struct altq_module_data *mdata)
    341 {
    342 	int type = mdata->type;
    343 
    344 	if (type < 0 || type >= ALTQT_MAX)
    345 		return (EINVAL);
    346 	if (mdata != altq_modules[type])
    347 		return (EINVAL);
    348 	if (altq_modules[type]->ref > 0)
    349 		return (EBUSY);
    350 #if (__FreeBSD_version < 502103)
    351 	destroy_dev(makedev(CDEV_MAJOR, type));
    352 #else
    353 	destroy_dev(altqsw[type].dev);
    354 #endif
    355 	altqsw[type] = noqdisc;
    356 	altq_modules[type] = NULL;
    357 	return (0);
    358 }
    359 
    360 int
    361 altq_module_handler(module_t mod, int cmd, void *arg)
    362 {
    363 	struct altq_module_data *data = (struct altq_module_data *)arg;
    364 	int	error = 0;
    365 
    366 	switch (cmd) {
    367 	case MOD_LOAD:
    368 		error = altq_module_register(data);
    369 		break;
    370 
    371 	case MOD_UNLOAD:
    372 		error = altq_module_deregister(data);
    373 		break;
    374 
    375 	default:
    376 		error = EINVAL;
    377 		break;
    378 	}
    379 
    380 	return (error);
    381 }
    382 
    383 #endif  /* ALTQ_KLD */
    384 #endif /* ALTQ3_COMPAT */
    385