From da6436f423b94a44a14fd48b06f6831cd2141762 Mon Sep 17 00:00:00 2001
From: malmert <marcus.almert@gmail.com>
Date: Mon, 30 Mar 2026 22:38:03 +0200
Subject: [PATCH] stream flow proto fix

---
 backend/src/utilities/database.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/backend/src/utilities/database.py b/backend/src/utilities/database.py
index 4635f9d..b4b1e7f 100644
--- a/backend/src/utilities/database.py
+++ b/backend/src/utilities/database.py
@@ -522,12 +522,16 @@ class DatabasePool:
                             AND (capture_iface = $2 OR ingress_if = $2 OR egress_if = $2)
                             AND timestamp BETWEEN $5 AND $6
                             AND (
-                                CASE
-                                    WHEN $3 = 'tcp' THEN COALESCE(dpi_metadata -> 'tcp' ->> 'stream', '')
-                                    WHEN $3 = 'udp' THEN COALESCE(dpi_metadata -> 'udp' ->> 'stream', '')
-                                    ELSE ''
-                                END
-                            ) = $4
+                                (
+                                    CASE
+                                        WHEN $3 = 'tcp' THEN COALESCE(dpi_metadata -> 'tcp' ->> 'stream', '')
+                                        WHEN $3 = 'udp' THEN COALESCE(dpi_metadata -> 'udp' ->> 'stream', '')
+                                        ELSE ''
+                                    END
+                                ) = $4
+                                OR flow_id = ($3 || ':' || $4)
+                                OR flow_id LIKE ('%:' || $3 || ':' || $4)
+                            )
                             AND (
                                 app_protocol IS NULL
                                 OR UPPER(app_protocol) IN ('TCP', 'UDP', 'IP', 'IPV4', 'IPV6', 'ETH', 'ETHERNET', 'ETHERTYPE', 'FRAME', 'DATA')