Thieu Le | e41a72d | 2012-02-06 20:46:51 +0000 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #ifndef SHILL_WIFI_ENDPOINT_ |
| 6 | #define SHILL_WIFI_ENDPOINT_ |
| 7 | |
| 8 | #include <map> |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 9 | #include <set> |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 10 | #include <string> |
| 11 | #include <vector> |
| 12 | |
| 13 | #include <base/memory/ref_counted.h> |
| 14 | #include <dbus-c++/dbus.h> |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 15 | #include <gtest/gtest_prod.h> // for FRIEND_TEST |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 16 | |
| 17 | #include "shill/endpoint.h" |
Paul Stewart | 26b327e | 2011-10-19 11:38:09 -0700 | [diff] [blame] | 18 | #include "shill/event_dispatcher.h" |
Thieu Le | 1df7f4e | 2012-02-10 15:21:45 -0800 | [diff] [blame] | 19 | #include "shill/metrics.h" |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 20 | #include "shill/refptr_types.h" |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 21 | |
| 22 | namespace shill { |
| 23 | |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 24 | class ProxyFactory; |
| 25 | class SupplicantBSSProxyInterface; |
| 26 | |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 27 | class WiFiEndpoint : public Endpoint { |
| 28 | public: |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 29 | struct VendorInformation { |
| 30 | std::string wps_manufacturer; |
| 31 | std::string wps_model_name; |
| 32 | std::string wps_model_number; |
| 33 | std::string wps_device_name; |
| 34 | std::set<uint32_t> oui_list; |
| 35 | }; |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 36 | WiFiEndpoint(ProxyFactory *proxy_factory, |
| 37 | const WiFiRefPtr &device, |
| 38 | const std::string &rpc_id, |
| 39 | const std::map<std::string, ::DBus::Variant> &properties); |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 40 | virtual ~WiFiEndpoint(); |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 41 | |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 42 | // Set up RPC channel. Broken out from the ctor, so that WiFi can |
| 43 | // look over the Endpoint details before commiting to setting up |
| 44 | // RPC. |
| 45 | virtual void Start(); |
| 46 | |
| 47 | // Called by SupplicantBSSProxy, in response to events from |
| 48 | // wpa_supplicant. |
| 49 | void PropertiesChanged( |
| 50 | const std::map<std::string, ::DBus::Variant> &properties); |
| 51 | |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 52 | // Maps mode strings from flimflam's nomenclature, as defined |
| 53 | // in chromeos/dbus/service_constants.h, to uints used by supplicant |
| 54 | static uint32_t ModeStringToUint(const std::string &mode_string); |
| 55 | |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 56 | // Returns a stringmap containing information gleaned about the |
| 57 | // vendor of this AP. |
| 58 | std::map<std::string, std::string> GetVendorInformation() const; |
| 59 | |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 60 | const std::vector<uint8_t> &ssid() const; |
| 61 | const std::string &ssid_string() const; |
| 62 | const std::string &ssid_hex() const; |
| 63 | const std::string &bssid_string() const; |
| 64 | const std::string &bssid_hex() const; |
| 65 | int16_t signal_strength() const; |
Thieu Le | e41a72d | 2012-02-06 20:46:51 +0000 | [diff] [blame] | 66 | uint16 frequency() const; |
Thieu Le | 1df7f4e | 2012-02-10 15:21:45 -0800 | [diff] [blame] | 67 | uint16 physical_mode() const; |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 68 | const std::string &network_mode() const; |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 69 | const std::string &security_mode() const; |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 70 | const VendorInformation &vendor_information() const; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 71 | |
| 72 | private: |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 73 | friend class WiFiEndpointTest; |
Paul Stewart | e369ece | 2012-05-22 09:11:03 -0700 | [diff] [blame] | 74 | friend class WiFiObjectTest; // for MakeOpenEndpoint |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 75 | friend class WiFiServiceTest; // for MakeOpenEndpoint |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 76 | // these test cases need access to the KeyManagement enum |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 77 | FRIEND_TEST(WiFiEndpointTest, DeterminePhyModeFromFrequency); |
| 78 | FRIEND_TEST(WiFiEndpointTest, ParseIEs); |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 79 | FRIEND_TEST(WiFiEndpointTest, ParseKeyManagementMethodsEAP); |
| 80 | FRIEND_TEST(WiFiEndpointTest, ParseKeyManagementMethodsPSK); |
| 81 | FRIEND_TEST(WiFiEndpointTest, ParseKeyManagementMethodsEAPAndPSK); |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 82 | FRIEND_TEST(WiFiEndpointTest, ParseVendorIEs); |
mukesh agrawal | e1d90e9 | 2012-02-15 17:36:08 -0800 | [diff] [blame] | 83 | FRIEND_TEST(WiFiServiceUpdateFromEndpointsTest, EndpointModified); |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 84 | |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 85 | enum KeyManagement { |
| 86 | kKeyManagement802_1x, |
| 87 | kKeyManagementPSK |
| 88 | }; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 89 | |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 90 | // Build a simple WiFiEndpoint, for testing purposes. |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 91 | static WiFiEndpoint *MakeOpenEndpoint(ProxyFactory *proxy_factory, |
| 92 | const WiFiRefPtr &wifi, |
| 93 | const std::string &ssid, |
mukesh agrawal | e1d90e9 | 2012-02-15 17:36:08 -0800 | [diff] [blame] | 94 | const std::string &bssid, |
| 95 | uint16 frequency, |
| 96 | int16 signal_dbm); |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 97 | // Maps mode strings from supplicant into flimflam's nomenclature, as defined |
| 98 | // in chromeos/dbus/service_constants.h |
| 99 | static const char *ParseMode(const std::string &mode_string); |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 100 | // Parses an Endpoint's properties to identify approprirate flimflam |
| 101 | // security property value, as defined in chromeos/dbus/service_constants.h |
| 102 | static const char *ParseSecurity( |
| 103 | const std::map<std::string, ::DBus::Variant> &properties); |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 104 | // Parses an Endpoint's properties' "RSN" or "WPA" sub-dictionary, to |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 105 | // identify supported key management methods (802.1x or PSK). |
| 106 | static void ParseKeyManagementMethods( |
| 107 | const std::map<std::string, ::DBus::Variant> &security_method_properties, |
| 108 | std::set<KeyManagement> *key_management_methods); |
Thieu Le | 1df7f4e | 2012-02-10 15:21:45 -0800 | [diff] [blame] | 109 | // Determine the negotiated operating mode for the channel by looking at |
| 110 | // the information elements, frequency and data rates. The information |
| 111 | // elements and data rates live in |properties|. |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 112 | static Metrics::WiFiNetworkPhyMode DeterminePhyModeFromFrequency( |
Thieu Le | 1df7f4e | 2012-02-10 15:21:45 -0800 | [diff] [blame] | 113 | const std::map<std::string, ::DBus::Variant> &properties, |
| 114 | uint16 frequency); |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 115 | // Parse information elements to determine the physical mode and vendor |
| 116 | // information associated with the AP. Returns true if a physical mode |
| 117 | // was determined from the IE elements, false otherwise. |
| 118 | static bool ParseIEs(const std::map<std::string, ::DBus::Variant> &properties, |
| 119 | Metrics::WiFiNetworkPhyMode *phy_mode, |
| 120 | VendorInformation *vendor_information); |
| 121 | // Parse a single vendor information element. |
| 122 | static void ParseVendorIE(std::vector<uint8_t>::const_iterator ie, |
| 123 | std::vector<uint8_t>::const_iterator end, |
| 124 | VendorInformation *vendor_information); |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 125 | |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 126 | // TODO(quiche): make const? |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 127 | std::vector<uint8_t> ssid_; |
| 128 | std::vector<uint8_t> bssid_; |
| 129 | std::string ssid_string_; |
| 130 | std::string ssid_hex_; |
| 131 | std::string bssid_string_; |
| 132 | std::string bssid_hex_; |
mukesh agrawal | e1d90e9 | 2012-02-15 17:36:08 -0800 | [diff] [blame] | 133 | int16 signal_strength_; |
Thieu Le | e41a72d | 2012-02-06 20:46:51 +0000 | [diff] [blame] | 134 | uint16 frequency_; |
Thieu Le | 1df7f4e | 2012-02-10 15:21:45 -0800 | [diff] [blame] | 135 | uint16 physical_mode_; |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 136 | // network_mode_ and security_mode_ are represented as flimflam names |
| 137 | // (not necessarily the same as wpa_supplicant names) |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 138 | std::string network_mode_; |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 139 | std::string security_mode_; |
Paul Stewart | 72b2fdc | 2012-06-02 08:58:51 -0700 | [diff] [blame] | 140 | VendorInformation vendor_information_; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 141 | |
mukesh agrawal | b20776f | 2012-02-10 16:00:36 -0800 | [diff] [blame] | 142 | ProxyFactory *proxy_factory_; |
| 143 | WiFiRefPtr device_; |
| 144 | std::string rpc_id_; |
| 145 | scoped_ptr<SupplicantBSSProxyInterface> supplicant_bss_proxy_; |
| 146 | |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 147 | DISALLOW_COPY_AND_ASSIGN(WiFiEndpoint); |
| 148 | }; |
| 149 | |
| 150 | } // namespace shill |
| 151 | |
| 152 | #endif // SHILL_WIFI_ENDPOINT_ |