Python学习之socket实现ssh

Windows环境下
服务器端步骤
# family address
#     AF.INET   ipv4
#     AF.INET6  ipv6
#     AF.UNIX   local
# socket protocol
#     socket.SOCK_STREAM    tcp
#     socket.SOCK_DGRAM     udp
# 声明socket实例,包括地址簇，协议类型，默认是ipv4,tcp
server = socket.socket()
#绑定要监听的端口
server.bind(('localhost',6969)) 
#开始监听
server.listen()
#等别人连接进来
conn, addr = server.accept() 
#接收客户端发送的东西
data = conn.recv(1024)
#也可以发送
conn.send(data.upper())
#关闭服务器端
server.close()

# 客户端步骤
# 实例化一个socket，作用是声明socket实例，同时生成socket连接对象
client = socket.socket() 
#开始连接
client.connect(('localhost',6969)) 
#发送数据
client.send()
#关闭客户端
client.close()

python3中
把字符串的编码变成了unicode,文件默认编码变成了utf-8。
把str 和bytes 做了明确区分， str 就是unicode格式的字符， bytes就是单纯二进制
bytes主要是给计算机看的，string主要是给人看的

字符串encode将获得一个bytes对象
将字节对象decode将获得一个str对象

import socket
import os
server = socket.socket()
server.bind(('localhost',9999))
#server.bind(('0.0.0.0',9999))

server.listen()
while True:
    conn,addr= server.accept()
    #每进来一个连接，服务器端都会为这个连接生成一个实例，赋值给conn
    print("new conn",addr)
    while True:
        print("等待新指令")
        data= conn.recv(1024)  #接受客户端发送的指令,建议不超过8192
        #这里默认是阻塞的，即客户端如果不发数据，会一直卡住
        #解决死循环问题
　　　　 if not data:   #客户端一断开，conn.recv收到的都是空数据，会陷入死循环
            print("客户端已断开")
            break
        print("执行指令：",data)
        cmd_res = os.popen(data.decode()).read()  # 接收字符串，执行结果也是字符串
        print("before send", len(cmd_res))
        if len(cmd_res) == 0:
            cmd_res = "cmd has no output"
            #整数不能encode,先转成字符串
            # a = "我"
            # len(a)   字符串长度是1
            # len(a.encode())   bytes长度是3
            # 所以应该先encode再判断长度，因为发送的是bytes类型的
        conn.send(str(len(cmd_res.encode())).encode("utf-8")) #先发送大小给客户端,二进制形式发送
        conn.send(cmd_res.encode("utf-8"))
        print("send done")

server.close()

# 获取控制台输出的内容，用os.popen方法

import socket
client = socket.socket()
#client.connect(('192.168.16.200',9999))
client.connect(('localhost',9999))

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  #存放收到的数据大小,bytes类型
    received_data = b'' #存放收到的命令执行结果
    while received_size < int(cmd_res_size.decode()):
        data = client.recv(1024)
        received_size += len(data) #每次收到的有可能小于1024，所以必须用len判断
        received_data += data

        #测试，只打印收到了多少
        # print(received_size)
        #发现，实际的命令大小和收到的大小不一致
        #带中文的导致的

    else:
        print("cmd res receive done...",received_size)
        print(received_data.decode())

client.close()