| OLD | NEW |
| 1 // Copyright 2008, Google Inc. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // Redistribution and use in source and binary forms, with or without | 3 // found in the LICENSE file. |
| 4 // modification, are permitted provided that the following conditions are met: | |
| 5 // | |
| 6 // 1. Redistributions of source code must retain the above copyright notice, | |
| 7 // this list of conditions and the following disclaimer. | |
| 8 // 2. Redistributions in binary form must reproduce the above copyright notice, | |
| 9 // this list of conditions and the following disclaimer in the documentation | |
| 10 // and/or other materials provided with the distribution. | |
| 11 // 3. Neither the name of Google Inc. nor the names of its contributors may be | |
| 12 // used to endorse or promote products derived from this software without | |
| 13 // specific prior written permission. | |
| 14 // | |
| 15 // THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED | |
| 16 // WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF | |
| 17 // MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO | |
| 18 // EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
| 19 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
| 20 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
| 21 // OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
| 22 // WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
| 23 // OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
| 24 // ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| 25 | 4 |
| 26 #ifndef GEARS_GEOLOCATION_WIFI_DATA_PROVIDER_WIN32_H__ | 5 #ifndef CHROME_BROWSER_GEOLOCATION_WIFI_DATA_PROVIDER_WIN_H_ |
| 27 #define GEARS_GEOLOCATION_WIFI_DATA_PROVIDER_WIN32_H__ | 6 #define CHROME_BROWSER_GEOLOCATION_WIFI_DATA_PROVIDER_WIN_H_ |
| 28 | 7 |
| 29 #include <vector> | 8 #include "base/task.h" |
| 30 #include "gears/base/common/common.h" | 9 #include "base/thread.h" |
| 31 #include "gears/base/common/thread.h" | 10 #include "chrome/browser/geolocation/device_data_provider.h" |
| 32 #include "gears/geolocation/device_data_provider.h" | 11 |
| 33 #include "gears/geolocation/wlanapi.h" | 12 class PollingPolicyInterface; |
| 34 | 13 |
| 35 class Win32WifiDataProvider | 14 class Win32WifiDataProvider |
| 36 : public WifiDataProviderImplBase, | 15 : public WifiDataProviderImplBase, |
| 37 public Thread { | 16 private base::Thread { |
| 38 public: | 17 public: |
| 18 // Interface to abstract the low level data OS library call, and to allow |
| 19 // mocking (hence public). |
| 20 class WlanApiInterface { |
| 21 public: |
| 22 virtual ~WlanApiInterface() {} |
| 23 // Gets wifi data for all visible access points. |
| 24 virtual bool GetAccessPointData(WifiData::AccessPointDataSet *data) = 0; |
| 25 }; |
| 26 |
| 39 Win32WifiDataProvider(); | 27 Win32WifiDataProvider(); |
| 40 virtual ~Win32WifiDataProvider(); | 28 virtual ~Win32WifiDataProvider(); |
| 41 | 29 |
| 30 // Takes ownership of wlan_api. Must be called before Start(). |
| 31 void inject_mock_wlan_api(WlanApiInterface* wlan_api); |
| 32 |
| 33 // Takes ownership of polic. Must be called before Start(). |
| 34 void inject_mock_polling_policy(PollingPolicyInterface* policy); |
| 35 |
| 42 // WifiDataProviderImplBase implementation | 36 // WifiDataProviderImplBase implementation |
| 37 virtual bool StartDataProvider(); |
| 43 virtual bool GetData(WifiData *data); | 38 virtual bool GetData(WifiData *data); |
| 44 | 39 |
| 45 private: | 40 private: |
| 46 // Thread implementation | 41 // Thread implementation |
| 47 virtual void Run(); | 42 virtual void Init(); |
| 43 // Called just after the message loop ends |
| 44 virtual void CleanUp(); |
| 48 | 45 |
| 49 // WLAN methods | 46 // Task which run in the child thread. |
| 50 // Loads the required functions from the DLL. | 47 void DoWifiScanTask(); |
| 51 void GetWLANFunctions(HINSTANCE wlan_library); | |
| 52 // Gets wifi data for all visible access points. | |
| 53 bool GetAccessPointDataWLAN(WifiData::AccessPointDataSet *data); | |
| 54 int GetInterfaceDataWLAN(HANDLE wlan_handle, | |
| 55 const GUID &interface_id, | |
| 56 WifiData::AccessPointDataSet *data); | |
| 57 | 48 |
| 58 // NDIS methods. | 49 // Will schedule a scan; i.e. enqueue DoWifiScanTask deferred task. |
| 59 bool GetInterfacesNDIS(); | 50 void ScheduleNextScan(); |
| 60 bool GetAccessPointDataNDIS(WifiData::AccessPointDataSet *data); | |
| 61 bool GetInterfaceDataNDIS(HANDLE adapter_handle, | |
| 62 WifiData::AccessPointDataSet *data); | |
| 63 | |
| 64 // Function pointers for WLAN | |
| 65 WlanOpenHandleFunction WlanOpenHandle_function_; | |
| 66 WlanEnumInterfacesFunction WlanEnumInterfaces_function_; | |
| 67 WlanGetNetworkBssListFunction WlanGetNetworkBssList_function_; | |
| 68 WlanFreeMemoryFunction WlanFreeMemory_function_; | |
| 69 WlanCloseHandleFunction WlanCloseHandle_function_; | |
| 70 | |
| 71 // NDIS variables. | |
| 72 std::vector<std::string16> interface_service_names_; | |
| 73 int oid_buffer_size_; | |
| 74 | 51 |
| 75 WifiData wifi_data_; | 52 WifiData wifi_data_; |
| 76 Mutex data_mutex_; | 53 Lock data_mutex_; |
| 77 // Events signalled to shut down the thread that polls for wifi data. | |
| 78 Event stop_event_; | |
| 79 | 54 |
| 80 // Whether we've successfully completed a scan for WiFi data (or the polling | 55 // Whether we've successfully completed a scan for WiFi data (or the polling |
| 81 // thread has terminated early). | 56 // thread has terminated early). |
| 82 bool is_first_scan_complete_; | 57 bool is_first_scan_complete_; |
| 83 | 58 |
| 84 DISALLOW_EVIL_CONSTRUCTORS(Win32WifiDataProvider); | 59 // Underlying OS wifi API. |
| 60 scoped_ptr<WlanApiInterface> wlan_api_; |
| 61 |
| 62 // Controls the polling update interval. |
| 63 scoped_ptr<PollingPolicyInterface> polling_policy_; |
| 64 |
| 65 // Holder for the tasks which run on the thread; takes care of cleanup. |
| 66 ScopedRunnableMethodFactory<Win32WifiDataProvider> task_factory_; |
| 67 |
| 68 DISALLOW_COPY_AND_ASSIGN(Win32WifiDataProvider); |
| 85 }; | 69 }; |
| 86 | 70 |
| 87 #endif // GEARS_GEOLOCATION_WIFI_DATA_PROVIDER_WIN32_H__ | 71 #endif // CHROME_BROWSER_GEOLOCATION_WIFI_DATA_PROVIDER_WIN_H_ |
| OLD | NEW |