mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 1 | // Copyright (c) 2011 The Chromium OS 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 | #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" |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 19 | |
| 20 | namespace shill { |
| 21 | |
| 22 | class WiFiEndpoint : public Endpoint { |
| 23 | public: |
| 24 | WiFiEndpoint(const std::map<std::string, ::DBus::Variant> &properties); |
| 25 | virtual ~WiFiEndpoint(); |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 26 | |
| 27 | // Maps mode strings from flimflam's nomenclature, as defined |
| 28 | // in chromeos/dbus/service_constants.h, to uints used by supplicant |
| 29 | static uint32_t ModeStringToUint(const std::string &mode_string); |
| 30 | |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 31 | const std::vector<uint8_t> &ssid() const; |
| 32 | const std::string &ssid_string() const; |
| 33 | const std::string &ssid_hex() const; |
| 34 | const std::string &bssid_string() const; |
| 35 | const std::string &bssid_hex() const; |
| 36 | int16_t signal_strength() const; |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 37 | const std::string &network_mode() const; |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 38 | const std::string &security_mode() const; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 39 | |
| 40 | private: |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 41 | friend class WiFiEndpointTest; |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 42 | friend class WiFiMainTest; // for MakeOpenEndpoint |
| 43 | friend class WiFiServiceTest; // for MakeOpenEndpoint |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 44 | // these test cases need access to the KeyManagement enum |
| 45 | FRIEND_TEST(WiFiEndpointTest, ParseKeyManagementMethodsEAP); |
| 46 | FRIEND_TEST(WiFiEndpointTest, ParseKeyManagementMethodsPSK); |
| 47 | FRIEND_TEST(WiFiEndpointTest, ParseKeyManagementMethodsEAPAndPSK); |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 48 | |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 49 | enum KeyManagement { |
| 50 | kKeyManagement802_1x, |
| 51 | kKeyManagementPSK |
| 52 | }; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 53 | |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 54 | // Build a simple WiFiEndpoint, for testing purposes. |
| 55 | static WiFiEndpoint *MakeOpenEndpoint( |
| 56 | const std::string &ssid, const std::string &bssid); |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 57 | // Maps mode strings from supplicant into flimflam's nomenclature, as defined |
| 58 | // in chromeos/dbus/service_constants.h |
| 59 | static const char *ParseMode(const std::string &mode_string); |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 60 | // Parses an Endpoint's properties to identify approprirate flimflam |
| 61 | // security property value, as defined in chromeos/dbus/service_constants.h |
| 62 | static const char *ParseSecurity( |
| 63 | const std::map<std::string, ::DBus::Variant> &properties); |
| 64 | // Parses and Endpoint's properties' "RSN" or "WPA" sub-dictionary, to |
| 65 | // identify supported key management methods (802.1x or PSK). |
| 66 | static void ParseKeyManagementMethods( |
| 67 | const std::map<std::string, ::DBus::Variant> &security_method_properties, |
| 68 | std::set<KeyManagement> *key_management_methods); |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 69 | |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 70 | // TODO(quiche): make const? |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 71 | std::vector<uint8_t> ssid_; |
| 72 | std::vector<uint8_t> bssid_; |
| 73 | std::string ssid_string_; |
| 74 | std::string ssid_hex_; |
| 75 | std::string bssid_string_; |
| 76 | std::string bssid_hex_; |
| 77 | int16_t signal_strength_; |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 78 | // network_mode_ and security_mode_ are represented as flimflam names |
| 79 | // (not necessarily the same as wpa_supplicant names) |
Chris Masone | 092df3e | 2011-08-22 09:41:39 -0700 | [diff] [blame] | 80 | std::string network_mode_; |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 81 | std::string security_mode_; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 82 | |
| 83 | DISALLOW_COPY_AND_ASSIGN(WiFiEndpoint); |
| 84 | }; |
| 85 | |
| 86 | } // namespace shill |
| 87 | |
| 88 | #endif // SHILL_WIFI_ENDPOINT_ |