network_interfaces_win.h 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. // Copyright (c) 2014 The Chromium 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. #ifndef NET_BASE_NETWORK_INTERFACES_WIN_H_
  5. #define NET_BASE_NETWORK_INTERFACES_WIN_H_
  6. // This file is only used to expose some of the internals
  7. // of network_interfaces_win.cc to tests.
  8. #include <winsock2.h>
  9. #include <iphlpapi.h>
  10. #include <wlanapi.h>
  11. #include "base/win/scoped_handle.h"
  12. #include "net/base/net_export.h"
  13. #include "net/base/network_interfaces.h"
  14. namespace net::internal {
  15. struct NET_EXPORT WlanApi {
  16. typedef DWORD (WINAPI *WlanOpenHandleFunc)(
  17. DWORD, VOID*, DWORD*, HANDLE*);
  18. typedef DWORD (WINAPI *WlanEnumInterfacesFunc)(
  19. HANDLE, VOID*, WLAN_INTERFACE_INFO_LIST**);
  20. typedef DWORD (WINAPI *WlanQueryInterfaceFunc)(
  21. HANDLE, const GUID*, WLAN_INTF_OPCODE, VOID*, DWORD*, VOID**,
  22. WLAN_OPCODE_VALUE_TYPE*);
  23. typedef DWORD (WINAPI *WlanSetInterfaceFunc)(
  24. HANDLE, const GUID*, WLAN_INTF_OPCODE, DWORD, const VOID*, VOID*);
  25. typedef VOID (WINAPI *WlanFreeMemoryFunc)(VOID*);
  26. typedef DWORD (WINAPI *WlanCloseHandleFunc)(HANDLE, VOID*);
  27. WlanApi();
  28. static WlanApi& GetInstance();
  29. template <typename T>
  30. DWORD OpenHandle(DWORD client_version, DWORD* cur_version, T* handle) const {
  31. HANDLE temp_handle;
  32. DWORD result =
  33. open_handle_func(client_version, nullptr, cur_version, &temp_handle);
  34. if (result != ERROR_SUCCESS)
  35. return result;
  36. handle->Set(temp_handle);
  37. return ERROR_SUCCESS;
  38. }
  39. WlanOpenHandleFunc open_handle_func;
  40. WlanEnumInterfacesFunc enum_interfaces_func;
  41. WlanQueryInterfaceFunc query_interface_func;
  42. WlanSetInterfaceFunc set_interface_func;
  43. WlanFreeMemoryFunc free_memory_func;
  44. WlanCloseHandleFunc close_handle_func;
  45. bool initialized;
  46. };
  47. struct WlanApiHandleTraits {
  48. typedef HANDLE Handle;
  49. static bool CloseHandle(HANDLE handle) {
  50. return WlanApi::GetInstance().close_handle_func(handle, nullptr) ==
  51. ERROR_SUCCESS;
  52. }
  53. static bool IsHandleValid(HANDLE handle) {
  54. return base::win::HandleTraits::IsHandleValid(handle);
  55. }
  56. static HANDLE NullHandle() {
  57. return base::win::HandleTraits::NullHandle();
  58. }
  59. };
  60. typedef base::win::GenericScopedHandle<
  61. WlanApiHandleTraits,
  62. base::win::DummyVerifierTraits> WlanHandle;
  63. struct WlanApiDeleter {
  64. inline void operator()(void* ptr) const {
  65. WlanApi::GetInstance().free_memory_func(ptr);
  66. }
  67. };
  68. NET_EXPORT bool GetNetworkListImpl(
  69. NetworkInterfaceList* networks,
  70. int policy,
  71. const IP_ADAPTER_ADDRESSES* ip_adapter_addresses);
  72. } // namespace net::internal
  73. #endif // NET_BASE_NETWORK_INTERFACES_WIN_H_