source: python/vendor/Python-2.7.6/Demo/sockets/mcast.py

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

python: Update vendor to 2.7.6.

  • Property svn:eol-style set to native
File size: 2.2 KB
Line 
1#!/usr/bin/env python
2#
3# Send/receive UDP multicast packets.
4# Requires that your OS kernel supports IP multicast.
5#
6# Usage:
7# mcast -s (sender, IPv4)
8# mcast -s -6 (sender, IPv6)
9# mcast (receivers, IPv4)
10# mcast -6 (receivers, IPv6)
11
12MYPORT = 8123
13MYGROUP_4 = '225.0.0.250'
14MYGROUP_6 = 'ff15:7079:7468:6f6e:6465:6d6f:6d63:6173'
15MYTTL = 1 # Increase to reach other networks
16
17import time
18import struct
19import socket
20import sys
21
22def main():
23 group = MYGROUP_6 if "-6" in sys.argv[1:] else MYGROUP_4
24
25 if "-s" in sys.argv[1:]:
26 sender(group)
27 else:
28 receiver(group)
29
30
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)
40 else:
41 s.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_MULTICAST_HOPS, ttl_bin)
42
43 while True:
44 data = repr(time.time())
45 s.sendto(data + '\0', (addrinfo[4][0], MYPORT))
46 time.sleep(1)
47
48
49def receiver(group):
50 # Look up multicast group address in name server and find out IP version
51 addrinfo = socket.getaddrinfo(group, None)[0]
52
53 # Create a socket
54 s = socket.socket(addrinfo[0], socket.SOCK_DGRAM)
55
56 # Allow multiple copies of this program on one machine
57 # (not strictly needed)
58 s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
59
60 # Bind it to the port
61 s.bind(('', MYPORT))
62
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)
71
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.