运维开发网

102 模拟ssh远程执行命令

运维开发网 https://www.qedev.com 2020-07-05 12:06 出处:网络 作者:运维开发网整理
目录 一、subprocess模块 1.1 使用方法 二、模拟实现SSH远程执行命令 服务器 客户端

目录

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

一、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'))

二、模拟实现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:
    cmd = input("请输入需要执行的命令")
    soc.send(cmd.encode("utf8"))

    data = soc.recv(1024)
    print(data.decode("gbk"))

输入dir命令,由于服务端发送字节少于1024字节,客户端可以接受。

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

0

精彩评论

暂无评论...
验证码 换一张
取 消