CLI WAP_Scanner Rev 1

/* I know the interface is ugly and not very usable at the moment, the point is to learn from the code, the usable prettyful one will be Rev 2*/
 
#include <iostream>
#include <Windows.h>
#include <wlanapi.h> //Wifi functions
#include <strsafe.h> //Safe String Conversion, Wanted to make sure this is Unicode compat. so it adds a little more work
 
#pragma comment(lib, "wlanapi.lib")
 
using namespace std;
 
/** A command line interface WAP scanner built with MSVC using the unicode character set. If you do not recognize the functions or members
 * its probably because they are the unicode versions for MSVC
 * @author BinaryShinigami
 */
 
int wmain(int argc, WCHAR **argv)
{
    HANDLE wifiHandle = 0;
    DWORD wifiVersion = 0;
    PWLAN_INTERFACE_INFO_LIST pInterfaceInfoList = 0;
    PWLAN_INTERFACE_INFO pInterfaceInfo = 0;
    PWLAN_AVAILABLE_NETWORK_LIST pNetworkList = 0;
    PWLAN_AVAILABLE_NETWORK pNetwork = 0;
    GUID guidSelectedInterface;
    DWORD nInterface;
    bool bIsRunning = true;
 
    //Get a handle to use with the Native Wifi API functions
    if (WlanOpenHandle(2, NULL, &wifiVersion, &wifiHandle) != ERROR_SUCCESS)
    {
        wcout <<L"Unable to get wifi Handle!" <<endl;
        return 0;
    }
 
    //Enumerate Network Interfaces to get the one we want to use.
    if (WlanEnumInterfaces(wifiHandle, NULL,&pInterfaceInfoList) != ERROR_SUCCESS)
    {
        wcout <<L"Unable to enumerate interface list!" <<endl;
        return 0;
    }
 
    wcout <<L"Interfaces Found: " <<pInterfaceInfoList->dwNumberOfItems <<endl <<endl <<L"Listing Interfaces: " <<endl;
    for (unsigned int i = 0; i < pInterfaceInfoList->dwNumberOfItems; i++)
    {
        pInterfaceInfo = (PWLAN_INTERFACE_INFO)&pInterfaceInfoList->InterfaceInfo[i];
        wcout <<L"Interface Number: " <<i <<endl;
        wcout <<L"Interface Description: " <<pInterfaceInfo->strInterfaceDescription <<endl;
        wcout <<L"Interface State: ";
        switch (pInterfaceInfo->isState)
        {
 
            case wlan_interface_state_not_ready:
                wcout <<L"Not Ready" <<endl;
                break;
            case wlan_interface_state_connected:
                wcout <<L"Connected" <<endl;
                break;
            case wlan_interface_state_ad_hoc_network_formed:
                wcout <<L"In Ad Hoc Network" <<endl;
                break;
            case wlan_interface_state_disconnecting:
                wcout <<L"Disconnecting" <<endl;
                break;
            case wlan_interface_state_disconnected:
                wcout <<L"Disconnected" <<endl;
                break;
            case wlan_interface_state_associating:
                wcout <<L"Associating" <<endl;
                break;
            case wlan_interface_state_discovering:
                wcout<<L"Discovering Networks" <<endl;
                break;
            case wlan_interface_state_authenticating:
                wcout <<L"Authenticating Network" <<endl;
                break;
            default:
                wcout <<L"WTF! Unknown State, Your Computer Is Fucked!" <<endl;
 
        }
    }
 
    wcout <<"Please select an interface to utilize for scanning: ";
    wcin >>nInterface;
 
    while (nInterface >= pInterfaceInfoList->dwNumberOfItems)
    {
        wcout <<"Invalid Interface Number! Please Try Again: ";
        wcin >>nInterface;
    }
 
    guidSelectedInterface = pInterfaceInfoList->InterfaceInfo[nInterface].InterfaceGuid; //Pull the GUID of the interface we want to use.
 
    //We are done with the interface list free it
    WlanFreeMemory(pInterfaceInfoList);
 
    //Keep listing the available wifi networks ^_^
    while (bIsRunning) {
        if (WlanGetAvailableNetworkList(wifiHandle, &guidSelectedInterface, 0, NULL, &pNetworkList) != ERROR_SUCCESS)
        {
            wcout <<L"Error! Quitting!" <<endl;
            WlanCloseHandle(wifiHandle,NULL);
            return 0;
        }
 
        wcout <<L"Networks Found: " <<pNetworkList->dwNumberOfItems <<endl <<endl;
 
        for(unsigned int i = 0; i < pNetworkList->dwNumberOfItems; i++)
        {
            pNetwork = (PWLAN_AVAILABLE_NETWORK)&pNetworkList->Network[i];
            //WCHAR wcSSID[DOT11_SSID_MAX_LENGTH] = { 0 };
 
            wcout <<L"Network SSID: ";
 
            //Convert the UCHAR SSID to a Unicode (WCHAR) ssid
            for (unsigned int a = 0; a < pNetwork->dot11Ssid.uSSIDLength; a++)
                wcout <<(wchar_t)pNetwork->dot11Ssid.ucSSID[a];
            wcout <<endl;
            //This block may cause some problems, will come back for future look if it causes errors.
 
            wcout <<L"Network Type: ";
            if (pNetwork->dot11BssType == dot11_BSS_type_infrastructure)
            {
                wcout <<L"Infrastructure" <<endl;
            }
            else if (pNetwork->dot11BssType == dot11_BSS_type_independent)
            {
                wcout <<L"Ad-Hoc" <<endl;
            }
            else
            {
                wcout <<L"Any" <<endl;
            }
 
            wcout <<L"Network Connectable: ";
            if (pNetwork->bNetworkConnectable) {
                wcout <<L"True"  <<endl;
            }
            else {
                wcout << L"False" <<endl;
            }
 
            wcout <<L"Signal Quality: " <<pNetwork->wlanSignalQuality <<endl;
 
            wcout <<L"Auth Algorithm: ";
            switch (pNetwork->dot11DefaultAuthAlgorithm)
            {
                case DOT11_AUTH_ALGO_80211_OPEN:
                    wcout <<L"None -- Open" <<endl;
                    break;
                case DOT11_AUTH_ALGO_80211_SHARED_KEY:
                    wcout <<L"WEP" <<endl;
                    break;
                case DOT11_AUTH_ALGO_WPA:
                    wcout <<L"WPA" <<endl;
                    break;
                case DOT11_AUTH_ALGO_WPA_PSK:
                    wcout <<L"WPA-PSK" <<endl;
                    break;
                case DOT11_AUTH_ALGO_RSNA:
                    wcout <<L"RSNA" <<endl;
                    break;
                case DOT11_AUTH_ALGO_RSNA_PSK:
                    wcout <<L"RSNA-PSK" <<endl;
                    break;
                case DOT11_AUTH_ALGO_IHV_START:
                case DOT11_AUTH_ALGO_IHV_END:
                    wcout <<L"IHV Proprietary" <<endl;
                    break;
                default:
                    wcout <<L"Unknown Algorithhm" <<endl;
 
            }
 
            wcout <<L"Cipher Algorithm: ";
            switch (pNetwork->dot11DefaultCipherAlgorithm)
            {
            case DOT11_CIPHER_ALGO_NONE:
                wcout <<L"None" <<endl;
                break;
            case DOT11_CIPHER_ALGO_WEP40:
                wcout <<L"WEP40" <<endl;
                break;
            case DOT11_CIPHER_ALGO_TKIP:
                wcout <<L"TKIP" <<endl;
                break;
            case DOT11_CIPHER_ALGO_CCMP:
                wcout <<L"CCMP" <<endl;
                break;
            case DOT11_CIPHER_ALGO_WEP104:
                wcout <<L"WEP104" <<endl;
                break;
            case DOT11_CIPHER_ALGO_WPA_USE_GROUP:
                wcout <<L"WPA-Group OR RSN-Group" <<endl;
                break;
            case DOT11_CIPHER_ALGO_WEP:
                wcout <<L"WEP-Any" <<endl;
                break;
            case DOT11_CIPHER_ALGO_IHV_START:
            case DOT11_CIPHER_ALGO_IHV_END:
                wcout <<L"IHV-Proprietary" <<endl;
                break;
            default:
                wcout <<L"Unknown" <<endl;
            }
 
            wcout <<endl;
 
        }
 
        WlanFreeMemory(pNetworkList); //Free the memory, no fucking memory leaks for me :p
        wcout <<endl;
    }
 
    //Close the wifi handle
    WlanCloseHandle(wifiHandle,NULL);
 
    wcin.get();
}