gecko-dev/netwerk/wifi/win_wlanLibrary.cpp

156 lines
3.9 KiB
C++

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "win_wlanLibrary.h"
// Moz headers (alphabetical)
WinWLANLibrary*
WinWLANLibrary::Load()
{
WinWLANLibrary *ret = new WinWLANLibrary();
if (!ret) {
return nullptr;
}
if (!ret->Initialize()) {
delete ret;
return nullptr;
}
return ret;
}
WinWLANLibrary::WinWLANLibrary()
: mWlanLibrary(nullptr),
mWlanHandle(nullptr),
mWlanEnumInterfacesPtr(nullptr),
mWlanGetNetworkBssListPtr(nullptr),
mWlanFreeMemoryPtr(nullptr),
mWlanCloseHandlePtr(nullptr),
mWlanOpenHandlePtr(nullptr),
mWlanRegisterNotificationPtr(nullptr),
mWlanScanPtr(nullptr)
{
}
HANDLE
WinWLANLibrary::GetWLANHandle() const
{
return mWlanHandle;
}
decltype(::WlanEnumInterfaces)*
WinWLANLibrary::GetWlanEnumInterfacesPtr() const
{
return mWlanEnumInterfacesPtr;
}
decltype(::WlanGetNetworkBssList)*
WinWLANLibrary::GetWlanGetNetworkBssListPtr() const
{
return mWlanGetNetworkBssListPtr;
}
decltype(::WlanFreeMemory)*
WinWLANLibrary::GetWlanFreeMemoryPtr() const
{
return mWlanFreeMemoryPtr;
}
decltype(::WlanCloseHandle)*
WinWLANLibrary::GetWlanCloseHandlePtr() const
{
return mWlanCloseHandlePtr;
}
decltype(::WlanOpenHandle)*
WinWLANLibrary::GetWlanOpenHandlePtr() const
{
return mWlanOpenHandlePtr;
}
decltype(::WlanRegisterNotification)*
WinWLANLibrary::GetWlanRegisterNotificationPtr() const
{
return mWlanRegisterNotificationPtr;
}
decltype(::WlanScan)*
WinWLANLibrary::GetWlanScanPtr() const
{
return mWlanScanPtr;
}
bool
WinWLANLibrary::Initialize()
{
mWlanLibrary = LoadLibrary("Wlanapi.dll");
if (!mWlanLibrary) {
return false;
}
mWlanOpenHandlePtr =
(decltype(::WlanOpenHandle)*) GetProcAddress(mWlanLibrary,
"WlanOpenHandle");
mWlanEnumInterfacesPtr =
(decltype(::WlanEnumInterfaces)*) GetProcAddress(mWlanLibrary,
"WlanEnumInterfaces");
mWlanRegisterNotificationPtr =
(decltype(::WlanRegisterNotification)*) GetProcAddress(mWlanLibrary,
"WlanRegisterNotification");
mWlanScanPtr =
(decltype(::WlanScan)*) GetProcAddress(mWlanLibrary, "WlanScan");
mWlanFreeMemoryPtr =
(decltype(::WlanFreeMemory)*) GetProcAddress(mWlanLibrary,
"WlanFreeMemory");
mWlanCloseHandlePtr =
(decltype(::WlanCloseHandle)*) GetProcAddress(mWlanLibrary,
"WlanCloseHandle");
mWlanGetNetworkBssListPtr =
(decltype(::WlanGetNetworkBssList)*) GetProcAddress(mWlanLibrary,
"WlanGetNetworkBssList");
if (!mWlanOpenHandlePtr ||
!mWlanEnumInterfacesPtr ||
!mWlanRegisterNotificationPtr ||
!mWlanGetNetworkBssListPtr ||
!mWlanScanPtr ||
!mWlanFreeMemoryPtr ||
!mWlanCloseHandlePtr) {
return false;
}
// Get the handle to the WLAN API.
DWORD negotiated_version;
// We could be executing on either Windows XP or Windows Vista, so use the
// lower version of the client WLAN API. It seems that the negotiated version
// is the Vista version irrespective of what we pass!
static const int kXpWlanClientVersion = 1;
if (ERROR_SUCCESS !=
(*mWlanOpenHandlePtr)(kXpWlanClientVersion,
nullptr,
&negotiated_version,
&mWlanHandle)) {
return false;
}
return true;
}
WinWLANLibrary::~WinWLANLibrary()
{
if (mWlanLibrary) {
if (mWlanHandle) {
(*mWlanCloseHandlePtr)(mWlanLibrary, mWlanHandle);
mWlanHandle = nullptr;
}
::FreeLibrary(mWlanLibrary);
mWlanLibrary = nullptr;
}
}