Python Socket 模拟SSH

学习记录之用

- 准备

• 不用解释啥是【SSH】吧。
• python里可以用commands.getoutput(‘whoami’)获取指令输出结果

derek@ubuntu:~$ python
Python 2.7.12 (default, Nov 19 2016, 06:48:10) 
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import commands
>>> commands.getoutput('whoami')
'derek'
#也可以用commands.getstatusoutput('whoami')

• 为了能多用户用这个【模拟SSH】，这里用到连接池，为此，需要实现类【SocketServer.BaseRequestHandler】重写其方法【handle】。为每一个建立连接的客服端开一个线程，专门为其服务。

class MyTCPServer(SocketServer.BaseRequestHandler):
    def handle(self):
        pass

- 客户端

import socket
from math import ceil

def recv_all(obj, length):
    raw_data=''
    for i in range(int(ceil(length/4096.0))):
        data=obj.recv(4096)
        raw_data+=data
    return raw_data

HOST,PORT='192.168.180.128',5200
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

s.connect((HOST,PORT))
print '---welcome to my ssh---'
while True:
    #美化的代码【\033[34;1m \033[0m】在windows下不适用
    key=raw_input('\033[34;1mPlease Input Key:\033[0m')
    if key == '':
        continue
    elif key in ['quit','exit']:
        break

    #发送请求指令
    s.sendall(key)

    #获取接受的数据的大小
    recv_size=int(s.recv(1024))

    #由于数据量可能太大，所以分多次接受
    recv_data=recv_all(s, recv_size)
    print recv_data
s.close()
print 'goodbay(t_t)'

- 服务器端

import SocketServer
import commands, time

class MyTCPServer(SocketServer.BaseRequestHandler):
    def handle(self):
        #提示建立新连接
        print 'Get a new connection from ',self.client_address
        #起一个死循环，不断侦听来自客户端的请求
        while True:
            #接受客户端来的指令
            key = self.request.recv(1024).strip()
            #接受的数据为空，跳出死循环
            if not key:
                break
            cmd_result = commands.getoutput(key)

            #send data_size发送数据大小
            self.request.sendall( str(len(cmd_result)) )
            #这里是为了防止粘包，是因为底层会把前者小的数据包连着后者一起发送
            time.sleep(0.2)
            #send raw_data发送原数据
            self.request.sendall(cmd_result)
        #提示丢失已有连接
        print 'Lost connection with ',self.client_address


if __name__=='__main__':
    HOST,PORT='192.168.180.128',5200
    server=SocketServer.ThreadingTCPServer((HOST,PORT),MyTCPServer)
    server.serve_forever()

注意：
服务器端必须要Linux环境下运行
服务器端和客服端需要在同一网段
手机上有一款软件【termux】可以模拟Linux环境，不妨作为客户端测试下