source: python/trunk/Demo/sockets/mcast.py

Last change on this file was 391, checked in by dmik, 11 years ago

python: Merge vendor 2.7.6 to trunk.

  • Property svn:eol-style set to native
File size: 2.2 KB
RevLine 
[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
12MYPORT = 8123
[391]13MYGROUP_4 = '225.0.0.250'
14MYGROUP_6 = 'ff15:7079:7468:6f6e:6465:6d6f:6d63:6173'
15MYTTL = 1 # Increase to reach other networks
[2]16
17import time
18import struct
[391]19import socket
20import sys
[2]21
[391]22def 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]31def 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]49def 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
79if __name__ == '__main__':
80 main()
Note: See TracBrowser for help on using the repository browser.