58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
import socket
|
|
import argparse
|
|
import threading
|
|
import random
|
|
import time
|
|
|
|
def handle_client(conn):
|
|
while True:
|
|
try:
|
|
num = random.randint(0, 100)
|
|
conn.send(f"number={num}".encode())
|
|
time.sleep(1)
|
|
except:
|
|
print("Disconnected.")
|
|
conn.close()
|
|
break
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('type', type=str, help="'client' or 'server'", default='server')
|
|
parser.add_argument('--host', type=str, help="server's IPv4 address", default='0.0.0.0')
|
|
parser.add_argument('--port', type=int, help="server's listening port", default=8888)
|
|
|
|
args = parser.parse_args()
|
|
socket_type = args.type
|
|
host, port = args.host, args.port
|
|
|
|
if socket_type == 'server':
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.bind((host, port))
|
|
|
|
print("Server listening...")
|
|
|
|
while True:
|
|
data, addr = sock.recvfrom(1024)
|
|
print(f"Accept connection from {addr}")
|
|
|
|
sock.sendto(host.encode(), addr)
|
|
|
|
# thread = threading.Thread(target=handle_client, args=(conn, ))
|
|
# thread.start()
|
|
|
|
else:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
|
sock.settimeout(2)
|
|
|
|
sock.sendto('discover'.encode(), ('255.255.255.255', port))
|
|
|
|
while True:
|
|
data, addr = sock.recvfrom(1024)
|
|
print(data.decode(), addr)
|
|
|
|
|
|
|
|
|