/**
* 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;
}
}
/**
* 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;
}