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

写个东西并非无聊或者练手，而是厌烦了每次都得重头写。我已经不是第一次碰到下面的情况：远程到一台可以连接内网的机器，结果发现其环境极为恶劣，没有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
 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)