大聊Python----通过Socket实现简单的ssh客户端

光只是简单的发消息、收消息没意思，干点正事，可以做一个极简版的ssh，就是客户端连接上服务器后，让服务器执行命令，并返回结果给客户端。

#ssh_client.py

import socket

client = socket.socket()  # 生命socket类型 同时 生成socket连接# 对相
client.connect(('HW-20180425SPSL',6969))  # 连接6969端口
while True:
    msg = input(">>:").strip()
    if len(msg) == 0:continue
    client.send(msg.encode("utf-8"))
    data = client.recv(512) # 接收到 512 个字节
    print(data.decode())


client.close()

#ssh_server

import socket
import os

server = socket.socket()
server.bind(("HW-20180425SPSL",6969)) # 绑定要监听的端口
server.listen(5) # 监听   最大允许多少监听

while True:  # 大循环
    conn, addr = server.accept()

    while True:
        data = conn.recv(1024)
        print(data.decode())
        if not data:
            print("client has lost....")
            break
        print("执行指令",data)
        cmd_res = os.popen(data.decode()).read()   # 接收字符串 执行结果也是字符串
        if len(cmd_res) == 0:
            cmd_res = "cmd has no output..."
        conn.send(cmd_res.encode("utf-8"))

server.close()

执行dir后，你就会发现执行命令后，返回结果不全！然后再执行pwd，你会发现执行pwd后，会返回dir没有显示的剩余数据！这是为什么呢？

这是因为，我们的客户写client.recv(512)，即客户端一次最多只接收512个字节，如果服务器端返回的数据是2000字节，那有至少1400多字节是客户端第一次接收不了的，那怎么办呢，服务器端此时不能把数据直接扔了呀，so它会暂时存在服务器的io发送缓冲区里，等客户端下次再接收数据的时候再发送给客户端。这就是为什么客户端执行第2条命令时，却接收到了第一条命令的结果的原因。这时有同学说了，那我直接在客户端把client.recv(1024)改大一点不就好了么，改成一次接收个100mb,哈哈，这是不行的，因为socket每次接收和发送都有最大数据量限制的，毕竟网络带宽也是有限的呀，不能一次发太多，发送的数据最大量的限制就是缓冲区能缓存的数据的最大量，这个缓冲区的最大值在不同的系统上是不一样的，我实在查不到一个具体的数字，但测试的结果是，在linux上最大一次可接收10mb左右的数据，不过官方的建议是不超过8k,也就是8192，并且数据要可以被2整除，不要问为什么。anyway , 如果一次只能接收最多不超过8192的数据，那服务端返回的数据超过了这个数字怎么办呢？比如让服务器端打开一个5mb的文件并返回，客户端怎么才能完整的接受到呢？那就只能循环收取啦。

## ssh_client ##

import socket

client = socket.socket()  # 生命socket类型 同时 生成socket连接# 对相
client.connect(('HW-20180425SPSL',6969))  # 连接6969端口
while True:
    msg = input(">>:").strip()
    if len(msg) == 0:continue  # 为空继续
    client.send(msg.encode("utf-8")) # 将 字符串 进行编码 发送给 服务器端
    data_size = client.recv(512)  # 接收 服务端向客户端 发送的 cmd 操作命令后生成的数据的长度
    print("命令的长度为：",data_size) # unicode
    reserver_size = 0
    while reserver_size < int(data_size.decode()):  # 0 < 512
        data = client.recv(512)           # 循环第一次 len(data) = 512 , 循环第二次 len(data) = 201
        reserver_size += len(data)  # 每次收到有可能小于512 ，所以必须用len()判断
        print(reserver_size) # 循环第一次 len(reserver_size) = 512  , 循环第二次 len(reserver_size) = 713
        print(data.decode())
    else: # 数据发送完毕
        print("data res receive down...",reserver_size)

client.close()

## ssh_server.py ##

import socket
import os

server = socket.socket()
server.bind(("HW-20180425SPSL",6969)) # 绑定要监听的端口
server.listen(5) # 监听   最大允许多少监听

while True:  # 大循环
    conn, addr = server.accept()

    while True:
        data = conn.recv(512)   # 客户端 向 服务端 发送过来的数据（也就是字符串）
        print(data.decode())    #  将该字符串进行解码 打印该字符串
        if not data:           # 数据为空
            print("client has lost....")
            break
        print("执行指令",data)       # unicode
        cmd_res = os.popen(data.decode()).read()   # 接收cmd的字符串 执行结果也是字符串
        if len(cmd_res) == 0:  # cmd 发送过来的 数据长度为0
            cmd_res = "cmd has no output..."
        conn.send(  str(len(cmd_res.encode())  ).encode("utf-8")  )  #先发大小给客户端 因为 len(cmd_res)结果是整数，所以 需要将它变为字符串 ，因为只有字符串才可以encode
        conn.send(cmd_res.encode("utf-8"))

server.close()

ssh_server所显示出来的结果为：

dir
执行指令 b'dir'

ssh_client所显示出来的结果为：

>>:dir
命令的长度为： b'713'
512
驱动器 E 中的卷是 Lunix
卷的序列号是 561D-6560

E:\week_27_1 的目录

2018/07/03 20:30 <DIR> .
2018/07/03 20:30 <DIR> ..
2018/07/03 20:34 <DIR> .idea
2018/06/19 18:45 <DIR> app01
2018/07/03 20:30 1,070 client.py
2018/06/14 09:45 556 manage.py
2018/07/03 20:18 1,094 server.py
2018/06/14 09:48 <DIR> static
2018/06/19 15:53 <DIR> templates
2018/07/01 15:41 62
713
text.text
2018/06/14 09:44 <DIR> venv
2018/06/19 15:26 <DIR> week_27_1
4 个文件 2,782 字节
8 个目录 22,906,576,896 可用字节

data res receive down... 713

大聊Python----通过Socket实现简单的ssh客户端

猜你喜欢