#! /usr/bin/env python3
# -*- coding: utf-8 -*-
# Multicast Python Module
# ..................................
# Copyright (c) 2017-2026, Mr. Walls
# ..................................
# Licensed under MIT (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# ..........................................
# https://github.com/reactive-firewall-org/multicast/tree/HEAD/LICENSE.md
# ..........................................
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Multicast RECV functionality implementation.
This module provides functionality for receiving multicast messages. It contains classes
and functions to handle receiving messages from multicast groups, with support for
single-message reception modes.
Functions:
joinstep: Configure socket for joining multicast groups.
tryrecv: Attempt to receive data on a socket.
recvstep: Receive messages continuously until interrupted.
Classes:
McastRECV: Main tool class for RECV operations.
Caution: See details regarding dynamic imports [documented](../__init__.py) in this module.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
Testcase 0: Multicast should be importable.
>>> import multicast
>>> multicast.recv is not None
True
>>> multicast.__doc__ is not None
True
>>>
Testcase 1: Recv should be automatically imported.
A: Test that the multicast component is initialized.
B: Test that the recv component is initialized.
C: Test that the recv component has __doc__
>>> multicast is not None
True
>>> multicast.recv is not None
True
>>> multicast.recv.__doc__ is not None
True
>>> type(multicast.recv.__doc__) == type(str(''''''))
True
>>>
Testcase 2: Recv should be detailed with some metadata.
A: Test that the __MAGIC__ variables are initialized.
B: Test that the __MAGIC__ variables are strings.
>>> multicast.recv is not None
True
>>> multicast.recv.__module__ is not None
True
>>> multicast.recv.__package__ is not None
True
>>> type(multicast.recv.__doc__) == type(multicast.recv.__module__)
True
>>>
Testcase 3: main should return an int.
A: Test that the multicast component is initialized.
B: Test that the recv component is initialized.
C: Test that the main(RECV) function-flow is initialized.
D: Test that the main(RECV) function-flow returns an int 0-255.
>>> multicast.__main__ is not None
True
>>> multicast.__main__.main is not None
True
>>> tst_fxtr_args = ['''RECV''', '''--port=1234''']
>>> (test_fixture, ignored_value) = multicast.__main__.main(tst_fxtr_args)
>>> test_fixture is not None
True
>>> type(test_fixture) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<...int...>
>>> int(test_fixture) >= int(0)
True
>>> type(test_fixture) is type(0)
True
>>> int(test_fixture) < int(256)
True
>>> (int(test_fixture) >= int(0)) and (int(test_fixture) < int(256))
True
>>>
"""
__package__ = "multicast" # skipcq: PYL-W0622
"""The package of this program.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
Testcase 0: Multicast should be importable.
>>> import multicast
>>>
Testcase 1: Recv should be automatically imported.
>>> multicast.recv.__package__ is not None
True
>>>
>>> multicast.recv.__package__ == multicast.__package__
True
>>>
"""
__module__ = "multicast"
"""The module of this program.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
Testcase 0: Multicast should be importable.
>>> import multicast
>>>
Testcase 1: Recv should be automatically imported.
>>> multicast.recv.__module__ is not None
True
>>>
"""
__file__ = "multicast/recv.py"
"""The file of this component."""
__name__ = "multicast.recv" # skipcq: PYL-W0622
"""The name of this component.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
Testcase 0: Multicast should be importable.
>>> import multicast
>>>
Testcase 1: Recv should be automatically imported.
>>> multicast.recv.__name__ is not None
True
>>>
"""
try:
import sys
import warnings
if 'multicast' not in sys.modules:
# skipcq
from . import multicast as multicast # pylint: disable=cyclic-import - skipcq: PYL-C0414
else: # pragma: no branch
multicast = sys.modules["multicast"]
_BLANK = multicast._BLANK # skipcq: PYL-W0212 - module ok
except Exception as _cause:
del _cause # skipcq - cleanup any error leaks early
# skipcq
import multicast as multicast # pylint: disable=cyclic-import - skipcq: PYL-R0401, PYL-C0414
try:
from multicast import logging
from multicast import argparse as _argparse
from multicast import unicodedata as _unicodedata
from multicast import socket as _socket
from multicast import struct as _struct
depends = [_unicodedata, _socket, _struct, _argparse]
for unit in depends:
if unit.__name__ is None: # pragma: no branch
_root_cause = ModuleNotFoundError(
f"[CWE-440] module failed to import {str(unit)}.",
) # pragma: no cover
_root_cause.module = unit # pragma: no cover
raise _root_cause from None # pragma: no cover
except Exception as _cause: # pragma: no branch
baton = ImportError(_cause, str("[CWE-758] Module failed completely."))
baton.module = __module__
baton.path = __file__
baton.__cause__ = _cause
raise baton from _cause
module_logger = logging.getLogger(__name__)
module_logger.debug(
"Loading %s", # lazy formatting to avoid PYL-W1203
__name__,
)
_w_prefix: str = "Unusual call to multicast.joinstep with no groups."
_w_example_code: str = """
sock = multicast.genSocket() if isock is None else isock
sock.bind((multicast._MCAST_DEFAULT_GROUP, port))
"""
_w_advice: str = "...".join([
"Consider using",
_w_example_code,
"instead, for improved performance. Otherwise specify the multicast bind group.",
])
_w_empty_join_warning: str = "\n".join([
_w_prefix,
_w_advice,
])
_w_unspec_bind: str = "\n".join([
"Lazy call to multicast.joinstep with unspecified bind group.",
f"Will bind to {multicast._MCAST_DEFAULT_BIND_IP}.", # skipcq: PYL-W0212 - module ok
"Tip: Pass a value for bind_group to suppress this message",
"(such as 'bind_group=multicast._MCAST_DEFAULT_BIND_IP')",
])
_w_non_multicast = f"{_w_prefix}\nJust use socket.Socket.bind(...) for non-multicast networking."
[docs]
def _validate_join_args(groups=None, port=None, iface=None, bind_group=None, isock=None) -> tuple:
"""Validates joinstep arguments.
This is a helper function and should NOT be called directly.
Args:
groups (list): List of multicast group addresses to join.
port (int): Port number to bind the socket to.
iface (str, optional): Network interface to use.
bind_group (str, optional): Specific group address to bind to.
isock (socket.socket, optional): Existing socket to configure.
Note:
All warning messages are only emitted when __debug__ is True
(i.e., when Python is not running with -O or -OO).
Returns:
A tuple of (groups, port, iface, bind_group, isock) after normalization.
"""
if not groups:
groups = []
if __debug__: # pragma: no branch
if not bind_group:
warnings.warn(
_w_empty_join_warning,
category=SyntaxWarning,
stacklevel=3,
)
else:
if multicast.env.validate_multicast_address(bind_group):
groups = [bind_group]
else:
warnings.warn(
_w_non_multicast,
category=SyntaxWarning,
stacklevel=3,
)
else:
if multicast.env.validate_multicast_address(bind_group):
groups = [bind_group]
else:
if __debug__ and not bind_group: # pragma: no branch
# skipcq: PYL-W0212
if (len(groups) == 1) and (
groups[0] != multicast._MCAST_DEFAULT_BIND_IP # skipcq: PYL-W0212 - module ok
):
warnings.warn(
_w_unspec_bind,
category=ResourceWarning,
stacklevel=3,
)
return (groups, port, iface, bind_group, isock)
[docs]
def joinstep(groups, port, iface=None, bind_group=None, isock=None) -> _socket.socket:
"""
Join multicast groups to prepare for receiving messages.
Configures the socket to join specified multicast groups on a given port.
The JOIN function. Will start to listen on the given port of an interface for multicast
messages to the given group(s).
Args:
groups (list): List of multicast group addresses to join.
port (int): Port number to bind the socket to.
iface (str, optional): Network interface to use.
bind_group (str, optional): Specific group address to bind to.
isock (socket.socket, optional): Existing socket to configure.
Returns:
socket.socket: Configured socket ready to receive multicast messages.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
>>> import multicast
>>> multicast.recv is not None
True
>>>
Testcase 1: Stability testing.
A: Verify the multicast.recv module is properly initialized.
B: Verify the joinstep function exists and has the expected type.
C: Test socket creation with no groups (default behavior).
D: Test socket creation with a specified multicast group.
E: Test socket creation with a multicast group and binding to that group.
F: Test socket creation using an existing socket handle.
>>> import multicast
>>>
>>> multicast.recv is None
False
>>>
>>> multicast.recv.joinstep is None
False
>>> type(multicast.recv.joinstep)
<class 'function'>
>>> tst_sk = multicast.recv.joinstep(None, 59991)
>>> tst_sk #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket...laddr=...>
>>> multicast.endSocket(tst_sk)
>>> tst_sk #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket [closed]...>
>>> tst_fxtr = multicast._MCAST_DEFAULT_GROUP
>>> tst_sk_2 = multicast.recv.joinstep([tst_fxtr], 59991)
>>> tst_sk_2 #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket...>
>>> multicast.endSocket(tst_sk_2)
>>> tst_sk_2 #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket [closed]...>
>>> tst_sk_3 = multicast.recv.joinstep(
... [tst_fxtr], 59991, None, tst_fxtr
... ) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
>>> tst_sk_3 #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket...>
>>> multicast.endSocket(tst_sk_3)
>>> tst_sk_3 #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket [closed]...>
>>> sk_fxtr = multicast.genSocket()
>>> tst_sk_4 = multicast.recv.joinstep(
... [tst_fxtr], 59991, None, tst_fxtr, sk_fxtr
... )
>>> tst_sk_4 #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket...>
>>> multicast.endSocket(tst_sk_4)
>>> tst_sk_4 #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<socket.socket [closed]...>
>>> sk_fxtr.close()
>>>
"""
groups, _, _, bind_group, _ = _validate_join_args(groups=groups, bind_group=bind_group)
sock = multicast.genSocket() if isock is None else isock.dup()
try:
# skipcq: PYL-W0212
sock.bind((multicast._MCAST_DEFAULT_BIND_IP if bind_group is None else bind_group, port))
for group in groups:
mreq = _struct.pack(
'4sl' if iface is None else '4s4s',
_socket.inet_aton(group),
_socket.INADDR_ANY if iface is None else _socket.inet_aton(iface)
)
sock.setsockopt(_socket.IPPROTO_IP, _socket.IP_ADD_MEMBERSHIP, mreq)
except Exception as _cause: # pragma: no branch
raise OSError("[CWE-440] Socket operation failed.") from _cause # pragma: no cover
return sock
[docs]
def tryrecv(msgbuffer: list, chunk: bytes, sock: _socket.socket) -> str:
"""
Attempt to receive data on the given socket and decode it into the message buffer.
Will try to listen on the given socket directly into the given chunk for decoding.
If the read into the chunk results in content, the chunk will be decoded and appended
to the caller-instantiated `msgbuffer`, which is a collection of utf8 strings (or None).
After decoding, `chunk` is zeroed for memory efficiency and security. Either way the
message buffer will be returned.
Tries to receive data without blocking and appends it to the message buffer.
Individual chunk sizes are controlled by the module attribute `_MCAST_DEFAULT_BUFFER_SIZE` set
at module's load-time. It is possible to override the buffer size via the environment variable
"MULTICAST_BUFFER_SIZE" if available at load-time. However changing the value is not recommended
unless absolutely needed, and can be done on the sender side too.
Args:
msgbuffer (list or None): Caller-instantiated collection to store received messages.
chunk (variable or None): Caller-instantiated variable for raw received data.
sock (socket.socket): The socket to receive data from.
Returns:
list: The message buffer possibly updated with any newly received data.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
>>> import multicast
>>> multicast.recv is not None
True
>>> multicast.recv.tryrecv is not None
True
>>>
Testcase 1: Stability testing.
>>> import multicast
>>>
>>> multicast.recv is None
False
>>> multicast.recv.tryrecv is None
False
>>> type(multicast.recv.tryrecv) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<class 'function'>
>>> sk_fxtr = multicast.genSocket()
>>> tst_args = ("test pass-through", None, sk_fxtr)
>>> multicast.recv.recvstep(*tst_args) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
'test pass-through'
>>> sk_fxtr.close()
>>>
Testcase 2: Mock overflow testing.
>>> import multicast
>>>
>>> multicast.recv is None
False
>>> multicast.recv.tryrecv is None
False
>>> type(multicast.recv.tryrecv) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<class 'function'>
>>> class mockSocket():
... def recv(self, *args, **kwargs):
... return b'it worked'
...
... def close(self):
... pass
...
... def shutdown(self, *args, **kwargs):
... pass
...
>>>
>>> sk_fxtr = mockSocket()
>>> tst_args = ("test added: ", None, sk_fxtr)
>>> multicast.recv.recvstep(*tst_args) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
'test added: it worked'
>>> sk_fxtr.close()
>>>
"""
chunk = sock.recv(multicast._MCAST_DEFAULT_BUFFER_SIZE) # skipcq: PYL-W0212 - module ok
if not (chunk is None): # skipcq: PYL-C0325 -- avoids flake E714 anti-pattern
msgbuffer += str(chunk, encoding='utf8') # pragma: no cover
chunk = None # pragma: no cover
return msgbuffer
[docs]
def recvstep(msgbuffer: list, chunk: bytes, sock: _socket.socket) -> str:
"""
Receive messages continuously until interrupted.
Listens on the socket and accumulates messages into the buffer.
Args:
msgbuffer (list): Buffer to store received messages.
chunk (int): Maximum number of bytes to read per message.
sock (socket.socket): The socket to receive data from.
Returns:
list: Updated message buffer with received messages.
"""
try:
msgbuffer = tryrecv(msgbuffer, chunk, sock)
except KeyboardInterrupt: # pragma: no branch
if sys.stdout.isatty(): # pragma: no cover
module_logger.warning("User Interrupted")
except OSError: # pragma: no branch
module_logger.debug("[CWE-440] Nothing happened. There seems to have been an OS error.")
finally:
if sock: # pragma: no branch
sock = multicast.endSocket(sock)
if not (chunk is None): # skipcq: PYL-C0325 -- avoids flake E714 anti-pattern
msgbuffer += str(chunk, encoding='utf8') # pragma: no cover
chunk = None # pragma: no cover
# about 969 bytes in base64 encoded as chars
return msgbuffer
[docs]
class McastRECV(multicast.mtool):
"""
Subclasses the multicast.mtool to provide the RECV functions.
Testing:
Testcase 0: First set up test fixtures by importing multicast.
>>> import multicast
>>> multicast.recv is not None
True
>>> multicast._MCAST_DEFAULT_PORT is not None
True
>>> multicast._MCAST_DEFAULT_GROUP is not None
True
>>> multicast._MCAST_DEFAULT_TTL is not None
True
>>> multicast.recv.McastRECV is not None
True
>>>
Testcase 2: Recv should be detailed with some metadata.
A: Test that the __MAGIC__ variables are initialized.
B: Test that the __MAGIC__ variables are strings.
>>> multicast.recv is not None
True
>>> multicast.recv.McastRECV is not None
True
>>> multicast.recv.McastRECV.__module__ is not None
True
>>> multicast.recv.McastRECV.__proc__ is not None
True
>>> multicast.recv.McastRECV.__epilogue__ is not None
True
>>> multicast.recv.McastRECV.__prologue__ is not None
True
>>>
"""
__module__ = "multicast.recv"
__name__ = "multicast.recv.McastRECV"
__proc__ = "RECV"
__epilogue__ = """Generally speaking you want to bind to one of the groups you joined in
this module/instance, but it is also possible to bind to group which
is added by some other programs (like another python program instance of this)
"""
__prologue__ = "Python Multicast Receiver. Primitives for a listener for multicast data."
[docs]
@classmethod
def setupArgs(cls, parser):
pass # skipcq - Optional abstract method
[docs]
@staticmethod
def _hearstep(groups, port, iface=None, bind_group=None):
"""
Will listen on the given port of an interface for multicast messages to the given group(s).
The work-horse function.
Internal method to set up receiving multicast messages.
Args:
groups (list): Multicast groups to join.
port (int): Port number for receiving messages.
iface (str, optional): Network interface to use.
bind_group (str, optional): Specific group address to bind to.
Returns:
str: Any received message buffer as a string. May be empty.
Raises:
NotImplementedError: if joining the multicast group is unsupported on the current system.
Minimal Acceptance Testing:
First set up test fixtures by importing multicast.
>>> import multicast
>>> multicast.recv is not None
True
>>> multicast.recv.McastRECV is not None
True
>>>
Testcase 1: Stability testing.
>>> import multicast
>>>
>>> multicast.recv is None
False
>>> multicast.recv.McastRECV is None
False
>>> test_RCEV = multicast.recv.McastRECV()
>>> type(test_RCEV) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
<class ...McastRECV...>
>>> type(test_RCEV._hearstep)
<class 'function'>
>>> test_RCEV._hearstep(None, 59991) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
'...'
>>> tst_fxtr = multicast._MCAST_DEFAULT_GROUP
>>> test_RCEV._hearstep([tst_fxtr], 59991) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
'...'
>>> test_RCEV._hearstep(
... [tst_fxtr], 59991, None, tst_fxtr
... ) #doctest: -DONT_ACCEPT_BLANKLINE, +ELLIPSIS
'...'
>>>
"""
module_logger.debug(
"Joining %s on port %d using %s as %s", # lazy formatting to avoid PYL-W1203
str(groups), port, str(iface), bind_group,
)
sock = joinstep(groups, port, iface, bind_group, None)
module_logger.debug("Opened %s", sock) # lazy formatting to avoid PYL-W1203
msgbuffer = str(multicast._BLANK) # skipcq: PYL-W0212 - module ok
chunk = None
module_logger.debug("Ready.")
msgbuffer = recvstep(msgbuffer, chunk, sock)
# about 969 bytes in base64 encoded as chars
module_logger.debug("Closing.")
multicast.endSocket(sock)
module_logger.debug("Done.")
return msgbuffer
[docs]
def doStep(self, *args, **kwargs):
"""
Execute the RECV operation to receive multicast messages.
Overrides the `doStep` method from `mtool` to start receiving messages
based on provided arguments.
Args:
*args: Variable length argument list containing command-line arguments.
**kwargs: Arbitrary keyword arguments.
Returns:
tuple: A tuple containing received data and a status indicator.
"""
module_logger.debug("RECV")
response = self._hearstep(
kwargs.get(
"groups",
[multicast._MCAST_DEFAULT_GROUP] # skipcq: PYL-W0212 - module ok
),
kwargs.get("port", multicast._MCAST_DEFAULT_PORT), # skipcq: PYL-W0212 - module ok
kwargs.get("iface", None), # skipcq: PTC-W0039 - ensure None by default
kwargs.get("group", multicast._MCAST_DEFAULT_GROUP), # skipcq: PYL-W0212 - module ok
)
_is_std = kwargs.get("is_std", False)
if (sys.stdout.isatty() or _is_std) and (len(response) > 0): # pragma: no cover
module_logger.debug("Will Print to Console.")
print(multicast._BLANK) # skipcq: PYL-W0212 - module ok
print(str(response))
print(multicast._BLANK) # skipcq: PYL-W0212 - module ok
_result = (len(response) > 0) is True
if _result:
module_logger.info("Success")
else:
module_logger.debug("Nothing Received.")
return (_result, None if not _result else response) # skipcq: PTC-W0020 - intended