OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/base/net_util.h" | 5 #include "net/base/net_util.h" |
6 | 6 |
7 #include <iphlpapi.h> | 7 #include <iphlpapi.h> |
8 #include <wlanapi.h> | 8 #include <wlanapi.h> |
9 | 9 |
10 #include <algorithm> | 10 #include <algorithm> |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
54 ::GetProcAddress(module, "WlanQueryInterface")); | 54 ::GetProcAddress(module, "WlanQueryInterface")); |
55 free_memory_func = reinterpret_cast<WlanFreeMemoryFunc>( | 55 free_memory_func = reinterpret_cast<WlanFreeMemoryFunc>( |
56 ::GetProcAddress(module, "WlanFreeMemory")); | 56 ::GetProcAddress(module, "WlanFreeMemory")); |
57 close_handle_func = reinterpret_cast<WlanCloseHandleFunc>( | 57 close_handle_func = reinterpret_cast<WlanCloseHandleFunc>( |
58 ::GetProcAddress(module, "WlanCloseHandle")); | 58 ::GetProcAddress(module, "WlanCloseHandle")); |
59 initialized = open_handle_func && enum_interfaces_func && | 59 initialized = open_handle_func && enum_interfaces_func && |
60 query_interface_func && free_memory_func && | 60 query_interface_func && free_memory_func && |
61 close_handle_func; | 61 close_handle_func; |
62 } | 62 } |
63 | 63 |
64 template <typename T> | |
65 DWORD OpenHandle(DWORD client_version, DWORD* cur_version, T* handle) const { | |
66 HANDLE temp_handle; | |
67 DWORD result = open_handle_func(client_version, NULL, cur_version, | |
68 &temp_handle); | |
69 if (result != ERROR_SUCCESS) | |
70 return result; | |
71 handle->Set(temp_handle); | |
72 return ERROR_SUCCESS; | |
73 } | |
74 | |
75 HMODULE module; | 64 HMODULE module; |
76 WlanOpenHandleFunc open_handle_func; | 65 WlanOpenHandleFunc open_handle_func; |
77 WlanEnumInterfacesFunc enum_interfaces_func; | 66 WlanEnumInterfacesFunc enum_interfaces_func; |
78 WlanQueryInterfaceFunc query_interface_func; | 67 WlanQueryInterfaceFunc query_interface_func; |
79 WlanFreeMemoryFunc free_memory_func; | 68 WlanFreeMemoryFunc free_memory_func; |
80 WlanCloseHandleFunc close_handle_func; | 69 WlanCloseHandleFunc close_handle_func; |
81 bool initialized; | 70 bool initialized; |
82 }; | 71 }; |
83 | 72 |
84 } // namespace | 73 } // namespace |
(...skipping 155 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
240 } | 229 } |
241 }; | 230 }; |
242 | 231 |
243 const WlanApi& wlanapi = lazy_wlanapi.Get(); | 232 const WlanApi& wlanapi = lazy_wlanapi.Get(); |
244 if (!wlanapi.initialized) | 233 if (!wlanapi.initialized) |
245 return WIFI_PHY_LAYER_PROTOCOL_NONE; | 234 return WIFI_PHY_LAYER_PROTOCOL_NONE; |
246 | 235 |
247 WlanHandle client; | 236 WlanHandle client; |
248 DWORD cur_version = 0; | 237 DWORD cur_version = 0; |
249 const DWORD kMaxClientVersion = 2; | 238 const DWORD kMaxClientVersion = 2; |
250 DWORD result = wlanapi.OpenHandle(kMaxClientVersion, &cur_version, &client); | 239 DWORD result = wlanapi.open_handle_func(kMaxClientVersion, NULL, &cur_version, |
| 240 client.Receive()); |
251 if (result != ERROR_SUCCESS) | 241 if (result != ERROR_SUCCESS) |
252 return WIFI_PHY_LAYER_PROTOCOL_NONE; | 242 return WIFI_PHY_LAYER_PROTOCOL_NONE; |
253 | 243 |
254 WLAN_INTERFACE_INFO_LIST* interface_list_ptr = NULL; | 244 WLAN_INTERFACE_INFO_LIST* interface_list_ptr = NULL; |
255 result = wlanapi.enum_interfaces_func(client, NULL, &interface_list_ptr); | 245 result = wlanapi.enum_interfaces_func(client, NULL, &interface_list_ptr); |
256 if (result != ERROR_SUCCESS) | 246 if (result != ERROR_SUCCESS) |
257 return WIFI_PHY_LAYER_PROTOCOL_NONE; | 247 return WIFI_PHY_LAYER_PROTOCOL_NONE; |
258 scoped_ptr<WLAN_INTERFACE_INFO_LIST, WlanApiDeleter> interface_list( | 248 scoped_ptr<WLAN_INTERFACE_INFO_LIST, WlanApiDeleter> interface_list( |
259 interface_list_ptr); | 249 interface_list_ptr); |
260 | 250 |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
296 case dot11_phy_type_erp: | 286 case dot11_phy_type_erp: |
297 return WIFI_PHY_LAYER_PROTOCOL_G; | 287 return WIFI_PHY_LAYER_PROTOCOL_G; |
298 case dot11_phy_type_ht: | 288 case dot11_phy_type_ht: |
299 return WIFI_PHY_LAYER_PROTOCOL_N; | 289 return WIFI_PHY_LAYER_PROTOCOL_N; |
300 default: | 290 default: |
301 return WIFI_PHY_LAYER_PROTOCOL_UNKNOWN; | 291 return WIFI_PHY_LAYER_PROTOCOL_UNKNOWN; |
302 } | 292 } |
303 } | 293 } |
304 | 294 |
305 } // namespace net | 295 } // namespace net |
OLD | NEW |