socket笔记

version： python3

TCP

创建服务器

from socket import * from time import ctime # 设置HOST，PORT HOST = "" PORT = 20000 # 设置缓冲区大小 BUFSIZE = 1024 ADDR = (HOST, PORT) # 创建服务端socket tcpServerSock = socket(AF_INET, SOCK_STREAM) # 绑定到ADDR上 tcpServerSock.bind(ADDR) #监听, 最大监听数为5 tcpServerSock.listen(5) while True: print("waiting for connection...") # 等待连接 tcpClientSock, addr = tcpServerSock.accept() print("connected from: ", addr) # 设置了一个变量error用于区分exit退出 error = False while True: # 获取客户端发送的数据 data = tcpClientSock.recv(BUFSIZE) # 为空时退出循环 if not data: break # 接收到exit字样时，退出循环 if data.decode('utf-8').lower() == "exit": break # 接收到KeyboardInterrupt or EOFError 的错误信息时，退出循环并将error设为True，用于退出服务器 if data == bytes('Program interrupted by user.', encoding="utf-8") or \ data == bytes('Read beyond end of file.', encoding="utf-8"): error = True break # 发送数据到客户端 tcpClientSock.send(bytes("[{} {}]".format(ctime(), data), encoding="utf-8")) # 断开客户端连接 tcpClientSock.close() # 接收到KeyboardInterrupt or EoFError时退出服务器 if error: break # 关闭服务器 tcpServerSock.close()

创建客户端

from socket import * HOST = "127.0.0.1" PORT = 20000 BUFSIZE = 1024 ADDR = (HOST, PORT) # 创建客户端socket tcpClientSock = socket(AF_INET, SOCK_STREAM) # 连接服务器 tcpClientSock.connect(ADDR) while True: try: data = input(">") except KeyboardInterrupt or EOFError as e: data = e.__doc__ # 输入为空时，跳出循环 if not data: break tcpClientSock.send(bytes(data, encoding="utf-8")) data = tcpClientSock.recv(BUFSIZE) # 接收到数据为空时，跳出循环 if not data: break print(data.decode('utf-8')) tcpClientSock.close()

UDP

创建服务器

from socket import * from time import ctime HOST = "" PORT = 30000 BUFSIZE = 1024 ADDR = (HOST, PORT) udpServerSock = socket(AF_INET, SOCK_DGRAM) udpServerSock.bind(ADDR) while True: print("waiting for connection...") data, addr = udpServerSock.recvfrom(BUFSIZE) udpServerSock.sendto(bytes("[{}] {}".format(ctime(), data), encoding="utf-8"), addr) print("...received from and returned to: ", addr) udpServerSock.close()

创建客户端

from socket import * HOST = "127.0.0.1" PORT = 30000 BUFSIZE = 1024 ADDR = (HOST, PORT) udpClientSock = socket(AF_INET, SOCK_DGRAM) while True: data = input(">") if not data: break udpClientSock.sendto(bytes(data, encoding="utf-8"), ADDR) data, ADDR = udpClientSock.recvfrom(BUFSIZE) if not data: break print(data.decode("utf-8")) udpClientSock.close()