Examples with NetFlowV9OptionTemplate org.graylog.plugins.netflow.v9.NetFlowV9OptionTemplate used on opensource projects

Example 1 with NetFlowV9OptionTemplate

use of org.graylog.plugins.netflow.v9.NetFlowV9OptionTemplate in project graylog2-server by Graylog2.

the class NetFlowCodec method decodeV9Packets.

@VisibleForTesting
List<NetFlowV9Packet> decodeV9Packets(ByteBuf buffer) throws InvalidProtocolBufferException {
    byte[] v9JournalEntry = new byte[buffer.readableBytes()];
    buffer.readBytes(v9JournalEntry);
    final NetFlowV9Journal.RawNetflowV9 rawNetflowV9 = NetFlowV9Journal.RawNetflowV9.parseFrom(v9JournalEntry);
    // parse all templates used in the packet
    final Map<Integer, NetFlowV9Template> templateMap = Maps.newHashMap();
    rawNetflowV9.getTemplatesMap().forEach((templateId, byteString) -> {
        final NetFlowV9Template netFlowV9Template = NetFlowV9Parser.parseTemplate(Unpooled.wrappedBuffer(byteString.toByteArray()), typeRegistry);
        templateMap.put(templateId, netFlowV9Template);
    });
    final NetFlowV9OptionTemplate[] optionTemplate = { null };
    rawNetflowV9.getOptionTemplateMap().forEach((templateId, byteString) -> {
        optionTemplate[0] = NetFlowV9Parser.parseOptionTemplate(Unpooled.wrappedBuffer(byteString.toByteArray()), typeRegistry);
    });
    return rawNetflowV9.getPacketsList().stream().map(bytes -> Unpooled.wrappedBuffer(bytes.toByteArray())).map(buf -> NetFlowV9Parser.parsePacket(buf, typeRegistry, templateMap, optionTemplate[0])).collect(Collectors.toList());
}