100 模拟ssh远程执行命令

目录

• 1、subprocess模块
  • 1.1 使用方法
• 2、模拟实现SSH远程执行命令
  • 服务器
  • 客户端

1、subprocess模块

subprocess 模块容许你去建立一个新的进程让其执行另外的程序，并与它进行通讯，获取标准的输入、标准输出、标准错误以及返回码等。更多查看官网：https://docs.python.org/2/library/subprocess.html?highlight=subprocess#frequently-used-arguments

能够经过subprocess 模块去执行命令

1.1 使用方法

执行命令，把执行的正确结果放到管道中，把执行错误的结果也放到管道中

obj=subprocess.Popen(命令, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

注意：的结果的编码是以当前所在的系统为准的，若是是windows，那么res.stdout.read()读出的就是GBK编码的，在接收端需要用GBK解码，正所谓用什么格式编码就用什么格式解码

# subprocess 执行系统命令的模块
import subprocess
# 把执行的正确结果放到管道中
obj=subprocess.Popen('tasklist',shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
#拿到正确结果的管道,读出里面的内容
ss=obj.stdout.read()
err=obj.stderr.read()
print('错误信息',str(err,encoding='gbk'))
#由于windows用的gbk编码,用gbk解码
print(str(ss,encoding='gbk'))

2、模拟实现SSH远程执行命令

服务器

import socket
import subprocess
HOST = "192.168.11.237"
PORT = 8081

soc = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
soc.bind((HOST,PORT))
soc.listen(5)

while 1:
    print("等待链接。。。")
    conn,addr = soc.accept()
    print("链接成功。。。\n")
    while 1:
        try:
            data = conn.recv(1024)
            if len(data)==0:    # 长度0说明断开了链接。在windows中没用，在linux中才有用
                break

            # 把执行正确的内容放到管道中
            obj = subprocess.Popen(str(data,encoding="utf8"),shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
            # 执行的结果 b 格式,gbk编码(windows平台)
            suc_msg = obj.stdout.read()
            if suc_msg:
                conn.send(suc_msg)
            else:
                fail_msg = obj.stderr.read()
                conn.send(fail_msg)

        except:
            print("客户机链接中断。。。")
            break
    conn.close()

soc.close()

客户端

import socket
HOST = "192.168.11.237"
PORT = 8081

soc = socket.socket()
soc.connect((HOST,PORT))

while 1:
    try:
        cmd = input("请输入须要执行的命令")
        soc.send(cmd.encode("utf8"))

        data = soc.recv(1024)
        print(data.decode("gbk"))
    except Exception as a:
        print("服务器关闭了:",a)
        break
        
# 4.关闭链接
soc.close()

输入dir命令，因为服务端发送字节少于1024字节，客户端能够接受。

输入tasklist命令，因为服务端发送字节多于1024字节，客户端只接受部分数据，而且当你再次输入dir命令的时候，客户端会接收dir命令的结果，可是会打印上一次的剩余未发送完的数据，这就是粘包问题。