day47——使用 Socket 实现 FTP Server

使用 Socket 实现 FTP Server ，实现如下功能：

get /tmp/1.txt /tmp/1.txt ：把服务端 /tmp/1.txt 下载到客户端 /tmp/1.txt
put /tmp/1.txt /tmp/1.txt ：把客户端 /tmp/1.txt 上传到服务端 /tmp/1.txt

 1 [root@localhost ~]# cat ftp_server.py 
 2 #!/usr/bin/env python
 3 #-*- coding: utf8 -*-
 4 
 5 import socket
 6 from subprocess import Popen, PIPE
 7 
 8 HOST = ''
 9 PORT = 1234
10 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
11 s.bind((HOST, PORT))
12 s.listen(1)
13 conn, addr = s.accept()
14 print 'conn', conn
15 print 'Connected by', addr
16 while 1:
17     cmd = conn.recv(1024)
18     cmd_list = cmd.split()
19     if cmd_list[0] == 'get':
20         with open(cmd_list[1]) as fd:
21             while True:
22                 data = fd.read(1024)
23                 conn.sendall(data)
24                 if not data:
25                     conn.sendall('EOF')
26                     break
27     if not cmd: break
28 conn.close()

 1 [root@localhost ~]# cat ftp_client.py 
 2 #!/usr/bin/env python
 3 #-*- coding:utf-8 -*-
 4 
 5 import os
 6 import sys
 7 import tab
 8 import socket
 9 
10 HOST = '192.168.5.131'
11 PORT = 1234
12 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
13 s.connect((HOST, PORT))
14 while True:
15     cmd = raw_input("Please input cmd: ").strip()
16     if cmd.lower()== 'exit' or cmd.lower() == 'quit':
17         break
18     cmd_list = cmd.split()
19     if len(cmd_list) != 3:
20         print "Ex: get file1 file2"
21         continue
22     else:
23         s.sendall(cmd)
24         if not os.path.exists(cmd_list[2]):
25             dst_file = cmd_list[2]
26         else:
27             dst_file = cmd_list[2]+'.new'
28         n = 1
29         while True:
30             data_rev = s.recv(1024)
31             if data_rev.endswith('EOF'):
32                 data = data_rev[:-3]
33             else:
34                 data = data_rev
35             if n == 1:
36                 with open(dst_file, 'wb') as fd:
37                     fd.write(data) 
38             else:
39                 with open(dst_file, 'a') as fd:
40                     fd.write(data)
41             n += 1
42             if data_rev[-3:] == 'EOF':
43                 print "Success! destination file is %s" % dst_file
44                 break
45 s.close()