Problema com trançado python - envio de dados binários

Question

O que eu estou tentando fazer é bastante simples: enviar um arquivo do cliente para o servidor. Primeiro, o cliente envia informações sobre o arquivo - o tamanho do que é. Em seguida, ele envia o arquivo real.

Este é o que eu fiz até agora:

Server.py

from twisted.internet import reactor, protocol
from twisted.protocols.basic import LineReceiver

import pickle
import sys

class Echo(LineReceiver):

    def connectionMade(self):
        self.factory.clients.append(self)
        self.setRawMode()

    def connectionLost(self, reason):
        self.factory.clients.remove(self)

    def lineReceived(self, data):
        print "line", data

    def rawDataReceived(self, data):
            try:
                obj = pickle.loads(data)
                print obj
            except:
                print data

        #self.transport.write("wa2")

def main():
    """This runs the protocol on port 8000"""
    factory = protocol.ServerFactory()
    factory.protocol = Echo
    factory.clients = []
    reactor.listenTCP(8000,factory)
    reactor.run()

# this only runs if the module was *not* imported
if __name__ == '__main__':
    main()

Client.py

import pickle

from twisted.internet import reactor, protocol
import time
import os.path
from twisted.protocols.basic import LineReceiver

class EchoClient(LineReceiver):

    def connectionMade(self):
        file = "some file that is a couple of megs"
        filesize = os.path.getsize(file)
        self.sendLine(pickle.dumps({"size":filesize}))

        f = open(file, "rb")
        contents = f.read()
        print contents[:20]
        self.sendLine(contents[:20])
        f.close()

#        self.sendLine("hej")
#        self.sendLine("wa")

    def connectionLost(self, reason):
        print "connection lost"

class EchoFactory(protocol.ClientFactory):
    protocol = EchoClient

    def clientConnectionFailed(self, connector, reason):
        print "Connection failed - goodbye!"
        reactor.stop()

    def clientConnectionLost(self, connector, reason):
        print "Connection lost - goodbye!"
        reactor.stop()


# this connects the protocol to a server runing on port 8000
def main():
    f = EchoFactory()
    reactor.connectTCP("localhost", 8000, f)
    reactor.run()

# this only runs if the module was *not* imported
if __name__ == '__main__':
    main()

O servidor exibirá somente a saída do objeto desserializado:

{ 'size': 183574528L}

Como assim? O que aconteceu com os 20 caracteres do arquivo que eu queria para enviar?

Se o uso do "hej" e "wa" envia vez, vou levá-los ambos (na mesma mensagem, não duas vezes).

Alguém?

Answer 1

Você configurou seu servidor para o modo raw com setRawMode (), de modo que o retorno de chamada rawDataReceived está sendo chamado com os dados de entrada (não lineReceived). Se você imprimir os dados que você recebe em rawDataReceived, você vê tudo, incluindo o conteúdo do arquivo, mas como você chama de picles para desserializar os dados, ele está sendo ignorado.

Ou você mudar a maneira de enviar dados para o servidor (eu sugiro o formato netstring) ou você passar o conteúdo dentro do objeto picles serializado, e fazer isso em uma chamada.

self.sendLine(pickle.dumps({"size":filesize, 'content': contents[:20]}))