[terry笔记]python FTP

如下是作业，用python做一个ftp，主要利用socket。

server端在linux下运行，在client端可以执行shell命令（静态的）

在client端输入get xxx，即可下载。

在client端输入put xxx，即可上传。

server端：

import socket
import subprocess
import os
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.bind(("0.0.0.0",8000))
server.listen(5)
print("start to listen".center(30,"-"))

while True:
    conn,client_addr = server.accept()
    print(conn,client_addr)
    while True:
        try:
            data = conn.recv(1024)
            print("receive from client :",data)
            if (data.decode()).startswith("get"):
                data_cmd = data.decode()
                data_cmd_list = data_cmd.split(" ")
                file_name = data_cmd_list[-1]
                conn.send(str(os.path.getsize(file_name)).encode())
                f = open(file_name,"rb")
                f_data = f.readlines()
                for line in f_data:
                    conn.send(line)
            elif (data.decode()).startswith("put"):
                data_size = conn.recv(1024)
                if data_size.decode() == "no file":
                    continue
                print("The file's size is %sM" % round(int(data_size.decode()) / 1024 / 1024, 2))
                data_list = (data.decode()).split(" ")
                file_name = data_list[-1]
                f = open(file_name, "wb")
                total_size = int(data_size.decode())
                file_size = 0
                while True:
                    data2 = conn.recv(1024)
                    f.write(data2)
                    file_size += len(data2)
                    if file_size == total_size:
                        break
                f.close()
                print("file upload done")
            else:
                res_obj = subprocess.Popen(data,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
                res = res_obj.stdout.read()
                conn.send(str(len(res)).encode())
                print("--res len :",len(res))
                conn.send(res)
        except ConnectionResetError as e:
            print(client_addr,"is break")
            break
        except FileNotFoundError as e:
            print("there is no such file!")
            conn.send(b"no file")
            continue

client端：

import socket
import os
client = socket.socket()
client.connect(("192.168.168.128",8000))
# client.connect(("localhost",8000))

while True:
    try:
        msg = input(">>>>>").strip()
        if len(msg) == 0: continue
        client.send(msg.encode())
        if msg.startswith("get"):
            data = client.recv(1024)
            if data.decode() == "no file":
                print("there is no such file!")
                continue
            else:
                print("The file's size is %sM" % round(int(data.decode())/1024/1024,2))
                msg_list = msg.split(" ")
                file_name = msg_list[-1]
                f = open(file_name,"wb")
                total_size = int(data.decode())
                file_size = 0
                while True:
                    data2 = client.recv(1024)
                    f.write(data2)
                    file_size += len(data2)
                    if file_size == total_size:
                        break
                f.close()
        elif msg.startswith("put"):
            msg_cmd_list = msg.split(" ")
            file_name = msg_cmd_list[-1]
            client.send(str(os.path.getsize(file_name)).encode())
            f = open(file_name, "rb")
            f_data = f.readlines()
            for line in f_data:
                client.send(line)
        else:
            data = client.recv(1024)
            print("client receive:", data.decode())
            total_size = int(data.decode())
            received_size = 0
            res = b""
            while received_size < total_size:
                d = client.recv(1024)
                res += d
                received_size += len(d)
            print("-----receive done-----")
            print(res.decode())
    except FileNotFoundError as e:
        print("There is no such file!")
        client.send(b"no file")
        continue