Class DceRpcSocket

scapy/layers/dcerpc.py:3284–3309

A Wrapper around StreamSocket that uses a DceRpcSession

class DceRpcSocket(StreamSocket):
    """
    A Wrapper around StreamSocket that uses a DceRpcSession
    """

    def __init__(self, *args, **kwargs):
        self.transport = kwargs.pop("transport", None)
        self.session = DceRpcSession(
            ssp=kwargs.pop("ssp", None),
            auth_level=kwargs.pop("auth_level", None),
            support_header_signing=kwargs.pop("support_header_signing", True),
        )
        super(DceRpcSocket, self).__init__(*args, **kwargs)

    def send(self, x, **kwargs):
        for pkt in self.session.out_pkt(x):
            if self.transport == DCERPC_Transport.NCACN_NP:
                # In this case DceRpcSocket wraps a SMB_RPC_SOCKET, call it directly.
                self.ins.send(pkt, **kwargs)
            else:
                super(DceRpcSocket, self).send(pkt, **kwargs)

    def recv(self, x=None):
        pkt = super(DceRpcSocket, self).recv(x)
        if pkt is not None:
            return self.session.in_pkt(pkt)


# --- TODO cleanup below