route:act: add vlan action

For more information please see:
http://man7.org/linux/man-pages/man8/tc-vlan.8.html

Signed-off-by: Volodymyr Bendiuga <volodymyr.bendiuga@westermo.se>

1 files changed, 416 insertions, 0 deletions

diff --git a/lib/route/act/vlan.c b/lib/route/act/vlan.c
new file mode 100644
index 00000000..e570ae38
--- /dev/null
+++ b/lib/route/act/vlan.c
@@ -0,0 +1,416 @@
+/*
+ * lib/route/act/vlan.c        vlan 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) 2018 Volodymyr Bendiuga <volodymyr.bendiuga@gmail.com>
+ */
+
+/**
+ * @ingroup act
+ * @defgroup act_vlan VLAN Manipulation
+ *
+ * @{
+ */
+
+#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/vlan.h>
+
+
+#define VLAN_F_VID   1 << 0
+#define VLAN_F_PROTO 1 << 1
+#define VLAN_F_PRIO  1 << 2
+#define VLAN_F_ACT   1 << 3
+#define VLAN_F_MODE  1 << 4
+
+static struct nla_policy vlan_policy[TCA_VLAN_MAX + 1] = {
+	[TCA_VLAN_PARMS]		= { .minlen = sizeof(struct tc_vlan) },
+	[TCA_VLAN_PUSH_VLAN_ID]		= { .type = NLA_U16 },
+	[TCA_VLAN_PUSH_VLAN_PROTOCOL]	= { .type = NLA_U16 },
+	[TCA_VLAN_PUSH_VLAN_PRIORITY]	= { .type = NLA_U8 },
+};
+
+static int vlan_msg_parser(struct rtnl_tc *tc, void *data)
+{
+        struct rtnl_vlan *v = data;
+	struct nlattr *tb[TCA_VLAN_MAX + 1];
+	int err;
+
+	err = tca_parse(tb, TCA_VLAN_MAX, tc, vlan_policy);
+	if (err < 0)
+		return err;
+
+	v->v_flags = 0;
+	if (!tb[TCA_VLAN_PARMS])
+		return -NLE_MISSING_ATTR;
+	else {
+	        nla_memcpy(&v->v_parm, tb[TCA_VLAN_PARMS], sizeof(v->v_parm));
+		v->v_flags |= VLAN_F_ACT;
+		v->v_flags |= VLAN_F_MODE;
+	}
+
+	if (tb[TCA_VLAN_PUSH_VLAN_ID]) {
+	        v->v_vid = nla_get_u16(tb[TCA_VLAN_PUSH_VLAN_ID]);
+		v->v_flags |= VLAN_F_VID;
+	}
+
+	if (tb[TCA_VLAN_PUSH_VLAN_PROTOCOL]) {
+	        v->v_proto = nla_get_u16(tb[TCA_VLAN_PUSH_VLAN_PROTOCOL]);
+		v->v_flags |= VLAN_F_PROTO;
+	}
+
+	if (tb[TCA_VLAN_PUSH_VLAN_PRIORITY]) {
+	        v->v_prio = nla_get_u8(tb[TCA_VLAN_PUSH_VLAN_PRIORITY]);
+		v->v_flags |= VLAN_F_PRIO;
+	}
+
+        return 0;
+}
+
+static int vlan_msg_fill(struct rtnl_tc *tc, void *data, struct nl_msg *msg)
+{
+        struct rtnl_vlan *v = data;
+
+	if (!v)
+	        return 0;
+	if (!(v->v_flags & VLAN_F_MODE))
+	        return -NLE_MISSING_ATTR;
+
+	NLA_PUT(msg, TCA_VLAN_PARMS, sizeof(v->v_parm), &v->v_parm);
+
+	/* vid is required for PUSH & MODIFY modes */
+	if ((v->v_parm.v_action != TCA_VLAN_ACT_POP) && !(v->v_flags & VLAN_F_VID))
+	        return -NLE_MISSING_ATTR;
+
+	if (v->v_flags & VLAN_F_VID)
+		NLA_PUT_U16(msg, TCA_VLAN_PUSH_VLAN_ID, v->v_vid);
+
+	if (v->v_flags & VLAN_F_PROTO)
+		NLA_PUT_U16(msg, TCA_VLAN_PUSH_VLAN_PROTOCOL, v->v_proto);
+
+	if (v->v_flags & VLAN_F_PRIO)
+		NLA_PUT_U8(msg, TCA_VLAN_PUSH_VLAN_PRIORITY, v->v_prio);
+
+        return 0;
+
+ nla_put_failure:
+	return -NLE_NOMEM;
+}
+
+static void vlan_free_data(struct rtnl_tc *tc, void *data)
+{
+}
+
+static int vlan_clone(void *_dst, void *_src)
+{
+        struct rtnl_vlan *dst = _dst, *src = _src;
+
+	memcpy(&dst->v_parm, &src->v_parm, sizeof(src->v_parm));
+	return 0;
+}
+
+static void vlan_dump_line(struct rtnl_tc *tc, void *data,
+			   struct nl_dump_params *p)
+{
+	struct rtnl_vlan *v = data;
+
+	if (!v)
+		return;
+
+	if (!(v->v_flags & VLAN_F_ACT))
+	        return;
+
+	if (TC_ACT_EXT_CMP(v->v_parm.action, TC_ACT_GOTO_CHAIN))
+	        nl_dump(p, " goto chain %u", v->v_parm.action & TC_ACT_EXT_VAL_MASK);
+
+	if (TC_ACT_EXT_CMP(v->v_parm.action, TC_ACT_JUMP))
+	        nl_dump(p, " jump %u", v->v_parm.action & TC_ACT_EXT_VAL_MASK);
+
+	switch(v->v_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 vlan_dump_details(struct rtnl_tc *tc, void *data,
+			      struct nl_dump_params *p)
+{
+        struct rtnl_vlan *v = data;
+
+	if (!v)
+		return;
+
+	if (v->v_flags & VLAN_F_MODE) {
+	        switch (v->v_parm.v_action) {
+		case TCA_VLAN_ACT_POP:
+		        nl_dump(p, " mode POP");
+			break;
+		case TCA_VLAN_ACT_PUSH:
+	                nl_dump(p, " mode PUSH");
+			break;
+		case TCA_VLAN_ACT_MODIFY:
+	                nl_dump(p, " mode MODIFY");
+			break;
+		}
+	}
+
+	if (v->v_flags & VLAN_F_VID)
+	        nl_dump(p, " vlan id %u", v->v_vid);
+
+	if (v->v_flags & VLAN_F_PRIO)
+		nl_dump(p, " priority %u", v->v_prio);
+
+	if (v->v_flags & VLAN_F_PROTO)
+		nl_dump(p, " protocol %u", v->v_proto);
+}
+
+/**
+ * @name Attribute Modifications
+ * @{
+ */
+
+/**
+ * Set vlan mode
+ * @arg act		vlan action
+ * @arg mode		one of (TCA_VLAN_ACT_*: POP, PUSH, MODIFY)
+ * @return 0 on success or a negative error code.
+ */
+int rtnl_vlan_set_mode(struct rtnl_act *act, int mode)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (mode > TCA_VLAN_ACT_MODIFY)
+	        return -NLE_RANGE;
+
+	v->v_parm.v_action = mode;
+	v->v_flags |= VLAN_F_MODE;
+
+	return 0;
+}
+
+/**
+ * Get vlan mode
+ * @arg act	        vlan action
+ * @return vlan mode on success or a negative error code.
+*/
+int rtnl_vlan_get_mode(struct rtnl_act *act)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data_peek(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (!(v->v_flags & VLAN_F_MODE))
+	        return -NLE_MISSING_ATTR;
+
+	return v->v_parm.v_action;
+}
+
+/**
+ * Set general action
+ * @arg act		vlan action
+ * @arg action		one of (TCA_ACT_*: PIPE, SHOT, GOTO_CHAIN, etc)
+ * @return 0 on success or a negative error code.
+ */
+int rtnl_vlan_set_action(struct rtnl_act *act, int action)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	v->v_parm.action = action;
+	v->v_flags |= VLAN_F_ACT;
+
+	return 0;
+}
+
+/**
+ * Get general action
+ * @arg act	        vlan action
+ * @return general action on success or a negative error code.
+*/
+int rtnl_vlan_get_action(struct rtnl_act *act)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data_peek(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (!(v->v_flags & VLAN_F_ACT))
+	        return -NLE_MISSING_ATTR;
+
+	return v->v_parm.action;
+}
+
+/**
+ * Set protocol
+ * @arg act		vlan action
+ * @arg protocol	one of (ETH_P_8021Q || ETH_P_8021AD)
+ * @return 0 on success or a negative error code.
+ */
+int rtnl_vlan_set_protocol(struct rtnl_act *act, uint16_t protocol)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	v->v_proto = protocol;
+	v->v_flags |= VLAN_F_PROTO;
+
+	return 0;
+}
+
+/**
+ * Get protocol
+ * @arg act	        vlan action
+ * @return protocol on success or a negative error code.
+*/
+uint16_t rtnl_vlan_get_protocol(struct rtnl_act *act)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data_peek(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (!(v->v_flags & VLAN_F_PROTO))
+	        return -NLE_MISSING_ATTR;
+
+	return v->v_proto;
+}
+
+/**
+ * Set vlan id
+ * @arg act		vlan action
+ * @arg vid	        vlan id
+ * @return 0 on success or a negative error code.
+ */
+int rtnl_vlan_set_vlan_id(struct rtnl_act *act, uint16_t vid)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (vid > 4095)
+	        return -NLE_RANGE;
+
+	v->v_vid = vid;
+	v->v_flags |= VLAN_F_VID;
+
+	return 0;
+}
+
+/**
+ * Get vlan id
+ * @arg act	        vlan action
+ * @return vlan id on success or a negative error code.
+*/
+uint16_t rtnl_vlan_get_vlan_id(struct rtnl_act *act)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data_peek(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (!(v->v_flags & VLAN_F_VID))
+	        return -NLE_MISSING_ATTR;
+
+	return v->v_vid;
+}
+
+/**
+ * Set vlan prio
+ * @arg act		vlan action
+ * @arg prio	        vlan priority (0 - 7)
+ * @return 0 on success or a negative error code.
+ */
+int rtnl_vlan_set_vlan_prio(struct rtnl_act *act, uint8_t prio)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (prio > 7)
+	        return -NLE_RANGE;
+
+	v->v_prio = prio;
+	v->v_flags |= VLAN_F_PRIO;
+
+	return 0;
+}
+
+/**
+ * Get vlan prio
+ * @arg act	        vlan action
+ * @return vlan prio on success or a negative error code.
+*/
+uint8_t rtnl_vlan_get_vlan_prio(struct rtnl_act *act)
+{
+        struct rtnl_vlan *v;
+
+	if (!(v = (struct rtnl_vlan *) rtnl_tc_data_peek(TC_CAST(act))))
+		return -NLE_NOMEM;
+
+	if (!(v->v_flags & VLAN_F_PRIO))
+	        return -NLE_MISSING_ATTR;
+
+	return v->v_prio;
+}
+
+/** @} */
+
+static struct rtnl_tc_ops vlan_ops = {
+	.to_kind		= "vlan",
+	.to_type		= RTNL_TC_TYPE_ACT,
+	.to_size		= sizeof(struct rtnl_vlan),
+	.to_msg_parser		= vlan_msg_parser,
+	.to_free_data		= vlan_free_data,
+	.to_clone		= vlan_clone,
+	.to_msg_fill		= vlan_msg_fill,
+	.to_dump = {
+	    [NL_DUMP_LINE]	= vlan_dump_line,
+	    [NL_DUMP_DETAILS]	= vlan_dump_details,
+	},
+};
+
+static void __init vlan_init(void)
+{
+	rtnl_tc_register(&vlan_ops);
+}
+
+static void __exit vlan_exit(void)
+{
+	rtnl_tc_unregister(&vlan_ops);
+}
+
+/** @} */