Python学习之路-28 (基于socket实现文件上传)

粘包

在实现发送文件功能之前我们先来理解下粘包的问题，下面有两张图，我觉得很清晰的就可以理解到了。

• 正常情况下发送文件

1. 第一步： 客户端把获取到的文件总大小(size=65426)先放到缓冲区，然后发送给服务端
2. 第二步： 此时客户端接收到的文件总大小就是65426

• 粘包的问题下发送文件

1. 第一步： 客户端把获取到的文件总大小(size=65426)先放到缓冲区
2. 第二步： 此时可能由于文件读取太快，导致缓存区的内容还没有发送到服务端，客户端就把读取到的文件内容(hello)也放到缓存区；
3. 第三步： 然后客户端就把缓存区的全部内容都发送到服务端，那么客户端本来第一次应该接收到的数据室文件大小(size=65426)，但实际接收到的数据确实：65426+hello，那么这个流程就是粘包的问题；

• 解决粘包问题

如果出现粘包的问题，那么传输的数据就有问题了，如何解决这个问题呢？看下图：

1. 第一步： 客户端把文件大小放到缓冲区
2. 第二步： 放入缓冲区之后立刻陷入阻塞的状态，登台服务端回复已收到文件大小，此时是不会再向服务端发送任何数据的
3. 第三步： 缓存区的数据会发送到服务端
4. 第四步： 服务端接收到客户端发来的文件大小之后立刻回复客户端，说我收到你发过来的文件大小

文件上传：

server.py

import SocketServer
import subprocess
import string
import time


class MyTcpServer(SocketServer.BaseRequestHandler):
    def recvfile(self, filename):
        print "starting  file!"
        f = open(filename, 'wb')
        self.request.send('ready')
        while True:
            data = self.request.recv(4096)
            if data == 'EOF':
                print "recv file success!"
                break
            f.write(data)
        f.close()

    def sendfile(self, filename):
        print "starting send file!"
        self.request.send('ready')
        time.sleep(1)
        f = open(filename, 'rb')
        while True:
            data = f.read(4096)
            if not data:
                break
            self.request.send(data)
        f.close()
        time.sleep(1)
        self.request.send('EOF')
        print "send file success!"

    def handle(self):
        print "get connection from :", self.client_address
        while True:
            try:
                data = self.request.recv(4096)
                print "getreve data:", data
                if not data:
                    print "break the connection!"
                    break
                else:
                    action, filename = data.split()
                    if action == "put":
                        self.recvfile(filename)
                    elif action == 'get':
                        self.sendfile(filename)
                    else:
                        print "get error!"
                        continue
            except Exception, e:
                print "get error at:", e


if __name__ == "__main__":
    host = ''
    port = 60000
    s = SocketServer.ThreadingTCPServer((host, port), MyTcpServer)
    s.serve_forever()

client.py

import socket
import sys
import time

ip = '127.0.0.1'
port = 60000
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)


def recvfile(filename):
    print "server ready, now client rece file~~"
    f = open(filename, 'wb')
    while True:
        data = s.recv(4096)
        if data == 'EOF':
            print "recv file success!"
            break
        f.write(data)
    f.close()


def sendfile(filename):
    print "server ready, now client sending file~~"
    f = open(filename, 'rb')
    while True:
        data = f.read(4096)
        if not data:
            break
        s.sendall(data)
    f.close()
    time.sleep(1)
    s.sendall('EOF')
    print "send file success!"


def confirm(s, client_command):
    s.send(client_command)
    data = s.recv(4096)
    if data == 'ready':
        return True


try:
    s.connect((ip, port))
    while 1:
        client_command = raw_input(">>")
        if not client_command:
            continue
        action, filename = client_command.split()
        if action == 'put':
            if confirm(s, client_command):
                sendfile(filename)
            else:
                print "server get error!"
        elif action == 'get':
            if confirm(s, client_command):
                recvfile(filename)
            else:
                print "server get error!"
        else:
            print "command error!"
except socket.error, e:
    print "get error as", e
finally:
    s.close()