luaskb: add priority getter and setter

Signed-off-by: Ashwani Kumar Kamal <ashwanikamal.im421@gmail.com>

Author: sneaky-potato

Makefile

@@ -80,6 +80,7 @@ scripts_install:
 	${MKDIR} ${SCRIPTS_INSTALL_PATH}
 	${MKDIR} ${SCRIPTS_INSTALL_PATH}/lunatik
 	${MKDIR} ${SCRIPTS_INSTALL_PATH}/socket
+	${MKDIR} ${SCRIPTS_INSTALL_PATH}/skb
 	${MKDIR} ${SCRIPTS_INSTALL_PATH}/syscall
 	${MKDIR} ${SCRIPTS_INSTALL_PATH}/crypto
 	${MKDIR} ${SCRIPTS_INSTALL_PATH}/linux
@@ -91,6 +92,7 @@ scripts_install:
 	${INSTALL} -m 0644 lib/lighten.lua ${SCRIPTS_INSTALL_PATH}/
 	${INSTALL} -m 0644 lib/lunatik/*.lua ${SCRIPTS_INSTALL_PATH}/lunatik
 	${INSTALL} -m 0644 lib/socket/*.lua ${SCRIPTS_INSTALL_PATH}/socket
+	${INSTALL} -m 0644 lib/skb/*.lua ${SCRIPTS_INSTALL_PATH}/skb
 	${INSTALL} -m 0644 lib/syscall/*.lua ${SCRIPTS_INSTALL_PATH}/syscall
 	${INSTALL} -m 0644 lib/crypto/*.lua ${SCRIPTS_INSTALL_PATH}/crypto
 	# NOTE: `lib/linux/` exists only as LDoc stubs (see doc-stubs); never install it.

config.ld

@@ -38,6 +38,7 @@ file = {
 	'./lib/net.lua',
 	'./lib/luanetfilter.c',
 	'./lib/luaskb.c',
+	'./lib/skb/attr.lua',
 	'./lib/luanotifier.c',
 	'./lib/luaprobe.c',
 	'./lib/luarcu.c',

lib/luaskb.c

@@ -195,6 +195,55 @@ static int luaskb_forward(lua_State *L)
 	return 0;
 }
 
+static int luaskb_getmark(lua_State *L)
+{
+	luaskb_t *lskb = luaskb_check(L, 1);
+	lua_pushinteger(L, lskb->skb->mark);
+	return 1;
+}
+
+static int luaskb_setmark(lua_State *L)
+{
+	luaskb_t *lskb = luaskb_check(L, 1);
+	lskb->skb->mark = (u32)luaL_checkinteger(L, 2);
+	return 0;
+}
+
+static int luaskb_getpriority(lua_State *L)
+{
+	luaskb_t *lskb = luaskb_check(L, 1);
+	lua_pushinteger(L, lskb->skb->priority);
+	return 1;
+}
+
+static int luaskb_setpriority(lua_State *L)
+{
+	luaskb_t *lskb = luaskb_check(L, 1);
+	lskb->skb->priority = (u32)luaL_checkinteger(L, 2);
+	return 0;
+}
+
+static int luaskb_protocol(lua_State *L)
+{
+	luaskb_t *lskb = luaskb_check(L, 1);
+	lua_pushinteger(L, ntohs(lskb->skb->protocol));
+	return 1;
+}
+
+static int luaskb_proto(lua_State *L)
+{
+	luaskb_t *lskb = luaskb_check(L, 1);
+	struct sk_buff *skb = lskb->skb;
+
+	if (skb->protocol == htons(ETH_P_IP))
+		lua_pushinteger(L, ip_hdr(skb)->protocol);
+	else if (skb->protocol == htons(ETH_P_IPV6))
+		lua_pushinteger(L, ipv6_hdr(skb)->nexthdr);
+	else
+		lua_pushnil(L);
+	return 1;
+}
+
 static int luaskb_copy(lua_State *L);
 
 static void luaskb_release(void *private)
@@ -211,18 +260,25 @@ static const luaL_Reg luaskb_lib[] = {
 };
 
 static const luaL_Reg luaskb_mt[] = {
-	{"__gc",     lunatik_deleteobject},
-	{"__len",    luaskb_len},
-	{"ifindex",  luaskb_ifindex},
-	{"vlan",     luaskb_vlan},
-	{"data",     luaskb_data},
-	{"resize",   luaskb_resize},
-	{"checksum", luaskb_checksum},
-	{"forward",  luaskb_forward},
-	{"copy",     luaskb_copy},
+	{"__gc",        lunatik_deleteobject},
+	{"__len",       luaskb_len},
+	{"ifindex",     luaskb_ifindex},
+	{"vlan",        luaskb_vlan},
+	{"data",        luaskb_data},
+	{"resize",      luaskb_resize},
+	{"checksum",    luaskb_checksum},
+	{"forward",     luaskb_forward},
+	{"copy",        luaskb_copy},
+	{"protocol",    luaskb_protocol},
+	{"proto",       luaskb_proto},
+	{"getmark",     luaskb_getmark},
+	{"getpriority", luaskb_getpriority},
+	{"setmark",     luaskb_setmark},
+	{"setpriority", luaskb_setpriority},
 	{NULL, NULL}
 };
 
+
 LUNATIK_OPENER(skb);
 static const lunatik_class_t luaskb_class = {
 	.name    = "skb",