OLD | NEW |
---|---|
(Empty) | |
1 /* | |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 | |
11 #include <iostream> | |
12 // #include <memory> | |
ivoc
2017/02/06 16:38:11
Please remove if it's not needed.
terelius
2017/02/06 17:09:43
Done.
| |
13 #include <sstream> | |
14 #include <string> | |
15 | |
16 #include "gflags/gflags.h" | |
17 #include "webrtc/base/checks.h" | |
18 #include "webrtc/call/call.h" | |
19 #include "webrtc/common_types.h" | |
20 #include "webrtc/logging/rtc_event_log/rtc_event_log_parser.h" | |
21 #include "webrtc/modules/rtp_rtcp/source/rtp_utility.h" | |
22 | |
23 namespace { | |
24 | |
25 DEFINE_bool(noincoming, false, "Excludes incoming packets."); | |
26 DEFINE_bool(nooutgoing, false, "Excludes incoming packets."); | |
ivoc
2017/02/06 16:38:11
I assume this should be "Excludes outgoing packets
terelius
2017/02/06 17:09:43
Done.
| |
27 // TODO(terelius): Note that the media type don't work with outgoing packets. | |
ivoc
2017/02/06 16:38:11
doesn't
terelius
2017/02/06 17:09:43
Done.
| |
28 DEFINE_bool(noaudio, false, "Excludes audio packets."); | |
29 // TODO(terelius): Note that the media type don't work with outgoing packets. | |
30 DEFINE_bool(novideo, false, "Excludes video packets."); | |
31 // TODO(terelius): Note that the media type don't work with outgoing packets. | |
32 DEFINE_bool(nodata, false, "Excludes data packets."); | |
33 DEFINE_bool(nortp, false, "Excludes RTP packets."); | |
34 DEFINE_bool(nortcp, false, "Excludes RTCP packets."); | |
35 // TODO(terelius): Allow a list of SSRCs. | |
36 DEFINE_string(ssrc, | |
37 "", | |
38 "Print only packets with this SSRC (decimal or hex, the latter " | |
39 "starting with 0x)."); | |
40 | |
41 // Parses the input string for a valid SSRC. If a valid SSRC is found, it is | |
42 // written to the output variable |ssrc|, and true is returned. Otherwise, | |
43 // false is returned. | |
44 // The empty string must be validated as true, because it is the default value | |
45 // of the command-line flag. In this case, no value is written to the output | |
46 // variable. | |
47 bool ParseSsrc(std::string str, uint32_t* ssrc) { | |
48 // If the input string starts with 0x or 0X it indicates a hexadecimal number. | |
49 auto read_mode = std::dec; | |
50 if (str.size() > 2 && | |
51 (str.substr(0, 2) == "0x" || str.substr(0, 2) == "0X")) { | |
52 read_mode = std::hex; | |
53 str = str.substr(2); | |
54 } | |
55 std::stringstream ss(str); | |
56 ss >> read_mode >> *ssrc; | |
57 return str.empty() || (!ss.fail() && ss.eof()); | |
58 } | |
59 | |
60 bool ExcludePacket(webrtc::PacketDirection direction, | |
61 webrtc::MediaType media_type, | |
62 uint32_t packet_ssrc, | |
63 uint32_t filtered_ssrc) { | |
64 if (FLAGS_nooutgoing && direction == webrtc::kOutgoingPacket) | |
65 return true; | |
66 if (FLAGS_noincoming && direction == webrtc::kIncomingPacket) | |
67 return true; | |
68 if (FLAGS_noaudio && media_type == webrtc::MediaType::AUDIO) | |
69 return true; | |
70 if (FLAGS_novideo && media_type == webrtc::MediaType::VIDEO) | |
71 return true; | |
72 if (FLAGS_nodata && media_type == webrtc::MediaType::DATA) | |
73 return true; | |
74 if (!FLAGS_ssrc.empty() && packet_ssrc != filtered_ssrc) | |
75 return true; | |
76 return false; | |
77 } | |
78 | |
79 std::string StreamInfo(webrtc::PacketDirection direction, | |
80 webrtc::MediaType media_type) { | |
81 if (direction == webrtc::kOutgoingPacket) { | |
82 if (media_type == webrtc::MediaType::AUDIO) | |
83 return "(out,audio)"; | |
84 if (media_type == webrtc::MediaType::VIDEO) | |
85 return "(out,video)"; | |
86 if (media_type == webrtc::MediaType::DATA) | |
87 return "(out,data)"; | |
88 } | |
89 if (direction == webrtc::kIncomingPacket) { | |
90 if (media_type == webrtc::MediaType::AUDIO) | |
91 return "(in,audio)"; | |
92 if (media_type == webrtc::MediaType::VIDEO) | |
93 return "(in,video)"; | |
94 if (media_type == webrtc::MediaType::DATA) | |
95 return "(in,data)"; | |
96 } | |
97 return "()"; | |
ivoc
2017/02/06 16:38:11
I think it would be good to add an RTC_NOTREACHED
terelius
2017/02/06 17:09:43
It isn't unreachable; both direction and media_typ
ivoc
2017/02/07 07:56:52
Acknowledged.
| |
98 } | |
99 | |
100 } // namespace | |
101 | |
102 // This utility will convert a stored event log to the rtpdump format. | |
ivoc
2017/02/06 16:38:11
Please update comment :-)
terelius
2017/02/06 17:09:43
Done.
| |
103 int main(int argc, char* argv[]) { | |
104 std::string program_name = argv[0]; | |
105 std::string usage = | |
106 "Tool for printing packet information from an RtcEventLog as text.\n" | |
107 "Run " + | |
108 program_name + | |
109 " --helpshort for usage.\n" | |
110 "Example usage:\n" + | |
111 program_name + " input.rel\n"; | |
112 google::SetUsageMessage(usage); | |
113 google::ParseCommandLineFlags(&argc, &argv, true); | |
114 | |
115 if (argc != 2) { | |
116 std::cout << google::ProgramUsage(); | |
117 return 0; | |
118 } | |
119 std::string input_file = argv[1]; | |
120 | |
121 uint32_t filtered_ssrc = 0; | |
122 if (!FLAGS_ssrc.empty()) | |
123 RTC_CHECK(ParseSsrc(FLAGS_ssrc, &filtered_ssrc)) | |
124 << "Flag verification has failed."; | |
125 | |
126 webrtc::ParsedRtcEventLog parsed_stream; | |
127 if (!parsed_stream.ParseFile(input_file)) { | |
128 std::cerr << "Error while parsing input file: " << input_file << std::endl; | |
129 return -1; | |
130 } | |
131 | |
132 for (size_t i = 0; i < parsed_stream.GetNumberOfEvents(); i++) { | |
133 // The parsed_stream will assert if the protobuf event is missing | |
134 // some required fields and we attempt to access them. We could consider | |
135 // a softer failure option, but it does not seem useful to generate | |
136 // RTP dumps based on broken event logs. | |
ivoc
2017/02/06 16:38:11
Please update comment.
terelius
2017/02/06 17:09:43
Done.
| |
137 if (!FLAGS_nortp && | |
138 parsed_stream.GetEventType(i) == webrtc::ParsedRtcEventLog::RTP_EVENT) { | |
139 size_t header_length; | |
140 size_t total_length; | |
141 uint8_t header[IP_PACKET_SIZE]; | |
142 webrtc::PacketDirection direction; | |
143 webrtc::MediaType media_type; | |
144 parsed_stream.GetRtpHeader(i, &direction, &media_type, header, | |
145 &header_length, &total_length); | |
146 | |
147 // Parse header to get SSRC and RTP time. | |
148 webrtc::RtpUtility::RtpHeaderParser rtp_parser(header, header_length); | |
149 webrtc::RTPHeader parsed_header; | |
150 rtp_parser.Parse(&parsed_header); | |
151 | |
152 if (ExcludePacket(direction, media_type, parsed_header.ssrc, | |
153 filtered_ssrc)) | |
154 continue; | |
155 | |
156 std::cout << parsed_stream.GetTimestamp(i) << "\tRTP" | |
157 << StreamInfo(direction, media_type) | |
158 << "\tSSRC=" << parsed_header.ssrc | |
159 << "\ttimestamp =" << parsed_header.timestamp << std::endl; | |
160 } | |
161 if (!FLAGS_nortcp && | |
162 parsed_stream.GetEventType(i) == | |
163 webrtc::ParsedRtcEventLog::RTCP_EVENT) { | |
164 size_t length; | |
165 uint8_t packet[IP_PACKET_SIZE]; | |
166 webrtc::PacketDirection direction; | |
167 webrtc::MediaType media_type; | |
168 parsed_stream.GetRtcpPacket(i, &direction, &media_type, packet, &length); | |
169 | |
170 // Parse header to get SSRC and RTP time. | |
171 webrtc::RtpUtility::RtpHeaderParser rtp_parser(packet, length); | |
172 webrtc::RTPHeader parsed_header; | |
173 rtp_parser.Parse(&parsed_header); | |
174 | |
175 if (ExcludePacket(direction, media_type, parsed_header.ssrc, | |
176 filtered_ssrc)) | |
177 continue; | |
178 | |
179 std::cout << parsed_stream.GetTimestamp(i) << "\tRTCP" | |
180 << StreamInfo(direction, media_type) | |
181 << "\tSSRC=" << parsed_header.ssrc | |
182 << "\ttimestamp =" << parsed_header.timestamp << std::endl; | |
183 } | |
184 } | |
185 return 0; | |
186 } | |
OLD | NEW |