EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS

[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Demo / sockets / mcast.py

#!/usr/bin/env python\r
#\r
# Send/receive UDP multicast packets.\r
# Requires that your OS kernel supports IP multicast.\r
#\r
# Usage:\r
#   mcast -s (sender, IPv4)\r
#   mcast -s -6 (sender, IPv6)\r
#   mcast    (receivers, IPv4)\r
#   mcast  -6  (receivers, IPv6)\r
\r
MYPORT = 8123\r
MYGROUP_4 = '225.0.0.250'\r
MYGROUP_6 = 'ff15:7079:7468:6f6e:6465:6d6f:6d63:6173'\r
MYTTL = 1 # Increase to reach other networks\r
\r
import time\r
import struct\r
import socket\r
import sys\r
\r
def main():\r
    group = MYGROUP_6 if "-6" in sys.argv[1:] else MYGROUP_4\r
\r
    if "-s" in sys.argv[1:]:\r
        sender(group)\r
    else:\r
        receiver(group)\r
\r
\r
def sender(group):\r
    addrinfo = socket.getaddrinfo(group, None)[0]\r
\r
    s = socket.socket(addrinfo[0], socket.SOCK_DGRAM)\r
\r
    # Set Time-to-live (optional)\r
    ttl_bin = struct.pack('@i', MYTTL)\r
    if addrinfo[0] == socket.AF_INET: # IPv4\r
        s.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, ttl_bin)\r
    else:\r
        s.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_MULTICAST_HOPS, ttl_bin)\r
\r
    while True:\r
        data = repr(time.time())\r
        s.sendto(data + '\0', (addrinfo[4][0], MYPORT))\r
        time.sleep(1)\r
\r
\r
def receiver(group):\r
    # Look up multicast group address in name server and find out IP version\r
    addrinfo = socket.getaddrinfo(group, None)[0]\r
\r
    # Create a socket\r
    s = socket.socket(addrinfo[0], socket.SOCK_DGRAM)\r
\r
    # Allow multiple copies of this program on one machine\r
    # (not strictly needed)\r
    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)\r
\r
    # Bind it to the port\r
    s.bind(('', MYPORT))\r
\r
    group_bin = socket.inet_pton(addrinfo[0], addrinfo[4][0])\r
    # Join group\r
    if addrinfo[0] == socket.AF_INET: # IPv4\r
        mreq = group_bin + struct.pack('=I', socket.INADDR_ANY)\r
        s.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)\r
    else:\r
        mreq = group_bin + struct.pack('@I', 0)\r
        s.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_JOIN_GROUP, mreq)\r
\r
    # Loop, printing any data we receive\r
    while True:\r
        data, sender = s.recvfrom(1500)\r
        while data[-1:] == '\0': data = data[:-1] # Strip trailing \0's\r
        print (str(sender) + '  ' + repr(data))\r
\r
\r
if __name__ == '__main__':\r
    main()\r