| # Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | |
| # | |
| # Use of this source code is governed by a BSD-style license | |
| # that can be found in the LICENSE file in the root of the source | |
| # tree. An additional intellectual property rights grant can be found | |
| # in the file PATENTS. All contributing project authors may | |
| # be found in the AUTHORS file in the root of the source tree. | |
| """Parses protobuf RTC dumps.""" | |
| from __future__ import division | |
| import struct | |
| import pyproto.logging.rtc_event_log.rtc_event_log_pb2 as rtc_pb | |
| class DataPoint(object): | |
| """Simple container class for RTP events.""" | |
| def __init__(self, rtp_header_str, packet_size, arrival_timestamp_us, | |
| incoming): | |
| """Builds a data point by parsing an RTP header, size and arrival time. | |
| RTP header structure is defined in RFC 3550 section 5.1. | |
| """ | |
| self.size = packet_size | |
| self.arrival_timestamp_ms = arrival_timestamp_us / 1000 | |
| self.incoming = incoming | |
| header = struct.unpack_from("!HHII", rtp_header_str, 0) | |
| (first2header_bytes, self.sequence_number, self.timestamp, | |
| self.ssrc) = header | |
| self.payload_type = first2header_bytes & 0b01111111 | |
| self.marker_bit = (first2header_bytes & 0b10000000) >> 7 | |
| def ParseProtobuf(file_path): | |
| """Parses RTC event log from protobuf file. | |
| Args: | |
| file_path: path to protobuf file of RTC event stream | |
| Returns: | |
| all RTP packet events from the event stream as a list of DataPoints | |
| """ | |
| event_stream = rtc_pb.EventStream() | |
| with open(file_path, "rb") as f: | |
| event_stream.ParseFromString(f.read()) | |
| return [ | |
| DataPoint(event.rtp_packet.header, event.rtp_packet.packet_length, | |
| event.timestamp_us, event.rtp_packet.incoming) | |
| for event in event_stream.stream if event.HasField("rtp_packet") | |
| ] |