mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
Paul Stewart | b50f0b9 | 2011-05-16 16:31:42 -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_ |
| 6 | #define SHILL_WIFI_ |
| 7 | |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 8 | #include <map> |
Chris Masone | 46eaaf5 | 2011-05-24 13:08:30 -0700 | [diff] [blame] | 9 | #include <string> |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 10 | #include <vector> |
Chris Masone | 46eaaf5 | 2011-05-24 13:08:30 -0700 | [diff] [blame] | 11 | |
mukesh agrawal | af57195 | 2011-07-14 14:31:12 -0700 | [diff] [blame] | 12 | #include <dbus-c++/dbus.h> |
Paul Stewart | 6ab23a9 | 2011-11-09 17:17:47 -0800 | [diff] [blame] | 13 | #include <gtest/gtest_prod.h> // for FRIEND_TEST |
mukesh agrawal | af57195 | 2011-07-14 14:31:12 -0700 | [diff] [blame] | 14 | |
Paul Stewart | b50f0b9 | 2011-05-16 16:31:42 -0700 | [diff] [blame] | 15 | #include "shill/device.h" |
Paul Stewart | 26b327e | 2011-10-19 11:38:09 -0700 | [diff] [blame] | 16 | #include "shill/event_dispatcher.h" |
Chris Masone | 2b10554 | 2011-06-22 10:58:09 -0700 | [diff] [blame] | 17 | #include "shill/refptr_types.h" |
Paul Stewart | b50f0b9 | 2011-05-16 16:31:42 -0700 | [diff] [blame] | 18 | |
| 19 | namespace shill { |
| 20 | |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 21 | class Error; |
| 22 | class KeyValueStore; |
Darin Petkov | ab565bb | 2011-10-06 02:55:51 -0700 | [diff] [blame] | 23 | class ProxyFactory; |
mukesh agrawal | af57195 | 2011-07-14 14:31:12 -0700 | [diff] [blame] | 24 | class SupplicantInterfaceProxyInterface; |
| 25 | class SupplicantProcessProxyInterface; |
mukesh agrawal | 445e72c | 2011-06-22 11:13:50 -0700 | [diff] [blame] | 26 | class WiFiService; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 27 | |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 28 | // WiFi class. Specialization of Device for WiFi. |
Paul Stewart | b50f0b9 | 2011-05-16 16:31:42 -0700 | [diff] [blame] | 29 | class WiFi : public Device { |
| 30 | public: |
Paul Stewart | f1ce5d2 | 2011-05-19 13:10:20 -0700 | [diff] [blame] | 31 | WiFi(ControlInterface *control_interface, |
| 32 | EventDispatcher *dispatcher, |
| 33 | Manager *manager, |
Chris Masone | 626719f | 2011-08-18 16:58:48 -0700 | [diff] [blame] | 34 | const std::string &link, |
| 35 | const std::string &address, |
Paul Stewart | f1ce5d2 | 2011-05-19 13:10:20 -0700 | [diff] [blame] | 36 | int interface_index); |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 37 | virtual ~WiFi(); |
Darin Petkov | c086531 | 2011-09-16 15:31:20 -0700 | [diff] [blame] | 38 | |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 39 | virtual void Start(); |
| 40 | virtual void Stop(); |
Paul Stewart | a41e38d | 2011-11-11 07:47:29 -0800 | [diff] [blame] | 41 | virtual bool Load(StoreInterface *storage); |
Darin Petkov | c086531 | 2011-09-16 15:31:20 -0700 | [diff] [blame] | 42 | virtual void Scan(Error *error); |
Paul Stewart | fdd1607 | 2011-09-16 12:41:35 -0700 | [diff] [blame] | 43 | virtual bool TechnologyIs(const Technology::Identifier type) const; |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 44 | virtual bool IsConnectingTo(const WiFiService &service) const; |
mukesh agrawal | f2f68a5 | 2011-09-01 12:15:48 -0700 | [diff] [blame] | 45 | virtual void LinkEvent(unsigned int flags, unsigned int change); |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 46 | |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 47 | // Called by SupplicantInterfaceProxy, in response to events from |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 48 | // wpa_supplicant. |
| 49 | void BSSAdded(const ::DBus::Path &BSS, |
Chris Masone | a82b711 | 2011-05-25 15:16:29 -0700 | [diff] [blame] | 50 | const std::map<std::string, ::DBus::Variant> &properties); |
mukesh agrawal | 261daca | 2011-12-02 18:56:56 +0000 | [diff] [blame] | 51 | void BSSRemoved(const ::DBus::Path &BSS); |
mukesh agrawal | 7ec7131 | 2011-11-10 02:08:26 +0000 | [diff] [blame] | 52 | void PropertiesChanged( |
| 53 | const std::map<std::string, ::DBus::Variant> &properties); |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 54 | void ScanDone(); |
| 55 | |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 56 | // Called by WiFiService. |
mukesh agrawal | 6e27777 | 2011-09-29 15:04:23 -0700 | [diff] [blame] | 57 | virtual void ConnectTo( |
| 58 | WiFiService *service, |
mukesh agrawal | 6489632 | 2011-12-01 01:13:10 +0000 | [diff] [blame] | 59 | std::map<std::string, ::DBus::Variant> service_params); |
mukesh agrawal | 0ed0f2e | 2011-12-05 20:36:17 +0000 | [diff] [blame] | 60 | virtual void DisconnectFrom(WiFiService *service); |
mukesh agrawal | 8a3188d | 2011-12-01 20:56:44 +0000 | [diff] [blame] | 61 | virtual bool IsIdle() const; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 62 | |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 63 | // Called by Manager. |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 64 | virtual WiFiServiceRefPtr GetService(const KeyValueStore &args, Error *error); |
| 65 | |
Paul Stewart | b50f0b9 | 2011-05-16 16:31:42 -0700 | [diff] [blame] | 66 | private: |
mukesh agrawal | 7ec7131 | 2011-11-10 02:08:26 +0000 | [diff] [blame] | 67 | friend class WiFiMainTest; // access to supplicant_*_proxy_, link_up_ |
mukesh agrawal | 7ec7131 | 2011-11-10 02:08:26 +0000 | [diff] [blame] | 68 | FRIEND_TEST(WiFiMainTest, InitialSupplicantState); // kInterfaceStateUnknown |
mukesh agrawal | 165e614 | 2011-11-22 02:22:56 +0000 | [diff] [blame] | 69 | FRIEND_TEST(WiFiMainTest, ScanResultsWithUpdates); // EndpointMap |
Paul Stewart | 6ab23a9 | 2011-11-09 17:17:47 -0800 | [diff] [blame] | 70 | |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 71 | typedef std::map<const std::string, WiFiEndpointRefPtr> EndpointMap; |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 72 | typedef std::map<WiFiService *, std::string> ReverseServiceMap; |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 73 | |
mukesh agrawal | 4d0401c | 2012-01-06 16:05:31 -0800 | [diff] [blame] | 74 | static const char *kDefaultBgscanMethod; |
| 75 | static const uint16 kDefaultBgscanShortIntervalSeconds; |
| 76 | static const int32 kDefaultBgscanSignalThresholdDbm; |
| 77 | static const uint16 kDefaultScanIntervalSeconds; |
mukesh agrawal | 7a4e400 | 2011-09-06 11:26:05 -0700 | [diff] [blame] | 78 | static const char kManagerErrorPassphraseRequired[]; |
| 79 | static const char kManagerErrorSSIDTooLong[]; |
| 80 | static const char kManagerErrorSSIDTooShort[]; |
| 81 | static const char kManagerErrorSSIDRequired[]; |
| 82 | static const char kManagerErrorTypeRequired[]; |
| 83 | static const char kManagerErrorUnsupportedSecurityMode[]; |
| 84 | static const char kManagerErrorUnsupportedServiceType[]; |
| 85 | static const char kManagerErrorUnsupportedServiceMode[]; |
mukesh agrawal | 7ec7131 | 2011-11-10 02:08:26 +0000 | [diff] [blame] | 86 | static const char kInterfaceStateUnknown[]; |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 87 | |
mukesh agrawal | 4d0401c | 2012-01-06 16:05:31 -0800 | [diff] [blame] | 88 | std::string CreateBgscanConfigString(); |
| 89 | std::string GetBgscanMethod(Error */* error */) { return bgscan_method_; } |
| 90 | uint16 GetBgscanShortInterval(Error */* error */) { |
| 91 | return bgscan_short_interval_seconds_; |
| 92 | } |
| 93 | int32 GetBgscanSignalThreshold(Error */* error */) { |
| 94 | return bgscan_signal_threshold_dbm_; |
| 95 | } |
| 96 | uint16 GetScanInterval(Error */* error */) { return scan_interval_seconds_; } |
| 97 | void SetBgscanMethod(const std::string &method, Error *error); |
| 98 | void SetBgscanShortInterval(const uint16 &seconds, Error *error); |
| 99 | void SetBgscanSignalThreshold(const int32 &dbm, Error *error); |
| 100 | void SetScanInterval(const uint16 &seconds, Error *error); |
| 101 | |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 102 | WiFiServiceRefPtr CreateServiceForEndpoint( |
| 103 | const WiFiEndpoint &endpoint, bool hidden_ssid); |
| 104 | void CurrentBSSChanged(const ::DBus::Path &new_bss); |
Paul Stewart | 6ab23a9 | 2011-11-09 17:17:47 -0800 | [diff] [blame] | 105 | WiFiServiceRefPtr FindService(const std::vector<uint8_t> &ssid, |
| 106 | const std::string &mode, |
| 107 | const std::string &security) const; |
mukesh agrawal | 165e614 | 2011-11-22 02:22:56 +0000 | [diff] [blame] | 108 | WiFiServiceRefPtr FindServiceForEndpoint(const WiFiEndpoint &endpoint); |
Paul Stewart | ced6a0b | 2011-11-08 15:32:04 -0800 | [diff] [blame] | 109 | ByteArrays GetHiddenSSIDList(); |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 110 | void HandleDisconnect(); |
| 111 | void HandleRoam(const ::DBus::Path &new_bssid); |
Paul Stewart | a41e38d | 2011-11-11 07:47:29 -0800 | [diff] [blame] | 112 | // Create services for hidden networks stored in |storage|. Returns true |
| 113 | // if any were found, otherwise returns false. |
| 114 | bool LoadHiddenServices(StoreInterface *storage); |
mukesh agrawal | b4bc57d | 2011-12-07 01:07:47 +0000 | [diff] [blame] | 115 | void BSSAddedTask(const ::DBus::Path &BSS, |
| 116 | const std::map<std::string, ::DBus::Variant> &properties); |
| 117 | void BSSRemovedTask(const ::DBus::Path &BSS); |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 118 | void PropertiesChangedTask( |
| 119 | const std::map<std::string, ::DBus::Variant> &properties); |
mukesh agrawal | dc42bb3 | 2011-07-28 10:40:26 -0700 | [diff] [blame] | 120 | void ScanDoneTask(); |
mukesh agrawal | 3239932 | 2011-09-01 10:53:43 -0700 | [diff] [blame] | 121 | void ScanTask(); |
mukesh agrawal | 7ec7131 | 2011-11-10 02:08:26 +0000 | [diff] [blame] | 122 | void StateChanged(const std::string &new_state); |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 123 | |
mukesh agrawal | 4d0401c | 2012-01-06 16:05:31 -0800 | [diff] [blame] | 124 | void HelpRegisterDerivedInt32( |
| 125 | PropertyStore *store, |
| 126 | const std::string &name, |
| 127 | int32(WiFi::*get)(Error *error), |
| 128 | void(WiFi::*set)(const int32 &value, Error *error)); |
| 129 | void HelpRegisterDerivedString( |
| 130 | PropertyStore *store, |
| 131 | const std::string &name, |
| 132 | std::string(WiFi::*get)(Error *error), |
| 133 | void(WiFi::*set)(const std::string &value, Error *error)); |
| 134 | void HelpRegisterDerivedUint16( |
| 135 | PropertyStore *store, |
| 136 | const std::string &name, |
| 137 | uint16(WiFi::*get)(Error *error), |
| 138 | void(WiFi::*set)(const uint16 &value, Error *error)); |
| 139 | |
Darin Petkov | ab565bb | 2011-10-06 02:55:51 -0700 | [diff] [blame] | 140 | // Store cached copies of singletons for speed/ease of testing. |
| 141 | ProxyFactory *proxy_factory_; |
| 142 | |
mukesh agrawal | b54601c | 2011-06-07 17:39:22 -0700 | [diff] [blame] | 143 | ScopedRunnableMethodFactory<WiFi> task_factory_; |
mukesh agrawal | af57195 | 2011-07-14 14:31:12 -0700 | [diff] [blame] | 144 | scoped_ptr<SupplicantProcessProxyInterface> supplicant_process_proxy_; |
| 145 | scoped_ptr<SupplicantInterfaceProxyInterface> supplicant_interface_proxy_; |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 146 | // The rpcid used as the key is wpa_supplicant's D-Bus path for the |
| 147 | // Endpoint (BSS, in supplicant parlance). |
| 148 | EndpointMap endpoint_by_rpcid_; |
mukesh agrawal | 1590839 | 2011-11-16 18:29:25 +0000 | [diff] [blame] | 149 | // Map from Services to the D-Bus path for the corresponding wpa_supplicant |
| 150 | // Network. |
| 151 | ReverseServiceMap rpcid_by_service_; |
| 152 | bool link_up_; |
| 153 | std::vector<WiFiServiceRefPtr> services_; |
| 154 | // The Service we are presently connected to. May be NULL is we're not |
| 155 | // not connected to any Service. |
| 156 | WiFiServiceRefPtr current_service_; |
| 157 | // The Service we're attempting to connect to. May be NULL if we're |
| 158 | // not attempting to connect to a new Service. If non-NULL, should |
| 159 | // be distinct from |current_service_|. (A service should not |
| 160 | // simultaneously be both pending, and current.) |
| 161 | WiFiServiceRefPtr pending_service_; |
| 162 | std::string supplicant_state_; |
| 163 | std::string supplicant_bss_; |
mukesh agrawal | ab87ea4 | 2011-05-18 11:44:49 -0700 | [diff] [blame] | 164 | |
Chris Masone | 853b81b | 2011-06-24 14:11:41 -0700 | [diff] [blame] | 165 | // Properties |
| 166 | std::string bgscan_method_; |
mukesh agrawal | 4d0401c | 2012-01-06 16:05:31 -0800 | [diff] [blame] | 167 | uint16 bgscan_short_interval_seconds_; |
| 168 | int32 bgscan_signal_threshold_dbm_; |
Chris Masone | 853b81b | 2011-06-24 14:11:41 -0700 | [diff] [blame] | 169 | bool scan_pending_; |
mukesh agrawal | 4d0401c | 2012-01-06 16:05:31 -0800 | [diff] [blame] | 170 | uint16 scan_interval_seconds_; |
Chris Masone | 853b81b | 2011-06-24 14:11:41 -0700 | [diff] [blame] | 171 | |
Paul Stewart | b50f0b9 | 2011-05-16 16:31:42 -0700 | [diff] [blame] | 172 | DISALLOW_COPY_AND_ASSIGN(WiFi); |
| 173 | }; |
| 174 | |
| 175 | } // namespace shill |
| 176 | |
| 177 | #endif // SHILL_WIFI_ |