CNetConnectionManager Class Reference

#include <connectmanager.h>

Inheritance diagram for CNetConnectionManager:

Collaboration diagram for CNetConnectionManager:

Public Member Functions
	CNetConnectionManager ()
HRESULT	Initialize ()
HRESULT	EnumerateINetConnections ()
virtual HRESULT WINAPI	EnumConnections (NETCONMGR_ENUM_FLAGS Flags, IEnumNetConnection **ppEnum)
virtual HRESULT WINAPI	Next (ULONG celt, INetConnection **rgelt, ULONG *pceltFetched)
virtual HRESULT WINAPI	Skip (ULONG celt)
virtual HRESULT WINAPI	Reset ()
virtual HRESULT WINAPI	Clone (IEnumNetConnection **ppenum)
Public Member Functions inherited from ATL::CComObjectRootEx< CComMultiThreadModelNoCS >
	~CComObjectRootEx ()
ULONG	InternalAddRef ()
ULONG	InternalRelease ()
void	Lock ()
void	Unlock ()
HRESULT	_AtlInitialConstruct ()
Public Member Functions inherited from ATL::CComObjectRootBase
	CComObjectRootBase ()
	~CComObjectRootBase ()
void	SetVoid (void *)
HRESULT	_AtlFinalConstruct ()
HRESULT	FinalConstruct ()
void	InternalFinalConstructAddRef ()
void	InternalFinalConstructRelease ()
void	FinalRelease ()

Private Attributes
PINetConnectionItem	m_pHead
PINetConnectionItem	m_pCurrent

Additional Inherited Members
Static Public Member Functions inherited from ATL::CComObjectRootBase
static void WINAPI	ObjectMain (bool)
static const struct _ATL_CATMAP_ENTRY *	GetCategoryMap ()
static HRESULT WINAPI	InternalQueryInterface (void *pThis, const _ATL_INTMAP_ENTRY *pEntries, REFIID iid, void **ppvObject)
Public Attributes inherited from ATL::CComObjectRootBase
LONG	m_dwRef

Detailed Description

Definition at line 9 of file connectmanager.h.

Constructor & Destructor Documentation

CNetConnectionManager()

CNetConnectionManager::CNetConnectionManager

(

)

Definition at line 381 of file connectmanager.cpp.

                                             :
    m_pHead(NULL),
    m_pCurrent(NULL)
{
}

Member Function Documentation

Clone()

HRESULT WINAPI CNetConnectionManager::Clone ( IEnumNetConnection **  ppenum )

virtual

Definition at line 458 of file connectmanager.cpp.

{
    return E_NOTIMPL;
}

EnumConnections()

HRESULT WINAPI CNetConnectionManager::EnumConnections ( NETCONMGR_ENUM_FLAGS  Flags, IEnumNetConnection **  ppEnum  )

virtual

Definition at line 389 of file connectmanager.cpp.

{
    TRACE("EnumConnections\n");
 
    if (!ppEnum)
        return E_POINTER;
 
    if (Flags != NCME_DEFAULT)
        return E_FAIL;
 
    *ppEnum = static_cast<IEnumNetConnection*>(this);
    AddRef();
    return S_OK;
}

EnumerateINetConnections()

HRESULT CNetConnectionManager::EnumerateINetConnections

(

)

Definition at line 518 of file connectmanager.cpp.

{
    DWORD dwSize, dwResult, dwIndex, dwAdapterIndex, dwShowIcon, dwNotifyDisconnect;
    MIB_IFTABLE *pIfTable;
    MIB_IFROW IfEntry;
    IP_ADAPTER_INFO * pAdapterInfo;
    HDEVINFO hInfo;
    SP_DEVINFO_DATA DevInfo;
    HKEY hSubKey;
    WCHAR szNetCfg[50];
    WCHAR szAdapterNetCfg[50];
    WCHAR szDetail[200] = L"SYSTEM\\CurrentControlSet\\Control\\Class\\";
    WCHAR szName[130] = L"SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\";
    PINetConnectionItem pCurrent = NULL;
 
    /* get the IfTable */
    dwSize = 0;
    if (GetIfTable(NULL, &dwSize, TRUE) != ERROR_INSUFFICIENT_BUFFER)
        return E_FAIL;
 
    pIfTable = static_cast<PMIB_IFTABLE>(CoTaskMemAlloc(dwSize));
    if (!pIfTable)
        return E_OUTOFMEMORY;
 
    dwResult = GetIfTable(pIfTable, &dwSize, TRUE);
    if (dwResult != NO_ERROR)
    {
        CoTaskMemFree(pIfTable);
        return HRESULT_FROM_WIN32(dwResult);
    }
 
    dwSize = 0;
    dwResult = GetAdaptersInfo(NULL, &dwSize);
    if (dwResult!= ERROR_BUFFER_OVERFLOW)
    {
        CoTaskMemFree(pIfTable);
        return HRESULT_FROM_WIN32(dwResult);
    }
 
    pAdapterInfo = static_cast<PIP_ADAPTER_INFO>(CoTaskMemAlloc(dwSize));
    if (!pAdapterInfo)
    {
        CoTaskMemFree(pIfTable);
        return E_OUTOFMEMORY;
    }
 
    dwResult = GetAdaptersInfo(pAdapterInfo, &dwSize);
    if (dwResult != NO_ERROR)
    {
        CoTaskMemFree(pIfTable);
        CoTaskMemFree(pAdapterInfo);
        return HRESULT_FROM_WIN32(dwResult);
    }
 
    hInfo = SetupDiGetClassDevs(&GUID_DEVCLASS_NET, NULL, NULL, DIGCF_PRESENT );
    if (!hInfo)
    {
        CoTaskMemFree(pIfTable);
        CoTaskMemFree(pAdapterInfo);
        return E_FAIL;
    }
 
    dwIndex = 0;
    do
    {
        ZeroMemory(&DevInfo, sizeof(SP_DEVINFO_DATA));
        DevInfo.cbSize = sizeof(DevInfo);
 
        /* get device info */
        if (!SetupDiEnumDeviceInfo(hInfo, dwIndex++, &DevInfo))
            break;
 
        /* get device software registry path */
        if (!SetupDiGetDeviceRegistryPropertyW(hInfo, &DevInfo, SPDRP_DRIVER, NULL, (LPBYTE)&szDetail[39], sizeof(szDetail)/sizeof(WCHAR) - 40, &dwSize))
            break;
 
        /* open device registry key */
        if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szDetail, 0, KEY_READ, &hSubKey) != ERROR_SUCCESS)
            break;
 
        /* query NetCfgInstanceId for current device */
        dwSize = sizeof(szNetCfg);
        if (RegQueryValueExW(hSubKey, L"NetCfgInstanceId", NULL, NULL, (LPBYTE)szNetCfg, &dwSize) != ERROR_SUCCESS)
        {
            RegCloseKey(hSubKey);
            break;
        }
        RegCloseKey(hSubKey);
 
        /* get the current adapter index from NetCfgInstanceId */
        if (!GetAdapterIndexFromNetCfgInstanceId(pAdapterInfo, szNetCfg, &dwAdapterIndex))
            continue;
 
        /* get detailed adapter info */
        ZeroMemory(&IfEntry, sizeof(IfEntry));
        IfEntry.dwIndex = dwAdapterIndex;
        if (GetIfEntry(&IfEntry) != NO_ERROR)
            break;
 
        /* allocate new INetConnectionItem */
        PINetConnectionItem pNew = static_cast<PINetConnectionItem>(CoTaskMemAlloc(sizeof(INetConnectionItem)));
        if (!pNew)
            break;
 
        ZeroMemory(pNew, sizeof(INetConnectionItem));
        pNew->dwAdapterIndex = dwAdapterIndex;
        /* store NetCfgInstanceId */
        CLSIDFromString(szNetCfg, &pNew->Props.guidId);
        NormalizeOperStatus(&IfEntry, &pNew->Props);
 
        switch (IfEntry.dwType)
        {
            case IF_TYPE_ETHERNET_CSMACD:
                pNew->Props.MediaType = NCM_LAN;
                break;
            case IF_TYPE_IEEE80211:
                pNew->Props.MediaType = NCM_SHAREDACCESSHOST_RAS;
                break;
            default:
                break;
        }
        /*  open network connections details */
        wcscpy(&szName[80], szNetCfg);
        wcscpy(&szName[118], L"\\Connection");
 
        if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szName, 0, KEY_READ, &hSubKey) == ERROR_SUCCESS)
        {
            /* retrieve name of connection */
            dwSize = sizeof(szAdapterNetCfg);
            if (RegQueryValueExW(hSubKey, L"Name", NULL, NULL, (LPBYTE)szAdapterNetCfg, &dwSize) == ERROR_SUCCESS)
            {
                pNew->Props.pszwName = static_cast<PWSTR>(CoTaskMemAlloc((wcslen(szAdapterNetCfg)+1) * sizeof(WCHAR)));
                if (pNew->Props.pszwName)
                    wcscpy(pNew->Props.pszwName, szAdapterNetCfg);
            }
            dwSize = sizeof(dwShowIcon);
            if (RegQueryValueExW(hSubKey, L"ShowIcon", NULL, NULL, (LPBYTE)&dwShowIcon, &dwSize) == ERROR_SUCCESS)
            {
                if (dwShowIcon)
                    pNew->Props.dwCharacter |= NCCF_SHOW_ICON;
            }
            dwSize = sizeof(dwNotifyDisconnect);
            if (RegQueryValueExW(hSubKey, L"IpCheckingEnabled", NULL, NULL, (LPBYTE)&dwNotifyDisconnect, &dwSize) == ERROR_SUCCESS)
            {
                if (dwNotifyDisconnect)
                    pNew->Props.dwCharacter |= NCCF_NOTIFY_DISCONNECTED;
            }
            RegCloseKey(hSubKey);
        }
 
        /* Get the adapter device description */
        dwSize = 0;
        SetupDiGetDeviceRegistryPropertyW(hInfo, &DevInfo, SPDRP_DEVICEDESC, NULL, NULL, 0, &dwSize);
        if (dwSize != 0)
        {
            pNew->Props.pszwDeviceName = static_cast<PWSTR>(CoTaskMemAlloc(dwSize));
            if (pNew->Props.pszwDeviceName)
                SetupDiGetDeviceRegistryPropertyW(hInfo, &DevInfo, SPDRP_DEVICEDESC, NULL, (PBYTE)pNew->Props.pszwDeviceName, dwSize, &dwSize);
        }
 
        if (pCurrent)
            pCurrent->Next = pNew;
        else
            m_pHead = pNew;
 
        pCurrent = pNew;
    } while (TRUE);
 
    CoTaskMemFree(pIfTable);
    CoTaskMemFree(pAdapterInfo);
    SetupDiDestroyDeviceInfoList(hInfo);
 
    m_pCurrent = m_pHead;
    return (m_pHead != NULL ? S_OK : S_FALSE);
}

Referenced by Initialize().

Initialize()

HRESULT CNetConnectionManager::Initialize

(

)

Definition at line 694 of file connectmanager.cpp.

{
    HRESULT hr = EnumerateINetConnections();
    if (FAILED_UNEXPECTEDLY(hr))
    {
        /* If something went wrong during the enumeration print an error don't enumerate anything */
        m_pCurrent = m_pHead = NULL;
        return S_FALSE;
    }
    return S_OK;
}

Next()

HRESULT WINAPI CNetConnectionManager::Next ( ULONG  celt, INetConnection **  rgelt, ULONG *  pceltFetched  )

virtual

Definition at line 412 of file connectmanager.cpp.

{
    HRESULT hr;
 
    if (!pceltFetched || !rgelt)
        return E_POINTER;
 
    if (celt != 1)
        return E_FAIL;
 
    if (!m_pCurrent)
        return S_FALSE;
 
    hr = CNetConnection_CreateInstance(m_pCurrent, IID_PPV_ARG(INetConnection, rgelt));
    m_pCurrent = m_pCurrent->Next;
 
    return hr;
}

Reset()

HRESULT WINAPI CNetConnectionManager::Reset ( )

virtual

Definition at line 450 of file connectmanager.cpp.

{
    m_pCurrent = m_pHead;
    return S_OK;
}

Skip()

HRESULT WINAPI CNetConnectionManager::Skip ( ULONG  celt )

virtual

Definition at line 436 of file connectmanager.cpp.

{
    while (m_pCurrent && celt-- > 0)
        m_pCurrent = m_pCurrent->Next;
 
    if (celt)
       return S_FALSE;
    else
       return S_OK;
 
}

Member Data Documentation

m_pCurrent

PINetConnectionItem CNetConnectionManager::m_pCurrent

private

Definition at line 31 of file connectmanager.h.

Referenced by EnumerateINetConnections(), Initialize(), Next(), Reset(), and Skip().

m_pHead

PINetConnectionItem CNetConnectionManager::m_pHead

private

Definition at line 30 of file connectmanager.h.

Referenced by EnumerateINetConnections(), Initialize(), and Reset().

---

The documentation for this class was generated from the following files:

• dll/shellext/netshell/connectmanager.h
• dll/shellext/netshell/connectmanager.cpp