cpython/Demo/sockets/mcast.py

95 lines
2.1 KiB
Python
Raw Normal View History

# Send/receive UDP multicast packets.
# Requires that your OS kernel supports IP multicast.
# This is built-in on SGI, still optional for most other vendors.
#
1992-09-08 18:19:55 -03:00
# Usage:
# mcast -s (sender)
1992-09-24 07:33:16 -03:00
# mcast -b (sender, using broadcast instead multicast)
1992-09-08 18:19:55 -03:00
# mcast (receivers)
MYPORT = 8123
MYGROUP = '225.0.0.250'
1992-09-08 18:19:55 -03:00
import sys
import time
import struct
import regsub
1992-09-08 18:19:55 -03:00
from socket import *
1992-11-16 12:54:45 -04:00
# Main program
def main():
flags = sys.argv[1:]
#
if flags:
sender(flags[0])
else:
receiver()
1992-09-08 18:19:55 -03:00
1992-11-16 12:54:45 -04:00
# Sender subroutine (only one per local area network)
def sender(flag):
s = socket(AF_INET, SOCK_DGRAM)
if flag == '-b':
1992-09-24 07:33:16 -03:00
s.setsockopt(SOL_SOCKET, SO_BROADCAST, 1)
mygroup = '<broadcast>'
else:
mygroup = MYGROUP
1992-09-24 07:33:16 -03:00
ttl = struct.pack('b', 1) # Time-to-live
s.setsockopt(IPPROTO_IP, IP_MULTICAST_TTL, ttl)
1992-09-08 18:19:55 -03:00
while 1:
1992-09-24 07:33:16 -03:00
data = `time.time()`
## data = data + (1400 - len(data)) * '\0'
1992-09-08 18:19:55 -03:00
s.sendto(data, (mygroup, MYPORT))
1992-09-24 07:33:16 -03:00
time.sleep(1)
1992-09-08 18:19:55 -03:00
1992-10-25 15:20:47 -04:00
1992-11-16 12:54:45 -04:00
# Receiver subroutine (as many as you like)
def receiver():
# Open and initialize the socket
s = openmcastsock(MYGROUP, MYPORT)
#
# Loop, printing any data we receive
while 1:
data, sender = s.recvfrom(1500)
while data[-1:] == '\0': data = data[:-1] # Strip trailing \0's
print sender, ':', `data`
# Open a UDP socket, bind it to a port and select a multicast group
def openmcastsock(group, port):
# Import modules used only here
import regsub
import struct
#
# Create a socket
s = socket(AF_INET, SOCK_DGRAM)
1992-11-16 12:54:45 -04:00
#
# Allow multiple copies of this program on one machine
# (not strictly needed)
s.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
1992-11-16 12:54:45 -04:00
#
# Bind it to the port
1994-10-08 16:13:48 -03:00
s.bind(('', port))
1992-11-16 12:54:45 -04:00
#
# Look up multicast group address in name server
# (doesn't hurt if it is already in ddd.ddd.ddd.ddd format)
group = gethostbyname(group)
1992-11-16 12:54:45 -04:00
#
# Construct binary group address
bytes = eval(regsub.gsub('\.', ',', group))
1992-09-08 18:19:55 -03:00
grpaddr = 0
for byte in bytes: grpaddr = (grpaddr << 8) | byte
1992-11-16 12:54:45 -04:00
#
1992-09-08 18:19:55 -03:00
# Construct struct mreq from grpaddr and ifaddr
ifaddr = INADDR_ANY
mreq = struct.pack('ll', grpaddr, ifaddr)
1992-11-16 12:54:45 -04:00
#
1992-09-08 18:19:55 -03:00
# Add group membership
s.setsockopt(IPPROTO_IP, IP_ADD_MEMBERSHIP, mreq)
1992-11-16 12:54:45 -04:00
#
return s
1992-09-08 18:19:55 -03:00
1992-11-16 12:54:45 -04:00
main()