[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