[PATCH v2 6/6] udp: Use epollfd from flow_common structure

[Laurent Vivier <lvivier@redhat.com>]

Update UDP flow handling to use the epollfd field from the flow_common
structure instead of accessing it from the context. This aligns UDP with
the recent changes to TCP and ICMP flow management.

Signed-off-by: Laurent Vivier <lvivier@redhat.com>
---
 udp_flow.c | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/udp_flow.c b/udp_flow.c
index caaf3dababb1..2814373e4173 100644
--- a/udp_flow.c
+++ b/udp_flow.c
@@ -52,7 +52,7 @@ void udp_flow_close(const struct ctx *c, struct udp_flow *uflow)
 	flow_foreach_sidei(sidei) {
 		flow_hash_remove(c, FLOW_SIDX(uflow, sidei));
 		if (uflow->s[sidei] >= 0) {
-			epoll_del(c->epollfd, uflow->s[sidei]);
+			epoll_del(uflow->f.epollfd, uflow->s[sidei]);
 			close(uflow->s[sidei]);
 			uflow->s[sidei] = -1;
 		}
@@ -92,7 +92,9 @@ static int udp_flow_sock(const struct ctx *c,
 	ref.data = fref.data;
 	ref.fd = s;
 
-	rc = epoll_add(c->epollfd, EPOLLIN, &ref);
+	flow_set_epollfd(&uflow->f, c->epollfd);
+
+	rc = epoll_add(uflow->f.epollfd, EPOLLIN, &ref);
 	if (rc < 0) {
 		close(s);
 		return rc;
@@ -101,7 +103,7 @@ static int udp_flow_sock(const struct ctx *c,
 	if (flowside_connect(c, s, pif, side) < 0) {
 		rc = -errno;
 
-		epoll_del(c->epollfd, s);
+		epoll_del(uflow->f.epollfd, s);
 		close(s);
 
 		flow_dbg_perror(uflow, "Couldn't connect flow socket");
-- 
2.50.1