156 lines
3.9 KiB
C++
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;
|
|
}
|
|
}
|