You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
291 lines
6.1 KiB
291 lines
6.1 KiB
/*
|
|
* lib/route/act/skbedit.c skbedit action
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* Copyright (c) 2015 Cong Wang <xiyou.wangcong@gmail.com>
|
|
*/
|
|
|
|
/**
|
|
* @ingroup act
|
|
* @defgroup act_skbedit SKB Editing
|
|
*
|
|
* @{
|
|
*/
|
|
|
|
#include <netlink-private/netlink.h>
|
|
#include <netlink-private/tc.h>
|
|
#include <netlink/netlink.h>
|
|
#include <netlink/attr.h>
|
|
#include <netlink/utils.h>
|
|
#include <netlink-private/route/tc-api.h>
|
|
#include <netlink/route/act/skbedit.h>
|
|
|
|
static struct nla_policy skbedit_policy[TCA_SKBEDIT_MAX + 1] = {
|
|
[TCA_SKBEDIT_PARMS] = { .minlen = sizeof(struct tc_skbedit) },
|
|
[TCA_SKBEDIT_PRIORITY] = { .type = NLA_U32 },
|
|
[TCA_SKBEDIT_QUEUE_MAPPING] = { .type = NLA_U16 },
|
|
[TCA_SKBEDIT_MARK] = { .type = NLA_U32 },
|
|
};
|
|
|
|
static int skbedit_msg_parser(struct rtnl_tc *tc, void *data)
|
|
{
|
|
struct rtnl_skbedit *u = data;
|
|
struct nlattr *tb[TCA_SKBEDIT_MAX + 1];
|
|
int err;
|
|
|
|
err = tca_parse(tb, TCA_SKBEDIT_MAX, tc, skbedit_policy);
|
|
if (err < 0)
|
|
return err;
|
|
|
|
if (!tb[TCA_SKBEDIT_PARMS])
|
|
return -NLE_MISSING_ATTR;
|
|
|
|
u->s_flags = 0;
|
|
if (tb[TCA_SKBEDIT_PRIORITY] != NULL) {
|
|
u->s_flags |= SKBEDIT_F_PRIORITY;
|
|
u->s_prio = nla_get_u32(tb[TCA_SKBEDIT_PRIORITY]);
|
|
}
|
|
|
|
if (tb[TCA_SKBEDIT_QUEUE_MAPPING] != NULL) {
|
|
u->s_flags |= SKBEDIT_F_QUEUE_MAPPING;
|
|
u->s_queue_mapping = nla_get_u16(tb[TCA_SKBEDIT_QUEUE_MAPPING]);
|
|
}
|
|
|
|
if (tb[TCA_SKBEDIT_MARK] != NULL) {
|
|
u->s_flags |= SKBEDIT_F_MARK;
|
|
u->s_mark = nla_get_u32(tb[TCA_SKBEDIT_MARK]);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void skbedit_free_data(struct rtnl_tc *tc, void *data)
|
|
{
|
|
}
|
|
|
|
static int skbedit_clone(void *_dst, void *_src)
|
|
{
|
|
struct rtnl_skbedit *dst = _dst, *src = _src;
|
|
|
|
memcpy(dst, src, sizeof(*src));
|
|
return 0;
|
|
}
|
|
|
|
static void skbedit_dump_line(struct rtnl_tc *tc, void *data,
|
|
struct nl_dump_params *p)
|
|
{
|
|
struct rtnl_skbedit *u = data;
|
|
|
|
if (!u)
|
|
return;
|
|
|
|
if (u->s_flags & SKBEDIT_F_PRIORITY)
|
|
nl_dump(p, " priority %u", u->s_prio);
|
|
|
|
if (u->s_flags & SKBEDIT_F_MARK)
|
|
nl_dump(p, " mark %u", u->s_mark);
|
|
|
|
if (u->s_flags & SKBEDIT_F_QUEUE_MAPPING)
|
|
nl_dump(p, " queue_mapping %u", u->s_queue_mapping);
|
|
|
|
switch(u->s_parm.action){
|
|
case TC_ACT_UNSPEC:
|
|
nl_dump(p, " unspecified");
|
|
break;
|
|
case TC_ACT_PIPE:
|
|
nl_dump(p, " pipe");
|
|
break;
|
|
case TC_ACT_STOLEN:
|
|
nl_dump(p, " stolen");
|
|
break;
|
|
case TC_ACT_SHOT:
|
|
nl_dump(p, " shot");
|
|
break;
|
|
case TC_ACT_QUEUED:
|
|
nl_dump(p, " queued");
|
|
break;
|
|
case TC_ACT_REPEAT:
|
|
nl_dump(p, " repeat");
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void skbedit_dump_details(struct rtnl_tc *tc, void *data,
|
|
struct nl_dump_params *p)
|
|
{
|
|
}
|
|
|
|
static void skbedit_dump_stats(struct rtnl_tc *tc, void *data,
|
|
struct nl_dump_params *p)
|
|
{
|
|
struct rtnl_skbedit *u = data;
|
|
|
|
if (!u)
|
|
return;
|
|
/* TODO */
|
|
}
|
|
|
|
|
|
static int skbedit_msg_fill(struct rtnl_tc *tc, void *data, struct nl_msg *msg)
|
|
{
|
|
struct rtnl_skbedit *u = data;
|
|
|
|
if (!u)
|
|
return 0;
|
|
|
|
NLA_PUT(msg, TCA_SKBEDIT_PARMS, sizeof(u->s_parm), &u->s_parm);
|
|
|
|
if (u->s_flags & SKBEDIT_F_MARK)
|
|
NLA_PUT_U32(msg, TCA_SKBEDIT_MARK, u->s_mark);
|
|
|
|
if (u->s_flags & SKBEDIT_F_PRIORITY)
|
|
NLA_PUT_U32(msg, TCA_SKBEDIT_PRIORITY, u->s_prio);
|
|
|
|
if (u->s_flags & SKBEDIT_F_QUEUE_MAPPING)
|
|
NLA_PUT_U32(msg, TCA_SKBEDIT_QUEUE_MAPPING, u->s_queue_mapping);
|
|
|
|
return 0;
|
|
|
|
nla_put_failure:
|
|
return -NLE_NOMEM;
|
|
}
|
|
|
|
/**
|
|
* @name Attribute Modifications
|
|
* @{
|
|
*/
|
|
|
|
int rtnl_skbedit_set_action(struct rtnl_act *act, int action)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
if (!(u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act))))
|
|
return -NLE_NOMEM;
|
|
|
|
if (action > TC_ACT_REPEAT || action < TC_ACT_UNSPEC)
|
|
return -NLE_INVAL;
|
|
|
|
u->s_parm.action = action;
|
|
return 0;
|
|
}
|
|
|
|
int rtnl_skbedit_get_action(struct rtnl_act *act)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
if (!(u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act))))
|
|
return -NLE_NOMEM;
|
|
return u->s_parm.action;
|
|
}
|
|
|
|
int rtnl_skbedit_set_queue_mapping(struct rtnl_act *act, uint16_t index)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
if (!(u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act))))
|
|
return -NLE_NOMEM;
|
|
|
|
u->s_queue_mapping = index;
|
|
u->s_flags |= SKBEDIT_F_QUEUE_MAPPING;
|
|
return 0;
|
|
}
|
|
|
|
int rtnl_skbedit_get_queue_mapping(struct rtnl_act *act, uint16_t *index)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act));
|
|
if (!u)
|
|
return -NLE_NOMEM;
|
|
if (!(u->s_flags & SKBEDIT_F_QUEUE_MAPPING))
|
|
return -NLE_NOATTR;
|
|
|
|
*index = u->s_queue_mapping;
|
|
return 0;
|
|
}
|
|
|
|
int rtnl_skbedit_set_mark(struct rtnl_act *act, uint32_t mark)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
if (!(u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act))))
|
|
return -NLE_NOMEM;
|
|
|
|
u->s_mark = mark;
|
|
u->s_flags |= SKBEDIT_F_MARK;
|
|
return 0;
|
|
}
|
|
|
|
int rtnl_skbedit_get_mark(struct rtnl_act *act, uint32_t *mark)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act));
|
|
if (!u)
|
|
return -NLE_NOMEM;
|
|
if (!(u->s_flags & SKBEDIT_F_MARK))
|
|
return -NLE_NOATTR;
|
|
|
|
*mark = u->s_mark;
|
|
return 0;
|
|
}
|
|
|
|
int rtnl_skbedit_set_priority(struct rtnl_act *act, uint32_t prio)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
if (!(u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act))))
|
|
return -NLE_NOMEM;
|
|
|
|
u->s_prio = prio;
|
|
u->s_flags |= SKBEDIT_F_PRIORITY;
|
|
return 0;
|
|
}
|
|
|
|
int rtnl_skbedit_get_priority(struct rtnl_act *act, uint32_t *prio)
|
|
{
|
|
struct rtnl_skbedit *u;
|
|
|
|
u = (struct rtnl_skbedit *) rtnl_tc_data(TC_CAST(act));
|
|
if (!u)
|
|
return -NLE_NOMEM;
|
|
if (!(u->s_flags & SKBEDIT_F_PRIORITY))
|
|
return -NLE_NOATTR;
|
|
|
|
*prio = u->s_prio;
|
|
return 0;
|
|
}
|
|
|
|
/** @} */
|
|
|
|
static struct rtnl_tc_ops skbedit_ops = {
|
|
.to_kind = "skbedit",
|
|
.to_type = RTNL_TC_TYPE_ACT,
|
|
.to_size = sizeof(struct rtnl_skbedit),
|
|
.to_msg_parser = skbedit_msg_parser,
|
|
.to_free_data = skbedit_free_data,
|
|
.to_clone = skbedit_clone,
|
|
.to_msg_fill = skbedit_msg_fill,
|
|
.to_dump = {
|
|
[NL_DUMP_LINE] = skbedit_dump_line,
|
|
[NL_DUMP_DETAILS] = skbedit_dump_details,
|
|
[NL_DUMP_STATS] = skbedit_dump_stats,
|
|
},
|
|
};
|
|
|
|
static void __init skbedit_init(void)
|
|
{
|
|
rtnl_tc_register(&skbedit_ops);
|
|
}
|
|
|
|
static void __exit skbedit_exit(void)
|
|
{
|
|
rtnl_tc_unregister(&skbedit_ops);
|
|
}
|
|
|
|
/** @} */
|