ssh+ftp+1

使用subprocess和ThreadingTCPServer实现并发的ssh与ftp服务器，可以实现常用的cd,ls,mkdir,touch等包括其带参数的命令，more命令在小文件上行，也就是不是一次性获取结果命令还不能实现，等再研究研究，能实现文件上传下载，有一定的容错能力。

server端
import json
import socketserver
import subprocess
import os
from itertools import chain

class MyServer(socketserver.BaseRequestHandler):

    def common_command(self,clean_data):
         print(clean_data["command"])
         if clean_data["command"] == "pwd":
             res = clean_data["cur_dir"]
             len_data = len(clean_data["cur_dir"])
         else:
             split_command = clean_data["command"].split()
             command = ""
             if len(split_command) == 1:
                 command = split_command[0] + " " + clean_data["cur_dir"]
             else:
                 command += split_command[0] + " "
                 for i in range(1,len(split_command)):
                     if split_command[i].startswith("-") or split_command[i].startswith("|"):
                         command += split_command[i] + " "
                 if split_command[-1].startswith("/"):
                     command += split_command[-1]
                 elif not split_command[-1].startswith("/") and not split_command[-1].startswith("-"):
                     command += clean_data["cur_dir"] + "/" + split_command[-1]
                 elif split_command[-1].startswith("-"):
                     command += clean_data["cur_dir"]
             print(command)
             res = subprocess.getoutput(command)
             len_data = len(res)
             if not res:
                 res = '%s successful'%command
                 len_data = len(res)
         self.request.sendall(("%s"%len_data).encode("utf-8"))
         self.request.sendall(res.encode("utf-8"))

    def chdir(self,clean_data):
        command = clean_data["command"]
        cur_dir = clean_data["cur_dir"]
        send_all_msg = {
            "reason": None,
            "cur_dir":cur_dir
        }
        if command == "cd":
            send_all_msg["cur_dir"] = "/root"
        elif len(command.split()) < 2:
            send_all_msg["reason"] = "请输入正确的格式"
        elif command.split()[1]== ".":
            send_all_msg["cur_dir"] = cur_dir
        elif command.split()[1] == "..":
            if cur_dir == "/":
                send_all_msg["cur_dir"] = cur_dir
            else:
                if len(cur_dir.split("/")) == 2:
                    send_all_msg["cur_dir"] = "/"
                else:
                    send_all_msg["cur_dir"] = "/".join(cur_dir.split("/")[:-1])
                print("else",cur_dir)
        elif command.split()[1].startswith("/"):
            send_all_msg["cur_dir"] = command.split()[1]
        else:
            send_all_msg["cur_dir"] = cur_dir + "/"+ command.split()[-1]
        if os.path.isdir(send_all_msg["cur_dir"]):
            send_all_msg = json.dumps(send_all_msg)
            self.request.sendall(send_all_msg.encode("utf-8"))
        else:
            send_all_msg["reason"] = "目录不存在或不是目录"
            send_all_msg["cur_dir"] = clean_data["cur_dir"]
            send_all_msg = json.dumps(send_all_msg)
            self.request.sendall(send_all_msg.encode("utf-8"))

    def rz(self,clean_data):
        command = clean_data["command"].split(" ")
        cur_dir = clean_data["cur_dir"]
        status = {
            "code":"0",
            "detail":"",
            "data_len":"0",
            "file_name":"",
        }
        if len(command) == 2 and command[0] == "rz":
            if command[1].startswith("/"):
                file_name = command[1]
            else:
                if cur_dir == "/":
                    file_name = "/"+command[1]
                else:
                    file_name = cur_dir+"/"+command[1]
            print(file_name)
            status["file_name"] = file_name
            if os.path.isfile(file_name):
                len_file = os.path.getsize(file_name)
                status["detail"] = "正在下载"
                status["data_len"] = str(len_file)
                status = json.dumps(status)
                self.server.sendall(status.encode("utf-8"))
                with open(file_name,'rb') as f:
                    for line in f:
                        self.server.sendall(line)
            else:
                status["code"] = "-1"
                status["detail"] = "%s不存在" % file_name
                status = json.dumps(status)
                self.server.sendall(status.encode("utf-8"))
        else:
            status["code"] = "-1"
            status["detail"] = "格式为'rz filename'"
            status = json.dumps(status)
            self.server.sendall(status.encode("utf-8"))
        print(status)

    def save(self,clean_data):
        file_name = clean_data["file_name"]
        cur_dir = clean_data["cur_dir"]
        recived_data = b""
        recived_size = 0
        file_len = int(clean_data["file_len"])
        with open(cur_dir + "/" + file_name, "wb") as f:
            while recived_size < file_len:
                data = self.server.recv(1024)
                recived_size += len(data)
                recived_data += data
                f.write(data)

    def sz(self,clean_data):
        file_name = clean_data["file_name"]
        cur_dir = clean_data["cur_dir"]
        if os.path.isfile(cur_dir+"/"+file_name):
            status = b"-1"
            self.server.sendall(status)
            if_sz = self.server.recv(1024)
            print(if_sz)
            if if_sz == b"0":
                self.save(clean_data)
                self.server.sendall("上传完成".encode("utf-8"))
            else:
                pass
        else:
            status = b"0"
            self.server.sendall(status)
            self.save(clean_data)
            self.server.sendall("上传完成".encode("utf-8"))

    def handle(self):
        self.server = self.request
        cur_dir = "%s"%subprocess.getoutput("pwd")
        self.server.sendall("这是我的socketserver,欢迎使用".encode("utf-8"))
        self.server.sendall(cur_dir.encode("utf-8"))
        while True:
            try:
                data = self.server.recv(1024).decode("utf-8")
                if data == "exit":
                    print("%s已断开"%self.client_address[0])
                    self.request.sendall("谢谢使用！".encode("utf-8"))
                    break
                clean_data = json.loads(data)
                if hasattr(self,clean_data["command_type"]):
                    getattr(self,clean_data["command_type"])(clean_data)
            except Exception:
                print("%s已断开" % self.client_address[0])
                break

if __name__ == "__main__":
    server = socketserver.ThreadingTCPServer(('0.0.0.0',9002),MyServer)
    server.allow_reuse_address = True
    print("启动服务器")
    server.serve_forever()

　　

　　

client端
import socket
import json
import os

class Client():
    def __init__(self,ip,port):
        self.client = socket.socket()
        self.ip = ip
        self.port = port

    def to_json(self,msg,command_type):
        all_msg = {
            "command_type": command_type,
            "command": msg,
            "cur_dir": self.cur_dir
        }
        all_msg = json.dumps(all_msg)
        return all_msg

    def common_command(self,msg):
        all_msg = self.to_json(msg,"common_command")
        self.client.sendall(all_msg.encode("utf-8"))
        len_data = self.client.recv(1024).decode("utf-8")
        recived_data = ""
        recived_size = 0
        while recived_size < int(len_data):
            data = self.client.recv(1024).decode("utf-8")
            recived_size += len(data)
            recived_data += data
        print(recived_data)

    def chdir(self,msg):
        all_msg = self.to_json(msg,"chdir")
        self.client.sendall(all_msg.encode("utf-8"))
        recv_data = self.client.recv(1024).decode("utf-8")
        recv_data = json.loads(recv_data)
        reason = recv_data["reason"]
        self.cur_dir = recv_data["cur_dir"]
        if reason != None:
            print(reason)

    def rz(self,msg):
        all_msg = self.to_json(msg,"rz")
        self.client.sendall(all_msg.encode("utf-8"))
        recv_data = self.client.recv(1024).decode("utf-8")
        recv_data = json.loads(recv_data)
        status_code = recv_data["code"]
        if status_code == "0":
            len_data = recv_data["data_len"]
            print(recv_data["detail"])
            recived_data = b""
            recived_size = 0
            with open(recv_data["file_name"].split("/")[-1],"wb") as f:
                while recived_size < int(len_data):
                    data = self.client.recv(1024)
                    recived_size += len(data)
                    recived_data += data
                    f.write(data)
            print("下载完成")
            print(len_data,recived_size)
        else:
            print(recv_data["detail"])

    def upload(self,file_name):
        with open(file_name, 'rb') as f:
            for line in f:
                self.client.sendall(line)
        res = self.client.recv(1024).decode("utf-8")
        print(res)

    def sz(self,msg):
        msg = msg.split(" ")
        if len(msg) == 2 and msg[0] == "sz":
            if os.path.isfile(msg[1]):
                file_len = os.path.getsize(msg[1])
                if "/" in msg[1]:
                    file_name = msg.split("/")[-1]
                elif "\\" in msg[1]:
                    file_name = msg.split("\\")[-1]
                else:
                    file_name = msg[1]
                all_msg = {
                    "command_type": "sz",
                    "file_len":str(file_len),
                    "cur_dir": self.cur_dir,
                    "file_name":file_name
                }
                all_msg = json.dumps(all_msg)
                self.client.sendall(all_msg.encode("utf-8"))
                status = self.client.recv(1024)
                if status == b"-1":
                    if_sz = input("文件已存在，是否继续上传，上传将覆盖 y/n?")
                    if if_sz == "y":
                        self.client.sendall(b"0")
                        print("正在上传")
                        self.upload(file_name)
                    else:
                        self.client.sendall(b"-1")
                        print("取消上传")
                else:
                    print("正在上传")
                    self.upload(file_name)
            else:
                print("%s不存在"%msg[1])
        else:
            print("格式为sz filename")

    def run_client(self):
        self.client.connect((self.ip,self.port))
        self.client.settimeout(5)
        data = self.client.recv(1024).decode("utf-8")
        self.cur_dir = self.client.recv(1024).decode("utf-8")
        print("server",data)
        while True:
            msg = input("[root@localhost %s]"%self.cur_dir)
            if not msg:
                continue
            if msg == "exit":
                self.client.sendall(msg.encode("utf-8"))
                data = self.client.recv(1024).decode("utf-8")
                print(data)
                break
            if msg.startswith("cd"):
                self.chdir(msg)
            elif msg.startswith("rz"):
                self.rz(msg)
            elif msg.startswith("sz"):
                self.sz(msg)
            else:
                self.common_command(msg)
        self.client.close()

if __name__ == "__main__":
    platform = "windows"
    client = Client('192.168.171.129',9002)
    client.run_client()