--- zzzz-none-000/linux-4.4.60/net/ipv4/udp.c	2017-04-08 07:53:53.000000000 +0000
+++ dragonfly-4020-701/linux-4.4.60/net/ipv4/udp.c	2018-11-08 13:36:17.000000000 +0000
@@ -577,11 +577,18 @@
 						 __be16 sport, __be16 dport,
 						 struct udp_table *udptable)
 {
+	struct sock *sk;
 	const struct iphdr *iph = ip_hdr(skb);
 
-	return __udp4_lib_lookup(dev_net(skb_dst(skb)->dev), iph->saddr, sport,
-				 iph->daddr, dport, inet_iif(skb),
-				 udptable);
+	sk = __udp4_lib_lookup(dev_net(skb_dst(skb)->dev), iph->saddr, sport,
+			       iph->daddr, dport, inet_iif(skb),
+			       udptable);
+
+#ifdef CONFIG_AVM_PA
+	if (sk)
+		avm_pa_add_local_session(skb, sk);
+#endif
+	return sk;
 }
 
 struct sock *udp4_lib_lookup(struct net *net, __be32 saddr, __be16 sport,
@@ -1626,6 +1633,10 @@
 	struct sk_buff *skb1 = NULL;
 	struct sock *sk;
 
+#ifdef CONFIG_AVM_PA
+	avm_pa_add_local_session(skb, stack[0]);
+#endif
+
 	for (i = 0; i < count; i++) {
 		sk = stack[i];
 		if (likely(!skb1))