golden hour

/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/internals

File/Folder	Size	Actions
__init__.py	0 B
__pycache__	-
auth_protocol.py	1.18 KB
cln.py	13.52 KB
deadlock_detecting_lock.py	755 B
feature_flags.py	9.47 KB
geo.py	2.39 KB
global_scope.py	462 B
iaid.py	14.32 KB
lazy_load.py	195 B
logger.py	14.81 KB
logging_protocol.py	1.07 KB
message_status_publisher.py	5.68 KB
persistent_message.py	3.02 KB
the_sink.py	12.09 KB

Edit: auth_protocol.py

import asyncio
import socket
import logging
import struct

logger = logging.getLogger(__name__)

class UnixSocketAuthProtocol(asyncio.Protocol):
    """
    This protocol uses SO_PEERCRED attribute of unix socket
    to get authentication data (pid, uid, gid)
    After connect, this values are stored in object's
    _pid, _uid, _gid attributes
    """

# ucred struct format (3 integers)
    # struct ucred
    # {
    #   pid_t pid;            /* PID of sending process.  */
    #   uid_t uid;            /* UID of sending process.  */
    #   gid_t gid;            /* GID of sending process.  */
    # };
    #
    STRUCT_FORMAT = "3i"

def connection_made(self, transport):
        self._transport = transport
        conn = self._transport.get_extra_info("socket")
        creds = conn.getsockopt(
            socket.SOL_SOCKET,
            socket.SO_PEERCRED,
            struct.calcsize(self.STRUCT_FORMAT),
        )
        self._pid, self._uid, self._gid = struct.unpack(
            self.STRUCT_FORMAT, creds
        )
        logger.debug(
            "New socket connection from pid=%s, uid=%s, gid=%s",
            self._pid,
            self._uid,
            self._gid,
        )