1213267Sluigi/*-
2204591Sluigi * Copyright (c) 2010 Riccardo Panicucci, Universita` di Pisa
3204591Sluigi * All rights reserved
4204591Sluigi *
5204591Sluigi * Redistribution and use in source and binary forms, with or without
6204591Sluigi * modification, are permitted provided that the following conditions
7204591Sluigi * are met:
8204591Sluigi * 1. Redistributions of source code must retain the above copyright
9204591Sluigi *    notice, this list of conditions and the following disclaimer.
10204591Sluigi * 2. Redistributions in binary form must reproduce the above copyright
11204591Sluigi *    notice, this list of conditions and the following disclaimer in the
12204591Sluigi *    documentation and/or other materials provided with the distribution.
13204591Sluigi *
14204591Sluigi * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15204591Sluigi * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16204591Sluigi * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17204591Sluigi * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18204591Sluigi * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19204591Sluigi * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20204591Sluigi * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21204591Sluigi * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22204591Sluigi * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23204591Sluigi * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24204591Sluigi * SUCH DAMAGE.
25204591Sluigi */
26204591Sluigi
27204591Sluigi/*
28204591Sluigi * $FreeBSD$
29204591Sluigi *
30204591Sluigi * Binary compatibility support for /sbin/ipfw RELENG_7 and RELENG_8
31204591Sluigi */
32204591Sluigi
33204591Sluigi#include "opt_inet6.h"
34204591Sluigi
35204591Sluigi#include <sys/param.h>
36204591Sluigi#include <sys/systm.h>
37204591Sluigi#include <sys/malloc.h>
38204591Sluigi#include <sys/mbuf.h>
39204591Sluigi#include <sys/kernel.h>
40204591Sluigi#include <sys/lock.h>
41204591Sluigi#include <sys/module.h>
42204591Sluigi#include <sys/priv.h>
43204591Sluigi#include <sys/proc.h>
44204591Sluigi#include <sys/rwlock.h>
45204591Sluigi#include <sys/socket.h>
46204591Sluigi#include <sys/socketvar.h>
47204591Sluigi#include <sys/time.h>
48204591Sluigi#include <sys/taskqueue.h>
49204591Sluigi#include <net/if.h>	/* IFNAMSIZ, struct ifaddr, ifq head, lock.h mutex.h */
50204591Sluigi#include <netinet/in.h>
51204591Sluigi#include <netinet/ip_var.h>	/* ip_output(), IP_FORWARDING */
52204591Sluigi#include <netinet/ip_fw.h>
53204591Sluigi#include <netinet/ip_dummynet.h>
54204591Sluigi
55240494Sglebius#include <netpfil/ipfw/ip_fw_private.h>
56240494Sglebius#include <netpfil/ipfw/dn_heap.h>
57240494Sglebius#include <netpfil/ipfw/ip_dn_private.h>
58240494Sglebius#include <netpfil/ipfw/dn_sched.h>
59240494Sglebius
60204591Sluigi/* FREEBSD7.2 ip_dummynet.h r191715*/
61204591Sluigi
62204591Sluigistruct dn_heap_entry7 {
63204591Sluigi	int64_t key;        /* sorting key. Topmost element is smallest one */
64204591Sluigi	void *object;      /* object pointer */
65204591Sluigi};
66204591Sluigi
67204591Sluigistruct dn_heap7 {
68204591Sluigi	int size;
69204591Sluigi	int elements;
70204591Sluigi	int offset; /* XXX if > 0 this is the offset of direct ptr to obj */
71204591Sluigi	struct dn_heap_entry7 *p;   /* really an array of "size" entries */
72204591Sluigi};
73204591Sluigi
74204591Sluigi/* Common to 7.2 and 8 */
75204591Sluigistruct dn_flow_set {
76204591Sluigi	SLIST_ENTRY(dn_flow_set)    next;   /* linked list in a hash slot */
77204591Sluigi
78204591Sluigi	u_short fs_nr ;             /* flow_set number       */
79204591Sluigi	u_short flags_fs;
80204591Sluigi#define DNOLD_HAVE_FLOW_MASK   0x0001
81204591Sluigi#define DNOLD_IS_RED       0x0002
82204591Sluigi#define DNOLD_IS_GENTLE_RED    0x0004
83204591Sluigi#define DNOLD_QSIZE_IS_BYTES   0x0008  /* queue size is measured in bytes */
84204591Sluigi#define DNOLD_NOERROR      0x0010  /* do not report ENOBUFS on drops  */
85204591Sluigi#define DNOLD_HAS_PROFILE      0x0020  /* the pipe has a delay profile. */
86204591Sluigi#define DNOLD_IS_PIPE      0x4000
87204591Sluigi#define DNOLD_IS_QUEUE     0x8000
88204591Sluigi
89204591Sluigi	struct dn_pipe7 *pipe ;  /* pointer to parent pipe */
90204591Sluigi	u_short parent_nr ;     /* parent pipe#, 0 if local to a pipe */
91204591Sluigi
92204591Sluigi	int weight ;        /* WFQ queue weight */
93204591Sluigi	int qsize ;         /* queue size in slots or bytes */
94204591Sluigi	int plr ;           /* pkt loss rate (2^31-1 means 100%) */
95204591Sluigi
96204591Sluigi	struct ipfw_flow_id flow_mask ;
97204591Sluigi
98204591Sluigi	/* hash table of queues onto this flow_set */
99204591Sluigi	int rq_size ;       /* number of slots */
100204591Sluigi	int rq_elements ;       /* active elements */
101204591Sluigi	struct dn_flow_queue7 **rq;  /* array of rq_size entries */
102204591Sluigi
103204591Sluigi	u_int32_t last_expired ;    /* do not expire too frequently */
104204591Sluigi	int backlogged ;        /* #active queues for this flowset */
105204591Sluigi
106204591Sluigi        /* RED parameters */
107204591Sluigi#define SCALE_RED               16
108204591Sluigi#define SCALE(x)                ( (x) << SCALE_RED )
109204591Sluigi#define SCALE_VAL(x)            ( (x) >> SCALE_RED )
110204591Sluigi#define SCALE_MUL(x,y)          ( ( (x) * (y) ) >> SCALE_RED )
111204591Sluigi	int w_q ;           /* queue weight (scaled) */
112204591Sluigi	int max_th ;        /* maximum threshold for queue (scaled) */
113204591Sluigi	int min_th ;        /* minimum threshold for queue (scaled) */
114204591Sluigi	int max_p ;         /* maximum value for p_b (scaled) */
115204591Sluigi	u_int c_1 ;         /* max_p/(max_th-min_th) (scaled) */
116204591Sluigi	u_int c_2 ;         /* max_p*min_th/(max_th-min_th) (scaled) */
117204591Sluigi	u_int c_3 ;         /* for GRED, (1-max_p)/max_th (scaled) */
118204591Sluigi	u_int c_4 ;         /* for GRED, 1 - 2*max_p (scaled) */
119204591Sluigi	u_int * w_q_lookup ;    /* lookup table for computing (1-w_q)^t */
120204591Sluigi	u_int lookup_depth ;    /* depth of lookup table */
121204591Sluigi	int lookup_step ;       /* granularity inside the lookup table */
122204591Sluigi	int lookup_weight ;     /* equal to (1-w_q)^t / (1-w_q)^(t+1) */
123204591Sluigi	int avg_pkt_size ;      /* medium packet size */
124204591Sluigi	int max_pkt_size ;      /* max packet size */
125204591Sluigi};
126204591SluigiSLIST_HEAD(dn_flow_set_head, dn_flow_set);
127204591Sluigi
128204591Sluigi#define DN_IS_PIPE		0x4000
129204591Sluigi#define DN_IS_QUEUE		0x8000
130204591Sluigistruct dn_flow_queue7 {
131204591Sluigi	struct dn_flow_queue7 *next ;
132204591Sluigi	struct ipfw_flow_id id ;
133204591Sluigi
134204591Sluigi	struct mbuf *head, *tail ;  /* queue of packets */
135204591Sluigi	u_int len ;
136204591Sluigi	u_int len_bytes ;
137204591Sluigi
138204591Sluigi	u_long numbytes;
139204591Sluigi
140204591Sluigi	u_int64_t tot_pkts ;    /* statistics counters  */
141204591Sluigi	u_int64_t tot_bytes ;
142204591Sluigi	u_int32_t drops ;
143204591Sluigi
144204591Sluigi	int hash_slot ;     /* debugging/diagnostic */
145204591Sluigi
146204591Sluigi	/* RED parameters */
147204591Sluigi	int avg ;                   /* average queue length est. (scaled) */
148204591Sluigi	int count ;                 /* arrivals since last RED drop */
149204591Sluigi	int random ;                /* random value (scaled) */
150204591Sluigi	u_int32_t q_time;      /* start of queue idle time */
151204591Sluigi
152204591Sluigi	/* WF2Q+ support */
153204591Sluigi	struct dn_flow_set *fs ;    /* parent flow set */
154204591Sluigi	int heap_pos ;      /* position (index) of struct in heap */
155204591Sluigi	int64_t sched_time ;     /* current time when queue enters ready_heap */
156204591Sluigi
157204591Sluigi	int64_t S,F ;        /* start time, finish time */
158204591Sluigi};
159204591Sluigi
160204591Sluigistruct dn_pipe7 {        /* a pipe */
161204591Sluigi	SLIST_ENTRY(dn_pipe7)    next;   /* linked list in a hash slot */
162204591Sluigi
163204591Sluigi	int pipe_nr ;       /* number   */
164204591Sluigi	int bandwidth;      /* really, bytes/tick.  */
165204591Sluigi	int delay ;         /* really, ticks    */
166204591Sluigi
167204591Sluigi	struct  mbuf *head, *tail ; /* packets in delay line */
168204591Sluigi
169204591Sluigi	/* WF2Q+ */
170204591Sluigi	struct dn_heap7 scheduler_heap ; /* top extract - key Finish time*/
171204591Sluigi	struct dn_heap7 not_eligible_heap; /* top extract- key Start time */
172204591Sluigi	struct dn_heap7 idle_heap ; /* random extract - key Start=Finish time */
173204591Sluigi
174204591Sluigi	int64_t V ;          /* virtual time */
175204591Sluigi	int sum;            /* sum of weights of all active sessions */
176204591Sluigi
177204591Sluigi	int numbytes;
178204591Sluigi
179204591Sluigi	int64_t sched_time ;     /* time pipe was scheduled in ready_heap */
180204591Sluigi
181204591Sluigi	/*
182204591Sluigi	* When the tx clock come from an interface (if_name[0] != '\0'), its name
183204591Sluigi	* is stored below, whereas the ifp is filled when the rule is configured.
184204591Sluigi	*/
185204591Sluigi	char if_name[IFNAMSIZ];
186204591Sluigi	struct ifnet *ifp ;
187204591Sluigi	int ready ; /* set if ifp != NULL and we got a signal from it */
188204591Sluigi
189204591Sluigi	struct dn_flow_set fs ; /* used with fixed-rate flows */
190204591Sluigi};
191204591SluigiSLIST_HEAD(dn_pipe_head7, dn_pipe7);
192204591Sluigi
193204591Sluigi
194204591Sluigi/* FREEBSD8 ip_dummynet.h r196045 */
195204591Sluigistruct dn_flow_queue8 {
196204591Sluigi	struct dn_flow_queue8 *next ;
197204591Sluigi	struct ipfw_flow_id id ;
198204591Sluigi
199204591Sluigi	struct mbuf *head, *tail ;  /* queue of packets */
200204591Sluigi	u_int len ;
201204591Sluigi	u_int len_bytes ;
202204591Sluigi
203204591Sluigi	uint64_t numbytes ;     /* credit for transmission (dynamic queues) */
204204591Sluigi	int64_t extra_bits;     /* extra bits simulating unavailable channel */
205204591Sluigi
206204591Sluigi	u_int64_t tot_pkts ;    /* statistics counters  */
207204591Sluigi	u_int64_t tot_bytes ;
208204591Sluigi	u_int32_t drops ;
209204591Sluigi
210204591Sluigi	int hash_slot ;     /* debugging/diagnostic */
211204591Sluigi
212204591Sluigi	/* RED parameters */
213204591Sluigi	int avg ;                   /* average queue length est. (scaled) */
214204591Sluigi	int count ;                 /* arrivals since last RED drop */
215204591Sluigi	int random ;                /* random value (scaled) */
216204591Sluigi	int64_t idle_time;       /* start of queue idle time */
217204591Sluigi
218204591Sluigi	/* WF2Q+ support */
219204591Sluigi	struct dn_flow_set *fs ;    /* parent flow set */
220204591Sluigi	int heap_pos ;      /* position (index) of struct in heap */
221204591Sluigi	int64_t sched_time ;     /* current time when queue enters ready_heap */
222204591Sluigi
223204591Sluigi	int64_t S,F ;        /* start time, finish time */
224204591Sluigi};
225204591Sluigi
226204591Sluigistruct dn_pipe8 {        /* a pipe */
227204591Sluigi	SLIST_ENTRY(dn_pipe8)    next;   /* linked list in a hash slot */
228204591Sluigi
229204591Sluigi	int pipe_nr ;       /* number   */
230204591Sluigi	int bandwidth;      /* really, bytes/tick.  */
231204591Sluigi	int delay ;         /* really, ticks    */
232204591Sluigi
233204591Sluigi	struct  mbuf *head, *tail ; /* packets in delay line */
234204591Sluigi
235204591Sluigi	/* WF2Q+ */
236204591Sluigi	struct dn_heap7 scheduler_heap ; /* top extract - key Finish time*/
237204591Sluigi	struct dn_heap7 not_eligible_heap; /* top extract- key Start time */
238204591Sluigi	struct dn_heap7 idle_heap ; /* random extract - key Start=Finish time */
239204591Sluigi
240204591Sluigi	int64_t V ;          /* virtual time */
241204591Sluigi	int sum;            /* sum of weights of all active sessions */
242204591Sluigi
243204591Sluigi	/* Same as in dn_flow_queue, numbytes can become large */
244204591Sluigi	int64_t numbytes;       /* bits I can transmit (more or less). */
245204591Sluigi	uint64_t burst;     /* burst size, scaled: bits * hz */
246204591Sluigi
247204591Sluigi	int64_t sched_time ;     /* time pipe was scheduled in ready_heap */
248204591Sluigi	int64_t idle_time;       /* start of pipe idle time */
249204591Sluigi
250204591Sluigi	char if_name[IFNAMSIZ];
251204591Sluigi	struct ifnet *ifp ;
252204591Sluigi	int ready ; /* set if ifp != NULL and we got a signal from it */
253204591Sluigi
254204591Sluigi	struct dn_flow_set fs ; /* used with fixed-rate flows */
255204591Sluigi
256204591Sluigi    /* fields to simulate a delay profile */
257204591Sluigi#define ED_MAX_NAME_LEN     32
258204591Sluigi	char name[ED_MAX_NAME_LEN];
259204591Sluigi	int loss_level;
260204591Sluigi	int samples_no;
261204591Sluigi	int *samples;
262204591Sluigi};
263204591Sluigi
264204591Sluigi#define ED_MAX_SAMPLES_NO   1024
265204591Sluigistruct dn_pipe_max8 {
266204591Sluigi	struct dn_pipe8 pipe;
267204591Sluigi	int samples[ED_MAX_SAMPLES_NO];
268204591Sluigi};
269204591SluigiSLIST_HEAD(dn_pipe_head8, dn_pipe8);
270204591Sluigi
271204591Sluigi/*
272204591Sluigi * Changes from 7.2 to 8:
273204591Sluigi * dn_pipe:
274204591Sluigi *      numbytes from int to int64_t
275204591Sluigi *      add burst (int64_t)
276204591Sluigi *      add idle_time (int64_t)
277204591Sluigi *      add profile
278204591Sluigi *      add struct dn_pipe_max
279204591Sluigi *      add flag DN_HAS_PROFILE
280204591Sluigi *
281204591Sluigi * dn_flow_queue
282204591Sluigi *      numbytes from u_long to int64_t
283204591Sluigi *      add extra_bits (int64_t)
284204591Sluigi *      q_time from u_int32_t to int64_t and name idle_time
285204591Sluigi *
286204591Sluigi * dn_flow_set unchanged
287204591Sluigi *
288204591Sluigi */
289204591Sluigi
290204591Sluigi/* NOTE:XXX copied from dummynet.c */
291204591Sluigi#define O_NEXT(p, len) ((void *)((char *)p + len))
292204591Sluigistatic void
293204591Sluigioid_fill(struct dn_id *oid, int len, int type, uintptr_t id)
294204591Sluigi{
295204591Sluigi	oid->len = len;
296204591Sluigi	oid->type = type;
297204591Sluigi	oid->subtype = 0;
298204591Sluigi	oid->id = id;
299204591Sluigi}
300204591Sluigi/* make room in the buffer and move the pointer forward */
301204591Sluigistatic void *
302204591Sluigio_next(struct dn_id **o, int len, int type)
303204591Sluigi{
304204591Sluigi	struct dn_id *ret = *o;
305204591Sluigi	oid_fill(ret, len, type, 0);
306204591Sluigi	*o = O_NEXT(*o, len);
307204591Sluigi	return ret;
308204591Sluigi}
309204591Sluigi
310204591Sluigi
311204591Sluigistatic size_t pipesize7 = sizeof(struct dn_pipe7);
312204591Sluigistatic size_t pipesize8 = sizeof(struct dn_pipe8);
313204591Sluigistatic size_t pipesizemax8 = sizeof(struct dn_pipe_max8);
314204591Sluigi
315204591Sluigi/* Indicate 'ipfw' version
316204591Sluigi * 1: from FreeBSD 7.2
317204591Sluigi * 0: from FreeBSD 8
318241369Skevlo * -1: unknown (for now is unused)
319204591Sluigi *
320204591Sluigi * It is update when a IP_DUMMYNET_DEL or IP_DUMMYNET_CONFIGURE request arrives
321241369Skevlo * NOTE: if a IP_DUMMYNET_GET arrives and the 'ipfw' version is unknown,
322204591Sluigi *       it is suppose to be the FreeBSD 8 version.
323204591Sluigi */
324204591Sluigistatic int is7 = 0;
325204591Sluigi
326204591Sluigistatic int
327204591Sluigiconvertflags2new(int src)
328204591Sluigi{
329204591Sluigi	int dst = 0;
330204591Sluigi
331204591Sluigi	if (src & DNOLD_HAVE_FLOW_MASK)
332204591Sluigi		dst |= DN_HAVE_MASK;
333204591Sluigi	if (src & DNOLD_QSIZE_IS_BYTES)
334204591Sluigi		dst |= DN_QSIZE_BYTES;
335204591Sluigi	if (src & DNOLD_NOERROR)
336204591Sluigi		dst |= DN_NOERROR;
337204591Sluigi	if (src & DNOLD_IS_RED)
338204591Sluigi		dst |= DN_IS_RED;
339204591Sluigi	if (src & DNOLD_IS_GENTLE_RED)
340204591Sluigi		dst |= DN_IS_GENTLE_RED;
341204591Sluigi	if (src & DNOLD_HAS_PROFILE)
342204591Sluigi		dst |= DN_HAS_PROFILE;
343204591Sluigi
344204591Sluigi	return dst;
345204591Sluigi}
346204591Sluigi
347204591Sluigistatic int
348204591Sluigiconvertflags2old(int src)
349204591Sluigi{
350204591Sluigi	int dst = 0;
351204591Sluigi
352204591Sluigi	if (src & DN_HAVE_MASK)
353204591Sluigi		dst |= DNOLD_HAVE_FLOW_MASK;
354204591Sluigi	if (src & DN_IS_RED)
355204591Sluigi		dst |= DNOLD_IS_RED;
356204591Sluigi	if (src & DN_IS_GENTLE_RED)
357204591Sluigi		dst |= DNOLD_IS_GENTLE_RED;
358204591Sluigi	if (src & DN_NOERROR)
359204591Sluigi		dst |= DNOLD_NOERROR;
360204591Sluigi	if (src & DN_HAS_PROFILE)
361204591Sluigi		dst |= DNOLD_HAS_PROFILE;
362204591Sluigi	if (src & DN_QSIZE_BYTES)
363204591Sluigi		dst |= DNOLD_QSIZE_IS_BYTES;
364204591Sluigi
365204591Sluigi	return dst;
366204591Sluigi}
367204591Sluigi
368204591Sluigistatic int
369204591Sluigidn_compat_del(void *v)
370204591Sluigi{
371204591Sluigi	struct dn_pipe7 *p = (struct dn_pipe7 *) v;
372204591Sluigi	struct dn_pipe8 *p8 = (struct dn_pipe8 *) v;
373204591Sluigi	struct {
374204591Sluigi		struct dn_id oid;
375204591Sluigi		uintptr_t a[1];	/* add more if we want a list */
376204591Sluigi	} cmd;
377204591Sluigi
378204591Sluigi	/* XXX DN_API_VERSION ??? */
379204591Sluigi	oid_fill((void *)&cmd, sizeof(cmd), DN_CMD_DELETE, DN_API_VERSION);
380204591Sluigi
381204591Sluigi	if (is7) {
382204591Sluigi		if (p->pipe_nr == 0 && p->fs.fs_nr == 0)
383204591Sluigi			return EINVAL;
384204591Sluigi		if (p->pipe_nr != 0 && p->fs.fs_nr != 0)
385204591Sluigi			return EINVAL;
386204591Sluigi	} else {
387204591Sluigi		if (p8->pipe_nr == 0 && p8->fs.fs_nr == 0)
388204591Sluigi			return EINVAL;
389204591Sluigi		if (p8->pipe_nr != 0 && p8->fs.fs_nr != 0)
390204591Sluigi			return EINVAL;
391204591Sluigi	}
392204591Sluigi
393204591Sluigi	if (p->pipe_nr != 0) { /* pipe x delete */
394204591Sluigi		cmd.a[0] = p->pipe_nr;
395204591Sluigi		cmd.oid.subtype = DN_LINK;
396204591Sluigi	} else { /* queue x delete */
397204591Sluigi		cmd.oid.subtype = DN_FS;
398204591Sluigi		cmd.a[0] = (is7) ? p->fs.fs_nr : p8->fs.fs_nr;
399204591Sluigi	}
400204591Sluigi
401204591Sluigi	return do_config(&cmd, cmd.oid.len);
402204591Sluigi}
403204591Sluigi
404204591Sluigistatic int
405204591Sluigidn_compat_config_queue(struct dn_fs *fs, void* v)
406204591Sluigi{
407204591Sluigi	struct dn_pipe7 *p7 = (struct dn_pipe7 *)v;
408204591Sluigi	struct dn_pipe8 *p8 = (struct dn_pipe8 *)v;
409204591Sluigi	struct dn_flow_set *f;
410204591Sluigi
411204591Sluigi	if (is7)
412204591Sluigi		f = &p7->fs;
413204591Sluigi	else
414204591Sluigi		f = &p8->fs;
415204591Sluigi
416204591Sluigi	fs->fs_nr = f->fs_nr;
417204591Sluigi	fs->sched_nr = f->parent_nr;
418204591Sluigi	fs->flow_mask = f->flow_mask;
419204591Sluigi	fs->buckets = f->rq_size;
420204591Sluigi	fs->qsize = f->qsize;
421204591Sluigi	fs->plr = f->plr;
422204591Sluigi	fs->par[0] = f->weight;
423204591Sluigi	fs->flags = convertflags2new(f->flags_fs);
424204591Sluigi	if (fs->flags & DN_IS_GENTLE_RED || fs->flags & DN_IS_RED) {
425204591Sluigi		fs->w_q = f->w_q;
426204591Sluigi		fs->max_th = f->max_th;
427204591Sluigi		fs->min_th = f->min_th;
428204591Sluigi		fs->max_p = f->max_p;
429204591Sluigi	}
430204591Sluigi
431204591Sluigi	return 0;
432204591Sluigi}
433204591Sluigi
434204591Sluigistatic int
435204591Sluigidn_compat_config_pipe(struct dn_sch *sch, struct dn_link *p,
436204591Sluigi		      struct dn_fs *fs, void* v)
437204591Sluigi{
438204591Sluigi	struct dn_pipe7 *p7 = (struct dn_pipe7 *)v;
439204591Sluigi	struct dn_pipe8 *p8 = (struct dn_pipe8 *)v;
440204591Sluigi	int i = p7->pipe_nr;
441204591Sluigi
442204591Sluigi	sch->sched_nr = i;
443204591Sluigi	sch->oid.subtype = 0;
444204591Sluigi	p->link_nr = i;
445204591Sluigi	fs->fs_nr = i + 2*DN_MAX_ID;
446204591Sluigi	fs->sched_nr = i + DN_MAX_ID;
447204591Sluigi
448204591Sluigi	/* Common to 7 and 8 */
449204591Sluigi	p->bandwidth = p7->bandwidth;
450204591Sluigi	p->delay = p7->delay;
451204591Sluigi	if (!is7) {
452204591Sluigi		/* FreeBSD 8 has burst  */
453204591Sluigi		p->burst = p8->burst;
454204591Sluigi	}
455204591Sluigi
456204591Sluigi	/* fill the fifo flowset */
457204591Sluigi	dn_compat_config_queue(fs, v);
458204591Sluigi	fs->fs_nr = i + 2*DN_MAX_ID;
459204591Sluigi	fs->sched_nr = i + DN_MAX_ID;
460204591Sluigi
461204591Sluigi	/* Move scheduler related parameter from fs to sch */
462204591Sluigi	sch->buckets = fs->buckets; /*XXX*/
463204591Sluigi	fs->buckets = 0;
464204591Sluigi	if (fs->flags & DN_HAVE_MASK) {
465204591Sluigi		sch->flags |= DN_HAVE_MASK;
466204591Sluigi		fs->flags &= ~DN_HAVE_MASK;
467204591Sluigi		sch->sched_mask = fs->flow_mask;
468204591Sluigi		bzero(&fs->flow_mask, sizeof(struct ipfw_flow_id));
469204591Sluigi	}
470204591Sluigi
471204591Sluigi	return 0;
472204591Sluigi}
473204591Sluigi
474204591Sluigistatic int
475204591Sluigidn_compat_config_profile(struct dn_profile *pf, struct dn_link *p,
476204591Sluigi			 void *v)
477204591Sluigi{
478204591Sluigi	struct dn_pipe8 *p8 = (struct dn_pipe8 *)v;
479204591Sluigi
480204591Sluigi	p8->samples = &(((struct dn_pipe_max8 *)p8)->samples[0]);
481204591Sluigi
482204591Sluigi	pf->link_nr = p->link_nr;
483204591Sluigi	pf->loss_level = p8->loss_level;
484204591Sluigi// 	pf->bandwidth = p->bandwidth; //XXX bandwidth redundant?
485204591Sluigi	pf->samples_no = p8->samples_no;
486204591Sluigi	strncpy(pf->name, p8->name,sizeof(pf->name));
487204591Sluigi	bcopy(p8->samples, pf->samples, sizeof(pf->samples));
488204591Sluigi
489204591Sluigi	return 0;
490204591Sluigi}
491204591Sluigi
492204591Sluigi/*
493204591Sluigi * If p->pipe_nr != 0 the command is 'pipe x config', so need to create
494204591Sluigi * the three main struct, else only a flowset is created
495204591Sluigi */
496204591Sluigistatic int
497204591Sluigidn_compat_configure(void *v)
498204591Sluigi{
499204954Sluigi	struct dn_id *buf = NULL, *base;
500204591Sluigi	struct dn_sch *sch = NULL;
501204591Sluigi	struct dn_link *p = NULL;
502204591Sluigi	struct dn_fs *fs = NULL;
503204591Sluigi	struct dn_profile *pf = NULL;
504204591Sluigi	int lmax;
505204591Sluigi	int error;
506204591Sluigi
507204591Sluigi	struct dn_pipe7 *p7 = (struct dn_pipe7 *)v;
508204591Sluigi	struct dn_pipe8 *p8 = (struct dn_pipe8 *)v;
509204591Sluigi
510204591Sluigi	int i; /* number of object to configure */
511204591Sluigi
512204591Sluigi	lmax = sizeof(struct dn_id);	/* command header */
513204591Sluigi	lmax += sizeof(struct dn_sch) + sizeof(struct dn_link) +
514204591Sluigi		sizeof(struct dn_fs) + sizeof(struct dn_profile);
515204591Sluigi
516243882Sglebius	base = buf = malloc(lmax, M_DUMMYNET, M_WAITOK|M_ZERO);
517204591Sluigi	o_next(&buf, sizeof(struct dn_id), DN_CMD_CONFIG);
518204591Sluigi	base->id = DN_API_VERSION;
519204591Sluigi
520204591Sluigi	/* pipe_nr is the same in p7 and p8 */
521204591Sluigi	i = p7->pipe_nr;
522204591Sluigi	if (i != 0) { /* pipe config */
523204591Sluigi		sch = o_next(&buf, sizeof(*sch), DN_SCH);
524204591Sluigi		p = o_next(&buf, sizeof(*p), DN_LINK);
525204591Sluigi		fs = o_next(&buf, sizeof(*fs), DN_FS);
526204591Sluigi
527204591Sluigi		error = dn_compat_config_pipe(sch, p, fs, v);
528204591Sluigi		if (error) {
529204591Sluigi			free(buf, M_DUMMYNET);
530204591Sluigi			return error;
531204591Sluigi		}
532204591Sluigi		if (!is7 && p8->samples_no > 0) {
533204591Sluigi			/* Add profiles*/
534204591Sluigi			pf = o_next(&buf, sizeof(*pf), DN_PROFILE);
535204591Sluigi			error = dn_compat_config_profile(pf, p, v);
536204591Sluigi			if (error) {
537204591Sluigi				free(buf, M_DUMMYNET);
538204591Sluigi				return error;
539204591Sluigi			}
540204591Sluigi		}
541204591Sluigi	} else { /* queue config */
542204591Sluigi		fs = o_next(&buf, sizeof(*fs), DN_FS);
543204591Sluigi		error = dn_compat_config_queue(fs, v);
544204591Sluigi		if (error) {
545204591Sluigi			free(buf, M_DUMMYNET);
546204591Sluigi			return error;
547204591Sluigi		}
548204591Sluigi	}
549204591Sluigi	error = do_config(base, (char *)buf - (char *)base);
550204591Sluigi
551204954Sluigi	if (buf)
552204954Sluigi		free(buf, M_DUMMYNET);
553204591Sluigi	return error;
554204591Sluigi}
555204591Sluigi
556204591Sluigiint
557206425Sluigidn_compat_calc_size(void)
558204591Sluigi{
559204591Sluigi	int need = 0;
560204591Sluigi	/* XXX use FreeBSD 8 struct size */
561204591Sluigi	/* NOTE:
562204591Sluigi	 * - half scheduler: 		schk_count/2
563204591Sluigi	 * - all flowset:		fsk_count
564204591Sluigi	 * - all flowset queues:	queue_count
565204591Sluigi	 * - all pipe queue:		si_count
566204591Sluigi	 */
567204591Sluigi	need += dn_cfg.schk_count * sizeof(struct dn_pipe8) / 2;
568204591Sluigi	need += dn_cfg.fsk_count * sizeof(struct dn_flow_set);
569204591Sluigi	need += dn_cfg.si_count * sizeof(struct dn_flow_queue8);
570204591Sluigi	need += dn_cfg.queue_count * sizeof(struct dn_flow_queue8);
571204591Sluigi
572204591Sluigi	return need;
573204591Sluigi}
574204591Sluigi
575204591Sluigiint
576204591Sluigidn_c_copy_q (void *_ni, void *arg)
577204591Sluigi{
578204591Sluigi	struct copy_args *a = arg;
579204591Sluigi	struct dn_flow_queue7 *fq7 = (struct dn_flow_queue7 *)*a->start;
580204591Sluigi	struct dn_flow_queue8 *fq8 = (struct dn_flow_queue8 *)*a->start;
581204591Sluigi	struct dn_flow *ni = (struct dn_flow *)_ni;
582204591Sluigi	int size = 0;
583204591Sluigi
584204591Sluigi	/* XXX hash slot not set */
585204591Sluigi	/* No difference between 7.2/8 */
586204591Sluigi	fq7->len = ni->length;
587204591Sluigi	fq7->len_bytes = ni->len_bytes;
588204591Sluigi	fq7->id = ni->fid;
589204591Sluigi
590204591Sluigi	if (is7) {
591204591Sluigi		size = sizeof(struct dn_flow_queue7);
592204591Sluigi		fq7->tot_pkts = ni->tot_pkts;
593204591Sluigi		fq7->tot_bytes = ni->tot_bytes;
594204591Sluigi		fq7->drops = ni->drops;
595204591Sluigi	} else {
596204591Sluigi		size = sizeof(struct dn_flow_queue8);
597204591Sluigi		fq8->tot_pkts = ni->tot_pkts;
598204591Sluigi		fq8->tot_bytes = ni->tot_bytes;
599204591Sluigi		fq8->drops = ni->drops;
600204591Sluigi	}
601204591Sluigi
602204591Sluigi	*a->start += size;
603204591Sluigi	return 0;
604204591Sluigi}
605204591Sluigi
606204591Sluigiint
607204591Sluigidn_c_copy_pipe(struct dn_schk *s, struct copy_args *a, int nq)
608204591Sluigi{
609204591Sluigi	struct dn_link *l = &s->link;
610204591Sluigi	struct dn_fsk *f = s->fs;
611204591Sluigi
612204591Sluigi	struct dn_pipe7 *pipe7 = (struct dn_pipe7 *)*a->start;
613204591Sluigi	struct dn_pipe8 *pipe8 = (struct dn_pipe8 *)*a->start;
614204591Sluigi	struct dn_flow_set *fs;
615204591Sluigi	int size = 0;
616204591Sluigi
617204591Sluigi	if (is7) {
618204591Sluigi		fs = &pipe7->fs;
619204591Sluigi		size = sizeof(struct dn_pipe7);
620204591Sluigi	} else {
621204591Sluigi		fs = &pipe8->fs;
622204591Sluigi		size = sizeof(struct dn_pipe8);
623204591Sluigi	}
624204591Sluigi
625204591Sluigi	/* These 4 field are the same in pipe7 and pipe8 */
626204591Sluigi	pipe7->next.sle_next = (struct dn_pipe7 *)DN_IS_PIPE;
627204591Sluigi	pipe7->bandwidth = l->bandwidth;
628221521Sae	pipe7->delay = l->delay * 1000 / hz;
629204591Sluigi	pipe7->pipe_nr = l->link_nr - DN_MAX_ID;
630204591Sluigi
631204591Sluigi	if (!is7) {
632204591Sluigi		if (s->profile) {
633204591Sluigi			struct dn_profile *pf = s->profile;
634204591Sluigi			strncpy(pipe8->name, pf->name, sizeof(pf->name));
635204591Sluigi			pipe8->loss_level = pf->loss_level;
636204591Sluigi			pipe8->samples_no = pf->samples_no;
637204591Sluigi		}
638204591Sluigi		pipe8->burst = div64(l->burst , 8 * hz);
639204591Sluigi	}
640204591Sluigi
641204591Sluigi	fs->flow_mask = s->sch.sched_mask;
642204591Sluigi	fs->rq_size = s->sch.buckets ? s->sch.buckets : 1;
643204591Sluigi
644204591Sluigi	fs->parent_nr = l->link_nr - DN_MAX_ID;
645204591Sluigi	fs->qsize = f->fs.qsize;
646204591Sluigi	fs->plr = f->fs.plr;
647204591Sluigi	fs->w_q = f->fs.w_q;
648204591Sluigi	fs->max_th = f->max_th;
649204591Sluigi	fs->min_th = f->min_th;
650204591Sluigi	fs->max_p = f->fs.max_p;
651204591Sluigi	fs->rq_elements = nq;
652204591Sluigi
653204591Sluigi	fs->flags_fs = convertflags2old(f->fs.flags);
654204591Sluigi
655204591Sluigi	*a->start += size;
656204591Sluigi	return 0;
657204591Sluigi}
658204591Sluigi
659204591Sluigi
660204591Sluigiint
661204591Sluigidn_compat_copy_pipe(struct copy_args *a, void *_o)
662204591Sluigi{
663204591Sluigi	int have = a->end - *a->start;
664204591Sluigi	int need = 0;
665204591Sluigi	int pipe_size = sizeof(struct dn_pipe8);
666204591Sluigi	int queue_size = sizeof(struct dn_flow_queue8);
667204591Sluigi	int n_queue = 0; /* number of queues */
668204591Sluigi
669204591Sluigi	struct dn_schk *s = (struct dn_schk *)_o;
670204591Sluigi	/* calculate needed space:
671204591Sluigi	 * - struct dn_pipe
672204591Sluigi	 * - if there are instances, dn_queue * n_instances
673204591Sluigi	 */
674204591Sluigi	n_queue = (s->sch.flags & DN_HAVE_MASK ? dn_ht_entries(s->siht) :
675204591Sluigi						(s->siht ? 1 : 0));
676204591Sluigi	need = pipe_size + queue_size * n_queue;
677204591Sluigi	if (have < need) {
678204591Sluigi		D("have %d < need %d", have, need);
679204591Sluigi		return 1;
680204591Sluigi	}
681204591Sluigi	/* copy pipe */
682204591Sluigi	dn_c_copy_pipe(s, a, n_queue);
683204591Sluigi
684204591Sluigi	/* copy queues */
685204591Sluigi	if (s->sch.flags & DN_HAVE_MASK)
686204591Sluigi		dn_ht_scan(s->siht, dn_c_copy_q, a);
687204591Sluigi	else if (s->siht)
688204591Sluigi		dn_c_copy_q(s->siht, a);
689204591Sluigi	return 0;
690204591Sluigi}
691204591Sluigi
692204591Sluigiint
693204591Sluigidn_c_copy_fs(struct dn_fsk *f, struct copy_args *a, int nq)
694204591Sluigi{
695204591Sluigi	struct dn_flow_set *fs = (struct dn_flow_set *)*a->start;
696204591Sluigi
697204591Sluigi	fs->next.sle_next = (struct dn_flow_set *)DN_IS_QUEUE;
698204591Sluigi	fs->fs_nr = f->fs.fs_nr;
699204591Sluigi	fs->qsize = f->fs.qsize;
700204591Sluigi	fs->plr = f->fs.plr;
701204591Sluigi	fs->w_q = f->fs.w_q;
702204591Sluigi	fs->max_th = f->max_th;
703204591Sluigi	fs->min_th = f->min_th;
704204591Sluigi	fs->max_p = f->fs.max_p;
705204591Sluigi	fs->flow_mask = f->fs.flow_mask;
706204591Sluigi	fs->rq_elements = nq;
707204591Sluigi	fs->rq_size = (f->fs.buckets ? f->fs.buckets : 1);
708204591Sluigi	fs->parent_nr = f->fs.sched_nr;
709204591Sluigi	fs->weight = f->fs.par[0];
710204591Sluigi
711204591Sluigi	fs->flags_fs = convertflags2old(f->fs.flags);
712204591Sluigi	*a->start += sizeof(struct dn_flow_set);
713204591Sluigi	return 0;
714204591Sluigi}
715204591Sluigi
716204591Sluigiint
717204591Sluigidn_compat_copy_queue(struct copy_args *a, void *_o)
718204591Sluigi{
719204591Sluigi	int have = a->end - *a->start;
720204591Sluigi	int need = 0;
721204591Sluigi	int fs_size = sizeof(struct dn_flow_set);
722204591Sluigi	int queue_size = sizeof(struct dn_flow_queue8);
723204591Sluigi
724204591Sluigi	struct dn_fsk *fs = (struct dn_fsk *)_o;
725204591Sluigi	int n_queue = 0; /* number of queues */
726204591Sluigi
727204591Sluigi	n_queue = (fs->fs.flags & DN_HAVE_MASK ? dn_ht_entries(fs->qht) :
728204591Sluigi						(fs->qht ? 1 : 0));
729204591Sluigi
730204591Sluigi	need = fs_size + queue_size * n_queue;
731204591Sluigi	if (have < need) {
732204591Sluigi		D("have < need");
733204591Sluigi		return 1;
734204591Sluigi	}
735204591Sluigi
736204591Sluigi	/* copy flowset */
737204591Sluigi	dn_c_copy_fs(fs, a, n_queue);
738204591Sluigi
739204591Sluigi	/* copy queues */
740204591Sluigi	if (fs->fs.flags & DN_HAVE_MASK)
741204591Sluigi		dn_ht_scan(fs->qht, dn_c_copy_q, a);
742204591Sluigi	else if (fs->qht)
743204591Sluigi		dn_c_copy_q(fs->qht, a);
744204591Sluigi
745204591Sluigi	return 0;
746204591Sluigi}
747204591Sluigi
748204591Sluigiint
749204591Sluigicopy_data_helper_compat(void *_o, void *_arg)
750204591Sluigi{
751204591Sluigi	struct copy_args *a = _arg;
752204591Sluigi
753204591Sluigi	if (a->type == DN_COMPAT_PIPE) {
754204591Sluigi		struct dn_schk *s = _o;
755204591Sluigi		if (s->sch.oid.subtype != 1 || s->sch.sched_nr <= DN_MAX_ID) {
756204591Sluigi			return 0;	/* not old type */
757204591Sluigi		}
758204591Sluigi		/* copy pipe parameters, and if instance exists, copy
759204591Sluigi		 * other parameters and eventually queues.
760204591Sluigi		 */
761204591Sluigi		if(dn_compat_copy_pipe(a, _o))
762204591Sluigi			return DNHT_SCAN_END;
763204591Sluigi	} else if (a->type == DN_COMPAT_QUEUE) {
764204591Sluigi		struct dn_fsk *fs = _o;
765204591Sluigi		if (fs->fs.fs_nr >= DN_MAX_ID)
766204591Sluigi			return 0;
767204591Sluigi		if (dn_compat_copy_queue(a, _o))
768204591Sluigi			return DNHT_SCAN_END;
769204591Sluigi	}
770204591Sluigi	return 0;
771204591Sluigi}
772204591Sluigi
773204591Sluigi/* Main function to manage old requests */
774204591Sluigiint
775204591Sluigiip_dummynet_compat(struct sockopt *sopt)
776204591Sluigi{
777204591Sluigi	int error=0;
778204591Sluigi	void *v = NULL;
779204591Sluigi	struct dn_id oid;
780204591Sluigi
781204591Sluigi	/* Lenght of data, used to found ipfw version... */
782204591Sluigi	int len = sopt->sopt_valsize;
783204591Sluigi
784204591Sluigi	/* len can be 0 if command was dummynet_flush */
785204591Sluigi	if (len == pipesize7) {
786204591Sluigi		D("setting compatibility with FreeBSD 7.2");
787204591Sluigi		is7 = 1;
788204591Sluigi	}
789204591Sluigi	else if (len == pipesize8 || len == pipesizemax8) {
790204591Sluigi		D("setting compatibility with FreeBSD 8");
791204591Sluigi		is7 = 0;
792204591Sluigi	}
793204591Sluigi
794204591Sluigi	switch (sopt->sopt_name) {
795204591Sluigi	default:
796204591Sluigi		printf("dummynet: -- unknown option %d", sopt->sopt_name);
797204591Sluigi		error = EINVAL;
798204591Sluigi		break;
799204591Sluigi
800204591Sluigi	case IP_DUMMYNET_FLUSH:
801204591Sluigi		oid_fill(&oid, sizeof(oid), DN_CMD_FLUSH, DN_API_VERSION);
802204591Sluigi		do_config(&oid, oid.len);
803204591Sluigi		break;
804204591Sluigi
805204591Sluigi	case IP_DUMMYNET_DEL:
806204591Sluigi		v = malloc(len, M_TEMP, M_WAITOK);
807204591Sluigi		error = sooptcopyin(sopt, v, len, len);
808204591Sluigi		if (error)
809204591Sluigi			break;
810204591Sluigi		error = dn_compat_del(v);
811220204Sae		free(v, M_TEMP);
812204591Sluigi		break;
813204591Sluigi
814204591Sluigi	case IP_DUMMYNET_CONFIGURE:
815204591Sluigi		v = malloc(len, M_TEMP, M_WAITOK);
816204591Sluigi		error = sooptcopyin(sopt, v, len, len);
817204591Sluigi		if (error)
818204591Sluigi			break;
819204591Sluigi		error = dn_compat_configure(v);
820220204Sae		free(v, M_TEMP);
821204591Sluigi		break;
822204591Sluigi
823204591Sluigi	case IP_DUMMYNET_GET: {
824204591Sluigi		void *buf;
825204591Sluigi		int ret;
826204591Sluigi		int original_size = sopt->sopt_valsize;
827204591Sluigi		int size;
828204591Sluigi
829204591Sluigi		ret = dummynet_get(sopt, &buf);
830204591Sluigi		if (ret)
831204591Sluigi			return 0;//XXX ?
832204591Sluigi		size = sopt->sopt_valsize;
833204591Sluigi		sopt->sopt_valsize = original_size;
834204591Sluigi		D("size=%d, buf=%p", size, buf);
835204591Sluigi		ret = sooptcopyout(sopt, buf, size);
836204591Sluigi		if (ret)
837204591Sluigi			printf("  %s ERROR sooptcopyout\n", __FUNCTION__);
838204591Sluigi		if (buf)
839204591Sluigi			free(buf, M_DUMMYNET);
840204591Sluigi	    }
841204591Sluigi	}
842204591Sluigi
843204591Sluigi	return error;
844204591Sluigi}
845204591Sluigi
846204591Sluigi
847