Index: webrtc/tools/network_tester/config_reader.cc |
diff --git a/webrtc/tools/network_tester/config_reader.cc b/webrtc/tools/network_tester/config_reader.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..21603e819f567f2c770bf164e63cd226ebe238c9 |
--- /dev/null |
+++ b/webrtc/tools/network_tester/config_reader.cc |
@@ -0,0 +1,71 @@ |
+/* |
+ * Copyright 2017 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 "webrtc/tools/network_tester/config_reader.h" |
+ |
+#include <string> |
+#include <vector> |
+ |
+#include "webrtc/base/ignore_wundef.h" |
+#include "webrtc/base/logging.h" |
+ |
+#ifdef WEBRTC_NETWORK_TESTER_PROTO |
+RTC_PUSH_IGNORING_WUNDEF() |
+#include "webrtc/tools/network_tester/network_tester_config.pb.h" |
+RTC_POP_IGNORING_WUNDEF() |
+#endif // WEBRTC_NETWORK_TESTER_PROTO |
+ |
+namespace webrtc { |
+ |
+namespace { |
+rtc::Optional<int> ParseMessageSize(std::istream& config_stream) { |
+ int byte = config_stream.get(); |
+ if (config_stream.eof()) { |
+ return rtc::Optional<int>(); |
+ } |
+ RTC_DCHECK(0 <= byte && byte <= 255); |
minyue-webrtc
2017/04/03 08:05:19
nit
RTC_DCHECK_LE
RTC_DCHECK_GE
for better error m
michaelt
2017/04/03 13:05:39
Done.
|
+ return rtc::Optional<int>(byte); |
+} |
+} |
+ |
+ConfigReader::ConfigReader(const std::string& config_file_path) |
+ : config_stream_(config_file_path, |
+ std::ios_base::in | std::ios_base::binary) { |
+ RTC_DCHECK(config_stream_.is_open()); |
+ RTC_DCHECK(config_stream_.good()); |
+} |
+ |
+ConfigReader::~ConfigReader() = default; |
+ |
+rtc::Optional<ConfigReader::Config> ConfigReader::GetNextConfig() { |
+#ifdef WEBRTC_NETWORK_TESTER_PROTO |
+ auto message_size = ParseMessageSize(config_stream_); |
+ if (!message_size) |
+ return rtc::Optional<Config>(); |
+ constexpr size_t kMaxEventSize = (1u << 16) - 1; |
+ std::vector<char> message_buffer(kMaxEventSize); |
+ config_stream_.read(message_buffer.data(), *message_size); |
+ RTC_DCHECK_EQ(config_stream_.gcount(), *message_size); |
+ webrtc::network_tester::config::NetworkTesterConfig proto_config; |
+ if (!proto_config.ParseFromArray(message_buffer.data(), *message_size)) |
+ return rtc::Optional<Config>(); |
+ RTC_DCHECK(proto_config.has_packet_send_interval_ms()); |
+ RTC_DCHECK(proto_config.has_packet_size()); |
+ RTC_DCHECK(proto_config.has_execution_time_ms()); |
+ Config config; |
+ config.packet_send_interval_ms = proto_config.packet_send_interval_ms(); |
+ config.packet_size = proto_config.packet_size(); |
+ config.execution_time_ms = proto_config.execution_time_ms(); |
+ return rtc::Optional<Config>(config); |
+#else |
+ return rtc::Optional<Config>(); |
+#endif // WEBRTC_NETWORK_TESTER_PROTO |
+} |
+ |
+} // namespace webrtc |