为什么我的Protobuf消息只有在从Windows框发送到Linux框时才会被截断?

2024-05-19 01:07:06 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用一个简单的jane,简化的,同步Python服务器和客户机的组合,通过TCP连接发送Google Protobuf消息。服务器使用Python2.5.4和Google Protobuf 2.4.1在Windows XP上运行。客户端运行在Ubuntu 12.04服务器上,使用Python 2.7.3和googleprotobuf2.4.1。

当我在XP机器上运行服务器并尝试使用Ubuntu机器上的客户机与之交谈时,服务器可以很好地解析请求消息并发送响应消息,但是客户机遇到了一个google.protobuf.message.DecodeError: Truncated message错误。

如果我在XP机器上运行客户机和服务器,我不会遇到这个错误;如果我在Ubuntu机器上运行服务器,在XP机器上运行客户机,我也不会遇到这个错误。

似乎它可能与实际的Protobuf消息本身有关。。。如果我使用一个非常简单的Protobuf消息,比如下面示例中的消息,那么一切似乎都很好。但是,一旦我开始使用我们在生产中使用的实际Protobuf消息,我就会看到错误。不过,当服务器在Windows上而客户机在Ubuntu上时,我只得到错误信息,这看起来还是很奇怪的。

你知道会发生什么吗?!考虑到一些环境限制,我必须在Windows机器上使用Python 2.5和googleprotobuf2.4.1。

protobuf.proto示例

message Foo {
  required string name = 1;
}

message Bar {
  required string name = 1;
}

message Sucka {
  required string name = 1;
  repeated Foo    foos = 2;
  repeated Bar    bars = 3;
}

message Result {
  enum Exception {
    NONE    = 0;
    GENERIC = 1;
  }

  required Exception exception = 1;
  optional string    message   = 2;
  optional Sucka     sucka     = 3;
}

服务器.py

import signal, socket, struct

from protobuf import Sucka, Result

class TestServer():
  def __init__(self, host = '0.0.0.0', port = 9989):
    self.host = host
    self.port = port

    self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    self.socket.bind((self.host, self.port))
    self.socket.listen(1)

    self.running = False
    self.conn    = None

    self.run()

  def run(self):
    def handle_exit(signum, stack):
      print 'Shutting down server in response to signal %s' % (str(signum))
      self.running = False

    signal.signal(signal.SIGINT,  handle_exit)
    signal.signal(signal.SIGTERM, handle_exit)

    print 'Waiting on a connection'
    self.running = True

    while self.running:
      try:
        self.conn, addr = self.socket.accept()

        print 'Connection from:', addr

        size_t = self.conn.recv(4)
        size   = struct.unpack('!I', size_t)
        data   = self.conn.recv(size[0])

        if not data:
          raise Exception('No data... did the client close the connection?')

        result = Result()

        try:
          sucka = Sucka()
          sucka.ParseFromString(data)

          result.exception = result.NONE
          result.sucka.CopyFrom(sucka)
        except Exception, ex:
          print str(ex)
          result.exception = result.GENERIC
          result.message   = 'Exception: %s' % (str(ex))
        finally:
          if self.conn: self.conn.close()

    self.socket.close()

if __name__ == '__main__':
  s = TestServer()

客户端.py

import socket, struct, time

from protobuf import Sucka, Result

class TestClient():
  def __init__(self, sucka, host = '127.0.0.1', port = 9989):
    self.addr = (host, port)
    self.__call(sucka)

  def __call(self, sucka):
    size   = sucka.ByteSize()
    size_t = struct.pack('!I', size)

    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    sock.connect(self.addr)
    sock.send(size_t)
    sock.send(sucka.SerializeToString())

    size_t = sock.recv(4)
    size   = struct.unpack('!I', size_t)
    data   = sock.recv(size[0])

    result = Result()
    result.ParseFromString(data)

    if result.exception != Result.NONE:
      raise Exception(result.message)

    print str(result.sucka)

if __name__ == '__main__':
  import sys, os
  sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))

  from protobuf import Sucka

  sucka      = Sucka()
  sucka.name = 'testing'

  for i in range(0, 25):
    foo      = sucka.foos.add()
    foo.name = 'foo-%i-tester' % i

  for i in range(0, 25):
    bar      = sucka.bars.add()
    bar.name = 'bar-%i-tester' % i

  c = TestClient(sucka, '127.0.0.1', 9989)

Tags: nameself服务器机器消息messagesize客户机

热门问题