コード例 #1
0
  /**
   * This function consumes the primary block of the bundle. It is a virtual from BlockProcessor.
   *
   * @param bundle Bundle to set data after consuming
   * @param blcok Primary block to set data after consuming
   * @param buf Populated buffer to read data from for consuming
   * @param len Number of bytes to consume
   * @return Return number of bytes successfully consumed, In case of error return -1
   */
  public int consume(Bundle bundle, BlockInfo block, IByteBuffer buffer, int len) {

    int consumed = buffer.position();

    PrimaryBlock primary = new PrimaryBlock();

    //        buf.position(0);
    assert (!block.complete()) : TAG + ": consume() block already complete";

    Dictionary dict = bundle.recv_blocks().dict();

    IByteBuffer byte_buffer_temp = new SerializableByteBuffer(len);

    //      	byte_buffer_temp = BufferHelper.reserve(byte_buffer_temp, len);
    block.set_contents(byte_buffer_temp);

    BufferHelper.copy_data(
        byte_buffer_temp, byte_buffer_temp.position(), buffer, buffer.position(), len);
    byte_buffer_temp.position(byte_buffer_temp.position() + len);

    IByteBuffer buf_block_content = block.contents();

    int primary_len = len = buf_block_content.capacity() - buf_block_content.remaining();
    buf_block_content.position(0);

    Log.d(TAG, " primary_len: " + primary_len + " : len:" + len);

    assert (primary_len == len) : TAG + ":  consume() primary!=len";

    primary.set_version(buf_block_content.get());

    if (primary.version() != BundleProtocol.CURRENT_VERSION) {
      Log.e(
          TAG,
          String.format(
              "protocol version mismatch %s != %s",
              primary.version, BundleProtocol.CURRENT_VERSION));
      return -1;
    }
    len -= 1;

    try {
      // Grab the SDNVs representing the flags and the block length.
      len -= read_sdnv(buf_block_content, primary.processing_flags());
      len -= read_sdnv(buf_block_content, primary.block_length());

      Log.d(
          TAG,
          String.format(
              "parsed primary block: version %s length %s",
              primary.version(), block.data_length()));

      // Parse the flags.
      parse_bundle_flags(bundle, primary.processing_flags_value());
      parse_cos_flags(bundle, primary.processing_flags_value());
      parse_srr_flags(bundle, primary.processing_flags_value());

      // What remains in the buffer should now be equal to what the block-length
      // field advertised./
      assert (len == block.data_length()) : TAG + ": consume() data and block length not equal";

      // set data_offset

      block.set_data_offset(buf_block_content.position());
      block.set_data_length((int) primary.block_length_value());

      len -= read_sdnv(buf_block_content, primary.dest_scheme_offset());
      len -= read_sdnv(buf_block_content, primary.dest_ssp_offset());
      len -= read_sdnv(buf_block_content, primary.source_scheme_offset());
      len -= read_sdnv(buf_block_content, primary.source_ssp_offset());
      len -= read_sdnv(buf_block_content, primary.replyto_scheme_offset());
      len -= read_sdnv(buf_block_content, primary.replyto_ssp_offset());
      len -= read_sdnv(buf_block_content, primary.custodian_scheme_offset());
      len -= read_sdnv(buf_block_content, primary.custodian_ssp_offset());

      len -= read_sdnv(buf_block_content, primary.creation_time());
      if (primary.creation_time_value() > Integer.MAX_VALUE) {
        Log.e(
            TAG,
            String.format(
                "creation timestamp time is too large: %s", primary.creation_time_value()));
        return -1;
      }

      len -= read_sdnv(buf_block_content, primary.creation_sequence());
      if (primary.creation_sequence_value() > Integer.MAX_VALUE) {
        Log.e(
            TAG,
            String.format(
                "creation timestamp sequence is too large: %s", primary.creation_sequence()));
        return -1;
      }

      len -= read_sdnv(buf_block_content, primary.lifetime());
      if (primary.lifetime_value() > Integer.MAX_VALUE) {
        Log.e(TAG, String.format("lifetime is too large: %s", primary.lifetime));
        return -1;
      }

      len -= read_sdnv(buf_block_content, primary.dictionary_length());

      // Make sure that the creation timestamp parts and the lifetime fit into
      // a 32 bit integer.

      bundle.set_creation_ts(
          new BundleTimestamp(primary.creation_time_value(), primary.creation_sequence_value()));
      bundle.set_expiration((int) primary.lifetime_value());

      /*
       * Verify that we have the whole dictionary.
       */
      if (len < primary.dictionary_length_value()) {

        Log.e(
            TAG,
            String.format("primary block advertised incorrect length %s", block.data_length()));

        return -1;
      }

      /*
       * Make sure that the dictionary ends with a null byte./
       */
      if (buf_block_content.get(
              (int) (buf_block_content.position() + primary.dictionary_length_value() - 1))
          != '\0') {
        Log.e(TAG, "dictionary does not end with a NULL character! " + primary_len);
        return -1;
      }

      /*
       * Now use the dictionary buffer to parse out the various endpoint
       * identifiers, making sure that none of them peeks past the end
       * of the dictionary block.
       */
      IByteBuffer dictionary = buf_block_content;

      len -= primary.dictionary_length_value();

      Log.d(TAG, "Dict starting point :" + (primary_len - primary.dictionary_length_value()));
      //        dictionary.position((int)(primary_len-primary.dictionary_length_value()));

      dict.set_dict(dictionary, (int) primary.dictionary_length_value());

      Log.d(TAG, "Extract source :" + (primary_len - primary.dictionary_length_value()));

      if (!dict.extract_eid(
          bundle.source(), primary.source_scheme_offset(), primary.source_ssp_offset())) {
        Log.e(TAG, "Extract source fail:");
      } else {
        block.eid_list().add(bundle.source());
        Log.d(TAG, "Extract source :" + bundle.source().str());
      }

      if (!dict.extract_eid(
          bundle.dest(), primary.dest_scheme_offset(), primary.dest_ssp_offset())) {
        Log.e(TAG, "Extract dest fail:");
      } else {
        block.eid_list().add(bundle.dest());
        Log.d(TAG, "Extract dest :" + bundle.dest().str());
      }

      if (!dict.extract_eid(
          bundle.replyto(), primary.replyto_scheme_offset(), primary.replyto_ssp_offset())) {

        Log.e(TAG, "Extract reply fail :");
      } else {
        block.eid_list().add(bundle.replyto());
        Log.d(TAG, "Extract reply :" + bundle.replyto().str());
      }

      if (!dict.extract_eid(
          bundle.custodian(), primary.custodian_scheme_offset(), primary.custodian_ssp_offset())) {
        Log.e(TAG, "Extract custodian fail:");
      } else {
        block.eid_list().add(bundle.custodian());
        Log.d(TAG, "Extract custodian :" + bundle.custodian().str());
      }

      buf_block_content.position(
          (int) (buf_block_content.position() + primary.dictionary_length_value()));
      // If the bundle is a fragment, grab the fragment offset and original
      // bundle size (and make sure they fit in a 32 bit integer).
      if (bundle.is_fragment()) {

        int[] sdnv_buf = new int[1];
        sdnv_buf[0] = 0;

        len -= read_sdnv(buf_block_content, sdnv_buf);
        if ((int) sdnv_buf[0] > Integer.MAX_VALUE) {
          Log.e(TAG, String.format("fragment offset is too large: %s", sdnv_buf));
          return -1;
        }

        bundle.set_frag_offset(sdnv_buf[0]);
        sdnv_buf[0] = 0;

        len -= read_sdnv(buf_block_content, sdnv_buf);

        if (sdnv_buf[0] > Integer.MAX_VALUE) {
          Log.e(TAG, String.format("fragment original length is too large: %s", sdnv_buf));
          return -1;
        }

        bundle.set_orig_length(sdnv_buf[0]);

        Log.d(
            TAG,
            String.format(
                TAG,
                "parsed fragmentation info: offset %s orig_len %s",
                bundle.frag_offset(),
                bundle.orig_length()));
      }

      Log.d(TAG, "primary_len: " + primary_len + " : ln" + len + ": Consumed" + consumed);

      block.set_complete(true);

      return primary_len - len;

    } catch (BlockProcessorTooShortException e) {
      // revert position
      buf_block_content.position();
      return -1;
    }
  }
コード例 #2
0
  /**
   * This function consumes the payload block of the bundle. It is a virtual from BlockProcessor.
   *
   * @param bundle Bundle to set data after consuming
   * @param blcok Payload block to set data after consuming
   * @param buf Populated buffer to read data from for consuming
   * @param len Number of bytes to consume
   * @return Return number of bytes successfully consumed
   */
  public int consume(Bundle bundle, BlockInfo block, IByteBuffer buf, int len) {

    BlockInfoVec recv_blocks = bundle.recv_blocks();
    int consumed = 0;

    int[] flags = new int[1];
    if (block.data_offset() == 0) {
      int cc = super.consume_preamble(recv_blocks, block, buf, len, flags);
      if (cc == -1) {
        return -1;
      }

      buf.position(buf.position() + cc);
      len -= cc;
      consumed += cc;
      Log.d(TAG, "in len: " + len + " : cc" + cc + " " + bundle.payload().length());
      assert (bundle.payload().length() == 0) : TAG + "consume() bundle payload length is not 0";
    }

    Log.d(TAG, "out len: " + len + " : cc" + consumed + " " + bundle.payload().length());

    if (block.data_offset() == 0) {
      assert (len == 0) : TAG + ": consume() len!=0";
      return consumed;
    }

    // Special case for the simulator -- if the payload location is
    // NODATA, then we're done.
    if (bundle.payload().location().getCode() == location_t.NODATA.getCode()) {
      block.set_complete(true);
      return consumed;
    }

    // If we've consumed the length (because the data_offset is
    // non-zero) and the length is zero, then we're done.
    if (block.data_offset() != 0 && block.data_length() == 0) {
      block.set_complete(true);
      return consumed;
    }

    // Also bail if there's nothing left to do
    if (len == 0) {
      return consumed;
    }

    // Otherwise, the buffer should always hold just the preamble
    // since we store the rest in the payload file
    assert (block.contents().capacity() == block.data_offset())
        : TAG + ": consume() data_offset not equal to content capacity";

    // Now make sure there's still something left to do for the block,
    // otherwise it should have been marked as complete
    assert (block.data_length() > bundle.payload().length());

    int rcvd = (int) bundle.payload().length();
    int remainder = block.data_length() - rcvd;
    int tocopy;

    if (len >= remainder) {
      block.set_complete(true);
      tocopy = remainder;
    } else {
      tocopy = len;
    }

    bundle.payload().set_length(rcvd + tocopy);
    bundle.payload().write_data(buf, rcvd, tocopy);

    consumed += tocopy;

    Log.d(
        TAG,
        String.format(
            "consumed %s/%s (%s)",
            consumed, block.full_length(), block.complete() ? "complete" : "not complete"));

    return consumed;
  }