Milter 协议(2)

class MilterServer(Protocol):
    def init_dataReceived(self):
        self.lastdata = ""
        self.len = 0

    def __init__(self):
        self.init_dataReceived()

    # 注意：实践中发现一次 recv 中包括了两个完整的 packet; 而且理论上存在
    # 一个 packet 需要两次 recv 才能取出来的可能
    def dataReceived(self, data):
        if len(data): print len(data)
        if self.lastdata != "":
            data = self.lastdata + data
        if self.len != 0:
            if self.len <= len(data):
                next = data[self.len:]
                self.proc(data[:self.len])
                self.init_dataReceived()
                if len(next) > 0:
                    self.dataReceived(next)
            else: #继续等待输入
                self.lastdata = data
        else:
            if len(data) < 4:
                self.lastdata = data
            else:
                self.len = struct.unpack(">i", data[:4])[0]
                self.lastdata = data[4:]
                print self.len, len(self.lastdata)
                self.dataReceived("")

    def proc(self, data):
        ...
        ...