Beispiel #1
0
    def phase_two(self):

        hash = None
        i = -1
        for message in self.phase_two_messages:
            i += 1

            # Sent ANP transport header
            log.debug("Phase 2: sending ANPT header %i, size %i." % ( i, message.size ))
            self.writer.send_command_header(kanp.KANP_CMD_KFS_PHASE_2, message.size)
            log.debug("Phase 2: sent ANPT header %i, size %i." % ( i, message.size ))

            # Send base message anp message.

            kanp.send_anpt_msg(self.writer, message.anpm)

            if not hash:
                hash = hashlib.md5()

            j = -1
            for subm in message.sub_messages:
                j += 1
                if isinstance(subm, PhaseTwoChunkSubMessage):
                    # send chunk
                    log.debug("Phase 2: preparing file %i chunk %i anp message." % ( i, j ))
                    bytes = subm.chunk.read()
                    hash.update(bytes)
                    subm.anpm.add_bin(bytes)
                    log.debug("Phase 2: sending file %i chunk %i anp message." % ( i, j ))
                    kanp.send_anpt_msg(self.writer, subm.anpm)
                    log.debug("Phase 2: sent file %i chunk %i anp message." % ( i, j ))

                else:
                    assert isinstance(subm, PhaseTwoCommitSubMessage)
                    # send commit
                    log.debug("Phase 2: preparing file %i commit anp message." % ( i ))
                    bytes = hash.digest()
                    subm.anpm.add_bin(bytes)
                    hash = hashlib.md5()
                    log.debug("Phase 2: sending file %i commit anp message." % ( i ))
                    kanp.send_anpt_msg(self.writer, subm.anpm)
                    log.debug("Phase 2: sent file %i commit anp message." % ( i ))

            # get response
            log.debug("Phase 2: getting %i reply." % ( i ))
            h, m = kanp.get_anpt_all(self.reader)
            log.debug("Phase 2: got %i reply." % ( i ))
            if h.type == kanp.KANP_RES_FAIL:
                raise kanp.KANPFailure(m.get_u32(), m.get_str())
            assert h.type == kanp.KANP_RES_OK

        # get response
        h, m = kanp.get_anpt_all(self.reader)
        log.debug("Phase 2: got final reply.")
        if h.type == kanp.KANP_RES_FAIL:
            raise kanp.KANPFailure(m.get_u32(), m.get_str())
        assert h.type == kanp.KANP_RES_OK

        log.debug("File upload finished.")
Beispiel #2
0
    def phase_2_send_message_with_one_submessage(self, subm):

        # Prepare ANP message.
        message = PhaseTwoMessage()
        message.anpm = kanp.ANP_msg()
        message.anpm.add_u32(1) # Send only one sub-message

        # Calculate base messasge size.
        message.size = message.anpm.get_payload_size()
        #log.debug("Base message size: %i bytes." % ( message.size ))

        # Calculate total sub-message size.
        subm.size = subm.anpm.get_payload_size()
        log.debug("Chunk sub-message size: %i bytes." % ( subm.size ))

        total_size = message.size + subm.size

        # Sent ANP transport header
        #log.debug("Phase 2: sending ANPT header with data size %i." % ( total_size ))
        self.writer.send_command_header(kanp.KANP_CMD_KFS_PHASE_2, total_size)
        #log.debug("Phase 2: sent ANPT header, size %i." % ( total_size ))

        # Send base message.
        kanp.send_anpt_msg(self.writer, message.anpm)

        # Send sub-message.
        kanp.send_anpt_msg(self.writer, subm.anpm)

        # get response
        #log.debug("Phase 2: getting reply.")
        h, m = kanp.get_anpt_all(self.reader)
        #log.debug("ANP RESPONSE DUMP: %s" % (str(m.dump())))
        #log.debug("Phase 2: got reply.")
        if h.type == kanp.KANP_RES_FAIL:
            raise kanp.KANPFailure(m.get_u32(), m.get_str())
        assert h.type == kanp.KANP_RES_OK
Beispiel #3
0
    def phase_2_send_message_with_one_submessage(self, subm):

        # Prepare ANP message.
        message = PhaseTwoMessage()
        message.anpm = kanp.ANP_msg()
        message.anpm.add_u32(1)  # Send only one sub-message

        # Calculate base messasge size.
        message.size = message.anpm.get_payload_size()
        #log.debug("Base message size: %i bytes." % ( message.size ))

        # Calculate total sub-message size.
        subm.size = subm.anpm.get_payload_size()
        log.debug("Chunk sub-message size: %i bytes." % (subm.size))

        total_size = message.size + subm.size

        # Sent ANP transport header
        #log.debug("Phase 2: sending ANPT header with data size %i." % ( total_size ))
        self.writer.send_command_header(kanp.KANP_CMD_KFS_PHASE_2, total_size)
        #log.debug("Phase 2: sent ANPT header, size %i." % ( total_size ))

        # Send base message.
        kanp.send_anpt_msg(self.writer, message.anpm)

        # Send sub-message.
        kanp.send_anpt_msg(self.writer, subm.anpm)

        # get response
        #log.debug("Phase 2: getting reply.")
        h, m = kanp.get_anpt_all(self.reader)
        #log.debug("ANP RESPONSE DUMP: %s" % (str(m.dump())))
        #log.debug("Phase 2: got reply.")
        if h.type == kanp.KANP_RES_FAIL:
            raise kanp.KANPFailure(m.get_u32(), m.get_str())
        assert h.type == kanp.KANP_RES_OK
Beispiel #4
0
    def phase_two(self):

        hash = None
        i = -1
        for message in self.phase_two_messages:
            i += 1

            # Sent ANP transport header
            log.debug("Phase 2: sending ANPT header %i, size %i." %
                      (i, message.size))
            self.writer.send_command_header(kanp.KANP_CMD_KFS_PHASE_2,
                                            message.size)
            log.debug("Phase 2: sent ANPT header %i, size %i." %
                      (i, message.size))

            # Send base message anp message.

            kanp.send_anpt_msg(self.writer, message.anpm)

            if not hash:
                hash = hashlib.md5()

            j = -1
            for subm in message.sub_messages:
                j += 1
                if isinstance(subm, PhaseTwoChunkSubMessage):
                    # send chunk
                    log.debug(
                        "Phase 2: preparing file %i chunk %i anp message." %
                        (i, j))
                    bytes = subm.chunk.read()
                    hash.update(bytes)
                    subm.anpm.add_bin(bytes)
                    log.debug(
                        "Phase 2: sending file %i chunk %i anp message." %
                        (i, j))
                    kanp.send_anpt_msg(self.writer, subm.anpm)
                    log.debug("Phase 2: sent file %i chunk %i anp message." %
                              (i, j))

                else:
                    assert isinstance(subm, PhaseTwoCommitSubMessage)
                    # send commit
                    log.debug(
                        "Phase 2: preparing file %i commit anp message." % (i))
                    bytes = hash.digest()
                    subm.anpm.add_bin(bytes)
                    hash = hashlib.md5()
                    log.debug("Phase 2: sending file %i commit anp message." %
                              (i))
                    kanp.send_anpt_msg(self.writer, subm.anpm)
                    log.debug("Phase 2: sent file %i commit anp message." %
                              (i))

            # get response
            log.debug("Phase 2: getting %i reply." % (i))
            h, m = kanp.get_anpt_all(self.reader)
            log.debug("Phase 2: got %i reply." % (i))
            if h.type == kanp.KANP_RES_FAIL:
                raise kanp.KANPFailure(m.get_u32(), m.get_str())
            assert h.type == kanp.KANP_RES_OK

        # get response
        h, m = kanp.get_anpt_all(self.reader)
        log.debug("Phase 2: got final reply.")
        if h.type == kanp.KANP_RES_FAIL:
            raise kanp.KANPFailure(m.get_u32(), m.get_str())
        assert h.type == kanp.KANP_RES_OK

        log.debug("File upload finished.")