--- /dev/null
+/*
+ * 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);
+}
+
+/** @} */
projects / libnl / commitdiff