22 lines
749 B
Python
22 lines
749 B
Python
import collections
|
|
|
|
from hippolyzer.lib.base.message.message import Message
|
|
from hippolyzer.lib.proxy.addon_utils import BaseAddon, GlobalProperty
|
|
from hippolyzer.lib.proxy.commands import handle_command
|
|
from hippolyzer.lib.proxy.region import ProxiedRegion
|
|
from hippolyzer.lib.proxy.sessions import Session
|
|
|
|
|
|
class PacketStatsAddon(BaseAddon):
|
|
packet_stats: collections.Counter = GlobalProperty(collections.Counter)
|
|
|
|
def handle_lludp_message(self, session: Session, region: ProxiedRegion, message: Message):
|
|
self.packet_stats[message.name] += 1
|
|
|
|
@handle_command()
|
|
async def print_packet_stats(self, _session: Session, _region: ProxiedRegion):
|
|
print(self.packet_stats.most_common(10))
|
|
|
|
|
|
addons = [PacketStatsAddon()]
|