[Python Study Notes]Socket模拟ssh执行cmd并记录遇到的问题

服务器端:

流程:

　　1.创建servert实例

　　2.绑定地址和端口

　　3.开始监听

　　4.创建客户端连接实例

　　5.等待客户端的消息

　　6..........

 # The_author = 'liu66'
 # By python3.x
 # -*- coding = utf-8 -*-

 import socket,os

 server=socket.socket()
 server.bind(("localhost",66))#绑定要监听的端口
 server.listen(5)#开始监听
 # 等电话打进来

 while True:
     print("我要开始等电话了")
     conn,addr=server.accept()#conn为连接的实例
     print("实例的连接和地址分别为:",conn,addr)
     while True:
         print("等待数据接收...")
         data=conn.recv(1024)
         if not data:
             print("client has closed...")
             break
         # print(data)
         # print(data.decode())
         # conn.send(data.upper())
         # 执行cmd命令
         try:
             res = os.popen(data.decode()).read()
         except TypeError:
             print("请输入正确的CMD")
         else:
             print("------------------")
             print(res)#打印返回的值
             print("------------------")
             '''一定要判断返回的长度'''
             print(len(res))
             received_size=''
             if len(res) == 0:
                 res = "cmd is not correct..."
             print(res)
             conn.send(str(len(res.encode("utf-8"))).encode())

             sleep_recv = conn.recv(1024)
             print("缓冲...", sleep_recv.decode())  # 两条send命令可能会出现粘包的情况

             conn.send(res.encode("utf-8"))

 print("server is closed...")
 server.close()

ssh_server

客户端:

　　1.创建client实例

　　2.连接服务器,地址端口

　　3.向服务器发送命令

　　4............

 # The_author = 'liu66'
 # By python3.x
 # -*- coding = utf-8 -*-

 import socket

 client=socket.socket()
 client.connect(("localhost",66))

 while True:
     send_data=input(">>:").strip()
     if len(send_data) == 0: continue
     client.send(send_data.encode("utf-8"))
     # data=client.recv(1024)
     # print("server接收到的数据:",data.decode())

     ############   数据初始化 ####################
     data_size=0#要接受的数据大小
     data_res=b''#要接受的数据的内容大小
     received_size=0
     received_data=b''#要接受的数据的内容
     ############################################
     '''先接收要接受的数据大小'''
     data_size=client.recv(1024)
     print("要接受的数据大小:", data_size)

     '''发送缓冲数据'''
     client.send("...".encode())

     while received_size != int(data_size):
         data_res=client.recv(1024)
         received_size+=len(data_res)
         received_data+=data_res
     else:
         print("已接收数据大小:",received_size)
         print(received_data.decode("utf-8","ignore"))
         '''decode()默认的参数就是strict，代表遇到非法字符时抛出异常；
            如果设置为ignore，则会忽略非法字符；
            如果设置为replace，则会用?号取代非法字符；
            如果设置为xmlcharrefreplace，则使用XML的字符引用。'''

     # print(data_res.decode("utf-8","replace"))

 print("client is closed...")
 client.close()

ssh_client

问题总结:

1. 绑定地址和端口时,需要合并为一个参数

client.connect(("localhost",66))

　　2.  发送cmd时,判断发送的字符串不能为空

send_data=input(">>:").strip()
if len(send_data) == 0: continue

　　3.  发送数据需要进行编码encode

client.send(send_data.encode("utf-8"))

　　4.  先发送需要的字符长度,再接收一个缓冲消息,最后再发送内容,防止两次连续发送的粘包

 conn.send(str(len(res.encode("utf-8"))).encode())
 sleep_recv = conn.recv(1024)
 print("缓冲...", sleep_recv.decode())  # 两条send命令可能会出现粘包的情况

 conn.send(res.encode("utf-8"))

　　5.  执行命令异常处理

        try:
            res = os.popen(data.decode()).read()
        except TypeError:
            print("请输入正确的CMD")
        else:
            print("------------------")
            print(res)#打印返回的值
            print("------------------")

　　6.  发送ls命令不报error,返回为空,判断返回是否为空

            print(len(res))
            received_size=''
            if len(res) == 0:
                res = "cmd is not correct..."

　　7.  对于数据超过1024时处理,判断每次接受的字节长度,累加,直到长度等于发送的总长度

    while received_size != int(data_size):
        data_res=client.recv(1024)
        received_size+=len(data_res)
        received_data+=data_res

　　8.  print(xx.decode())经常出现非法字符异常的解决办法

        print(received_data.decode("utf-8","ignore"))
        '''decode()默认的参数就是strict，代表遇到非法字符时抛出异常；
           如果设置为ignore，则会忽略非法字符；
           如果设置为replace，则会用?号取代非法字符；
           如果设置为xmlcharrefreplace，则使用XML的字符引用。'''