[391] | 1 | #!/usr/bin/env python
|
---|
| 2 | #
|
---|
[2] | 3 | # Send/receive UDP multicast packets.
|
---|
| 4 | # Requires that your OS kernel supports IP multicast.
|
---|
| 5 | #
|
---|
| 6 | # Usage:
|
---|
[391] | 7 | # mcast -s (sender, IPv4)
|
---|
| 8 | # mcast -s -6 (sender, IPv6)
|
---|
| 9 | # mcast (receivers, IPv4)
|
---|
| 10 | # mcast -6 (receivers, IPv6)
|
---|
[2] | 11 |
|
---|
| 12 | MYPORT = 8123
|
---|
[391] | 13 | MYGROUP_4 = '225.0.0.250'
|
---|
| 14 | MYGROUP_6 = 'ff15:7079:7468:6f6e:6465:6d6f:6d63:6173'
|
---|
| 15 | MYTTL = 1 # Increase to reach other networks
|
---|
[2] | 16 |
|
---|
| 17 | import time
|
---|
| 18 | import struct
|
---|
[391] | 19 | import socket
|
---|
| 20 | import sys
|
---|
[2] | 21 |
|
---|
[391] | 22 | def main():
|
---|
| 23 | group = MYGROUP_6 if "-6" in sys.argv[1:] else MYGROUP_4
|
---|
[2] | 24 |
|
---|
[391] | 25 | if "-s" in sys.argv[1:]:
|
---|
| 26 | sender(group)
|
---|
[2] | 27 | else:
|
---|
[391] | 28 | receiver(group)
|
---|
[2] | 29 |
|
---|
| 30 |
|
---|
[391] | 31 | def sender(group):
|
---|
| 32 | addrinfo = socket.getaddrinfo(group, None)[0]
|
---|
| 33 |
|
---|
| 34 | s = socket.socket(addrinfo[0], socket.SOCK_DGRAM)
|
---|
| 35 |
|
---|
| 36 | # Set Time-to-live (optional)
|
---|
| 37 | ttl_bin = struct.pack('@i', MYTTL)
|
---|
| 38 | if addrinfo[0] == socket.AF_INET: # IPv4
|
---|
| 39 | s.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, ttl_bin)
|
---|
[2] | 40 | else:
|
---|
[391] | 41 | s.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_MULTICAST_HOPS, ttl_bin)
|
---|
| 42 |
|
---|
| 43 | while True:
|
---|
[2] | 44 | data = repr(time.time())
|
---|
[391] | 45 | s.sendto(data + '\0', (addrinfo[4][0], MYPORT))
|
---|
[2] | 46 | time.sleep(1)
|
---|
| 47 |
|
---|
| 48 |
|
---|
[391] | 49 | def receiver(group):
|
---|
| 50 | # Look up multicast group address in name server and find out IP version
|
---|
| 51 | addrinfo = socket.getaddrinfo(group, None)[0]
|
---|
[2] | 52 |
|
---|
[391] | 53 | # Create a socket
|
---|
| 54 | s = socket.socket(addrinfo[0], socket.SOCK_DGRAM)
|
---|
[2] | 55 |
|
---|
| 56 | # Allow multiple copies of this program on one machine
|
---|
| 57 | # (not strictly needed)
|
---|
[391] | 58 | s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
---|
| 59 |
|
---|
[2] | 60 | # Bind it to the port
|
---|
[391] | 61 | s.bind(('', MYPORT))
|
---|
[2] | 62 |
|
---|
[391] | 63 | group_bin = socket.inet_pton(addrinfo[0], addrinfo[4][0])
|
---|
| 64 | # Join group
|
---|
| 65 | if addrinfo[0] == socket.AF_INET: # IPv4
|
---|
| 66 | mreq = group_bin + struct.pack('=I', socket.INADDR_ANY)
|
---|
| 67 | s.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
|
---|
| 68 | else:
|
---|
| 69 | mreq = group_bin + struct.pack('@I', 0)
|
---|
| 70 | s.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_JOIN_GROUP, mreq)
|
---|
[2] | 71 |
|
---|
[391] | 72 | # Loop, printing any data we receive
|
---|
| 73 | while True:
|
---|
| 74 | data, sender = s.recvfrom(1500)
|
---|
| 75 | while data[-1:] == '\0': data = data[:-1] # Strip trailing \0's
|
---|
| 76 | print (str(sender) + ' ' + repr(data))
|
---|
| 77 |
|
---|
| 78 |
|
---|
| 79 | if __name__ == '__main__':
|
---|
| 80 | main()
|
---|