TCP编程
在python中可以使用socket模块进行TCP编程
客户端
- 创建一个套接字socket
- 指定协议类型,使用协议IPv4或IPv6;AF_INET:表示IPv4;AF_INET6:表示IPv6
- 指定面向流的TCP协议:SOCKET_STREAM
import socket
s = socket.socket(socket.AF_INET, socket.SOCKET_STREAM)
建立连接,指定需要连接的服务器地址及端口
s.connect(('www.sina.com.cn', 80))
连接到www.sina.com.cn,端口为80数据收发
发送数据:使用send()
接收数据:使用recv(max) max参数表示一次最多接收的字节数调用完成后使用close()函数关闭连接
服务器端
服务器先要绑定一个端口进行监听
每来一个客户端连接就要创建该客户端的socket连接,因为服务器需要区分每个客户端,所以一个Socket依赖4项:服务器地址、服务器端口、客户端地址、客户端端口来唯一确定一个Socket。
实际应用中,并不只有一个客户端连接服务器,因此需要使用进程或线程来处理多个客户端的连接
- 创建一个socket
s = socket.socket(socket.AF_INET, socket.SOCKET_STREAM) - 绑定端口
s.bind(('127.0.0.1', 80))
注意:'127.0.0.1'表示的是本机的IP地址,80表示端口,当端口小于1024时,需要管理员权限 - 监听端口
s.listen(5) # 5表示等待连接的最大客户端数量 - 等待连接(该方法会阻塞,等待接收数据)
sock, addr = s.accept() # 返回客户端套接字和地址
代码示例:
服务器端
import threading
import socket
def tcplink(sock, addr):
print('Accept new connection from %s:%s...' % addr)
sock.send(b'Welcome!')
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)
s = socket.socket(socket.AF_INET, socket.SOCKET_STREAM)
s.bind(("127.0.0.1", 9999))
s.listen(5)
sock, addr = s.accept()
while True:
# 接受一个新连接:
sock, addr = s.accept()
# 创建新线程来处理TCP连接:
t = threading.Thread(target=tcplink, args=(sock, addr))
t.start()
客户端
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 建立连接:
s.connect(('127.0.0.1', 9999))
# 接收欢迎消息:
print(s.recv(1024).decode('utf-8'))
for data in [b'Michael', b'Tracy', b'Sarah']:
# 发送数据:
s.send(data)
print(s.recv(1024).decode('utf-8'))
s.send(b'exit')
s.close()