353 lines
9.2 KiB
C
353 lines
9.2 KiB
C
/* $NetBSD: qop_rio.c,v 1.5 2002/03/05 04:11:53 itojun Exp $ */
|
|
/* $KAME: qop_rio.c,v 1.6 2001/12/03 08:20:56 kjc Exp $ */
|
|
/*
|
|
* Copyright (C) 1999-2000
|
|
* Sony Computer Science Laboratories, Inc. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY SONY CSL AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL SONY CSL OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <sys/param.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/sockio.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/fcntl.h>
|
|
#include <net/if.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <stddef.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include <syslog.h>
|
|
#include <netdb.h>
|
|
|
|
#include <altq/altq.h>
|
|
#include <altq/altq_red.h>
|
|
#include <altq/altq_rio.h>
|
|
#include "altq_qop.h"
|
|
#include "qop_rio.h"
|
|
|
|
static int rio_attach(struct ifinfo *);
|
|
static int rio_detach(struct ifinfo *);
|
|
static int rio_enable(struct ifinfo *);
|
|
static int rio_disable(struct ifinfo *);
|
|
|
|
#define RIO_DEVICE "/dev/altq/rio"
|
|
|
|
static int rio_fd = -1;
|
|
static int rio_refcount = 0;
|
|
|
|
static struct qdisc_ops rio_qdisc = {
|
|
ALTQT_RIO,
|
|
"rio",
|
|
rio_attach,
|
|
rio_detach,
|
|
NULL, /* clear */
|
|
rio_enable,
|
|
rio_disable,
|
|
NULL, /* add class */
|
|
NULL, /* modify class */
|
|
NULL, /* delete class */
|
|
NULL, /* add filter */
|
|
NULL /* delete filter */
|
|
};
|
|
|
|
/*
|
|
* parser interface
|
|
*/
|
|
#define EQUAL(s1, s2) (strcmp((s1), (s2)) == 0)
|
|
|
|
int
|
|
rio_interface_parser(const char *ifname, int argc, char **argv)
|
|
{
|
|
u_int bandwidth = 100000000; /* 100Mbps */
|
|
u_int tbrsize = 0;
|
|
int weight = 0; /* 0: use default */
|
|
int lo_inv_pmax = 0; /* 0: use default */
|
|
int lo_th_min = 0; /* 0: use default */
|
|
int lo_th_max = 0; /* 0: use default */
|
|
int med_inv_pmax = 0; /* 0: use default */
|
|
int med_th_min = 0; /* 0: use default */
|
|
int med_th_max = 0; /* 0: use default */
|
|
int hi_inv_pmax = 0; /* 0: use default */
|
|
int hi_th_min = 0; /* 0: use default */
|
|
int hi_th_max = 0; /* 0: use default */
|
|
int qlimit = 60;
|
|
int pkttime = 0;
|
|
int flags = 0;
|
|
int packet_size = 1000;
|
|
|
|
/*
|
|
* process options
|
|
*/
|
|
while (argc > 0) {
|
|
if (EQUAL(*argv, "bandwidth")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
bandwidth = atobps(*argv);
|
|
} else if (EQUAL(*argv, "tbrsize")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
tbrsize = atobytes(*argv);
|
|
} else if (EQUAL(*argv, "packetsize")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
packet_size = atobytes(*argv);
|
|
} else if (EQUAL(*argv, "weight")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
weight = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "qlimit")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
qlimit = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "lo_thmin")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
lo_th_min = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "lo_thmax")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
lo_th_max = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "lo_invpmax")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
lo_inv_pmax = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "med_thmin")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
med_th_min = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "med_thmax")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
med_th_max = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "med_invpmax")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
med_inv_pmax = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "hi_thmin")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
hi_th_min = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "hi_thmax")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
hi_th_max = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "hi_invpmax")) {
|
|
argc--; argv++;
|
|
if (argc > 0)
|
|
hi_inv_pmax = (int)strtol(*argv, NULL, 0);
|
|
} else if (EQUAL(*argv, "rio")) {
|
|
/* just skip */
|
|
} else if (EQUAL(*argv, "ecn")) {
|
|
flags |= RIOF_ECN;
|
|
} else {
|
|
LOG(LOG_ERR, 0, "Unknown keyword '%s'", *argv);
|
|
return (0);
|
|
}
|
|
argc--; argv++;
|
|
}
|
|
|
|
if (qcmd_tbr_register(ifname, bandwidth, tbrsize) != 0)
|
|
return (0);
|
|
|
|
pkttime = packet_size * 8 * 1000 / (bandwidth / 1000);
|
|
if (weight != 0) {
|
|
/* check if weight is power of 2 */
|
|
int i, w;
|
|
|
|
w = weight;
|
|
for (i = 0; w > 1; i++)
|
|
w = w >> 1;
|
|
w = 1 << i;
|
|
if (weight != w) {
|
|
LOG(LOG_ERR, 0, "weight %d: should be power of 2",
|
|
weight);
|
|
return (0);
|
|
}
|
|
}
|
|
|
|
if (qcmd_rio_add_if(ifname, bandwidth, weight,
|
|
lo_inv_pmax, lo_th_min, lo_th_max,
|
|
med_inv_pmax, med_th_min, med_th_max,
|
|
hi_inv_pmax, hi_th_min, hi_th_max,
|
|
qlimit, pkttime, flags) != 0)
|
|
return (0);
|
|
return (1);
|
|
}
|
|
|
|
/*
|
|
* qcmd api
|
|
*/
|
|
int
|
|
qcmd_rio_add_if(const char *ifname, u_int bandwidth, int weight,
|
|
int lo_inv_pmax, int lo_th_min, int lo_th_max,
|
|
int med_inv_pmax, int med_th_min, int med_th_max,
|
|
int hi_inv_pmax, int hi_th_min, int hi_th_max,
|
|
int qlimit, int pkttime, int flags)
|
|
{
|
|
struct redparams red_params[RIO_NDROPPREC];
|
|
int error;
|
|
|
|
red_params[0].inv_pmax = lo_inv_pmax;
|
|
red_params[0].th_min = lo_th_min;
|
|
red_params[0].th_max = lo_th_max;
|
|
red_params[1].inv_pmax = med_inv_pmax;
|
|
red_params[1].th_min = med_th_min;
|
|
red_params[1].th_max = med_th_max;
|
|
red_params[2].inv_pmax = hi_inv_pmax;
|
|
red_params[2].th_min = hi_th_min;
|
|
red_params[2].th_max = hi_th_max;
|
|
|
|
error = qop_rio_add_if(NULL, ifname, bandwidth, weight, red_params,
|
|
qlimit, pkttime, flags);
|
|
if (error != 0)
|
|
LOG(LOG_ERR, errno, "%s: can't add rio on interface '%s'",
|
|
qoperror(error), ifname);
|
|
return (error);
|
|
}
|
|
|
|
/*
|
|
* qop api
|
|
*/
|
|
int
|
|
qop_rio_add_if(struct ifinfo **rp, const char *ifname,
|
|
u_int bandwidth, int weight, struct redparams *red_params,
|
|
int qlimit, int pkttime, int flags)
|
|
{
|
|
struct ifinfo *ifinfo = NULL;
|
|
struct rio_ifinfo *rio_ifinfo;
|
|
int i, error;
|
|
|
|
if ((rio_ifinfo = calloc(1, sizeof(*rio_ifinfo))) == NULL)
|
|
return (QOPERR_NOMEM);
|
|
for (i = 0; i < RIO_NDROPPREC; i++)
|
|
rio_ifinfo->red_params[i] = red_params[i];
|
|
rio_ifinfo->weight = weight;
|
|
rio_ifinfo->qlimit = qlimit;
|
|
rio_ifinfo->pkttime = pkttime;
|
|
rio_ifinfo->flags = flags;
|
|
|
|
error = qop_add_if(&ifinfo, ifname, bandwidth,
|
|
&rio_qdisc, rio_ifinfo);
|
|
if (error != 0) {
|
|
free(rio_ifinfo);
|
|
return (error);
|
|
}
|
|
|
|
if (rp != NULL)
|
|
*rp = ifinfo;
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* system call interfaces for qdisc_ops
|
|
*/
|
|
static int
|
|
rio_attach(struct ifinfo *ifinfo)
|
|
{
|
|
struct rio_interface iface;
|
|
struct rio_ifinfo *rio_ifinfo;
|
|
struct rio_conf conf;
|
|
int i;
|
|
|
|
if (rio_fd < 0 &&
|
|
(rio_fd = open(RIO_DEVICE, O_RDWR)) < 0 &&
|
|
(rio_fd = open_module(RIO_DEVICE, O_RDWR)) < 0) {
|
|
LOG(LOG_ERR, errno, "RIO open");
|
|
return (QOPERR_SYSCALL);
|
|
}
|
|
|
|
rio_refcount++;
|
|
memset(&iface, 0, sizeof(iface));
|
|
strncpy(iface.rio_ifname, ifinfo->ifname, IFNAMSIZ);
|
|
|
|
if (ioctl(rio_fd, RIO_IF_ATTACH, &iface) < 0)
|
|
return (QOPERR_SYSCALL);
|
|
|
|
/* set rio parameters */
|
|
rio_ifinfo = (struct rio_ifinfo *)ifinfo->private;
|
|
memset(&conf, 0, sizeof(conf));
|
|
strncpy(conf.iface.rio_ifname, ifinfo->ifname, IFNAMSIZ);
|
|
for (i = 0; i < RIO_NDROPPREC; i++)
|
|
conf.q_params[i] = rio_ifinfo->red_params[i];
|
|
conf.rio_weight = rio_ifinfo->weight;
|
|
conf.rio_limit = rio_ifinfo->qlimit;
|
|
conf.rio_flags = rio_ifinfo->flags;
|
|
if (ioctl(rio_fd, RIO_CONFIG, &conf) < 0)
|
|
return (QOPERR_SYSCALL);
|
|
|
|
#if 1
|
|
LOG(LOG_INFO, 0, "rio attached to %s", iface.rio_ifname);
|
|
#endif
|
|
return (0);
|
|
}
|
|
|
|
static int
|
|
rio_detach(struct ifinfo *ifinfo)
|
|
{
|
|
struct rio_interface iface;
|
|
|
|
memset(&iface, 0, sizeof(iface));
|
|
strncpy(iface.rio_ifname, ifinfo->ifname, IFNAMSIZ);
|
|
|
|
if (ioctl(rio_fd, RIO_IF_DETACH, &iface) < 0)
|
|
return (QOPERR_SYSCALL);
|
|
|
|
if (--rio_refcount == 0) {
|
|
close(rio_fd);
|
|
rio_fd = -1;
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
static int
|
|
rio_enable(struct ifinfo *ifinfo)
|
|
{
|
|
struct rio_interface iface;
|
|
|
|
memset(&iface, 0, sizeof(iface));
|
|
strncpy(iface.rio_ifname, ifinfo->ifname, IFNAMSIZ);
|
|
|
|
if (ioctl(rio_fd, RIO_ENABLE, &iface) < 0)
|
|
return (QOPERR_SYSCALL);
|
|
return (0);
|
|
}
|
|
|
|
static int
|
|
rio_disable(struct ifinfo *ifinfo)
|
|
{
|
|
struct rio_interface iface;
|
|
|
|
memset(&iface, 0, sizeof(iface));
|
|
strncpy(iface.rio_ifname, ifinfo->ifname, IFNAMSIZ);
|
|
|
|
if (ioctl(rio_fd, RIO_DISABLE, &iface) < 0)
|
|
return (QOPERR_SYSCALL);
|
|
return (0);
|
|
}
|