blob: 18748618fb3009f9f290c60f9c83b675ec9604b7 [file] [log] [blame]
// 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