socket_server_ssh.py
#!usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = "Samson"
import socket
import os
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.bind(("localhost",6969))
server.listen()
while True:
conn,addr = server.accept()
print("new conn:",addr)
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开!")
break
print("执行指令:",data)
cmd_res = os.popen(data.decode()).read()
if len(cmd_res) == 0:
print("Command has no output")
conn.send(str(len(cmd_res.encode())).encode("utf-8"))#连续两条send可能会出现粘包，建议send后立即recv一下
conn.send(cmd_res.encode("utf-8"))
server.close()
socket_server_client.py
#!usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = "Samson"
import socket
client = socket.socket()
client.connect(("localhost",6969))
while True:
cmd = input(">>:").strip()
if len(cmd) == 0:
continue
client.send(cmd.encode("utf-8"))
cmd_res_size = client.recv(1024)#接受命令结果的长度
print("命令结果大小:",cmd_res_size)
received_size = 0
while received_size < int(cmd_res_size.decode()):
data = client.recv(1024)
received_size += len(data)#接收到的数据可能小于1024
print(data.decode())
else:
print("cmd res receive done!")
client.close()