python --利用Socket多线程的FTP程序

python –利用Socket多线程的FTP程序

１.代码
服务器端代码：

root@kali:~/python/socket/ftp# pwd/root/python/socket/ftproot@kali:~/python/socket/ftp# lssocket_server_ftp1.py  socket_server_ftpok.py  tab.py  tab.pyc  xwb.txtroot@kali:~/python/socket/ftp# ls -a.  ..  socket_server_ftp1.py  socket_server_ftpok.py  tab.py  tab.pyc  xwb.txtroot@kali:~/python/socket/ftp# ls -la总用量 20drwxr-xr-x 2 root root 4096  7月 16 21:03 .drwxr-xr-x 3 root root 4096  7月 15 10:42 ..-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.py-rw-r--r-- 1 root root 3398  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pyc-rw-r--r-- 1 root root    0  7月 16 21:03 xwb.txtroot@kali:~/python/socket/ftp# cat socket_server_ftpok.py #!/usr/bin/python# --*-- coding:utf-8 --*--import SocketServerimport osfrom time import ctime,sleep#import socket#服务器端ＩＰ地址192.168.72.129HOST = ''PORT = 21567ADDR = (HOST,PORT)class MyRequestHandler(SocketServer.BaseRequestHandler):    def handle(self):#调用MyRequestHandler类必须使用handle函数        print '...connected from :', self.client_address        #print self.request.recv(1024)        #self.request.send('Username')        #----------Auth part------------------        if self.request.recv(1024) == 'auth':            print 'auth'            while 1:                self.request.send('Username')                username = self.request.recv(1024)                sleep(1)#可以处理验证账户延迟导致的死机                if username == 'alex':                    self.request.send('correct')                    print 'Correct! Welcome!!'                    break                else:                    self.request.send('incorrect')                    continue        def SendFromClient(filename):#服务器端发送文件函数，内置函数            print 'start receiving data'            f = file(filename,'wb')#二进制方式读取打开客户端刚刚发送过来的数据            while True:#一直循环接收客户端发送过来的数据                data = self.request.recv(4096)                if data == 'file_send_done':break#如果收到file_send_done信息，就结束                f.write(data)#否则把接收到的信息写入data文件中            f.close()            print 'file %s receive done!' % filename        def SendToClient(filename):#服务器端接收文件函数，内置函数            print 'start sending file to client....'            f = file(filename,'rb')            #while True:            file_data = f.read()            #if not file_data:break            self.request.sendall(file_data)            f.close()            print 'file %s sent to client finished!' % filename            sleep(0.5)            self.request.send('file_send_to_client_done')        #ftp()        while True:            try:                re_msg = self.request.recv(1024)                print 'get',re_msg                if re_msg.split()[0] == 'send':                    filename = re_msg.split()[1]                    self.request.send('ok2send')                    print 'ready to receive file from %s' % self.client_address[0]                    SendFromClient(filename)#调用发送文件函数                elif re_msg.split()[0] == 'get':                    filename = re_msg.split()[1]                    try:                        os.stat(filename)                    except OSError:                        msg = '\033[31;1mNo file %s found on FTP server\033[0m' % filename                        self.request.send(msg)                        print msg                    else:                        self.request.send('ok2get')                        sleep(0.5)                        print 'ready to send file to client %s' % self.client_address[0]                        SendToClient(filename)#调用接收文件函数                elif re_msg == 'help' or re_msg == '?':                    help_msg = '''\033[32;1m\nhelp\nget filename\tget file from FTP server\nsend filename\tsend file to FTP server\nls\t\tshow file list on FTP server\033[0m'''                    self.request.send(help_msg)                elif re_msg == 'ls':                    print 'print dir list',re_msg                    #file_list = os.listdir('.')                    #convert2string = '\t'.jion(file_list)                    file_list = os.popen('ls -lth')                    f_list = file_list.read()                    self.request.sendall(f_list)                else:                    print 'invalid instruction'                    self.request.send('\033[31;1minvalid_instruction\033[0m')                    print "get from %s : %s" % (self.client_address[0],re_msg)                    #self.request.sendall('[%s] %s' % (ctime(),re_msg))            except IndexError:                print "%s client %s logout !" % (ctime(),self.client_address[0])                breaktry:    tcpServ = SocketServer.ThreadingTCPServer(ADDR,MyRequestHandler)    print 'waiting for connection........'    tcpServ.serve_forever()except  socket.error,e:    print 'error happend!!'root@kali:~/python/socket/ftp# 

客户端代码：

root@kali:~/python/socket/ftp# pwd/root/python/socket/ftproot@kali:~/python/socket/ftp# ls -la总用量 36drwxr-xr-x 2 root root 4096  7月 16 21:02 .drwxr-xr-x 3 root root 4096  7月 15 09:39 ..-rw-r--r-- 1 root root  351  7月 15 17:46 socket_clent_ftp1.py-rw-r--r-- 1 root root 2311  7月 16 21:00 socket_clent_ftpok.py-rw-r--r-- 1 root root  382  7月 15 17:44 tab.py-rw-r--r-- 1 root root  547  7月 15 17:44 tab.pyc-rw-r--r-- 1 root root  408  7月 15 22:03 user_startup.py-rw-r--r-- 1 root root  555  7月 15 22:10 user_startup.pyc-rw-r--r-- 1 root root   14  7月 16 21:02 xwb.txtroot@kali:~/python/socket/ftp# cat socket_clent_ftpok.py #!/usr/bin/env python# --*-- coding:utf-8 --*--import tabimport user_startupimport osfrom time import sleepfrom socket import *#客户端Ip地址192.168.72.130HOST = '192.168.72.129'PORT = 21567BUFSIZ = 4096ADDR = (HOST,PORT)tcpCliSock = socket(AF_INET,SOCK_STREAM)tcpCliSock.connect(ADDR)tcpCliSock.send('auth')while 1:    #tcpCliSock.send('auth')    if tcpCliSock.recv(BUFSIZ) == 'Username':        print 'please input your username:'        while 1:            data = raw_input('Username:>').strip()            if len(data) == 0:continue            else:break        tcpCliSock.send(data)        if tcpCliSock.recv(BUFSIZ) == 'correct':            print 'welcome'            break        else:            print 'Wrong pass'            continuewhile 1:    data = raw_input('ftp>').strip()    if len(data) == 0:continue    if data == 'quit':        tcpCliSock.close()        break    if data == 'get' or data == 'send':        print '\033[31;1mNo file specified,usr %s filename \033[0m' % data        continue    if data == 'ls':        tcpCliSock.send(data)        file_list = tcpCliSock.recv(8096)        print file_list    if data.split()[0] == 'send':        try:            os.stat(data.split()[1])        except  OSError:            print '\033[31;1mNo file %s found on localhost\033[0m' % data.split()[1]            continue    tcpCliSock.send(data)    print 'send msg:', data    #tcpCliSock.send('%s\r\n' % data)    recv_data = tcpCliSock.recv(BUFSIZ)    if recv_data == 'ok2send':#向服务器上传输文件        file2send = data.split()[1]        f = open(file2send,'rb')#打开本地文件file2send        file_data = f.read()#把整个文件读取到内存中        f.close()        tcpCliSock.sendall(file_data)#sendall一次性读取的文件信息发送完成，以一条消息方式发送        print 'file sent finished!!'        sleep(0.5)#等待0.5秒        tcpCliSock.send('file_sent_done')    if recv_data == 'ok2get':#客户端发送文件到服务器        file2get = 'test/%s' % data.split()[1]        f = file(file2get,'wb')        file_get_done_mark = 0        while True:#不断给服务器发送文件            get_data = tcpCliSock.recv(1024)            if get_data == 'file_send_to_client_done':                file_get_done_mark = 1                break            f.write(get_data)        f.close()        if file_get_done_mark == 1:            print 'Download file %s from FTP server success!' % file2get            continue        else:            print 'wrong'        print 'File %s receive done!!' % filename    else:        #print 'invalid cmd'        print 'FTP server :',recv_dataroot@kali:~/python/socket/ftp# 

客户端导入user_start.py文件代码：

root@kali:~/python/socket/ftp# cat user_startup.py#!/usr/bin/python# --*-- coding:utf-8 --*--import sysimport readlineimport rlcompleterimport atexitimport os#tab completionreadline.parse_and_bind('tab: complete')#history filehistfile = os.path.join(os.environ['HOME'],'.pythonhistory')try:    readline.read_history_file(histfile)except IOError:    pass    atexit.register(readline.write_history_file,histfile)del os,histfile,readline,rlcompleterroot@kali:~/python/socket/ftp# 

２.运行情况
服务器端：

root@kali:~/python/socket/ftp# python socket_server_ftpok.py waiting for connection...........connected from : ('192.168.72.130', 53177)authCorrect! Welcome!!get lsprint dir list lsget lsprint dir list lsget ?get lsprint dir list lsget lsprint dir list lsget Sun Jul 16 21:02:20 2017 client 192.168.72.130 logout !...connected from : ('192.168.72.130', 53178)authCorrect! Welcome!!get lsprint dir list lsget lsprint dir list lsget ?get send xwb.txtready to receive file from 192.168.72.130start receiving data...connected from : ('192.168.72.130', 53181)authCorrect! Welcome!!get lsprint dir list lsget lsprint dir list ls＃运行前root@kali:~/python/socket/ftp# ls -la总用量 20drwxr-xr-x 2 root root 4096  7月 16 21:03 .drwxr-xr-x 3 root root 4096  7月 15 10:42 ..-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.py-rw-r--r-- 1 root root 3398  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pycroot@kali:~/python/socket/ftp# cat socket_server_ftpok.py ＃运行后root@kali:~/python/socket/ftp# ls -la总用量 20drwxr-xr-x 2 root root 4096  7月 16 21:03 .drwxr-xr-x 3 root root 4096  7月 15 10:42 ..-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.py-rw-r--r-- 1 root root 3398  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pyc-rw-r--r-- 1 root root    0  7月 16 21:03 xwb.txtroot@kali:~/python/socket/ftp# cat socket_server_ftpok.py 

客户端１：

root@kali:~/python/socket/ftp# python socket_clent_ftpok.py please input your username:Username:>ssWrong passplease input your username:Username:>alexwelcomeftp>ls总用量 12K-rw-r--r-- 1 root root 3.4K  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pyc-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.pysend msg: lsFTP server : 总用量 12K-rw-r--r-- 1 root root 3.4K  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pyc-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.pyftp>?send msg: ?FTP server : helpget filename    get file from FTP serversend filename   send file to FTP serverls      show file list on FTP serverftp>send xwb.txtsend msg: send xwb.txtfile sent finished!!FTP server : ok2sendftp>ls

客户端２：

root@kali:~/python/socket/ftp# python socket_clent_ftpok.py please input your username:Username:>alexwelcomeftp>ls总用量 12K-rw-r--r-- 1 root root    0  7月 16 21:03 xwb.txt-rw-r--r-- 1 root root 3.4K  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pyc-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.pysend msg: lsFTP server : 总用量 12K-rw-r--r-- 1 root root    0  7月 16 21:03 xwb.txt-rw-r--r-- 1 root root 3.4K  7月 16 21:00 socket_server_ftpok.py-rw-r--r-- 1 root root    0  7月 15 22:11 tab.py-rw-r--r-- 1 root root  547  7月 15 17:45 tab.pyc-rw-r--r-- 1 root root  660  7月 15 17:43 socket_server_ftp1.pyftp>