OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/media/router/discovery/discovery_network_monitor.h" |
| 6 |
| 7 #include "base/lazy_instance.h" |
| 8 #include "base/logging.h" |
| 9 #include "base/sha1.h" |
| 10 #include "base/strings/string_number_conversions.h" |
| 11 #include "base/strings/string_util.h" |
| 12 #include "chrome/browser/media/router/discovery/discovery_network_list.h" |
| 13 #include "net/base/network_interfaces.h" |
| 14 |
| 15 namespace { |
| 16 |
| 17 using content::BrowserThread; |
| 18 |
| 19 std::vector<NetworkInfo> GetNetworkInfo() { |
| 20 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 21 |
| 22 net::NetworkInterfaceList interface_list; |
| 23 if (!net::GetNetworkList(&interface_list, |
| 24 net::INCLUDE_HOST_SCOPE_VIRTUAL_INTERFACES)) { |
| 25 DVLOG(2) << "net::GetNetworkList failed after a network change event"; |
| 26 return std::vector<NetworkInfo>(); |
| 27 } |
| 28 |
| 29 std::map<std::string, NetworkInfo> network_info_map; |
| 30 for (const auto& interface : interface_list) { |
| 31 if (interface.type != net::NetworkChangeNotifier::CONNECTION_ETHERNET && |
| 32 interface.type != net::NetworkChangeNotifier::CONNECTION_WIFI) { |
| 33 continue; |
| 34 } |
| 35 // Skip duplicate addresses for the same interface. |
| 36 if (network_info_map.find(interface.name) != network_info_map.end()) { |
| 37 continue; |
| 38 } |
| 39 network_info_map.insert(std::make_pair( |
| 40 interface.name, NetworkInfo{interface.interface_index, interface.type, |
| 41 interface.name, std::string("")})); |
| 42 } |
| 43 |
| 44 auto network_ids = GetDiscoveryNetworkIdList(); |
| 45 for (const auto& network_id : network_ids) { |
| 46 auto matching_info_it = network_info_map.find(network_id.interface_name); |
| 47 if (matching_info_it != end(network_info_map)) { |
| 48 matching_info_it->second.network_id = network_id.network_id; |
| 49 } |
| 50 } |
| 51 |
| 52 std::vector<NetworkInfo> network_info_list; |
| 53 std::transform(network_info_map.begin(), network_info_map.end(), |
| 54 std::back_inserter(network_info_list), |
| 55 [](const decltype(network_info_map)::value_type& entry) { |
| 56 return entry.second; |
| 57 }); |
| 58 return network_info_list; |
| 59 } |
| 60 |
| 61 base::LazyInstance<DiscoveryNetworkMonitor>::Leaky g_discovery_monitor; |
| 62 |
| 63 } // namespace |
| 64 |
| 65 NetworkInfo::NetworkInfo() {} |
| 66 |
| 67 NetworkInfo::NetworkInfo( |
| 68 uint32_t index, |
| 69 net::NetworkChangeNotifier::ConnectionType connection_type, |
| 70 std::string name, |
| 71 std::string network_id) |
| 72 : index(index), |
| 73 connection_type(connection_type), |
| 74 name(name), |
| 75 network_id(network_id) {} |
| 76 |
| 77 NetworkInfo::NetworkInfo(const NetworkInfo&) = default; |
| 78 |
| 79 NetworkInfo& NetworkInfo::operator=(const NetworkInfo&) = default; |
| 80 |
| 81 bool NetworkInfo::operator==(const NetworkInfo& o) const { |
| 82 return index == o.index && connection_type == o.connection_type && |
| 83 name == o.name && network_id == o.network_id; |
| 84 } |
| 85 |
| 86 bool NetworkInfo::operator!=(const NetworkInfo& o) const { |
| 87 return !(*this == o); |
| 88 } |
| 89 |
| 90 DiscoveryNetworkMonitor::DiscoveryNetworkMonitor() |
| 91 : network_id_(kNetworkIdDisconnected), |
| 92 observers_(new base::ObserverListThreadSafe<Observer>( |
| 93 base::ObserverListThreadSafe< |
| 94 Observer>::NotificationType::NOTIFY_EXISTING_ONLY)) { |
| 95 net::NetworkChangeNotifier::AddNetworkChangeObserver(this); |
| 96 } |
| 97 |
| 98 DiscoveryNetworkMonitor::~DiscoveryNetworkMonitor() { |
| 99 net::NetworkChangeNotifier::RemoveNetworkChangeObserver(this); |
| 100 } |
| 101 |
| 102 // static |
| 103 DiscoveryNetworkMonitor* DiscoveryNetworkMonitor::GetInstance() { |
| 104 DiscoveryNetworkMonitor* monitor = g_discovery_monitor.Pointer(); |
| 105 if (!monitor->network_info_callback_) { |
| 106 monitor->network_info_callback_ = base::Bind(&GetNetworkInfo); |
| 107 } |
| 108 return monitor; |
| 109 } |
| 110 |
| 111 // static |
| 112 DiscoveryNetworkMonitor* DiscoveryNetworkMonitor::GetInstanceForTest( |
| 113 NetworkInfoCallback strategy) { |
| 114 DiscoveryNetworkMonitor* monitor = g_discovery_monitor.Pointer(); |
| 115 monitor->network_info_callback_ = std::move(strategy); |
| 116 return monitor; |
| 117 } |
| 118 |
| 119 void DiscoveryNetworkMonitor::RebindNetworkChangeObserverForTest() { |
| 120 net::NetworkChangeNotifier::AddNetworkChangeObserver(this); |
| 121 } |
| 122 |
| 123 void DiscoveryNetworkMonitor::AddObserver(Observer* const observer) { |
| 124 observers_->AddObserver(observer); |
| 125 } |
| 126 |
| 127 void DiscoveryNetworkMonitor::RemoveObserver(Observer* const observer) { |
| 128 observers_->RemoveObserver(observer); |
| 129 } |
| 130 |
| 131 void DiscoveryNetworkMonitor::Refresh(NetworkRefreshCompleteCallback callback) { |
| 132 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 133 |
| 134 BrowserThread::PostTask( |
| 135 BrowserThread::IO, FROM_HERE, |
| 136 base::Bind(&DiscoveryNetworkMonitor::UpdateNetworkInfo, |
| 137 base::Unretained(this), callback)); |
| 138 } |
| 139 |
| 140 void DiscoveryNetworkMonitor::OnNetworkChanged( |
| 141 net::NetworkChangeNotifier::ConnectionType) { |
| 142 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 143 |
| 144 BrowserThread::PostTask( |
| 145 BrowserThread::IO, FROM_HERE, |
| 146 base::Bind(&DiscoveryNetworkMonitor::UpdateNetworkInfo, |
| 147 base::Unretained(this), NetworkRefreshCompleteCallback())); |
| 148 } |
| 149 |
| 150 // static |
| 151 const char DiscoveryNetworkMonitor::kNetworkIdDisconnected[] = "disconnected"; |
| 152 // static |
| 153 const char DiscoveryNetworkMonitor::kNetworkIdUnknown[] = "unknown"; |
| 154 |
| 155 std::string DiscoveryNetworkMonitor::ComputeNetworkId( |
| 156 const std::vector<NetworkInfo>& network_info_list) { |
| 157 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 158 |
| 159 if (network_info_list.size() == 0) { |
| 160 return kNetworkIdDisconnected; |
| 161 } |
| 162 std::string network_ids; |
| 163 for (const auto& network_info : network_info_list) { |
| 164 network_ids += network_info.network_id; |
| 165 } |
| 166 if (network_ids.size() == 0) { |
| 167 return kNetworkIdUnknown; |
| 168 } |
| 169 |
| 170 std::vector<std::string> id_list; |
| 171 std::transform( |
| 172 network_info_list.begin(), network_info_list.end(), |
| 173 std::back_inserter(id_list), |
| 174 [](const NetworkInfo& network_info) { return network_info.network_id; }); |
| 175 std::sort(id_list.begin(), id_list.end()); |
| 176 std::string combined_ids; |
| 177 for (const auto& id : id_list) { |
| 178 combined_ids = combined_ids + "!" + id; |
| 179 } |
| 180 |
| 181 std::string hash = base::SHA1HashString(combined_ids); |
| 182 return base::ToLowerASCII(base::HexEncode(hash.data(), hash.length())); |
| 183 } |
| 184 |
| 185 void DiscoveryNetworkMonitor::UpdateNetworkInfo( |
| 186 const NetworkRefreshCompleteCallback& callback) { |
| 187 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 188 |
| 189 auto network_info_list = network_info_callback_.Run(); |
| 190 auto network_id = ComputeNetworkId(network_info_list); |
| 191 |
| 192 networks_.swap(network_info_list); |
| 193 network_id_.swap(network_id); |
| 194 |
| 195 if (!std::is_permutation(networks_.begin(), networks_.end(), |
| 196 network_info_list.begin())) { |
| 197 observers_->Notify(FROM_HERE, &Observer::OnNetworksChanged, |
| 198 base::ConstRef(*this)); |
| 199 } |
| 200 if (callback) { |
| 201 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, callback); |
| 202 } |
| 203 } |
OLD | NEW |