--- zzzz-none-000/linux-5.4.213/net/ipv4/udp.c	2022-09-15 10:04:56.000000000 +0000
+++ miami-7690-761/linux-5.4.213/net/ipv4/udp.c	2024-05-29 11:20:02.000000000 +0000
@@ -113,6 +113,8 @@
 #include <net/addrconf.h>
 #include <net/udp_tunnel.h>
 
+#include <avm/pa/avm_pa.h>
+
 struct udp_table udp_table __read_mostly;
 EXPORT_SYMBOL(udp_table);
 
@@ -2008,6 +2010,7 @@
 static int udp_queue_rcv_one_skb(struct sock *sk, struct sk_buff *skb)
 {
 	struct udp_sock *up = udp_sk(sk);
+	struct net *net = dev_net(skb->dev);
 	int is_udplite = IS_UDPLITE(sk);
 
 	/*
@@ -2017,6 +2020,8 @@
 		goto drop;
 	nf_reset_ct(skb);
 
+	avm_pa_add_local_session(AVM_PA_NET_IP_DEVINFO(net), skb, sk);
+
 	if (static_branch_unlikely(&udp_encap_needed_key) && up->encap_type) {
 		int (*encap_rcv)(struct sock *sk, struct sk_buff *skb);