用Python实现一个简单的文件传输协议

用Python实现一个简单的文件传输协议 - 但是博客很漂亮 - 博客频道 - CSDN.NET

用Python实现一个简单的文件传输协议

2012-11-04 00:09 36人阅读 评论(0) 收藏 举报

写个东西并非无聊或者练手，而是厌烦了每次都得重头写。我已经不是第一次碰到下面的情况：远程到一台可以连接内网的机器，结果发现其环境极为恶劣，没有scp。最夸张的一次，我见过一台机器连man都没装。所幸装了ssh可以让我远程。但没有scp怎么传文件呢？ftp？试了几个命令，没有。HTTP？客户端或者服务器端软件都没有。python？有的。尼玛，最后在远程机器上用python写了个简单的服务器，终于可以传文件了。当然，首先要传scp。

被整了两三回以后，我开始考虑把写的东西保存起来。这个程序最好小于一屏，以便我直接粘贴到远程终端。我考虑过实现一个已有协议，这样，在客户端可以用现成的工具。一番搜索，无果，看起来最简单的TFTP我都嫌太复杂了。于是我放弃这一想法，直接实现一个超原始的协议。最后写出来的客户端和服务器端代码长度都在一屏之内(23行;校准终端24行;vi的状态栏占了一行)。

我还是希望以后别遇到这种问题：用不能自动解决依赖的方法装软件太痛苦了。Python现在也像Perl一样无处不在，不得不感叹，有了这些脚本语言几乎相当于有了一切。

服务器端：

#!/usr/bin/python
import SocketServer

# Format: name_len      --- one byte
#         name          --- name_len bytes
#         data          --- variable length
# Save data to name into current directory
addr = ('', 1234)
class MyTCPHandler (SocketServer.StreamRequestHandler):
        def handle (self):
                name_len = ord(self.rfile.read(1))
                name = self.rfile.read(name_len)
                print "Get request:%s"%name
                fd = open(name, 'w')
                cont = self.rfile.read(4096)
                while cont:
                        fd.write(cont)
                        cont = self.rfile.read(4096)
                fd.close()
                print "Out :%s"%name

server = SocketServer.TCPServer(addr, MyTCPHandler)
server.serve_forever()

#!/usr/bin/python
import SocketServer

# Format: name_len      --- one byte
#         name          --- name_len bytes
#         data          --- variable length
# Save data to name into current directory
addr = ('', 1234)
class MyTCPHandler (SocketServer.StreamRequestHandler):
        def handle (self):
                name_len = ord(self.rfile.read(1))
                name = self.rfile.read(name_len)
                print "Get request:%s"%name
                fd = open(name, 'w')
                cont = self.rfile.read(4096)    
                while cont:
                        fd.write(cont)
                        cont = self.rfile.read(4096)
                fd.close()
                print "Out :%s"%name

server = SocketServer.TCPServer(addr, MyTCPHandler)
server.serve_forever()

客户端：

#!/usr/bin/python
from socket import *
import os.path
import sys
target = ('',1234)
def get_header (name):
        leng = len(name)
        assert leng < 250
        return chr(leng) + name

def send_file (name):
        basename = os.path.basename(name)
        header = get_header(basename)
        cont = open(name).read()
        s = socket (AF_INET, SOCK_STREAM)
        s.connect(target)
        s.sendall (header)
        s.sendall (cont)
        s.close()

for i in sys.argv[1:]:
        print i
        send_file (i)