简单的 socket 代码

TCP 编程

客户端代码

将键盘输入的字符发送到服务端，并将从服务端接收到的字符输出到终端

 #!/usr/python3

 import socket

 def socket_client():
     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

     #获取主机名
     host = socket.gethostname()
     port = 9999

     #建立连接
     s.connect((host, port))

     #接受欢迎消息
     print(s.recv(1024).decode('utf-8'))
     while True:
         data = input()
         #发送数据
         s.send(data.encode('utf-8'))
         print(s.recv(1024).decode('utf-8'))
     s.send(b'exit')
     s.close()

 def main():
     socket_client()

 if __name__ == '__main__':
     main()

注意：若服务器需要同时响应多个客户端请求，每个连接都需要一个新进程或新线程处理，否则服务器一次只能服务一个客户端

服务端代码

将从客户端接收到的字符前面加上 HELLO, 再发送回给客户端

 #!/usr/bin/python3

 import socket
 import threading
 import time

 def tcp_link(sock, addr):
     print('Accept new connection from %s %s...' % addr)
     sock.send('欢迎学习 python 网络编程'.encode('utf-8'))
     while True:
         data = sock.recv(1024)
         time.sleep(1)
         if not data or data.decode('utf-8') == 'exit':
             break
         sock.send(('HELLO, %s' % data.decode('utf-8')).encode('utf-8'))
     sock.close()
     print('Connection from %s : %s closed.' % addr)

 def socket_server():
     #创建一个 socket
     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)#SOCK_STREAM 指定使用面向流的 TCP 协议
     #其中 AF_INET 指定使用 IPv4 协议，若要使用 IPv6 则指定该参数为 AF_INET6

     #获取本地主机名
     host = socket.gethostname()

     #获取本地主机名
     host = socket.gethostname()

     #设置端口号
     port = 9999

     #监听端口
     s.bind((host, port))

     #设置最大连接数，超过后排队
     s.listen(5)

     while True:
         #接受一个新连接
         sock, addr = s.accept()
         # 创建新线程处理TCP连接
         t = threading.Thread(target = tcp_link, args = (sock, addr))
         t.start()    

 def main():
     socket_server()

 if __name__ == '__main__':
     main()

UDP 编程

 客户端代码

将键盘输入的字符发送到服务端，并将从服务端接收到的字符输出到终端

 #!/usr/bin/python3

 import socket

 def socket_udp_client():
     host = socket.gethostname()
     port = 9999
     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
     while True:
         data = input()

         #发送数据
         s.sendto(data.encode('utf-8'), (host, port))

         #接收数据
         print(s.recv(1024).decode('utf-8'))

 def main():
     socket_udp_client()

 if __name__ == '__main__':
     main()

服务端代码

将从客户端接收到的字符前面加上 hello, 再发送回给客户端

 #!/usr/bin/python3

 import socket

 def socket_udp_server():
     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #SOCK_DGRAM 参数指定 socket 的类型是 udp
     host = socket.gethostname()
     port = 9999

     #绑定端口
     s.bind((host, port))

     #不需要调用 listen 方法，直接接收来自任何客户端的数据

     while True:
         #接收数据
         data, addr = s.recvfrom(1024)
         #recvfrom 方法返回数据和客户端的地址与端口。这样，服务器收到数据后，直接调用 sendto 就可以把数据用 udp 发送给客户端
         print('received from %s: %s.' % addr)
         s.sendto(b'hello, %s' % data, addr)

 def main():
     socket_udp_server()

 if __name__ == '__main__':
     main()