blob: 16ae458d50a9acb65325edbae09496ab12b72bac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
/*
* 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 "rtc_tools/network_tester/config_reader.h"
#include <fstream>
#include <iterator>
#include <string>
#include "rtc_base/checks.h"
namespace webrtc {
ConfigReader::ConfigReader(const std::string& config_file_path)
: proto_config_index_(0) {
std::ifstream config_stream(config_file_path,
std::ios_base::in | std::ios_base::binary);
RTC_DCHECK(config_stream.is_open())
<< "Config " << config_file_path << " open failed";
RTC_DCHECK(config_stream.good());
std::string config_data((std::istreambuf_iterator<char>(config_stream)),
(std::istreambuf_iterator<char>()));
if (config_data.size() > 0) {
proto_all_configs_.ParseFromString(config_data);
}
}
ConfigReader::~ConfigReader() = default;
absl::optional<ConfigReader::Config> ConfigReader::GetNextConfig() {
#ifdef WEBRTC_NETWORK_TESTER_PROTO
if (proto_config_index_ >= proto_all_configs_.configs_size())
return absl::nullopt;
auto proto_config = proto_all_configs_.configs(proto_config_index_++);
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 config;
#else
return absl::nullopt;
#endif // WEBRTC_NETWORK_TESTER_PROTO
}
} // namespace webrtc
|
