代码清单2-4 UDP广播

代码清单2-4 UDP广播

import argparse
import socket

MAX_BYTES = 65535


def server(interface, port):
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.bind((interface, port))
    print('Listening for datagrams at {}'.format(sock.getsockname()))
    while True:
        data, address = sock.recvfrom(MAX_BYTES)
        text = data.decode('ascii')
        print('The client at {} says {!r}'.format(address, text))


def client(network, port):  # 将这里的network设置为本机的广播地址，就可以进行广播发送
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)  # 允许广播
    text = 'Broadcast datagram!'
    sock.sendto(text.encode('ascii'), (network, port))


choices = {'client': client, 'server': server}
parser = argparse.ArgumentParser(description='Send, receive UDP broadcast')
parser.add_argument('role', choices=choices, help='Which role to take')
parser.add_argument('host', help='interface the server listens at;''network the server send to')
parser.add_argument('-p', metavar='port', type=int, default=1060, help='UDP port(default 1060)')
args = parser.parse_args()
function = choices[args.role]
function(args.host, args.p)