зеркало из https://github.com/mozilla/gecko-dev.git
130 строки
4.5 KiB
C++
130 строки
4.5 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 "windows.h"
|
|
#include "wlanapi.h"
|
|
|
|
#include "stdlib.h"
|
|
|
|
#include "nsWifiMonitor.h"
|
|
#include "nsWifiAccessPoint.h"
|
|
|
|
#include "nsServiceManagerUtils.h"
|
|
#include "nsComponentManagerUtils.h"
|
|
#include "nsIMutableArray.h"
|
|
|
|
using namespace mozilla;
|
|
|
|
nsresult
|
|
nsWifiMonitor::DoScan()
|
|
{
|
|
HINSTANCE wlan_library = LoadLibrary("Wlanapi.dll");
|
|
if (!wlan_library)
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
|
|
WlanOpenHandleFunction WlanOpenHandle = (WlanOpenHandleFunction) GetProcAddress(wlan_library, "WlanOpenHandle");
|
|
WlanEnumInterfacesFunction WlanEnumInterfaces = (WlanEnumInterfacesFunction) GetProcAddress(wlan_library, "WlanEnumInterfaces");
|
|
WlanGetNetworkBssListFunction WlanGetNetworkBssList = (WlanGetNetworkBssListFunction) GetProcAddress(wlan_library, "WlanGetNetworkBssList");
|
|
WlanFreeMemoryFunction WlanFreeMemory = (WlanFreeMemoryFunction) GetProcAddress(wlan_library, "WlanFreeMemory");
|
|
WlanCloseHandleFunction WlanCloseHandle = (WlanCloseHandleFunction) GetProcAddress(wlan_library, "WlanCloseHandle");
|
|
|
|
if (!WlanOpenHandle ||
|
|
!WlanEnumInterfaces ||
|
|
!WlanGetNetworkBssList ||
|
|
!WlanFreeMemory ||
|
|
!WlanCloseHandle)
|
|
return NS_ERROR_FAILURE;
|
|
|
|
// Regularly get the access point data.
|
|
|
|
nsCOMArray<nsWifiAccessPoint> lastAccessPoints;
|
|
nsCOMArray<nsWifiAccessPoint> accessPoints;
|
|
|
|
do {
|
|
accessPoints.Clear();
|
|
|
|
// Get the handle to the WLAN API.
|
|
DWORD negotiated_version;
|
|
HANDLE wlan_handle = NULL;
|
|
// 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 ((*WlanOpenHandle)(kXpWlanClientVersion,
|
|
NULL,
|
|
&negotiated_version,
|
|
&wlan_handle) != ERROR_SUCCESS) {
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
}
|
|
|
|
// try again later.
|
|
if (!wlan_handle)
|
|
return NS_ERROR_FAILURE;
|
|
|
|
// Get the list of interfaces. WlanEnumInterfaces allocates interface_list.
|
|
WLAN_INTERFACE_INFO_LIST *interface_list = NULL;
|
|
if ((*WlanEnumInterfaces)(wlan_handle, NULL, &interface_list) != ERROR_SUCCESS) {
|
|
// try again later
|
|
(*WlanCloseHandle)(wlan_handle, NULL);
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
// Go through the list of interfaces and get the data for each.
|
|
for (int i = 0; i < static_cast<int>(interface_list->dwNumberOfItems); ++i) {
|
|
|
|
WLAN_BSS_LIST *bss_list;
|
|
HRESULT rv = (*WlanGetNetworkBssList)(wlan_handle,
|
|
&interface_list->InterfaceInfo[i].InterfaceGuid,
|
|
NULL, // Use all SSIDs.
|
|
DOT11_BSS_TYPE_UNUSED,
|
|
false, // bSecurityEnabled - unused
|
|
NULL, // reserved
|
|
&bss_list);
|
|
if (rv != ERROR_SUCCESS) {
|
|
continue;
|
|
}
|
|
|
|
for (int j = 0; j < static_cast<int>(bss_list->dwNumberOfItems); ++j) {
|
|
|
|
nsWifiAccessPoint* ap = new nsWifiAccessPoint();
|
|
if (!ap)
|
|
continue;
|
|
|
|
const WLAN_BSS_ENTRY bss_entry = bss_list->wlanBssEntries[j];
|
|
|
|
ap->setMac(bss_entry.dot11Bssid);
|
|
ap->setSignal(bss_entry.lRssi);
|
|
ap->setSSID((char*) bss_entry.dot11Ssid.ucSSID,
|
|
bss_entry.dot11Ssid.uSSIDLength);
|
|
|
|
accessPoints.AppendObject(ap);
|
|
}
|
|
(*WlanFreeMemory)(bss_list);
|
|
}
|
|
|
|
// Free interface_list.
|
|
(*WlanFreeMemory)(interface_list);
|
|
|
|
// Close the handle.
|
|
(*WlanCloseHandle)(wlan_handle, NULL);
|
|
|
|
|
|
bool accessPointsChanged = !AccessPointsEqual(accessPoints, lastAccessPoints);
|
|
ReplaceArray(lastAccessPoints, accessPoints);
|
|
|
|
nsresult rv = CallWifiListeners(lastAccessPoints, accessPointsChanged);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// wait for some reasonable amount of time. pref?
|
|
LOG(("waiting on monitor\n"));
|
|
|
|
ReentrantMonitorAutoEnter mon(mReentrantMonitor);
|
|
mon.Wait(PR_SecondsToInterval(60));
|
|
}
|
|
while (mKeepGoing);
|
|
|
|
return NS_OK;
|
|
}
|