socket编程之实现简单的ssh

服务器代码：

#-*- coding:utf-8 -*-
#edited by python3.6
#
import socket,os

'''
创建socket对象
'''
server = socket.socket()

'''
绑定ip和端口
'''
server.bind(('localhost',23233))

'''
监听端口
'''
server.listen()

while True:
conn,addr = server.accept()
print('======welcome to use leo\'s ssh========')
print('new connection:',addr)
while True:
print('等待新指令-->\r')
data = conn.recv(1024)
if not data:
print('客户端已断开!')
break
print('执行命令：',data)
cmd_result = os.popen(data.decode()).read() #接收来自客户端的数据
print('发送前的数据大小(byte)：',len(cmd_result))
if len(cmd_result) == 0:
cmd_result = 'cmd has not output...'
conn.send(str(len(cmd_result.encode())).encode('utf-8'))
conn.send(cmd_result.encode('utf-8'))
print('发送完成。')
print('=========end==============\r')
'''
关闭程序
'''
server.close()

 注：上述代码可能会造成粘包问题（即缓冲区将2行代码合并成一行执行），解决方式如下：

服务器：
conn.send(str(len(cmd_result.encode())).encode('utf-8'))
client_ack = conn.recv(1024)
conn.send(cmd_result.encode('utf-8'))
客户端：
print('命令执行结果大小为(byte)：',cmd_result_size)
client.send('可以开始发送数据了！'.encode('utf-8'))
recv_size = 0
recv_data = b''

客户端代码：

#-*- coding:utf-8 -*-
#edited by python3.6
#
import socket

'''
创建socket对象
'''
client = socket.socket()

'''
连接到服务器
'''
client.connect(('localhost',23233))

while True:
cmd = input('[client]$ ').strip()
if len(cmd) == 0:continue
client.send(cmd.encode('utf-8'))
cmd_result_size = client.recv(1024)
print('命令执行结果大小为(byte)：',cmd_result_size)
recv_size = 0
recv_data = b''

while recv_size < int(cmd_result_size.decode()):
data = client.recv(1024)
recv_size += len(data)
recv_data += data
else:
print('cmd result receive done...',recv_size)
print(recv_data.decode())

'''
关闭程序
'''
client.close()