socket实现多个连接
前戏很重要~~
在实现多个连接之前,先实现下多次发送和接收数据。
如果要多次接收数据,那么在服务器端的接收和客户端的发送部分就必须使用循环。
以下代码在python3.5下运行。
服务器端:
#服务器端 import socket
server = socket.socket()
server.bind(('localhost',6969))#绑定要监听的端口
server.listen() #监听 conn,address = server.accept() #等待接收数据 返回两个值,一个是对方的标记位,一个是对方的地址 while True:
data = conn.recv(1024) #接收数据,谁发数据给我就用谁的标记位接收
print(data)
conn.send(data.upper()) #发送数据,要发给谁就用谁的标记位 server.close()
客户端:
#客户端
import socket client = socket.socket()#声明socket类型,同时生成socket连接对象 client.connect(('localhost',6969))
while True:
msg = input('>>').strip()
client.send(msg.encode("utf-8")) #发送数据
data = client.recv(1024) #接收指定大小的字节
print(data.decode()) #
client.close()
直接把用户输入的内容发送给服务器。
执行结果:
#=========客户端========
>>你好
你好
>>中国
中国
>>hello world
HELLO WORLD
>>
Process finished with exit code 1 #========服务器端=========
b'\xe4\xbd\xa0\xe5\xa5\xbd'
b'\xe4\xb8\xad\xe5\x9b\xbd'
b'hello world'
Traceback (most recent call last):
...
ConnectionResetError: [WinError 10054] 远程主机强迫关闭了一个现有的连接。 Process finished with exit code 1
可以看到这里已经实现了多少发送和接收的效果,但是如果客户端断开连接,服务器端也*中断。
在python2下如果客户端突然中断,服务器端并不会直接中断。
以下实验在Python2环境下。
单个客户端通信
服务器端代码:
#-*-coding:utf-8 -*- import socket
server = socket.socket()
server.bind(('localhost',6969))#绑定要监听的端口
server.listen(1) #监听 最多可以挂起多少个连接 conn,address = server.accept() #等待接收数据 返回两个值,一个是对方的标记位,一个是对方的地址 while True:
data = conn.recv(1024) #接收数据,谁发数据给我就用谁的标记位接收
print('recv:',data)
conn.send(data.upper()) #发送数据,要发给谁就用谁的标记位 server.close()
客户端代码:
#-*- coding=utf-8 -*-
#客户端
import socket client = socket.socket()#声明socket类型,同时生成socket连接对象 client.connect(('localhost',6969))
while True:
msg = raw_input('>>').strip() client.send(msg.encode("utf-8")) #发送数据 data = client.recv(1024) #接收指定大小的字节
print('recv:',data.decode()) #
client.close()
开始通信:
首先客户端先发送数据:
在看服务器端接收:
发送接收正常。
现在用CTRL+c断开客户端之后服务器端的情况。
修改服务器端代码, 加一个统计,查看下什么时候开始死循环的。
import socket
server = socket.socket()
server.bind(('localhost',6969))#绑定要监听的端口
server.listen(1) #监听 conn,address = server.accept() #等待接收数据 返回两个值,一个是对方的标记位,一个是对方的地址
count=0 #统计 条件
while True:
data = conn.recv(1024) #接收数据,谁发数据给我就用谁的标记位接收
print('recv:',data)
conn.send(data.upper()) #发送数据,要发给谁就用谁的标记位
count+=1
if count >10:
break
server.close()
客户端发送接收数据结果。
服务器端发送接收数据结果。
因为客户端断开,服务器端接收的都是空。
优化服务器端,断开就退出
#-*-coding:utf-8 -*- import socket
server = socket.socket()
server.bind(('localhost',6969))#绑定要监听的端口
server.listen(1) #监听 conn,address = server.accept() #等待接收数据 返回两个值,一个是对方的标记位,一个是对方的地址 while True:
data = conn.recv(1024) #接收数据,谁发数据给我就用谁的标记位接收 print('recv:',data)
if not data:
print('客户端断开连接...') #客户端断开就退出
break
conn.send(data.upper()) #发送数据,要发给谁就用谁的标记位 server.close()
客户端:
服务器端:
客户端断开,服务器直接退出。
多个客户端通信
怎么让服务器端一直保持接收状态呢?
想要 一直保持接收状态就要在客户端断开之后,继续执行server.accept()。
对的。在server.accept()之前在加一个while,当一个客户端断开了之后break,跳出最里层的循环的时候,又继续server.accept()。
#-*-coding:utf-8 -*-
import socket
server = socket.socket()
server.bind(('localhost',6969))#绑定要监听的端口
server.listen(1) #监听
while True:
conn,address = server.accept() #等待接收数据 返回两个值,一个是对方的标记位,一个是对方的地址 while True:
data = conn.recv(1024) #接收数据,谁发数据给我就用谁的标记位接收
print('recv:',data)
if not data:
print('客户端断开连接...')
break
conn.send(data.upper()) #发送数据,要发给谁就用谁的标记位 server.close()
修改并启动服务器端代码。
客户端1
客户端2
服务器端
断开客户端1之后:
服务器端
客户端2
简单的实现了能多个连接,但是只能同时跟一个客户端通信的功能。。
模拟SSH
既然可以发数据过来,那么也可以把这个数据做为命令执行。
首先修改服务器端,发送过来的命令执行并把结果返回给客户端。
服务器端代码:
#-*-coding:utf-8 -*-
import socket
import os
server = socket.socket()
server.bind(('localhost',6969))#绑定要监听的端口
server.listen(1) #监听
while True:
conn,address = server.accept() #等待接收数据 返回两个值,一个是对方的标记位,一个是对方的地址 while True:
data = conn.recv(1024) #接收数据,谁发数据给我就用谁的标记位接收
print('recv:',data)
if not data:
print('客户端断开连接...')
break
res = os.popen(data).read()
conn.send(res) #发送数据,要发给谁就用谁的标记位 server.close()
客户端代码:
#-*- coding=utf-8 -*-
#客户端
import socket client = socket.socket()#声明socket类型,同时生成socket连接对象 client.connect(('localhost',6969))
while True:
msg = raw_input('>>').strip() client.send(msg.encode("utf-8")) #发送数据 data = client.recv(1024) #接收指定大小的字节
print(data) #
client.close()
执行客户端
服务器端
这里是有返回的,如果一条命令的返回数据大于了定义的接收的大小,那么这次客户端接收的数据就不全,剩余的数据要等到服务器下次发送才能接收过来。比如定义的接收数据大小是1024,那么一次只能接收1024字节的数据。其余的数据都是在缓冲区里面。因为客户端不知道要接收几次。所以服务器端应该把本次要发送的数据大小先发过来,客户端就知道要接收几次了。