/* 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;
  static const int kXpWlanClientVersion = 2;
  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;
  }
}