bool dissect_protobuf_field(const FieldDescriptor* field, const Message* message, tvbuff_t *tvb, guint* offset, proto_tree *leaf)
{
int len = WireFormat::FieldByteSize( field, *message );
map<string, Handles*>::iterator it = g_mapHandles.find( field->full_name() );
if( it == g_mapHandles.end() )
{
return false; // bug
}
Handles* handles = it->second;
const Reflection *reflection = message->GetReflection();
const EnumValueDescriptor* enumDesc = NULL;
switch( field->cpp_type() )
{
case FieldDescriptor::CPPTYPE_UINT32:
proto_tree_add_uint( leaf, handles->p_id, tvb, *offset, len,
reflection->GetUInt32( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_INT32:
proto_tree_add_int( leaf, handles->p_id, tvb, *offset, len,
reflection->GetInt32( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_FLOAT:
proto_tree_add_float( leaf, handles->p_id, tvb, *offset, len,
reflection->GetFloat( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_UINT64:
proto_tree_add_uint64( leaf, handles->p_id, tvb, *offset, len,
reflection->GetUInt64( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_INT64:
proto_tree_add_int64( leaf, handles->p_id, tvb, *offset, len,
reflection->GetInt64( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_DOUBLE:
proto_tree_add_double( leaf, handles->p_id, tvb, *offset, len,
reflection->GetDouble( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_BOOL:
proto_tree_add_boolean( leaf, handles->p_id, tvb, *offset, len,
reflection->GetBool( *message, field ) );
break;
case FieldDescriptor::CPPTYPE_ENUM:
enumDesc = reflection->GetEnum( *message, field );
proto_tree_add_int_format_value( leaf, handles->p_id, tvb, *offset, len,
enumDesc->number(), "%d ( %s )", enumDesc->number(), enumDesc->name().c_str() );
break;
case FieldDescriptor::CPPTYPE_STRING:
proto_tree_add_string( leaf, handles->p_id, tvb, *offset, len,
reflection->GetString( *message, field ).c_str() );
break;
default:
proto_tree_add_item( leaf, handles->p_id, tvb, *offset, len, true );
};
*offset += len;
return true;
}
static guint
riemann_dissect_int64(proto_tree *riemann_tree, tvbuff_t *tvb, guint offset, int hf_index)
{
guint64 num;
guint len = 0;
num = riemann_get_guint64(tvb, offset, &len);
proto_tree_add_int64(riemann_tree, hf_index, tvb, offset, len, num);
return len;
}
bool dissect_protobuf_repeated_field(const FieldDescriptor* field, const Message* message, tvbuff_t *tvb, guint* offset, proto_tree *leaf, int iRepeatedIndex)
{
int len = 0;
string scratch;
if( !field->options().packed() )
{
len += WireFormat::TagSize( field->number(), field->type() );
}
map<string, Handles*>::iterator it = g_mapHandles.find( field->full_name() );
if( it == g_mapHandles.end() )
{
return false; // bug
}
Handles* handles = it->second;
const Reflection *reflection = message->GetReflection();
const EnumValueDescriptor* enumDesc = NULL;
switch( field->cpp_type() )
{
case FieldDescriptor::CPPTYPE_UINT32:
if( field->type() == FieldDescriptor::TYPE_FIXED32 )
{
len += WireFormatLite::kFixed32Size;
}
else
{
len += WireFormatLite::UInt32Size( reflection->GetRepeatedUInt32( *message, field, iRepeatedIndex ) );
}
proto_tree_add_uint( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedUInt32( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_INT32:
if( field->type() == FieldDescriptor::TYPE_SFIXED32 )
{
len += WireFormatLite::kSFixed32Size;
}
else if( field->type() == FieldDescriptor::TYPE_SINT32 )
{
len += WireFormatLite::SInt32Size( reflection->GetRepeatedInt32( *message, field, iRepeatedIndex ) );
}
else
{
len += WireFormatLite::Int32Size( reflection->GetRepeatedInt32( *message, field, iRepeatedIndex ) );
}
proto_tree_add_int( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedInt32( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_FLOAT:
len += WireFormatLite::kFloatSize;
proto_tree_add_float( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedFloat( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_UINT64:
if( field->type() == FieldDescriptor::TYPE_FIXED64 )
{
len += WireFormatLite::kFixed64Size;
}
else
{
len += WireFormatLite::UInt64Size( reflection->GetRepeatedUInt64( *message, field, iRepeatedIndex ) );
}
proto_tree_add_uint64( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedUInt64( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_INT64:
if( field->type() == FieldDescriptor::TYPE_SFIXED64 )
{
len += WireFormatLite::kSFixed64Size;
}
else if( field->type() == FieldDescriptor::TYPE_SINT64 )
{
len += WireFormatLite::SInt64Size( reflection->GetRepeatedInt64( *message, field, iRepeatedIndex ) );
}
else
{
len += WireFormatLite::Int64Size( reflection->GetRepeatedInt64( *message, field, iRepeatedIndex ) );
}
proto_tree_add_int64( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedInt64( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_DOUBLE:
len += WireFormatLite::kDoubleSize;
proto_tree_add_double( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedDouble( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_BOOL:
len += WireFormatLite::kBoolSize;
proto_tree_add_boolean( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedBool( *message, field, iRepeatedIndex ) );
break;
case FieldDescriptor::CPPTYPE_ENUM:
enumDesc = reflection->GetRepeatedEnum( *message, field, iRepeatedIndex );
len += WireFormatLite::EnumSize( enumDesc->number() );
proto_tree_add_int_format_value( leaf, handles->p_id, tvb, *offset, len,
enumDesc->number(), "%d ( %s )", enumDesc->number(),
enumDesc->name().c_str() );
break;
case FieldDescriptor::CPPTYPE_STRING:
len += WireFormatLite::StringSize( reflection->GetRepeatedStringReference( *message, field, iRepeatedIndex, &scratch ) );
proto_tree_add_string( leaf, handles->p_id, tvb, *offset, len,
reflection->GetRepeatedString( *message, field, iRepeatedIndex ).c_str() );
break;
default:
proto_tree_add_item( leaf, handles->p_id, tvb, *offset, len, true );
};
*offset += len;
return true;
}
