webrtc/rtc_tools/rtp_generator/main.cc
Benjamin Wright 87bbb91469 Add rtp_generator utility to rtc_tools.
This CL introduces a new rtp_generator tool that can be utilized to generate
.rtpdump files that can be replayed by the video_replayer. This allows
automated generation of corpus material for the new WebRTC RTP fuzzers in
addition to allowing anyone who is experimenting with a new RTP feature to
quickly debug issues.

It can be used as follows:
./rtp_generator --input_config=./rtc_tools/rtp_generator/configs/vp8.json --output_rtpdump=/tmp/vp8.rtpdump
./video_replay --config_file test/fuzzers/configs/replay_packet_fuzzer/vp8_config.json --input_file /tmp/vp8.rtpdump

It works by generating squares randomly on the screen for a given duration. This
initial version is very limited and doesn't support FEC, RED and other
configurations. I plan to extend it to support these in future CLs.

Bug: webrtc:10117
Change-Id: I31d3dbb6fad73c727145ead4e7d085113d11fc51
Reviewed-on: https://webrtc-review.googlesource.com/c/119964
Commit-Queue: Benjamin Wright <benwright@webrtc.org>
Reviewed-by: Fredrik Solenberg <solenberg@webrtc.org>
Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#26517}
2019-02-01 18:36:19 +00:00

50 lines
1.6 KiB
C++

/*
* Copyright (c) 2019 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.
*/
#include <stdlib.h>
#include <string>
#include "rtc_tools/rtp_generator/rtp_generator.h"
#include "rtc_tools/simple_command_line_parser.h"
int main(int argc, char* argv[]) {
const std::string usage =
"Generates custom configured rtpdumps for the purpose of testing.\n"
"Example Usage:\n"
"./rtp_generator --input_config=sender_config.json\n"
" --output_rtpdump=my.rtpdump\n";
webrtc::test::CommandLineParser cmd_parser;
cmd_parser.Init(argc, argv);
cmd_parser.SetUsageMessage(usage);
cmd_parser.SetFlag("input_config", "");
cmd_parser.SetFlag("output_rtpdump", "");
cmd_parser.ProcessFlags();
const std::string config_path = cmd_parser.GetFlag("input_config");
const std::string rtp_dump_path = cmd_parser.GetFlag("output_rtpdump");
if (cmd_parser.GetFlag("help") == "true" || rtp_dump_path.empty() ||
config_path.empty()) {
cmd_parser.PrintUsageMessage();
return EXIT_FAILURE;
}
absl::optional<webrtc::RtpGeneratorOptions> options =
webrtc::ParseRtpGeneratorOptionsFromFile(config_path);
if (!options.has_value()) {
return EXIT_FAILURE;
}
webrtc::RtpGenerator rtp_generator(*options);
rtp_generator.GenerateRtpDump(rtp_dump_path);
return EXIT_SUCCESS;
}