root/sys/net/dummynet/ip_dummynet.h
/*
 * Copyright (c) 1998-2002 Luigi Rizzo, Universita` di Pisa
 * Portions Copyright (c) 2000 Akamba Corp.
 * 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 THE AUTHOR 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 THE AUTHOR 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.
 *
 * $FreeBSD: src/sys/netinet/ip_dummynet.h,v 1.10.2.9 2003/05/13 09:31:06 maxim Exp $
 * $DragonFly: src/sys/net/dummynet/ip_dummynet.h,v 1.19 2008/09/20 04:36:51 sephe Exp $
 */

#ifndef _IP_DUMMYNET_H
#define _IP_DUMMYNET_H

/*
 * We start with a heap, which is used in the scheduler to decide when to
 * transmit packets etc.
 *
 * The key for the heap is used for two different values:
 *
 * 1. Timer ticks- max 10K/second, so 32 bits are enough;
 *
 * 2. Virtual times.  These increase in steps of len/x, where len is the
 *    packet length, and x is either the weight of the flow, or the sum
 *    of all weights.
 *    If we limit to max 1000 flows and a max weight of 100, then x needs
 *    17 bits.  The packet size is 16 bits, so we can easily overflow if
 *    we do not allow errors.
 *
 * So we use a key "dn_key" which is 64 bits.
 *
 * MY_M is used as a shift count when doing fixed point arithmetic
 * (a better name would be useful...).
 */
typedef uint64_t        dn_key; /* sorting key */

/*
 * Number of left shift to obtain a larger precision
 *
 * XXX With this scaling, max 1000 flows, max weight 100, 1Gbit/s, the
 * virtual time wraps every 15 days.
 */
#define MY_M            16

#ifdef _KERNEL

/*
 * A heap entry is made of a key and a pointer to the actual object stored
 * in the heap.
 *
 * The heap is an array of dn_heap_entry entries, dynamically allocated.
 * Current size is "size", with "elements" actually in use.
 *
 * The heap normally supports only ordered insert and extract from the top.
 * If we want to extract an object from the middle of the heap, we have to
 * know where the object itself is located in the heap (or we need to scan
 * the whole array).  To this purpose, an object has a field (int) which
 * contains the index of the object itself into the heap.  When the object
 * is moved, the field must also be updated.  The offset of the index in the
 * object is stored in the 'offset' field in the heap descriptor.  The
 * assumption is that this offset is non-zero if we want to support extract
 * from the middle.
 */
struct dn_heap_entry {
    dn_key key;         /* sorting key.  Topmost element is smallest one */
    void *object;       /* object pointer */
};

struct dn_heap {
    int size;
    int elements;
    int offset; /* XXX if > 0 this is the offset of direct ptr to obj */
    struct dn_heap_entry *p;    /* really an array of "size" entries */
};

struct dn_flow_id {
    uint16_t fid_type;  /* ETHERTYPE_ */
    uint16_t pad;
    union {
        struct {
            uint32_t dst_ip;
            uint32_t src_ip;
            uint16_t dst_port;
            uint16_t src_port;
            uint8_t proto;
            uint8_t flags;
        } inet;
    } fid_u;
#define fid_dst_ip      fid_u.inet.dst_ip
#define fid_src_ip      fid_u.inet.src_ip
#define fid_dst_port    fid_u.inet.dst_port
#define fid_src_port    fid_u.inet.src_port
#define fid_proto       fid_u.inet.proto
#define fid_flags       fid_u.inet.flags
};

typedef void    (*ip_dn_unref_priv_t)(void *);
struct lwkt_port;

/*
 * struct dn_pkt identifies a packet in the dummynet queue, but is also used
 * to tag packets passed back to the various destinations (ip_input(),
 * ip_output() and so on).
 *
 * It is a tag (PACKET_TAG_DUMMYNET) associated with the actual mbuf.
 */
struct dn_pkt {
    struct mbuf *dn_m;
    TAILQ_ENTRY(dn_pkt) dn_next;

    void *dn_priv;
    ip_dn_unref_priv_t dn_unref_priv;

    uint32_t dn_flags;          /* action when packet comes out. */
#define DN_FLAGS_IS_PIPE        0x10
#define DN_FLAGS_DIR_MASK       0x0f
#define DN_TO_IP_OUT            1
#define DN_TO_IP_IN             2
#define DN_TO_ETH_DEMUX         4
#define DN_TO_ETH_OUT           5
#define DN_TO_MAX               6

    dn_key output_time;         /* when the pkt is due for delivery */
    struct ifnet *ifp;          /* interface, for ip_output */
    struct sockaddr_in *dn_dst;
    struct route ro;            /* route, for ip_output. MUST COPY */
    int flags;                  /* flags, for ip_output (IPv6 ?) */

    u_short pipe_nr;            /* pipe/flow_set number */
    u_short pad;

    struct dn_flow_id id;       /* flow id */
    int cpuid;                  /* target cpuid, for assertion */
    struct lwkt_port *msgport;  /* target msgport */
};
TAILQ_HEAD(dn_pkt_queue, dn_pkt);

/*
 * Overall structure of dummynet (with WF2Q+):
 *
 * In dummynet, packets are selected with the firewall rules, and passed to
 * two different objects: PIPE or QUEUE.
 *
 * A QUEUE is just a queue with configurable size and queue management policy.
 * It is also associated with a mask (to discriminate among different flows),
 * a weight (used to give different shares of the bandwidth to different flows)
 * and a "pipe", which essentially supplies the transmit clock for all queues
 * associated with that pipe.
 *
 * A PIPE emulates a fixed-bandwidth link, whose bandwidth is configurable.
 * The "clock" for a pipe comes from an internal timer.  A pipe is also
 * associated with one (or more, if masks are used) queue, where all packets
 * for that pipe are stored.
 *
 * The bandwidth available on the pipe is shared by the queues associated with
 * that pipe (only one in case the packet is sent to a PIPE) according to the
 * WF2Q+ scheduling algorithm and the configured weights.
 *
 * In general, incoming packets are stored in the appropriate queue, which is
 * then placed into one of a few heaps managed by a scheduler to decide when
 * the packet should be extracted.  The scheduler (a function called dummynet())
 * is run at every timer tick, and grabs queues from the head of the heaps when
 * they are ready for processing.
 *
 * There are three data structures definining a pipe and associated queues:
 *
 *  + dn_pipe, which contains the main configuration parameters related to
 *    delay and bandwidth;
 *  + dn_flow_set, which contains WF2Q+ configuration, flow masks, plr and
 *    RED configuration;
 *  + dn_flow_queue, which is the per-flow queue (containing the packets)
 *
 * Multiple dn_flow_set can be linked to the same pipe, and multiple
 * dn_flow_queue can be linked to the same dn_flow_set.
 * All data structures are linked in a linear list which is used for
 * housekeeping purposes.
 *
 * During configuration, we create and initialize the dn_flow_set and dn_pipe
 * structures (a dn_pipe also contains a dn_flow_set).
 *
 * At runtime: packets are sent to the appropriate dn_flow_set (either WFQ
 * ones, or the one embedded in the dn_pipe for fixed-rate flows), which in
 * turn dispatches them to the appropriate dn_flow_queue (created dynamically
 * according to the masks).
 *
 * The transmit clock for fixed rate flows (ready_event()) selects the
 * dn_flow_queue to be used to transmit the next packet. For WF2Q,
 * wfq_ready_event() extract a pipe which in turn selects the right flow using
 * a number of heaps defined into the pipe itself.
 */

/*
 * Per flow queue.  This contains the flow identifier, the queue of packets,
 * counters, and parameters used to support both RED and WF2Q+.
 *
 * A dn_flow_queue is created and initialized whenever a packet for a new
 * flow arrives.
 */
struct dn_flow_queue {
    struct dn_flow_id id;
    LIST_ENTRY(dn_flow_queue) q_link;

    struct dn_pkt_queue queue;  /* queue of packets */
    u_int len;
    u_int len_bytes;
    u_long numbytes;            /* credit for transmission (dynamic queues) */

    uint64_t tot_pkts;          /* statistics counters */
    uint64_t tot_bytes;
    uint32_t drops;

    int hash_slot;              /* debugging/diagnostic */

    /* RED parameters */
    int avg;                    /* average queue length est. (scaled) */
    int count;                  /* arrivals since last RED drop */
    int random;                 /* random value (scaled) */
    uint32_t q_time;            /* start of queue idle time */

    /* WF2Q+ support */
    struct dn_flow_set *fs;     /* parent flow set */
    int heap_pos;               /* position (index) of struct in heap */
    dn_key sched_time;          /* current time when queue enters ready_heap */

    dn_key S, F;                /* start time, finish time */
    /*
     * Setting F < S means the timestamp is invalid. We only need
     * to test this when the queue is empty.
     */
};
LIST_HEAD(dn_flowqueue_head, dn_flow_queue);

/*
 * flow_set descriptor.  Contains the "template" parameters for the queue
 * configuration, and pointers to the hash table of dn_flow_queue's.
 *
 * The hash table is an array of lists -- we identify the slot by hashing
 * the flow-id, then scan the list looking for a match.
 * The size of the hash table (buckets) is configurable on a per-queue basis.
 *
 * A dn_flow_set is created whenever a new queue or pipe is created (in the
 * latter case, the structure is located inside the struct dn_pipe).
 */
struct dn_flow_set {
    u_short fs_nr;              /* flow_set number */
    u_short flags_fs;           /* see 'Flow set flags' */

    LIST_ENTRY(dn_flow_set) fs_link;

    struct dn_pipe *pipe;       /* pointer to parent pipe */
    u_short parent_nr;          /* parent pipe#, 0 if local to a pipe */

    int weight;                 /* WFQ queue weight */
    int qsize;                  /* queue size in slots or bytes */
    int plr;                    /* pkt loss rate (2^31-1 means 100%) */

    struct dn_flow_id flow_mask;

    /* hash table of queues onto this flow_set */
    int rq_size;                /* number of slots */
    int rq_elements;            /* active elements */
    struct dn_flowqueue_head *rq;/* array of rq_size entries */

    uint32_t last_expired;      /* do not expire too frequently */
    int backlogged;             /* #active queues for this flowset */

    /* RED parameters */
    int w_q;                    /* queue weight (scaled) */
    int max_th;                 /* maximum threshold for queue (scaled) */
    int min_th;                 /* minimum threshold for queue (scaled) */
    int max_p;                  /* maximum value for p_b (scaled) */
    u_int c_1;                  /* max_p/(max_th-min_th) (scaled) */
    u_int c_2;                  /* max_p*min_th/(max_th-min_th) (scaled) */
    u_int c_3;                  /* for GRED, (1-max_p)/max_th (scaled) */
    u_int c_4;                  /* for GRED, 1 - 2*max_p (scaled) */
    u_int *w_q_lookup;          /* lookup table for computing (1-w_q)^t */
    u_int lookup_depth;         /* depth of lookup table */
    int lookup_step;            /* granularity inside the lookup table */
    int lookup_weight;          /* equal to (1-w_q)^t / (1-w_q)^(t+1) */
    int avg_pkt_size;           /* medium packet size */
    int max_pkt_size;           /* max packet size */
};
LIST_HEAD(dn_flowset_head, dn_flow_set);

/*
 * Pipe descriptor. Contains global parameters, delay-line queue, and the
 * flow_set used for fixed-rate queues.
 *
 * For WF2Q+ support it also has 3 heaps holding dn_flow_queue:
 *  + not_eligible_heap, for queues whose start time is higher than the
 *    virtual time. Sorted by start time.
 *  + scheduler_heap, for queues eligible for scheduling.  Sorted by finish
 *    time.
 *  + idle_heap, all flows that are idle and can be removed.  We do that on
 *    each tick so we do not slow down too much operations during forwarding.
 */
struct dn_pipe {                /* a pipe */
    int pipe_nr;                /* number */
    int bandwidth;              /* really, bytes/tick. */
    int delay;                  /* really, ticks */

    struct dn_pkt_queue p_queue;/* packets in delay line */
    LIST_ENTRY(dn_pipe) p_link;

    /* WF2Q+ */
    struct dn_heap scheduler_heap; /* top extract - key Finish time*/
    struct dn_heap not_eligible_heap; /* top extract- key Start time */
    struct dn_heap idle_heap;   /* random extract - key Start=Finish time */

    dn_key V;                   /* virtual time */
    int sum;                    /* sum of weights of all active sessions */
    int numbytes;               /* bits I can transmit (more or less). */

    dn_key sched_time;          /* time pipe was scheduled in ready_heap */

    struct dn_flow_set fs;      /* used with fixed-rate flows */
};
LIST_HEAD(dn_pipe_head, dn_pipe);

struct dn_sopt {
        int     dn_sopt_name;
        void    *dn_sopt_arg;
        size_t  dn_sopt_arglen;
};

typedef int     ip_dn_ctl_t(struct dn_sopt *);
typedef int     ip_dn_io_t(struct mbuf *);

extern ip_dn_ctl_t      *ip_dn_ctl_ptr;
extern ip_dn_io_t       *ip_dn_io_ptr;

void    ip_dn_queue(struct mbuf *);
void    ip_dn_packet_free(struct dn_pkt *);
void    ip_dn_packet_redispatch(struct dn_pkt *);
int     ip_dn_sockopt(struct sockopt *);

#define DUMMYNET_LOADED (ip_dn_io_ptr != NULL)

#endif  /* _KERNEL */

struct dn_ioc_flowid {
    uint16_t type;      /* ETHERTYPE_ */
    uint16_t pad;
    union {
        struct {
            uint32_t dst_ip;
            uint32_t src_ip;
            uint16_t dst_port;
            uint16_t src_port;
            uint8_t proto;
            uint8_t flags;
        } ip;
        uint8_t pad[64];
    } u;
};

struct dn_ioc_flowqueue {
    u_int len;
    u_int len_bytes;

    uint64_t tot_pkts;
    uint64_t tot_bytes;
    uint32_t drops;

    int hash_slot;              /* debugging/diagnostic */
    dn_key S;                   /* virtual start time */
    dn_key F;                   /* virtual finish time */

    struct dn_ioc_flowid id;
    uint8_t reserved[16];
};

struct dn_ioc_flowset {
    u_short fs_type;            /* DN_IS_{QUEUE,PIPE}, MUST be first */

    u_short fs_nr;              /* flow_set number */
    u_short flags_fs;           /* see 'Flow set flags' */
    u_short parent_nr;          /* parent pipe#, 0 if local to a pipe */

    int weight;                 /* WFQ queue weight */
    int qsize;                  /* queue size in slots or bytes */
    int plr;                    /* pkt loss rate (2^31-1 means 100%) */

    /* Hash table information */
    int rq_size;                /* number of slots */
    int rq_elements;            /* active elements */

    /* RED parameters */
    int w_q;                    /* queue weight (scaled) */
    int max_th;                 /* maximum threshold for queue (scaled) */
    int min_th;                 /* minimum threshold for queue (scaled) */
    int max_p;                  /* maximum value for p_b (scaled) */
    int lookup_step;            /* granularity inside the lookup table */
    int lookup_weight;          /* equal to (1-w_q)^t / (1-w_q)^(t+1) */

    struct dn_ioc_flowid flow_mask;
    uint8_t reserved[16];
};

struct dn_ioc_pipe {
    struct dn_ioc_flowset fs;   /* MUST be first */

    int pipe_nr;                /* pipe number */
    int bandwidth;              /* bit/second */
    int delay;                  /* milliseconds */

    dn_key V;                   /* virtual time */

    uint8_t reserved[16];
};

/*
 * Flow set flags
 */
#define DN_HAVE_FLOW_MASK       0x0001
#define DN_IS_RED               0x0002
#define DN_IS_GENTLE_RED        0x0004
#define DN_QSIZE_IS_BYTES       0x0008  /* queue size is measured in bytes */
#define DN_NOERROR              0x0010  /* do not report ENOBUFS on drops */
#define DN_IS_PIPE              0x4000
#define DN_IS_QUEUE             0x8000

/*
 * Macros for RED
 */
#define SCALE_RED               16
#define SCALE(x)                ((x) << SCALE_RED)
#define SCALE_VAL(x)            ((x) >> SCALE_RED)
#define SCALE_MUL(x, y)         (((x) * (y)) >> SCALE_RED)

/*
 * Maximum pipe number
 */
#define DN_PIPE_NR_MAX          65536

#endif /* !_IP_DUMMYNET_H */