电信软计网原理课程实验代码-UDP-HTTP等实验-内含源码和说明书(可自己修改).zip

#!/usr/bin/env python3 import sys import argparse import time import socket from socket import socket as Socket def main(): # Create args parser parser = argparse.ArgumentParser() # add '-p' arg, which specify the port the server will be listening at # and the client will connect to parser.add_argument('--server-port', '-p', default=2081, type=int, help='Server_Port to use') # add '-s' arg, which determine the program will run server or client parser.add_argument('--run-server', '-s', action='store_true', help='Run a ping server') # add '--server-address' arg, which specify the addr that client will connect to parser.add_argument('--server-address', default='localhost', help='Server to ping, no effect if running as a server.') args = parser.parse_args() # If '-s' is specified, run server, or run client if args.run_server: return run_server(args.server_port) else: return run_client(args.server_address, args.server_port, ) def run_server(server_port): # Create UDP socket with Socket(socket.AF_INET, socket.SOCK_DGRAM) as server_socket: # Set 'SO_REUSEADDR' so that we can reuse the same port immediately server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) # Bind the port to listening socket server_socket.bind(('', server_port)) # Starting receiving UDP packet print("Ping server ready on port", server_port) while True: # If receive the 'ping!' msg, client_address = server_socket.recvfrom(1024) if msg == b'ping!': server_socket.sendto(b"pong!", client_address) def run_client(server_address, server_port): # Measure RTT for 10 times for i in range(0, 10): # Create UDP socket with Socket(socket.AF_INET, socket.SOCK_DGRAM) as client_socket: # set timeout to 1s client_socket.settimeout(1) # Specify the addr and port to connect to client_socket.connect((server_address, server_port)) print('#%d ' % i, end='') # Measure RTT start = time.time_ns() # Send UDP ping packet client_socket.send(b'ping!') try: recv_ = client_socket.recv(4096) except socket.timeout: recv_ = None end = time.time_ns() elapsed = end - start # If pong packet is received normally if recv_ == b'pong!': print('RTT: %.2fms' % (elapsed / (10 ** 6))) else: print('Request Timeout') if __name__ == "__main__": sys.exit(main())