Python：socket实现ftp程序

刚开始学习socket编程，还不是特熟练，码了好长时间，中间遇到许多问题，记录一下用socketserver写ftp

server端：

 #!/usr/bin/env python

 import socketserver,os

 class FTP(socketserver.BaseRequestHandler):
     def handle(self):
         def clientsend(filename):
             f = open('/ftp/%s' %filename,'w')
         #    while 1:
             data = self.request.recv(4096).decode("utf-8")
         #    if not data:break
             f.write(data)
             f.close()
             print('File %s  receive done!' % filename)
         def clientdownload(filename):
             f = open('/ftp/%s' %filename,'r')
             data = f.read()
             f.close()
             self.request.sendall(data.encode("utf-8"))
         print('listening connect from:',self.client_address)
         listmsg = 'list:\n1.help\t\t\tdispaly the list\n2.send file\t\tsend + filename\n3.download file\t\tdownload + filename\n4.ls\t\t\tShow files and permissions\n5.exit\n'
         self.request.sendall("Welcome my ftp,input \033[34;1m'help'\033[0m dispaly meun".encode("utf-8"))
         while 1:
             try:
                 re_msg = self.request.recv(1024).decode("utf-8")
                 if not re_msg:break
                 if re_msg == 'help' or re_msg == '?':
                     self.request.sendall(listmsg.encode("utf-8"))
                 elif re_msg.split()[0] == 'send':
                     filename = re_msg.split()[1]
                     clientsend(filename)
                 elif re_msg.split()[0] == 'ls':
                     ls_list = os.popen('ls -l /ftp').read().encode("utf-8")
                     self.request.sendall(ls_list)
                 elif re_msg.split()[0] == 'download':
                     filename = re_msg.split()[1]
                     clientdownload(filename)
                 else:
                     self.request.sendall('Invalid value'.encode("utf-8"))
             finally:
                 pass
 if __name__ == '__main__':　#只有程序执行时才会触发下面的语句，程序被导入调用时，下面的语句不会触发
     host = '192.168.101.128'
     port = 2222
     server = socketserver.ThreadingTCPServer((host,port),FTP)
     server.serve_forever()

client端：

 #!/usr/bin/env python
 import socket,sys

 host = '192.168.101.128'
 port = 2222

 s = socket.socket()
 s.connect((host,port))
 info = s.recv(8096).decode("utf-8")
 print(info)
 while 1:
     INPUT = str(input('input:'))
     if INPUT.split()[0] == 'send':
         s.send(INPUT.encode("utf-8"))
         filename = INPUT.split()[1]
         f = open(filename,'r')
         file_data = f.read()
         f.close()
         s.sendall(file_data.encode("utf-8"))
         print('file sent finished!')
     elif INPUT.split()[0] == 'download':
         s.send(INPUT.encode("utf-8"))
         filename = INPUT.split()[1]
         f = open(filename,'w')
         data = s.recv(8096).decode("utf-8")
         f.write(data)
         f.close()
         print('%s has been download!' %filename)
     elif INPUT.split()[0] == 'exit':
         sys.exit()
     else:
         s.send(INPUT.encode("utf-8"))
         data = s.recv(8096).decode("utf-8")
         print(data)
 s.close()

socket 仅支持单线程，socketserver 支持多线程

socketserver（支持多线程！）：

定义请求处理类时，需要继承socketserver.BaseRequestHandler，并且需要创建新的handle()函数来处理请求

python中的socketserver模块已经为我们初始化好了，直接调用即可

self.request.recv(bufsize).decode("utf-8")　　读取请求数据

self.request.sendall().encode("utf-8")　　　　发送数据

self.client_address　　　　　　　　　　　　客户端请求的地址

server = socketserver.ThreadingTCPServer((Host,Port),FTP)　　多线程！！！

server = socketserver.TCPServer((Host,Port),FTP)　　　　　　　　单线程

server.serve_forever()　　　　　　　　　　　持续监听，服务端一直运行，知道收到shutdown信号

socket（仅支持单线程）：

socket.AF_UNIX　　　　　　只能用于单一的Unix系统进程通信

socket.AF_INET　　　　　　服务器之间网络通信

socket.AF_INET6　　　　　  IPv6

socket.SOCK_STREAM　　  流式socket，for TCP

socket.SOCK_DGRAM　　　数据报式socket，for UDP

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

s.bind((Host,Port))

s.listen()

s.connect((Host,Port))

conn,address = s.accept()　　conn是新的套接字对象，可以用来接受和发送数据，address是客户端的地址

s.close()

s.recv(bufsize)　　　　　　　接受套接字的数据，数据以字符串的形式返回，bufsize是接收的最大数据量

s.sendall(string)　　　　　　 尝试发送所有数据，成功返回None，失败则抛出异常

s.getpeername()　　　　　　返回连接套接字的远程地址，返回值通常为元组

s.gettimeout()　　　　　　　 返回当前超时值，单位为秒，如果没有设置超时，则为None