fixup! luaskb: add priority getter and setter

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

Author: sneaky-potato

lib/luaskb.c

@@ -195,23 +195,32 @@ static int luaskb_forward(lua_State *L)
 	return 0;
 }
 
-/***
-* Gets or sets skb->priority, used by TC classifiers to select an HTB class.
-* @function priority
-* @tparam[opt] integer priority TC handle in TC_H_MAKE(major, minor) format
-* @treturn integer current priority, or nothing if setting
-*/
-static int luaskb_priority(lua_State *L)
+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, 3); /* __newindex: key=2, val=3 */
+	return 0;
+}
+
+static int luaskb_getpriority(lua_State *L)
 {
-    luaskb_t *lskb = luaskb_check(L, 1);
-    struct sk_buff *skb = lskb->skb;
-
-    if (lua_gettop(L) >= 2) {
-        skb->priority = (u32)luaL_checkinteger(L, 2);
-        return 0;
-    }
-    lua_pushinteger(L, skb->priority);
-    return 1;
+	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, 3);
+	return 0;
 }
 
 static int luaskb_copy(lua_State *L);
@@ -225,24 +234,67 @@ static void luaskb_release(void *private)
 		luadata_close(lskb->data);
 }
 
+static const luaL_Reg luaskb_getters[] = {
+	{"mark",     luaskb_getmark},
+	{"priority", luaskb_getpriority},
+	{NULL, NULL}
+};
+
+static const luaL_Reg luaskb_setters[] = {
+	{"mark",     luaskb_setmark},
+	{"priority", luaskb_setpriority},
+	{NULL, NULL}
+};
+
+static int luaskb_index(lua_State *L)
+{
+	lua_getmetatable(L, 1);
+	lua_pushvalue(L, 2);
+	lua_rawget(L, -2);
+	if (!lua_isnil(L, -1))
+		return 1;
+	lua_pop(L, 2);
+
+	const char *key = luaL_checkstring(L, 2);
+	for (const luaL_Reg *f = luaskb_getters; f->name; f++)
+		if (strcmp(f->name, key) == 0)
+			return f->func(L);
+
+	lua_pushnil(L);
+	return 1;
+}
+
+static int luaskb_newindex(lua_State *L)
+{
+	const char *key = luaL_checkstring(L, 2);
+	for (const luaL_Reg *f = luaskb_setters; f->name; f++)
+		if (strcmp(f->name, key) == 0)
+			return f->func(L);
+
+	luaL_error(L, "skb field '%s' is read-only or does not exist", key);
+	return 0;
+}
+
 static const luaL_Reg luaskb_lib[] = {
 	{NULL, NULL}
 };
 
 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},
-	{"priority", luaskb_priority},
+	{"__gc",       lunatik_deleteobject},
+	{"__len",      luaskb_len},
+	{"__index",    luaskb_index},
+	{"__newindex", luaskb_newindex},
+	{"ifindex",    luaskb_ifindex},
+	{"vlan",       luaskb_vlan},
+	{"data",       luaskb_data},
+	{"resize",     luaskb_resize},
+	{"checksum",   luaskb_checksum},
+	{"forward",    luaskb_forward},
+	{"copy",       luaskb_copy},
 	{NULL, NULL}
 };
 
+
 LUNATIK_OPENER(skb);
 static const lunatik_class_t luaskb_class = {
 	.name    = "skb",