From 9bf424041114f1298ae5b063d11ded695c5e3cca Mon Sep 17 00:00:00 2001 From: Salad Dais Date: Fri, 18 Jun 2021 20:31:15 +0000 Subject: [PATCH] Allow tagging UDPPackets with arbitrary metadata The metadata should propagate to any Messages deserialized from the packet as well. --- hippolyzer/lib/base/network/transport.py | 1 + hippolyzer/lib/proxy/lludp_proxy.py | 1 + 2 files changed, 2 insertions(+) diff --git a/hippolyzer/lib/base/network/transport.py b/hippolyzer/lib/base/network/transport.py index 2a99c8c..245dbd8 100644 --- a/hippolyzer/lib/base/network/transport.py +++ b/hippolyzer/lib/base/network/transport.py @@ -30,6 +30,7 @@ class UDPPacket: self.dst_addr = dst_addr self.data = data self.direction = direction + self.meta = {} @property def outgoing(self): diff --git a/hippolyzer/lib/proxy/lludp_proxy.py b/hippolyzer/lib/proxy/lludp_proxy.py index 87d9b69..c842ed8 100644 --- a/hippolyzer/lib/proxy/lludp_proxy.py +++ b/hippolyzer/lib/proxy/lludp_proxy.py @@ -58,6 +58,7 @@ class InterceptingLLUDPProxyProtocol(UDPProxyProtocol): message = self.deserializer.deserialize(packet.data) message.direction = packet.direction message.sender = packet.src_addr + message.meta.update(packet.meta) assert message is not None # Check for UDP bans on inbound messages