/**
* Copy value into the vector at offset, using ensureLength to make sure
* the vector has enough size.
* @param value The buffer to copy from.
* @param valueLength The length of the value buffer.
* @param offset The offset in the vector to copy to.
* @return The new offset which is offset + valueLength.
*/
size_t
copy(const uint8_t *value, size_t valueLength, size_t offset)
{
ndn_Error error;
if ((error = ndn_DynamicUInt8Array_copy(this, value, valueLength, offset)))
throw std::runtime_error(ndn_getErrorString(error));
return offset + valueLength;
}
ndn_Error ndn_ElementReader_onReceivedData
(struct ndn_ElementReader *self, uint8_t *data, size_t dataLength)
{
// Process multiple objects in the data.
while(1) {
ndn_Error error;
int gotElementEnd;
size_t offset;
if (!self->usePartialData) {
// This is the beginning of an element. Check whether it is binaryXML or TLV.
if (dataLength <= 0)
// Wait for more data.
return NDN_ERROR_success;
// The type codes for TLV Interest and Data packets are chosen to not
// conflict with the first byte of a binary XML packet, so we can
// just look at the first byte.
if (data[0] == ndn_Tlv_Interest || data[0] == ndn_Tlv_Data ||
data[0] == 0x80)
self->useTlv = 1;
else
// Binary XML.
self->useTlv = 0;
}
if (self->useTlv) {
// Scan the input to check if a whole TLV element has been read.
ndn_TlvStructureDecoder_seek(&self->tlvStructureDecoder, 0);
if ((error = ndn_TlvStructureDecoder_findElementEnd(&self->tlvStructureDecoder, data, dataLength)))
return error;
gotElementEnd = self->tlvStructureDecoder.gotElementEnd;
offset = self->tlvStructureDecoder.offset;
}
else {
// Scan the input to check if a whole binary XML element has been read.
ndn_BinaryXmlStructureDecoder_seek(&self->binaryXmlStructureDecoder, 0);
if ((error = ndn_BinaryXmlStructureDecoder_findElementEnd(&self->binaryXmlStructureDecoder, data, dataLength)))
return error;
gotElementEnd = self->binaryXmlStructureDecoder.gotElementEnd;
offset = self->binaryXmlStructureDecoder.offset;
}
if (gotElementEnd) {
// Got the remainder of an element. Report to the caller.
if (self->usePartialData) {
// We have partial data from a previous call, so append this data and point to partialData.
if ((error = ndn_DynamicUInt8Array_copy(&self->partialData, data, offset, self->partialDataLength)))
return error;
self->partialDataLength += offset;
(*self->elementListener->onReceivedElement)(self->elementListener, self->partialData.array, self->partialDataLength);
// Assume we don't need to use partialData anymore until needed.
self->usePartialData = 0;
}
else
// We are not using partialData, so just point to the input data buffer.
(*self->elementListener->onReceivedElement)(self->elementListener, data, offset);
// Need to read a new object.
data += offset;
dataLength -= offset;
ndn_BinaryXmlStructureDecoder_initialize(&self->binaryXmlStructureDecoder);
ndn_TlvStructureDecoder_initialize(&self->tlvStructureDecoder);
if (dataLength == 0)
// No more data in the packet.
return NDN_ERROR_success;
// else loop back to decode.
}
else {
// Save remaining data for a later call.
if (!self->usePartialData) {
self->usePartialData = 1;
self->partialDataLength = 0;
}
if ((error = ndn_DynamicUInt8Array_copy(&self->partialData, data, dataLength, self->partialDataLength)))
return error;
self->partialDataLength += dataLength;
return NDN_ERROR_success;
}
}
}