add LuaData submodule

Author: VictorNogueiraRio

.gitmodules

@@ -1,3 +1,6 @@
 [submodule "lib/lunatik"]
 	path = lib/lunatik
 	url = https://github.com/luainkernel/lunatik.git
+[submodule "lib/luadata"]
+	path = lib/luadata
+	url = https://github.com/luainkernel/luadata

include/uapi/linux/bpf.h

@@ -2228,6 +2228,8 @@ union bpf_attr {
 	FN(sk_select_reuseport),	\
 	FN(skb_ancestor_cgroup_id),	\
 	/* #ifdef CONFIG_XDP_LUA */	\
+	FN(lua_dataref),		\
+	FN(lua_dataunref),		\
 	FN(lua_pcall),			\
 	FN(lua_pop),			\
 	FN(lua_pushinteger),		\

lib/Makefile

@@ -272,5 +272,7 @@ obj-$(CONFIG_GENERIC_LIB_CMPDI2) += cmpdi2.o
 obj-$(CONFIG_GENERIC_LIB_UCMPDI2) += ucmpdi2.o
 
 subdir-ccflags-y += -I$(srctree)/lib/lunatik/lua \
+	            -I$(srctree)/lib/luadata/ \
 	            -D_KERNEL
 obj-$(CONFIG_LUNATIK) += lunatik/
+obj-$(CONFIG_LUADATA) += luadata/

lib/luadata

@@ -8,8 +8,10 @@ obj-y := sock.o request_sock.o skbuff.o datagram.o stream.o scm.o \
 
 obj-$(CONFIG_SYSCTL) += sysctl_net_core.o
 
-CFLAGS_dev.o = -Ilib/lunatik/lua/ -D_KERNEL
-CFLAGS_filter.o = -Ilib/lunatik/lua/ -D_KERNEL
+CFLAGS_dev.o = -Ilib/lunatik/lua/ -D_KERNEL \
+		-Ilib/luadata/
+CFLAGS_filter.o = -Ilib/lunatik/lua/ -D_KERNEL \
+		-Ilib/luadata/
 obj-y		     += dev.o ethtool.o dev_addr_lists.o dst.o netevent.o \
 			neighbour.o rtnetlink.o utils.o link_watch.o filter.o \
 			sock_diag.o dev_ioctl.o tso.o sock_reuseport.o \

net/core/Makefile

@@ -76,6 +76,7 @@
 #include <lua.h>
 #include <lauxlib.h>
 #include <lualib.h>
+#include <luadata.h>
 #endif /* CONFIG_XDP_LUA */
 
 #include <linux/uaccess.h>

net/core/dev.c

@@ -4792,6 +4792,41 @@ static const struct bpf_func_proto bpf_lwt_seg6_adjust_srh_proto = {
 #endif /* CONFIG_IPV6_SEG6_BPF */
 
 #ifdef CONFIG_XDP_LUA
+BPF_CALL_2(bpf_lua_dataref, struct xdp_buff *, ctx, int, offset) {
+	if (offset + ctx->data < ctx->data_end) {
+		int data_ref;
+
+		data_ref = ldata_newref(ctx->L, ctx->data + offset,
+				ctx->data_end - ctx->data - offset);
+		return data_ref;
+	}
+
+	return -1;
+}
+
+static const struct bpf_func_proto bpf_lua_dataref_proto = {
+	.func		= bpf_lua_dataref,
+	.gpl_only	= false,
+	.pkt_access	= false,
+	.ret_type	= RET_INTEGER,
+	.arg1_type	= ARG_PTR_TO_CTX,
+	.arg1_type	= ARG_ANYTHING,
+};
+
+BPF_CALL_2(bpf_lua_dataunref, struct xdp_buff *, ctx, int, data_ref) {
+	ldata_unref(ctx->L, data_ref);
+	return 0;
+}
+
+static const struct bpf_func_proto bpf_lua_dataunref_proto = {
+	.func		= bpf_lua_dataunref,
+	.gpl_only	= false,
+	.pkt_access	= false,
+	.ret_type	= RET_VOID,
+	.arg1_type	= ARG_PTR_TO_CTX,
+	.arg2_type	= ARG_ANYTHING,
+};
+
 BPF_CALL_4(bpf_lua_pcall, struct xdp_buff *, ctx, char *, funcname,
 			int, num_args, int, num_rets) {
 	if (lua_getglobal(ctx->L, funcname) != LUA_TFUNCTION) {
@@ -5022,6 +5057,10 @@ bpf_base_func_proto(enum bpf_func_id func_id)
 			return bpf_get_trace_printk_proto();
 		/* else: fall through */
 #ifdef CONFIG_XDP_LUA
+	case BPF_FUNC_lua_dataref:
+		return &bpf_lua_dataref_proto;
+	case BPF_FUNC_lua_dataunref:
+		return &bpf_lua_dataunref_proto;
 	case BPF_FUNC_lua_pcall:
 		return &bpf_lua_pcall_proto;
 	case BPF_FUNC_lua_pop:

net/core/filter.c

@@ -2228,6 +2228,8 @@ union bpf_attr {
 	FN(sk_select_reuseport),	\
 	FN(skb_ancestor_cgroup_id),	\
 	/* #ifdef CONFIG_XDP_LUA */	\
+	FN(lua_dataref),		\
+	FN(lua_dataunref),		\
 	FN(lua_pcall),			\
 	FN(lua_pop),			\
 	FN(lua_pushinteger),		\

tools/include/uapi/linux/bpf.h

@@ -145,6 +145,10 @@ static unsigned long long (*bpf_skb_ancestor_cgroup_id)(void *ctx, int level) =
 	(void *) BPF_FUNC_skb_ancestor_cgroup_id;
 
 /* #ifdef CONFIG_XDP_LUA */
+static int (*bpf_lua_dataref)(void *ctx, int offset) =
+    (void *)BPF_FUNC_lua_dataref;
+static void (*bpf_lua_dataunref)(void *ctx, int data_ref) =
+    (void *)BPF_FUNC_lua_dataunref;
 static int (*bpf_lua_pcall)(void *ctx, char *funcname, int num_args,
 				   int num_rets) =
     (void *) BPF_FUNC_lua_pcall;