OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/test/field_trial.h" | 11 #include "webrtc/test/field_trial.h" |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 #include <cassert> | 14 #include <cassert> |
15 #include <cstdio> | 15 #include <cstdio> |
16 #include <cstdlib> | 16 #include <cstdlib> |
17 #include <map> | 17 #include <map> |
18 #include <string> | 18 #include <string> |
19 | 19 |
20 #include "webrtc/system_wrappers/include/field_trial.h" | 20 #include "webrtc/system_wrappers/include/field_trial.h" |
21 #include "webrtc/system_wrappers/include/field_trial_default.h" | |
22 | 21 |
23 namespace webrtc { | 22 namespace webrtc { |
24 namespace { | 23 namespace { |
| 24 // Clients of this library have show a clear intent to setup field trials by |
| 25 // linking with it. As so try to crash if they forget to call |
| 26 // InitFieldTrialsFromString before webrtc tries to access a field trial. |
25 bool field_trials_initiated_ = false; | 27 bool field_trials_initiated_ = false; |
| 28 std::map<std::string, std::string> field_trials_; |
26 } // namespace | 29 } // namespace |
27 | 30 |
| 31 namespace field_trial { |
| 32 std::string FindFullName(const std::string& trial_name) { |
| 33 assert(field_trials_initiated_); |
| 34 std::map<std::string, std::string>::const_iterator it = |
| 35 field_trials_.find(trial_name); |
| 36 if (it == field_trials_.end()) |
| 37 return std::string(); |
| 38 return it->second; |
| 39 } |
| 40 } // namespace field_trial |
| 41 |
28 namespace test { | 42 namespace test { |
29 // Note: this code is copied from src/base/metrics/field_trial.cc since the aim | 43 // Note: this code is copied from src/base/metrics/field_trial.cc since the aim |
30 // is to mimic chromium --force-fieldtrials. | 44 // is to mimic chromium --force-fieldtrials. |
31 void InitFieldTrialsFromString(const std::string& trials_string) { | 45 void InitFieldTrialsFromString(const std::string& trials_string) { |
32 static const char kPersistentStringSeparator = '/'; | 46 static const char kPersistentStringSeparator = '/'; |
33 | 47 |
34 // Catch an error if this is called more than once. | 48 // Catch an error if this is called more than once. |
35 assert(!field_trials_initiated_); | 49 assert(!field_trials_initiated_); |
36 field_trials_initiated_ = true; | 50 field_trials_initiated_ = true; |
37 | 51 |
38 if (trials_string.empty()) | 52 if (trials_string.empty()) |
39 return; | 53 return; |
40 | 54 |
41 size_t next_item = 0; | 55 size_t next_item = 0; |
42 std::map<std::string, std::string> field_trials; | |
43 while (next_item < trials_string.length()) { | 56 while (next_item < trials_string.length()) { |
44 size_t name_end = trials_string.find(kPersistentStringSeparator, next_item); | 57 size_t name_end = trials_string.find(kPersistentStringSeparator, next_item); |
45 if (name_end == trials_string.npos || next_item == name_end) | 58 if (name_end == trials_string.npos || next_item == name_end) |
46 break; | 59 break; |
47 size_t group_name_end = trials_string.find(kPersistentStringSeparator, | 60 size_t group_name_end = trials_string.find(kPersistentStringSeparator, |
48 name_end + 1); | 61 name_end + 1); |
49 if (group_name_end == trials_string.npos || name_end + 1 == group_name_end) | 62 if (group_name_end == trials_string.npos || name_end + 1 == group_name_end) |
50 break; | 63 break; |
51 std::string name(trials_string, next_item, name_end - next_item); | 64 std::string name(trials_string, next_item, name_end - next_item); |
52 std::string group_name(trials_string, name_end + 1, | 65 std::string group_name(trials_string, name_end + 1, |
53 group_name_end - name_end - 1); | 66 group_name_end - name_end - 1); |
54 next_item = group_name_end + 1; | 67 next_item = group_name_end + 1; |
55 | 68 |
56 // Fail if duplicate with different group name. | 69 // Fail if duplicate with different group name. |
57 if (field_trials.find(name) != field_trials.end() && | 70 if (field_trials_.find(name) != field_trials_.end() && |
58 field_trials.find(name)->second != group_name) { | 71 field_trials_.find(name)->second != group_name) |
59 break; | 72 break; |
60 } | |
61 | 73 |
62 field_trials[name] = group_name; | 74 field_trials_[name] = group_name; |
63 | 75 |
64 // Successfully parsed all field trials from the string. | 76 // Successfully parsed all field trials from the string. |
65 if (next_item == trials_string.length()) { | 77 if (next_item == trials_string.length()) |
66 webrtc::field_trial::InitFieldTrialsFromString(trials_string.c_str()); | |
67 return; | 78 return; |
68 } | |
69 } | 79 } |
70 // Using fprintf as LOG does not print when this is called early in main. | 80 // Using fprintf as LOG does not print when this is called early in main. |
71 fprintf(stderr, "Invalid field trials string.\n"); | 81 fprintf(stderr, "Invalid field trials string.\n"); |
72 | 82 |
73 // Using abort so it crashes in both debug and release mode. | 83 // Using abort so it crashes in both debug and release mode. |
74 abort(); | 84 abort(); |
75 } | 85 } |
76 | 86 |
77 ScopedFieldTrials::ScopedFieldTrials(const std::string& config) | 87 ScopedFieldTrials::ScopedFieldTrials(const std::string& config) |
78 : previous_field_trials_(webrtc::field_trial::GetFieldTrialString()) { | 88 : previous_field_trials_(field_trials_) { |
79 assert(field_trials_initiated_); | 89 assert(field_trials_initiated_); |
80 field_trials_initiated_ = false; | 90 field_trials_initiated_ = false; |
81 current_field_trials_ = config; | 91 field_trials_.clear(); |
82 InitFieldTrialsFromString(current_field_trials_); | 92 InitFieldTrialsFromString(config); |
83 } | 93 } |
84 | 94 |
85 ScopedFieldTrials::~ScopedFieldTrials() { | 95 ScopedFieldTrials::~ScopedFieldTrials() { |
86 // Should still be initialized, since InitFieldTrials is called from ctor. | 96 // Should still be initialized, since InitFieldTrials is called from ctor. |
87 // That's why we don't restore the flag. | 97 // That's why we don't restore the flag. |
88 assert(field_trials_initiated_); | 98 assert(field_trials_initiated_); |
89 webrtc::field_trial::InitFieldTrialsFromString(previous_field_trials_); | 99 field_trials_ = previous_field_trials_; |
90 } | 100 } |
91 | 101 |
92 } // namespace test | 102 } // namespace test |
93 } // namespace webrtc | 103 } // namespace webrtc |
OLD | NEW |