Server : Apache System : Linux indy02.toastserver.com 3.10.0-962.3.2.lve1.5.85.el7.x86_64 #1 SMP Thu Apr 18 15:18:36 UTC 2024 x86_64 User : palandch ( 1163) PHP Version : 7.1.33 Disable Function : NONE Directory :  /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/internals/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

import asyncio


class LoggingProtocol(asyncio.Protocol):
    def __init__(self, logger, network_logger, real_protocol):
        self._logger = logger
        self._network_logger = network_logger
        self._real_protocol = real_protocol

    def connection_made(self, transport):
        self._network_logger.debug("Connection made.")
        self._handle(lambda: self._real_protocol.connection_made(transport))

    def connection_lost(self, exc):
        self._network_logger.debug("Connection lost.")
        self._handle(lambda: self._real_protocol.connection_lost(exc))

    def datagram_received(self, data, addr):
        self._network_logger.debug("datagram_received: {!r}".format(data))
        self._handle(lambda: self._real_protocol.datagram_received(data, addr))

    def data_received(self, data):
        self._network_logger.debug("data_received: {!r}".format(data))
        self._handle(lambda: self._real_protocol.data_received(data))

    def _handle(self, impl):
        try:
            impl()
        except Exception as e:
            self._logger.exception(str(e))