#服务端
ftp
    - main
        - server.py  #代码程序文件
    - conf
        - user         #存储用户名和MD5加密密码的文件
    -home
        - dev          #dev用户的家目录
        - admin
        - user1
        - user2

# 100 防止粘包
# 404 文件未找到
# 101 登录成功
# 102 密码错误
# 103 用户不存在
# 104 注册成功
# 105 用户已存在
# 130 目录下没有文件

dev$e77989ed21758e78331b20e477fc5582
#用户名和加密的密码之间用$分隔开

import socketserver
import json
import os

class MyTCPHandler(socketserver.BaseRequestHandler):

    def put(self,*args):
        '''接收客户端文件'''
        file_dic = args[0]
        filename = file_dic["filename"]
        filesize = file_dic["filesize"]
        fileuser = file_dic["fileuser"]
        file_path = "../home/" + fileuser + "/" + filename
        if os.path.isfile(file_path):  # 判断文件是否存在
            f = open(file_path + ".new","wb")
        else:
            f = open(file_path,"wb")
        self.request.send(b"100")  # 防止粘包
        recvd_size = 0
        while recvd_size < filesize:  # 判断文件已经接收的大小和文件总大小
            data = self.request.recv(1024)
            f.write(data)
            recvd_size += len(data)
        else:
            f.close()
            print("file has uploaded")

    def get(self,*args):
        """发送文件到客户端"""
        file_dic = args[0]
        filename = file_dic["filename"]
        fileuser = file_dic["fileuser"]
        file_path = "../home/" + fileuser + "/" + filename
        if os.path.isfile(file_path):  # 判断文件是否存在
            file_size = os.stat(file_path).st_size  # 获取文件的大小
            self.request.send(str(file_size).encode())  # 发送文件大小给客户端
            self.request.recv(1024)
            f = open(file_path,"rb")
            for line in f:
                self.request.send(line)
            else:
                f.close()
                print("file download finish")
        else:
            self.request.send(b"404 none")  # 文件不存在

    def login(self,*args):
        """用户登录，用接收到的用户名和密码去和user文件遍历验证"""
        user_dic = args[0]
        username = user_dic["username"]
        password = user_dic["password"]
        f = open(r"../conf/user","r")
        for line in f:
            line = line.strip()
            line_list = line.split("$")
            # print(line_list[0],username)
            # print(line_list[1],password)
            if line_list[0] == username and line_list[1] == password:
                self.request.send(b"101")  # 101 登录成功
                return
            elif line_list[0] == username and line_list[1] != password:
                self.request.send(b"102")  # 102 密码错误
                f.close()
                return
        else:
            self.request.send(b"103")  # 103 用户不存在
        f.close()

    def regist(self,*args):
        """注册用户，主要是接收到传来的用户名和密码，将其写入到user文件中"""
        user_dic = args[0]
        username = user_dic["username"]
        password = user_dic["password"]
        f = open(r"../conf/user","r+")  # r+ 读写，可以先检查是否存在
        for line in f:
            line = line.strip()
            line_list = line.split("$")
            if line_list[0] == username:
                f.close()
                self.request.send(b"105")
                return
        temp = "
" + username + "$" + password  # 加上 
 换行符，不然所有的用户都在一行
        f.write(temp)
        f.close()
        os.mkdir("../home/{}".format(username))  # 在注册的同时就创建好相应的家目录
        self.request.send(b"104")

    def ls(self,*args):
        """查看指定目录下文件，并返回相应的文件列表"""
        username = args[0]["username"]
        data = os.listdir("../home/" + username)  # 列出指定目录下的所有文件
        if len(data) == 0:
            self.request.send(b"130 none")
        else:
            self.request.send(json.dumps(data).encode())

    def handle(self):
        print("{} wrote:".format(self.client_address[0]))
        while True:
            try:
                self.data = self.request.recv(1024).strip()
                file_dic = json.loads(self.data.decode())  # loads将字符串类型转化为字典类型（数据传输只能是byte类型，字典必须事先用json转）
                action = file_dic["action"]
                if hasattr(self,action):  # 这里使用反射，对客户端传来的数据action，告诉服务端该操作什么
                    func = getattr(self,action)
                    func(file_dic)
            except ConnectionResetError as e:
                    print("err:",e)
                    break

if __name__ == "__main__":
    HOST,POST = "localhost",9998
    server = socketserver.ThreadingTCPServer((HOST, POST), MyTCPHandler)  # 一个服务端处理多个客户端，多线程
    server.serve_forever()

import socket
import os
import json
import hashlib

user = "none"  # 当用户成功登录后，就会改变这个全局变量为当前用户名
class FtpClient(object):

    def __init__(self):
        self.client = socket.socket()

    def connection(self,ip,port):
        """
        客户端去连接服务端
        :param ip: 服务端IP地址
        :param port: 服务端端口号
        """
        self.client.connect((ip,port))

    def login(self):
        """
        用户登录验证
        :return:登录成功返回True，登录失败返回False
        """
        m = hashlib.md5()  # 设定加密类型
        username = input("[login]<username>:").strip()
        passw = input("[login]<password>:").strip()
        m.update(bytes(passw,encoding="utf-8"))  # 对密码进行加密
        password = m.hexdigest()  # 返回十六进制的加密后的密码，加密时必须对bytes类型数据加密，使用bytes时必须指定encoding
        user_dic = {
            "username":username,
            "password":password, # 这里的密码已经是加密了的
            "action":"login"
        }
        user_dic_byte = str(json.dumps(user_dic)).encode()  # 将这个字典转化为bytes类型，传给服务端
        self.client.send(user_dic_byte)
        login_res_byte = self.client.recv(1024)  # 返回用户验证后的状态码（用户名或者密码对不对）
        login_res = login_res_byte.decode()
        if login_res == "101":  # 用户名和密码正确
            global user
            user = username
            print("Login successed")
            return True
        elif login_res == "102":  # 用户存在，密码错误
            print("password error")
            return False
        elif login_res == "103":  # 用户不存在
            print("user {} does not exist".format(username))
            return False

    def regist(self):
        """用户注册"""
        m = hashlib.md5()
        username = input("[regist]<username>:").strip()
        passw = input("[regist]<password>:").strip()
        m.update(bytes(passw,encoding="utf-8"))  # 加密时必须对bytes类型数据加密，使用bytes时必须指定encoding
        password = m.hexdigest()  # 返回十六进制的加密后的密码
        user_dic = {
            "username": username,
            "password": password,  # 这里的密码已经是加密了的
            "action": "regist"
        }
        user_dic_byte = str(json.dumps(user_dic)).encode()
        self.client.send(user_dic_byte)
        regist_res_byte = self.client.recv(1024)
        regist_res = regist_res_byte.decode()
        if regist_res == "104":  # 用户注册成功
            print("Regist successed")
            return False
        elif regist_res == "105":  # 用户已经存在
            print("user {} exist".format(username))
            return False

    def userlogin(self):
        """实现用户登录或注册的交互"""
        while True:
            option = input("<login or regist>(L/l or G/g):").strip()
            if option.upper() == "L" and self.login():
                break
            elif option.upper() == "G":
                self.regist()
            else:
                continue
        self.interact()

    def interact(self):
        """
        实行客户端与服务端的交互
        """
        while True:
            opt = input("[{}@ftpserver ~]# ".format(user)).strip()
            if len(opt) == 0: continue
            cmd = opt.split()[0]  # 取出命令
            if hasattr(self,cmd):  # 通过反射，执行相应的函数（一个命令对应一个函数，服务端也是如此）
                func = getattr(self,cmd)
                func(opt)
            else:
                self.help()

    def help(self):
        """
        命令使用帮助信息
        """
        msg = """命令使用帮助:
            ls 查看家目录下的文件目录
            put file 上传文件
            get file 下载文件
            exit 退出当前用户
        """
        print(msg)

    def progress(self,passed,all):
        """
        进度条显示
        :param passed: 已经进行过的长度
        :param all: 进度条总长度
        """
        s = passed * 100 // all
        a = s * "#"
        b = (100 - s) * "-"
        print('
 %d%% [%s%s]' % (s, a, b), end="")  # 光标回到行首，两个百分号表示输出一个百分号，end默认为
，意为换行，改为空格就不换行

    def put(self,*args):
        """
        客户端上传文件
        """
        cmd_list = args[0].split()  # 将命令和文件名分开
        if len(cmd_list) > 1:  # 判断输入的命令是否正确，列表里命令加上文件名至少是两个元素
            filename = cmd_list[1]
            if os.path.isfile(filename):  # 判断文件是否存在
                filesize = os.stat(filename).st_size  # 获取文件的大小
                file_dic = {
                    "filename":filename,
                    "filesize":filesize,
                    "action":"put",  # action是为了告诉服务端该执行什么操作
                    "fileuser":user  # 传这个值是因为要告诉服务端把文件放到哪个用户的家目录
                }
                self.client.send(json.dumps(file_dic).encode())
                responce = self.client.recv(1024)  # 防止粘包,后续可用作验证返回
                f = open(filename,"rb")
                recvd_size = 0
                for line in f:
                    recvd_size += len(line)
                    self.progress(recvd_size,filesize)  # 进度条函数，传入已传文件大小和总文件大小
                    self.client.send(line)
                else:
                    print("
file upload finish")
                    f.close()
            else:
                print("'" + filename + "': No such file")

    def get(self,*args):
        """
        客户端下载文件
        """
        cmd_list = args[0].split()
        if len(cmd_list) > 1:
            filename = cmd_list[1]
        file_dic = {
            "filename":filename,
            "action":"get",
            "fileuser": user
        }
        self.client.send(json.dumps(file_dic).encode())
        file_size_byte = self.client.recv(1024)  # 传回的要么是文件大小，要么是状态码，下面就做判断
        if file_size_byte.decode() != "404 none":
            file_size = int(file_size_byte.decode())
            self.client.send(b"100")
            recvd_size = 0
            if os.path.isfile(filename):
                f = open(filename + ".new","wb")
            else:
                f = open(filename,"wb")
            while recvd_size < file_size:  # 判断已接收的文件大小和文件总大小
                data = self.client.recv(1024)
                f.write(data)
                recvd_size += len(data)
                self.progress(recvd_size,file_size)  # 进度条
            else:
                f.close()
                print("
file has download")
        else:
            print("file no exist")

    def exit(self,*args):
        # 退出当前用户
        self.userlogin()

    def ls(self,*args):
        """查看用户家目录下的所有文件"""
        ls_dic = {
            "username":user,
            "action":"ls",
        }
        self.client.send(json.dumps(ls_dic).encode())
        data = self.client.recv(1024)
        if data.decode() != "130 none":  # 要么返回列表，要么返回状态码
            ls_list = json.loads(data.decode())
            for i in ls_list:
                print(i)

ftp = FtpClient()
ftp.connection("localhost",9998)
ftp.userlogin()