| // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "shill/wifi_endpoint.h" |
| |
| #include <algorithm> |
| |
| #include <base/stl_util.h> |
| #include <base/stringprintf.h> |
| #include <base/string_number_conversions.h> |
| #include <base/string_util.h> |
| #include <chromeos/dbus/service_constants.h> |
| |
| #include "shill/ieee80211.h" |
| #include "shill/logging.h" |
| #include "shill/metrics.h" |
| #include "shill/proxy_factory.h" |
| #include "shill/supplicant_bss_proxy_interface.h" |
| #include "shill/wifi.h" |
| #include "shill/wifi_endpoint.h" |
| #include "shill/wpa_supplicant.h" |
| |
| using std::map; |
| using std::set; |
| using std::string; |
| using std::vector; |
| |
| namespace shill { |
| |
| WiFiEndpoint::WiFiEndpoint(ProxyFactory *proxy_factory, |
| const WiFiRefPtr &device, |
| const string &rpc_id, |
| const map<string, ::DBus::Variant> &properties) |
| : frequency_(0), |
| physical_mode_(Metrics::kWiFiNetworkPhyModeUndef), |
| ieee80211w_required_(false), |
| proxy_factory_(proxy_factory), |
| device_(device), |
| rpc_id_(rpc_id) { |
| // XXX will segfault on missing properties |
| ssid_ = |
| properties.find(WPASupplicant::kBSSPropertySSID)->second. |
| operator std::vector<uint8_t>(); |
| bssid_ = |
| properties.find(WPASupplicant::kBSSPropertyBSSID)->second. |
| operator std::vector<uint8_t>(); |
| signal_strength_ = |
| properties.find(WPASupplicant::kBSSPropertySignal)->second. |
| reader().get_int16(); |
| map<string, ::DBus::Variant>::const_iterator it = |
| properties.find(WPASupplicant::kBSSPropertyFrequency); |
| if (it != properties.end()) |
| frequency_ = it->second.reader().get_uint16(); |
| |
| Metrics::WiFiNetworkPhyMode phy_mode = Metrics::kWiFiNetworkPhyModeUndef; |
| if (!ParseIEs(properties, &phy_mode, &vendor_information_, |
| &ieee80211w_required_, &country_code_)) { |
| phy_mode = DeterminePhyModeFromFrequency(properties, frequency_); |
| } |
| physical_mode_ = phy_mode; |
| |
| network_mode_ = ParseMode( |
| properties.find(WPASupplicant::kBSSPropertyMode)->second); |
| set_security_mode(ParseSecurity(properties, &security_flags_)); |
| has_rsn_property_ = ContainsKey(properties, WPASupplicant::kPropertyRSN); |
| has_wpa_property_ = ContainsKey(properties, WPASupplicant::kPropertyWPA); |
| |
| if (network_mode_.empty()) { |
| // XXX log error? |
| } |
| |
| ssid_string_ = string(ssid_.begin(), ssid_.end()); |
| WiFi::SanitizeSSID(&ssid_string_); |
| ssid_hex_ = base::HexEncode(&(*ssid_.begin()), ssid_.size()); |
| bssid_string_ = StringPrintf("%02x:%02x:%02x:%02x:%02x:%02x", |
| bssid_[0], bssid_[1], bssid_[2], |
| bssid_[3], bssid_[4], bssid_[5]); |
| bssid_hex_ = base::HexEncode(&(*bssid_.begin()), bssid_.size()); |
| } |
| |
| WiFiEndpoint::~WiFiEndpoint() {} |
| |
| void WiFiEndpoint::Start() { |
| supplicant_bss_proxy_.reset( |
| proxy_factory_->CreateSupplicantBSSProxy( |
| this, rpc_id_, WPASupplicant::kDBusAddr)); |
| } |
| |
| void WiFiEndpoint::PropertiesChanged( |
| const map<string, ::DBus::Variant> &properties) { |
| SLOG(WiFi, 2) << __func__; |
| bool should_notify = false; |
| map<string, ::DBus::Variant>::const_iterator properties_it = |
| properties.find(WPASupplicant::kBSSPropertySignal); |
| if (properties_it != properties.end()) { |
| signal_strength_ = properties_it->second.reader().get_int16(); |
| SLOG(WiFi, 2) << "WiFiEndpoint " << bssid_string_ << " signal is now " |
| << signal_strength_; |
| should_notify = true; |
| } |
| |
| properties_it = properties.find(WPASupplicant::kBSSPropertyMode); |
| if (properties_it != properties.end()) { |
| string new_mode = ParseMode(properties_it->second); |
| if (new_mode != network_mode_) { |
| network_mode_ = new_mode; |
| SLOG(WiFi, 2) << "WiFiEndpoint " << bssid_string_ << " mode is now " |
| << network_mode_; |
| should_notify = true; |
| } |
| } |
| |
| const char *new_security_mode = ParseSecurity(properties, &security_flags_); |
| if (new_security_mode != security_mode()) { |
| set_security_mode(new_security_mode); |
| SLOG(WiFi, 2) << "WiFiEndpoint " << bssid_string_ << " security is now " |
| << security_mode(); |
| should_notify = true; |
| } |
| |
| if (should_notify) { |
| device_->NotifyEndpointChanged(this); |
| } |
| } |
| |
| void WiFiEndpoint::UpdateSignalStrength(int16 strength) { |
| if (signal_strength_ == strength ) { |
| return; |
| } |
| |
| SLOG(WiFi, 2) << __func__ << ": signal strength " |
| << signal_strength_ << " -> " << strength; |
| signal_strength_ = strength; |
| device_->NotifyEndpointChanged(this); |
| } |
| |
| map<string, string> WiFiEndpoint::GetVendorInformation() const { |
| map<string, string> vendor_information; |
| if (!vendor_information_.wps_manufacturer.empty()) { |
| vendor_information[kVendorWPSManufacturerProperty] = |
| vendor_information_.wps_manufacturer; |
| } |
| if (!vendor_information_.wps_model_name.empty()) { |
| vendor_information[kVendorWPSModelNameProperty] = |
| vendor_information_.wps_model_name; |
| } |
| if (!vendor_information_.wps_model_number.empty()) { |
| vendor_information[kVendorWPSModelNumberProperty] = |
| vendor_information_.wps_model_number; |
| } |
| if (!vendor_information_.wps_device_name.empty()) { |
| vendor_information[kVendorWPSDeviceNameProperty] = |
| vendor_information_.wps_device_name; |
| } |
| if (!vendor_information_.oui_list.empty()) { |
| vector<string> oui_list; |
| set<uint32_t>::const_iterator it; |
| for (it = vendor_information_.oui_list.begin(); |
| it != vendor_information_.oui_list.end(); |
| ++it) { |
| oui_list.push_back( |
| StringPrintf("%02x-%02x-%02x", |
| *it >> 16, (*it >> 8) & 0xff, *it & 0xff)); |
| } |
| vendor_information[kVendorOUIListProperty] = |
| JoinString(oui_list, ' '); |
| } |
| return vendor_information; |
| } |
| |
| // static |
| uint32_t WiFiEndpoint::ModeStringToUint(const string &mode_string) { |
| if (mode_string == flimflam::kModeManaged) |
| return WPASupplicant::kNetworkModeInfrastructureInt; |
| else if (mode_string == flimflam::kModeAdhoc) |
| return WPASupplicant::kNetworkModeAdHocInt; |
| else |
| NOTIMPLEMENTED() << "Shill dos not support " << mode_string |
| << " mode at this time."; |
| return 0; |
| } |
| |
| const vector<uint8_t> &WiFiEndpoint::ssid() const { |
| return ssid_; |
| } |
| |
| const string &WiFiEndpoint::ssid_string() const { |
| return ssid_string_; |
| } |
| |
| const string &WiFiEndpoint::ssid_hex() const { |
| return ssid_hex_; |
| } |
| |
| const string &WiFiEndpoint::bssid_string() const { |
| return bssid_string_; |
| } |
| |
| const string &WiFiEndpoint::bssid_hex() const { |
| return bssid_hex_; |
| } |
| |
| const string &WiFiEndpoint::country_code() const { |
| return country_code_; |
| } |
| |
| const WiFiRefPtr &WiFiEndpoint::device() const { |
| return device_; |
| } |
| |
| int16_t WiFiEndpoint::signal_strength() const { |
| return signal_strength_; |
| } |
| |
| uint16 WiFiEndpoint::frequency() const { |
| return frequency_; |
| } |
| |
| uint16 WiFiEndpoint::physical_mode() const { |
| return physical_mode_; |
| } |
| |
| const string &WiFiEndpoint::network_mode() const { |
| return network_mode_; |
| } |
| |
| const string &WiFiEndpoint::security_mode() const { |
| return security_mode_; |
| } |
| |
| bool WiFiEndpoint::ieee80211w_required() const { |
| return ieee80211w_required_; |
| } |
| |
| bool WiFiEndpoint::has_rsn_property() const { |
| return has_rsn_property_; |
| } |
| |
| bool WiFiEndpoint::has_wpa_property() const { |
| return has_wpa_property_; |
| } |
| |
| // static |
| WiFiEndpoint *WiFiEndpoint::MakeOpenEndpoint(ProxyFactory *proxy_factory, |
| const WiFiRefPtr &wifi, |
| const string &ssid, |
| const string &bssid, |
| const string &network_mode, |
| uint16 frequency, |
| int16 signal_dbm) { |
| return MakeEndpoint(proxy_factory, wifi, ssid, bssid, network_mode, |
| frequency, signal_dbm, false, false); |
| } |
| |
| // static |
| WiFiEndpoint *WiFiEndpoint::MakeEndpoint(ProxyFactory *proxy_factory, |
| const WiFiRefPtr &wifi, |
| const string &ssid, |
| const string &bssid, |
| const string &network_mode, |
| uint16 frequency, |
| int16 signal_dbm, |
| bool has_wpa_property, |
| bool has_rsn_property) { |
| map <string, ::DBus::Variant> args; |
| ::DBus::MessageIter writer; |
| |
| writer = args[WPASupplicant::kBSSPropertySSID].writer(); |
| writer << vector<uint8_t>(ssid.begin(), ssid.end()); |
| |
| string bssid_nosep; |
| RemoveChars(bssid, ":", &bssid_nosep); |
| vector<uint8_t> bssid_bytes; |
| base::HexStringToBytes(bssid_nosep, &bssid_bytes); |
| writer = args[WPASupplicant::kBSSPropertyBSSID].writer(); |
| writer << bssid_bytes; |
| |
| args[WPASupplicant::kBSSPropertySignal].writer().append_int16(signal_dbm); |
| args[WPASupplicant::kBSSPropertyFrequency].writer().append_uint16(frequency); |
| args[WPASupplicant::kBSSPropertyMode].writer().append_string( |
| network_mode.c_str()); |
| |
| if (has_wpa_property) { |
| ::DBus::MessageIter writer; // local is required; see HACKING |
| map <string, string> empty_dict; |
| writer = args[WPASupplicant::kPropertyWPA].writer(); |
| writer << empty_dict; |
| } |
| if (has_rsn_property) { |
| ::DBus::MessageIter writer; // local is required; see HACKING |
| map <string, string> empty_dict; |
| writer = args[WPASupplicant::kPropertyRSN].writer(); |
| writer << empty_dict; |
| } |
| |
| return new WiFiEndpoint( |
| proxy_factory, wifi, bssid, args); // |bssid| fakes an RPC ID |
| } |
| |
| // static |
| const char *WiFiEndpoint::ParseMode(const string &mode_string) { |
| if (mode_string == WPASupplicant::kNetworkModeInfrastructure) { |
| return flimflam::kModeManaged; |
| } else if (mode_string == WPASupplicant::kNetworkModeAdHoc) { |
| return flimflam::kModeAdhoc; |
| } else if (mode_string == WPASupplicant::kNetworkModeAccessPoint) { |
| NOTREACHED() << "Shill does not support AP mode at this time."; |
| return NULL; |
| } else { |
| NOTREACHED() << "Unknown WiFi endpoint mode!"; |
| return NULL; |
| } |
| } |
| |
| // static |
| const char *WiFiEndpoint::ParseSecurity( |
| const map<string, ::DBus::Variant> &properties, SecurityFlags *flags) { |
| if (ContainsKey(properties, WPASupplicant::kPropertyRSN)) { |
| // TODO(quiche): check type before casting |
| const map<string, ::DBus::Variant> rsn_properties( |
| properties.find(WPASupplicant::kPropertyRSN)->second. |
| operator map<string, ::DBus::Variant>()); |
| set<KeyManagement> key_management; |
| ParseKeyManagementMethods(rsn_properties, &key_management); |
| flags->rsn_8021x = ContainsKey(key_management, kKeyManagement802_1x); |
| flags->rsn_psk = ContainsKey(key_management, kKeyManagementPSK); |
| } |
| |
| if (ContainsKey(properties, WPASupplicant::kPropertyWPA)) { |
| // TODO(quiche): check type before casting |
| const map<string, ::DBus::Variant> rsn_properties( |
| properties.find(WPASupplicant::kPropertyWPA)->second. |
| operator map<string, ::DBus::Variant>()); |
| set<KeyManagement> key_management; |
| ParseKeyManagementMethods(rsn_properties, &key_management); |
| flags->wpa_8021x = ContainsKey(key_management, kKeyManagement802_1x); |
| flags->wpa_psk = ContainsKey(key_management, kKeyManagementPSK); |
| } |
| |
| if (ContainsKey(properties, WPASupplicant::kPropertyPrivacy)) { |
| flags->privacy = properties.find(WPASupplicant::kPropertyPrivacy)->second. |
| reader().get_bool(); |
| } |
| |
| if (flags->rsn_8021x || flags->wpa_8021x) { |
| return flimflam::kSecurity8021x; |
| } else if (flags->rsn_psk) { |
| return flimflam::kSecurityRsn; |
| } else if (flags->wpa_psk) { |
| return flimflam::kSecurityWpa; |
| } else if (flags->privacy) { |
| return flimflam::kSecurityWep; |
| } else { |
| return flimflam::kSecurityNone; |
| } |
| } |
| |
| // static |
| void WiFiEndpoint::ParseKeyManagementMethods( |
| const map<string, ::DBus::Variant> &security_method_properties, |
| set<KeyManagement> *key_management_methods) { |
| if (!ContainsKey(security_method_properties, |
| WPASupplicant::kSecurityMethodPropertyKeyManagement)) { |
| return; |
| } |
| |
| // TODO(quiche): check type before cast |
| const vector<string> key_management_vec = |
| security_method_properties. |
| find(WPASupplicant::kSecurityMethodPropertyKeyManagement)->second. |
| operator vector<string>(); |
| for (vector<string>::const_iterator it = key_management_vec.begin(); |
| it != key_management_vec.end(); |
| ++it) { |
| if (EndsWith(*it, WPASupplicant::kKeyManagementMethodSuffixEAP, true)) { |
| key_management_methods->insert(kKeyManagement802_1x); |
| } else if ( |
| EndsWith(*it, WPASupplicant::kKeyManagementMethodSuffixPSK, true)) { |
| key_management_methods->insert(kKeyManagementPSK); |
| } |
| } |
| } |
| |
| // static |
| Metrics::WiFiNetworkPhyMode WiFiEndpoint::DeterminePhyModeFromFrequency( |
| const map<string, ::DBus::Variant> &properties, uint16 frequency) { |
| uint32_t max_rate = 0; |
| map<string, ::DBus::Variant>::const_iterator it = |
| properties.find(WPASupplicant::kBSSPropertyRates); |
| if (it != properties.end()) { |
| vector<uint32_t> rates = it->second.operator vector<uint32_t>(); |
| if (rates.size() > 0) |
| max_rate = rates[0]; // Rates are sorted in descending order |
| } |
| |
| Metrics::WiFiNetworkPhyMode phy_mode = Metrics::kWiFiNetworkPhyModeUndef; |
| if (frequency < 3000) { |
| // 2.4GHz legacy, check for tx rate for 11b-only |
| // (note 22M is valid) |
| if (max_rate < 24000000) |
| phy_mode = Metrics::kWiFiNetworkPhyMode11b; |
| else |
| phy_mode = Metrics::kWiFiNetworkPhyMode11g; |
| } else { |
| phy_mode = Metrics::kWiFiNetworkPhyMode11a; |
| } |
| |
| return phy_mode; |
| } |
| |
| // static |
| bool WiFiEndpoint::ParseIEs( |
| const map<string, ::DBus::Variant> &properties, |
| Metrics::WiFiNetworkPhyMode *phy_mode, |
| VendorInformation *vendor_information, |
| bool *ieee80211w_required, string *country_code) { |
| |
| map<string, ::DBus::Variant>::const_iterator ies_property = |
| properties.find(WPASupplicant::kBSSPropertyIEs); |
| if (ies_property == properties.end()) { |
| SLOG(WiFi, 2) << __func__ << ": No IE property in BSS."; |
| return false; |
| } |
| |
| vector<uint8_t> ies = ies_property->second.operator vector<uint8_t>(); |
| |
| |
| // Format of an information element: |
| // 1 1 1 - 252 |
| // +------+--------+----------------+ |
| // | Type | Length | Data | |
| // +------+--------+----------------+ |
| *phy_mode = Metrics::kWiFiNetworkPhyModeUndef; |
| bool found_ht = false; |
| bool found_erp = false; |
| int ie_len = 0; |
| vector<uint8_t>::iterator it; |
| for (it = ies.begin(); |
| std::distance(it, ies.end()) > 1; // Ensure Length field is within PDU. |
| it += ie_len) { |
| ie_len = 2 + *(it + 1); |
| if (std::distance(it, ies.end()) < ie_len) { |
| LOG(ERROR) << __func__ << ": IE extends past containing PDU."; |
| break; |
| } |
| switch (*it) { |
| case IEEE_80211::kElemIdCountry: |
| // Retrieve 2-character country code from the beginning of the element. |
| if (ie_len >= 4) { |
| *country_code = string(it + 2, it + 4); |
| } |
| case IEEE_80211::kElemIdErp: |
| if (!found_ht) { |
| *phy_mode = Metrics::kWiFiNetworkPhyMode11g; |
| } |
| found_erp = true; |
| break; |
| case IEEE_80211::kElemIdHTCap: |
| case IEEE_80211::kElemIdHTInfo: |
| *phy_mode = Metrics::kWiFiNetworkPhyMode11n; |
| found_ht = true; |
| break; |
| case IEEE_80211::kElemIdRSN: |
| ParseWPACapabilities(it + 2, it + ie_len, ieee80211w_required); |
| break; |
| case IEEE_80211::kElemIdVendor: |
| ParseVendorIE(it + 2, it + ie_len, vendor_information, |
| ieee80211w_required); |
| break; |
| } |
| } |
| return found_ht || found_erp; |
| } |
| |
| // static |
| void WiFiEndpoint::ParseWPACapabilities( |
| vector<uint8_t>::const_iterator ie, |
| vector<uint8_t>::const_iterator end, |
| bool *ieee80211w_required) { |
| // Format of an RSN Information Element: |
| // 2 4 |
| // +------+--------------------+ |
| // | Type | Group Cipher Suite | |
| // +------+--------------------+ |
| // 2 4 * pairwise count |
| // +-----------------------+---------------------+ |
| // | Pairwise Cipher Count | Pairwise Ciphers... | |
| // +-----------------------+---------------------+ |
| // 2 4 * authkey count |
| // +-----------------------+---------------------+ |
| // | AuthKey Suite Count | AuthKey Suites... | |
| // +-----------------------+---------------------+ |
| // 2 |
| // +------------------+ |
| // | RSN Capabilities | |
| // +------------------+ |
| // 2 16 * pmkid count |
| // +------------------+-------------------+ |
| // | PMKID Count | PMKIDs... | |
| // +------------------+-------------------+ |
| // 4 |
| // +-------------------------------+ |
| // | Group Management Cipher Suite | |
| // +-------------------------------+ |
| if (std::distance(ie, end) < IEEE_80211::kRSNIECipherCountOffset) { |
| return; |
| } |
| ie += IEEE_80211::kRSNIECipherCountOffset; |
| |
| // Advance past the pairwise and authkey ciphers. Each is a little-endian |
| // cipher count followed by n * cipher_selector. |
| for (int i = 0; i < IEEE_80211::kRSNIENumCiphers; ++i) { |
| // Retrieve a little-endian cipher count. |
| if (std::distance(ie, end) < IEEE_80211::kRSNIECipherCountLen) { |
| return; |
| } |
| uint16 cipher_count = *ie | (*(ie + 1) << 8); |
| |
| // Skip over the cipher selectors. |
| int skip_length = IEEE_80211::kRSNIECipherCountLen + |
| cipher_count * IEEE_80211::kRSNIESelectorLen; |
| if (std::distance(ie, end) < skip_length) { |
| return; |
| } |
| ie += skip_length; |
| } |
| |
| if (std::distance(ie, end) < IEEE_80211::kRSNIECapabilitiesLen) { |
| return; |
| } |
| |
| // Retrieve a little-endian capabilities bitfield. |
| uint16 capabilities = *ie | (*(ie + 1) << 8); |
| |
| if (capabilities & IEEE_80211::kRSNCapabilityFrameProtectionRequired && |
| ieee80211w_required) { |
| // Never set this value to false, since there may be multiple RSN |
| // information elements. |
| *ieee80211w_required = true; |
| } |
| } |
| |
| |
| // static |
| void WiFiEndpoint::ParseVendorIE(vector<uint8_t>::const_iterator ie, |
| vector<uint8_t>::const_iterator end, |
| VendorInformation *vendor_information, |
| bool *ieee80211w_required) { |
| // Format of an vendor-specific information element (with type |
| // and length field for the IE removed by the caller): |
| // 3 1 1 - 248 |
| // +------------+----------+----------------+ |
| // | OUI | OUI Type | Data | |
| // +------------+----------+----------------+ |
| |
| if (std::distance(ie, end) < 4) { |
| LOG(ERROR) << __func__ << ": no room in IE for OUI and type field."; |
| return; |
| } |
| uint32_t oui = (*ie << 16) | (*(ie + 1) << 8) | *(ie + 2); |
| uint8_t oui_type = *(ie + 3); |
| ie += 4; |
| |
| if (oui == IEEE_80211::kOUIVendorMicrosoft && |
| oui_type == IEEE_80211::kOUIMicrosoftWPS) { |
| // Format of a WPS data element: |
| // 2 2 |
| // +------+--------+----------------+ |
| // | Type | Length | Data | |
| // +------+--------+----------------+ |
| while (std::distance(ie, end) >= 4) { |
| int element_type = (*ie << 8) | *(ie + 1); |
| int element_length = (*(ie + 2) << 8) | *(ie + 3); |
| ie += 4; |
| if (std::distance(ie, end) < element_length) { |
| LOG(ERROR) << __func__ << ": WPS element extends past containing PDU."; |
| break; |
| } |
| string s(ie, ie + element_length); |
| if (IsStringASCII(s)) { |
| switch (element_type) { |
| case IEEE_80211::kWPSElementManufacturer: |
| vendor_information->wps_manufacturer = s; |
| break; |
| case IEEE_80211::kWPSElementModelName: |
| vendor_information->wps_model_name = s; |
| break; |
| case IEEE_80211::kWPSElementModelNumber: |
| vendor_information->wps_model_number = s; |
| break; |
| case IEEE_80211::kWPSElementDeviceName: |
| vendor_information->wps_device_name = s; |
| break; |
| } |
| } |
| ie += element_length; |
| } |
| } else if (oui == IEEE_80211::kOUIVendorMicrosoft && |
| oui_type == IEEE_80211::kOUIMicrosoftWPA) { |
| ParseWPACapabilities(ie, end, ieee80211w_required); |
| } else if (oui != IEEE_80211::kOUIVendorEpigram && |
| oui != IEEE_80211::kOUIVendorMicrosoft) { |
| vendor_information->oui_list.insert(oui); |
| } |
| } |
| |
| } // namespace shill |