mirror of
https://github.com/reactos/reactos.git
synced 2024-10-29 19:13:58 +00:00
958 lines
28 KiB
C
958 lines
28 KiB
C
/*
|
|
* PROJECT: ReactOS WLAN command-line configuration utility
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: base/applications/network/wlanconf/wlanconf.c
|
|
* PURPOSE: Allows WLAN configuration via the command prompt
|
|
* COPYRIGHT: Copyright 2012 Cameron Gutman (cameron.gutman@reactos.org)
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include <tchar.h>
|
|
|
|
#include <windef.h>
|
|
#include <winbase.h>
|
|
#include <devioctl.h>
|
|
#include <ntddndis.h>
|
|
#include <nuiouser.h>
|
|
#include <iphlpapi.h>
|
|
|
|
#include <conutils.h>
|
|
|
|
#include "resource.h"
|
|
|
|
BOOL bScan = FALSE;
|
|
|
|
BOOL bConnect = FALSE;
|
|
WCHAR *sSsid = NULL;
|
|
WCHAR *sWepKey = NULL;
|
|
BOOL bAdhoc = FALSE;
|
|
|
|
BOOL bDisconnect = FALSE;
|
|
|
|
VOID DoFormatMessage(DWORD ErrorCode)
|
|
{
|
|
if (ErrorCode == ERROR_SUCCESS)
|
|
return;
|
|
|
|
ConMsgPuts(StdErr, FORMAT_MESSAGE_FROM_SYSTEM,
|
|
NULL, ErrorCode, LANG_USER_DEFAULT);
|
|
}
|
|
|
|
HANDLE
|
|
OpenDriverHandle(VOID)
|
|
{
|
|
HANDLE hDriver;
|
|
DWORD dwBytesReturned;
|
|
BOOL bSuccess;
|
|
|
|
/* Open a handle to the NDISUIO driver */
|
|
hDriver = CreateFileW(NDISUIO_DEVICE_NAME,
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL);
|
|
if (hDriver == INVALID_HANDLE_VALUE)
|
|
return INVALID_HANDLE_VALUE;
|
|
|
|
/* Wait for binds */
|
|
bSuccess = DeviceIoControl(hDriver,
|
|
IOCTL_NDISUIO_BIND_WAIT,
|
|
NULL,
|
|
0,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
CloseHandle(hDriver);
|
|
return INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
return hDriver;
|
|
}
|
|
|
|
BOOL
|
|
IsWlanAdapter(HANDLE hAdapter)
|
|
{
|
|
BOOL bSuccess;
|
|
DWORD dwBytesReturned;
|
|
NDISUIO_QUERY_OID QueryOid;
|
|
|
|
/* WLAN drivers must support this OID */
|
|
QueryOid.Oid = OID_GEN_PHYSICAL_MEDIUM;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
&QueryOid,
|
|
sizeof(QueryOid),
|
|
&QueryOid,
|
|
sizeof(QueryOid),
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess || *(PULONG)QueryOid.Data != NdisPhysicalMediumWirelessLan)
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
OpenAdapterHandle(DWORD Index, HANDLE *hAdapter, IP_ADAPTER_INDEX_MAP *IpInfo)
|
|
{
|
|
HANDLE hDriver;
|
|
BOOL bSuccess;
|
|
DWORD dwBytesReturned;
|
|
DWORD QueryBindingSize = sizeof(NDISUIO_QUERY_BINDING) + (1024 * sizeof(WCHAR));
|
|
PNDISUIO_QUERY_BINDING QueryBinding;
|
|
DWORD dwStatus, dwSize;
|
|
LONG i;
|
|
PIP_INTERFACE_INFO InterfaceInfo = NULL;
|
|
|
|
/* Open the driver handle */
|
|
hDriver = OpenDriverHandle();
|
|
if (hDriver == INVALID_HANDLE_VALUE)
|
|
return FALSE;
|
|
|
|
/* Allocate the binding struct */
|
|
QueryBinding = HeapAlloc(GetProcessHeap(), 0, QueryBindingSize);
|
|
if (!QueryBinding)
|
|
{
|
|
CloseHandle(hDriver);
|
|
return FALSE;
|
|
}
|
|
|
|
/* Query the adapter binding information */
|
|
QueryBinding->BindingIndex = Index;
|
|
bSuccess = DeviceIoControl(hDriver,
|
|
IOCTL_NDISUIO_QUERY_BINDING,
|
|
QueryBinding,
|
|
QueryBindingSize,
|
|
QueryBinding,
|
|
QueryBindingSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryBinding);
|
|
CloseHandle(hDriver);
|
|
return FALSE;
|
|
}
|
|
|
|
/* Bind to the adapter */
|
|
bSuccess = DeviceIoControl(hDriver,
|
|
IOCTL_NDISUIO_OPEN_DEVICE,
|
|
(PUCHAR)QueryBinding + QueryBinding->DeviceNameOffset,
|
|
QueryBinding->DeviceNameLength,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryBinding);
|
|
CloseHandle(hDriver);
|
|
return FALSE;
|
|
}
|
|
|
|
/* Get interface info from the IP helper */
|
|
dwSize = sizeof(IP_INTERFACE_INFO);
|
|
do {
|
|
if (InterfaceInfo) HeapFree(GetProcessHeap(), 0, InterfaceInfo);
|
|
InterfaceInfo = HeapAlloc(GetProcessHeap(), 0, dwSize);
|
|
if (!InterfaceInfo)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryBinding);
|
|
CloseHandle(hDriver);
|
|
return FALSE;
|
|
}
|
|
dwStatus = GetInterfaceInfo(InterfaceInfo, &dwSize);
|
|
} while (dwStatus == ERROR_INSUFFICIENT_BUFFER);
|
|
|
|
if (dwStatus != NO_ERROR)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryBinding);
|
|
HeapFree(GetProcessHeap(), 0, InterfaceInfo);
|
|
CloseHandle(hDriver);
|
|
return FALSE;
|
|
}
|
|
|
|
for (i = 0; i < InterfaceInfo->NumAdapters; i++)
|
|
{
|
|
if (wcsstr((PWCHAR)((PUCHAR)QueryBinding + QueryBinding->DeviceNameOffset),
|
|
InterfaceInfo->Adapter[i].Name))
|
|
{
|
|
*IpInfo = InterfaceInfo->Adapter[i];
|
|
*hAdapter = hDriver;
|
|
|
|
HeapFree(GetProcessHeap(), 0, QueryBinding);
|
|
HeapFree(GetProcessHeap(), 0, InterfaceInfo);
|
|
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, QueryBinding);
|
|
HeapFree(GetProcessHeap(), 0, InterfaceInfo);
|
|
CloseHandle(hDriver);
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
/* Only works with the first adapter for now */
|
|
BOOL
|
|
OpenWlanAdapter(HANDLE *hAdapter, IP_ADAPTER_INDEX_MAP *IpInfo)
|
|
{
|
|
DWORD dwCurrentIndex;
|
|
|
|
for (dwCurrentIndex = 0; ; dwCurrentIndex++)
|
|
{
|
|
if (!OpenAdapterHandle(dwCurrentIndex, hAdapter, IpInfo))
|
|
break;
|
|
|
|
if (IsWlanAdapter(*hAdapter))
|
|
return TRUE;
|
|
else
|
|
CloseHandle(*hAdapter);
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL
|
|
WlanDisconnect(HANDLE hAdapter, PIP_ADAPTER_INDEX_MAP IpInfo)
|
|
{
|
|
BOOL bSuccess;
|
|
DWORD dwBytesReturned;
|
|
NDISUIO_SET_OID SetOid;
|
|
|
|
/* Release this IP address */
|
|
IpReleaseAddress(IpInfo);
|
|
|
|
/* Disassociate from the AP */
|
|
SetOid.Oid = OID_802_11_DISASSOCIATE;
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
&SetOid,
|
|
FIELD_OFFSET(NDISUIO_SET_OID, Data),
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
return FALSE;
|
|
|
|
ConResPuts(StdOut, IDS_SUCCESS);
|
|
return TRUE;
|
|
}
|
|
|
|
static
|
|
UCHAR
|
|
CharToHex(WCHAR Char)
|
|
{
|
|
if ((Char >= L'0') && (Char <= L'9'))
|
|
return Char - L'0';
|
|
|
|
if ((Char >= L'a') && (Char <= L'f'))
|
|
return Char - L'a' + 10;
|
|
|
|
if ((Char >= L'A') && (Char <= L'F'))
|
|
return Char - L'A' + 10;
|
|
|
|
return 0;
|
|
}
|
|
|
|
BOOL
|
|
WlanPrintCurrentStatus(HANDLE hAdapter)
|
|
{
|
|
PNDISUIO_QUERY_OID QueryOid;
|
|
DWORD QueryOidSize;
|
|
BOOL bSuccess;
|
|
DWORD dwBytesReturned;
|
|
PNDIS_802_11_SSID SsidInfo;
|
|
CHAR SsidBuffer[NDIS_802_11_LENGTH_SSID + 1];
|
|
DWORD i;
|
|
WCHAR szMsgBuf[128];
|
|
|
|
QueryOidSize = FIELD_OFFSET(NDISUIO_QUERY_OID, Data) + sizeof(NDIS_802_11_SSID);
|
|
QueryOid = HeapAlloc(GetProcessHeap(), 0, QueryOidSize);
|
|
if (!QueryOid)
|
|
return FALSE;
|
|
|
|
QueryOid->Oid = OID_802_11_SSID;
|
|
SsidInfo = (PNDIS_802_11_SSID)QueryOid->Data;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
return FALSE;
|
|
}
|
|
|
|
/* Copy the SSID to our internal buffer and terminate it */
|
|
RtlCopyMemory(SsidBuffer, SsidInfo->Ssid, SsidInfo->SsidLength);
|
|
SsidBuffer[SsidInfo->SsidLength] = 0;
|
|
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
QueryOidSize = FIELD_OFFSET(NDISUIO_QUERY_OID, Data) + sizeof(NDIS_802_11_MAC_ADDRESS);
|
|
QueryOid = HeapAlloc(GetProcessHeap(), 0, QueryOidSize);
|
|
if (!QueryOid)
|
|
return FALSE;
|
|
|
|
QueryOid->Oid = OID_802_11_BSSID;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (SsidInfo->SsidLength == 0 || !bSuccess)
|
|
{
|
|
ConResPuts(StdOut, IDS_WLAN_DISCONNECT);
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
return TRUE;
|
|
}
|
|
else
|
|
{
|
|
ConResPuts(StdOut, IDS_MSG_CURRENT_WIRELESS);
|
|
}
|
|
|
|
printf("SSID: %s\n", SsidBuffer);
|
|
|
|
printf("BSSID: ");
|
|
for (i = 0; i < sizeof(NDIS_802_11_MAC_ADDRESS); i++)
|
|
{
|
|
UINT BssidData = QueryOid->Data[i];
|
|
|
|
printf("%.2x", BssidData);
|
|
|
|
if (i != sizeof(NDIS_802_11_MAC_ADDRESS) - 1)
|
|
printf(":");
|
|
}
|
|
printf("\n");
|
|
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
QueryOidSize = sizeof(NDISUIO_QUERY_OID);
|
|
QueryOid = HeapAlloc(GetProcessHeap(), 0, QueryOidSize);
|
|
if (!QueryOid)
|
|
return FALSE;
|
|
|
|
QueryOid->Oid = OID_802_11_INFRASTRUCTURE_MODE;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
return FALSE;
|
|
}
|
|
|
|
K32LoadStringW(GetModuleHandle(NULL),
|
|
*(PUINT)QueryOid->Data == Ndis802_11IBSS ? IDS_ADHOC : IDS_INFRASTRUCTURE,
|
|
szMsgBuf,
|
|
ARRAYSIZE(szMsgBuf));
|
|
ConResPrintf(StdOut, IDS_MSG_NETWORK_MODE, szMsgBuf);
|
|
|
|
QueryOid->Oid = OID_802_11_WEP_STATUS;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
return FALSE;
|
|
}
|
|
|
|
K32LoadStringW(GetModuleHandle(NULL),
|
|
*(PUINT)QueryOid->Data == Ndis802_11WEPEnabled ? IDS_YES : IDS_NO,
|
|
szMsgBuf,
|
|
ARRAYSIZE(szMsgBuf));
|
|
ConResPrintf(StdOut, IDS_MSG_WEP_ENABLED, szMsgBuf);
|
|
|
|
printf("\n");
|
|
QueryOid->Oid = OID_802_11_RSSI;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
/* This OID is optional */
|
|
printf("RSSI: %i dBm\n", *(PINT)QueryOid->Data);
|
|
}
|
|
|
|
QueryOid->Oid = OID_802_11_TX_POWER_LEVEL;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
/* This OID is optional */
|
|
ConResPrintf(StdOut, IDS_MSG_TRANSMISSION_POWER, *(PUINT)QueryOid->Data);
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
QueryOid->Oid = OID_802_11_NUMBER_OF_ANTENNAS;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
/* This OID is optional */
|
|
ConResPrintf(StdOut, IDS_MSG_ANTENNA_COUNT, *(PUINT)QueryOid->Data);
|
|
}
|
|
|
|
QueryOid->Oid = OID_802_11_TX_ANTENNA_SELECTED;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
UINT TransmitAntenna = *(PUINT)QueryOid->Data;
|
|
|
|
if (TransmitAntenna != 0xFFFFFFFF)
|
|
ConResPrintf(StdOut, IDS_MSG_TRANSMIT_ANTENNA, TransmitAntenna);
|
|
else
|
|
ConResPuts(StdOut, IDS_MSG_TRANSMIT_ANTENNA_ANY);
|
|
}
|
|
|
|
QueryOid->Oid = OID_802_11_RX_ANTENNA_SELECTED;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
UINT ReceiveAntenna = *(PUINT)QueryOid->Data;
|
|
|
|
if (ReceiveAntenna != 0xFFFFFFFF)
|
|
ConResPrintf(StdOut, IDS_MSG_RECEIVE_ANTENNA, ReceiveAntenna);
|
|
else
|
|
ConResPuts(StdOut, IDS_MSG_RECEIVE_ANTENNA_ANY);
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
QueryOid->Oid = OID_802_11_FRAGMENTATION_THRESHOLD;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
/* This OID is optional */
|
|
ConResPrintf(StdOut, IDS_MSG_FRAGMENT_THRESHOLD, *(PUINT)QueryOid->Data);
|
|
}
|
|
|
|
QueryOid->Oid = OID_802_11_RTS_THRESHOLD;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (bSuccess)
|
|
{
|
|
/* This OID is optional */
|
|
ConResPrintf(StdOut, IDS_MSG_RTS_THRESHOLD, *(PUINT)QueryOid->Data);
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
|
|
printf("\n");
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
WlanConnect(HANDLE hAdapter)
|
|
{
|
|
CHAR SsidBuffer[NDIS_802_11_LENGTH_SSID + 1];
|
|
BOOL bSuccess;
|
|
DWORD dwBytesReturned, SetOidSize;
|
|
PNDISUIO_SET_OID SetOid;
|
|
PNDIS_802_11_SSID Ssid;
|
|
DWORD i;
|
|
|
|
SetOidSize = sizeof(NDISUIO_SET_OID);
|
|
SetOid = HeapAlloc(GetProcessHeap(), 0, SetOidSize);
|
|
if (!SetOid)
|
|
return FALSE;
|
|
|
|
/* Set the network mode */
|
|
SetOid->Oid = OID_802_11_INFRASTRUCTURE_MODE;
|
|
*(PULONG)SetOid->Data = bAdhoc ? Ndis802_11IBSS : Ndis802_11Infrastructure;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
SetOid,
|
|
SetOidSize,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
return FALSE;
|
|
}
|
|
|
|
/* Set the authentication mode */
|
|
SetOid->Oid = OID_802_11_AUTHENTICATION_MODE;
|
|
*(PULONG)SetOid->Data = sWepKey ? Ndis802_11AuthModeShared : Ndis802_11AuthModeOpen;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
SetOid,
|
|
SetOidSize,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
return FALSE;
|
|
}
|
|
|
|
if (sWepKey)
|
|
{
|
|
PNDIS_802_11_WEP WepData;
|
|
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
|
|
SetOidSize = FIELD_OFFSET(NDISUIO_SET_OID, Data) +
|
|
FIELD_OFFSET(NDIS_802_11_WEP, KeyMaterial) +
|
|
(wcslen(sWepKey) >> 1);
|
|
SetOid = HeapAlloc(GetProcessHeap(), 0, SetOidSize);
|
|
if (!SetOid)
|
|
return FALSE;
|
|
|
|
/* Add the WEP key */
|
|
SetOid->Oid = OID_802_11_ADD_WEP;
|
|
WepData = (PNDIS_802_11_WEP)SetOid->Data;
|
|
|
|
WepData->KeyIndex = 0x80000000;
|
|
WepData->KeyLength = wcslen(sWepKey) >> 1;
|
|
WepData->Length = FIELD_OFFSET(NDIS_802_11_WEP, KeyMaterial) + WepData->KeyLength;
|
|
|
|
/* Assemble the hex key */
|
|
i = 0;
|
|
while (sWepKey[i << 1] != '\0')
|
|
{
|
|
WepData->KeyMaterial[i] = CharToHex(sWepKey[i << 1]) << 4;
|
|
WepData->KeyMaterial[i] |= CharToHex(sWepKey[(i << 1) + 1]);
|
|
i++;
|
|
}
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
SetOid,
|
|
SetOidSize,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
/* Set the encryption status */
|
|
SetOid->Oid = OID_802_11_WEP_STATUS;
|
|
*(PULONG)SetOid->Data = sWepKey ? Ndis802_11WEPEnabled : Ndis802_11WEPDisabled;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
SetOid,
|
|
SetOidSize,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
return FALSE;
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
SetOidSize = FIELD_OFFSET(NDISUIO_SET_OID, Data) + sizeof(NDIS_802_11_MAC_ADDRESS);
|
|
SetOid = HeapAlloc(GetProcessHeap(), 0, SetOidSize);
|
|
if (!SetOid)
|
|
return FALSE;
|
|
|
|
/* Set the BSSID */
|
|
SetOid->Oid = OID_802_11_BSSID;
|
|
RtlFillMemory(SetOid->Data, sizeof(NDIS_802_11_MAC_ADDRESS), 0xFF);
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
SetOid,
|
|
SetOidSize,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
return FALSE;
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
SetOidSize = FIELD_OFFSET(NDISUIO_SET_OID, Data) + sizeof(NDIS_802_11_SSID);
|
|
SetOid = HeapAlloc(GetProcessHeap(), 0, SetOidSize);
|
|
if (!SetOid)
|
|
return FALSE;
|
|
|
|
/* Finally, set the SSID */
|
|
SetOid->Oid = OID_802_11_SSID;
|
|
Ssid = (PNDIS_802_11_SSID)SetOid->Data;
|
|
|
|
snprintf(SsidBuffer, sizeof(SsidBuffer), "%S", sSsid);
|
|
RtlCopyMemory(Ssid->Ssid, SsidBuffer, strlen(SsidBuffer));
|
|
Ssid->SsidLength = strlen(SsidBuffer);
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
SetOid,
|
|
SetOidSize,
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
|
|
HeapFree(GetProcessHeap(), 0, SetOid);
|
|
|
|
if (!bSuccess)
|
|
return FALSE;
|
|
|
|
ConResPuts(StdOut, IDS_SUCCESS);
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
WlanScan(HANDLE hAdapter)
|
|
{
|
|
BOOL bSuccess;
|
|
DWORD dwBytesReturned;
|
|
NDISUIO_SET_OID SetOid;
|
|
PNDISUIO_QUERY_OID QueryOid;
|
|
DWORD QueryOidSize;
|
|
PNDIS_802_11_BSSID_LIST BssidList;
|
|
DWORD i, j;
|
|
DWORD dwNetworkCount;
|
|
WCHAR szMsgBuf[128];
|
|
|
|
SetOid.Oid = OID_802_11_BSSID_LIST_SCAN;
|
|
|
|
/* Send the scan OID */
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_SET_OID_VALUE,
|
|
&SetOid,
|
|
FIELD_OFFSET(NDISUIO_SET_OID, Data),
|
|
NULL,
|
|
0,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess)
|
|
return FALSE;
|
|
|
|
/* Wait 2 seconds for the scan to return some results */
|
|
Sleep(2000);
|
|
|
|
/* Allocate space for 10 networks to be returned initially */
|
|
QueryOid = NULL;
|
|
dwNetworkCount = 10;
|
|
for (;;)
|
|
{
|
|
if (QueryOid)
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
|
|
QueryOidSize = sizeof(NDISUIO_QUERY_OID) + (sizeof(NDIS_WLAN_BSSID) * dwNetworkCount);
|
|
QueryOid = HeapAlloc(GetProcessHeap(), 0, QueryOidSize);
|
|
if (!QueryOid)
|
|
return FALSE;
|
|
|
|
QueryOid->Oid = OID_802_11_BSSID_LIST;
|
|
BssidList = (PNDIS_802_11_BSSID_LIST)QueryOid->Data;
|
|
|
|
bSuccess = DeviceIoControl(hAdapter,
|
|
IOCTL_NDISUIO_QUERY_OID_VALUE,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
QueryOid,
|
|
QueryOidSize,
|
|
&dwBytesReturned,
|
|
NULL);
|
|
if (!bSuccess && GetLastError() == ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
/* Try allocating space for 10 more networks */
|
|
dwNetworkCount += 10;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!bSuccess)
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
return FALSE;
|
|
}
|
|
|
|
if (BssidList->NumberOfItems == 0)
|
|
{
|
|
ConResPuts(StdOut, IDS_NO_NETWORK);
|
|
}
|
|
else
|
|
{
|
|
PNDIS_WLAN_BSSID BssidInfo = BssidList->Bssid;
|
|
for (i = 0; i < BssidList->NumberOfItems; i++)
|
|
{
|
|
PNDIS_802_11_SSID Ssid = &BssidInfo->Ssid;
|
|
NDIS_802_11_RSSI Rssi = BssidInfo->Rssi;
|
|
NDIS_802_11_NETWORK_INFRASTRUCTURE NetworkType = BssidInfo->InfrastructureMode;
|
|
CHAR SsidBuffer[NDIS_802_11_LENGTH_SSID + 1];
|
|
UINT Rate;
|
|
|
|
/* SSID member is a non-null terminated ASCII string */
|
|
RtlCopyMemory(SsidBuffer, Ssid->Ssid, Ssid->SsidLength);
|
|
SsidBuffer[Ssid->SsidLength] = 0;
|
|
|
|
printf("\nSSID: %s\n", SsidBuffer);
|
|
|
|
printf("BSSID: ");
|
|
for (j = 0; j < sizeof(NDIS_802_11_MAC_ADDRESS); j++)
|
|
{
|
|
UINT BssidData = BssidInfo->MacAddress[j];
|
|
|
|
printf("%.2x", BssidData);
|
|
|
|
if (j != sizeof(NDIS_802_11_MAC_ADDRESS) - 1)
|
|
printf(":");
|
|
}
|
|
printf("\n");
|
|
|
|
K32LoadStringW(GetModuleHandle(NULL),
|
|
BssidInfo->Privacy == 0 ? IDS_NO : IDS_YES,
|
|
szMsgBuf,
|
|
ARRAYSIZE(szMsgBuf));
|
|
ConResPrintf(StdOut, IDS_MSG_ENCRYPTED, szMsgBuf);
|
|
K32LoadStringW(GetModuleHandle(NULL),
|
|
NetworkType == Ndis802_11IBSS ? IDS_ADHOC : IDS_INFRASTRUCTURE,
|
|
szMsgBuf,
|
|
ARRAYSIZE(szMsgBuf));
|
|
ConResPrintf(StdOut, IDS_MSG_NETWORK_TYPE, szMsgBuf);
|
|
ConResPrintf(StdOut, IDS_MSG_RSSI, (int)Rssi);
|
|
ConResPuts(StdOut, IDS_MSG_SUPPORT_RATE);
|
|
|
|
for (j = 0; j < NDIS_802_11_LENGTH_RATES; j++)
|
|
{
|
|
Rate = BssidInfo->SupportedRates[j];
|
|
if (Rate != 0)
|
|
{
|
|
/* Bit 7 is the basic rates bit */
|
|
Rate = Rate & 0x7F;
|
|
|
|
/* SupportedRates are in units of .5 */
|
|
if (Rate & 0x01)
|
|
{
|
|
/* Bit 0 is set so we need to add 0.5 */
|
|
printf("%u.5 ", (Rate >> 1));
|
|
}
|
|
else
|
|
{
|
|
/* Bit 0 is clear so just print the conversion */
|
|
printf("%u ", (Rate >> 1));
|
|
}
|
|
}
|
|
}
|
|
printf("\n");
|
|
|
|
/* Move to the next entry */
|
|
BssidInfo = (PNDIS_WLAN_BSSID)((PUCHAR)BssidInfo + BssidInfo->Length);
|
|
}
|
|
}
|
|
|
|
HeapFree(GetProcessHeap(), 0, QueryOid);
|
|
|
|
return bSuccess;
|
|
}
|
|
|
|
BOOL ParseCmdline(int argc, WCHAR *argv[])
|
|
{
|
|
INT i;
|
|
|
|
for (i = 1; i < argc; i++)
|
|
{
|
|
if (argv[i][0] == L'-')
|
|
{
|
|
switch (argv[i][1])
|
|
{
|
|
case L's':
|
|
bScan = TRUE;
|
|
break;
|
|
case L'd':
|
|
bDisconnect = TRUE;
|
|
break;
|
|
case L'c':
|
|
if (i == argc - 1)
|
|
{
|
|
ConResPuts(StdOut, IDS_USAGE);
|
|
return FALSE;
|
|
}
|
|
bConnect = TRUE;
|
|
sSsid = argv[++i];
|
|
break;
|
|
case L'w':
|
|
if (i == argc - 1)
|
|
{
|
|
ConResPuts(StdOut, IDS_USAGE);
|
|
return FALSE;
|
|
}
|
|
sWepKey = argv[++i];
|
|
break;
|
|
case L'a':
|
|
bAdhoc = TRUE;
|
|
break;
|
|
default :
|
|
ConResPuts(StdOut, IDS_USAGE);
|
|
return FALSE;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
ConResPuts(StdOut, IDS_USAGE);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
int wmain(int argc, WCHAR *argv[])
|
|
{
|
|
HANDLE hAdapter;
|
|
IP_ADAPTER_INDEX_MAP IpInfo;
|
|
|
|
/* Initialize the Console Standard Streams */
|
|
ConInitStdStreams();
|
|
|
|
if (!ParseCmdline(argc, argv))
|
|
return -1;
|
|
|
|
if (!OpenWlanAdapter(&hAdapter, &IpInfo))
|
|
{
|
|
ConResPuts(StdOut, IDS_NO_WLAN_ADAPTER);
|
|
return -1;
|
|
}
|
|
|
|
if (bScan)
|
|
{
|
|
if (!WlanScan(hAdapter))
|
|
{
|
|
DoFormatMessage(GetLastError());
|
|
CloseHandle(hAdapter);
|
|
return -1;
|
|
}
|
|
}
|
|
else if (bDisconnect)
|
|
{
|
|
if (!WlanDisconnect(hAdapter, &IpInfo))
|
|
{
|
|
DoFormatMessage(GetLastError());
|
|
CloseHandle(hAdapter);
|
|
return -1;
|
|
}
|
|
}
|
|
else if (bConnect)
|
|
{
|
|
if (!WlanConnect(hAdapter))
|
|
{
|
|
DoFormatMessage(GetLastError());
|
|
CloseHandle(hAdapter);
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (!WlanPrintCurrentStatus(hAdapter))
|
|
{
|
|
DoFormatMessage(GetLastError());
|
|
CloseHandle(hAdapter);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
CloseHandle(hAdapter);
|
|
return 0;
|
|
}
|