linux-imx

/*

 * net/sched/act_sample.c - Packet sampling tc action

 * Copyright (c) 2017 Yotam Gigi <yotamg@mellanox.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation.

 */

​

#include <linux/types.h>

#include <linux/kernel.h>

#include <linux/string.h>

#include <linux/errno.h>

#include <linux/skbuff.h>

#include <linux/rtnetlink.h>

#include <linux/module.h>

#include <linux/init.h>

#include <linux/gfp.h>

#include <net/net_namespace.h>

#include <net/netlink.h>

#include <net/pkt_sched.h>

#include <linux/tc_act/tc_sample.h>

#include <net/tc_act/tc_sample.h>

#include <net/psample.h>

#include <net/pkt_cls.h>

​

#include <linux/if_arp.h>

​

static unsigned int sample_net_id;

static struct tc_action_ops act_sample_ops;

​

static const struct nla_policy sample_policy[TCA_SAMPLE_MAX + 1] = {

    [TCA_SAMPLE_PARMS]      = { .len = sizeof(struct tc_sample) },

    [TCA_SAMPLE_RATE]       = { .type = NLA_U32 },

    [TCA_SAMPLE_TRUNC_SIZE]     = { .type = NLA_U32 },

    [TCA_SAMPLE_PSAMPLE_GROUP]  = { .type = NLA_U32 },

};

​

static int tcf_sample_init(struct net *net, struct nlattr *nla,

               struct nlattr *est, struct tc_action **a, int ovr,

               int bind, bool rtnl_held, struct tcf_proto *tp,

               struct netlink_ext_ack *extack)

{

    struct tc_action_net *tn = net_generic(net, sample_net_id);

    struct nlattr *tb[TCA_SAMPLE_MAX + 1];

    struct psample_group *psample_group;

    struct tcf_chain *goto_ch = NULL;

    u32 psample_group_num, rate;

    struct tc_sample *parm;

    struct tcf_sample *s;

    bool exists = false;

    int ret, err;

​

    if (!nla)

        return -EINVAL;

    ret = nla_parse_nested(tb, TCA_SAMPLE_MAX, nla, sample_policy, NULL);

    if (ret < 0)

        return ret;

    if (!tb[TCA_SAMPLE_PARMS] || !tb[TCA_SAMPLE_RATE] ||

        !tb[TCA_SAMPLE_PSAMPLE_GROUP])

        return -EINVAL;

​

    parm = nla_data(tb[TCA_SAMPLE_PARMS]);