CNetConnection Class Reference

#include <connectmanager.h>

Inheritance diagram for CNetConnection:

Collaboration diagram for CNetConnection:

Public Member Functions
	~CNetConnection ()
HRESULT WINAPI	Initialize (PINetConnectionItem pItem)
STDMETHOD()	Connect () override
STDMETHOD()	Disconnect () override
STDMETHOD()	Delete () override
STDMETHOD()	Duplicate (LPCWSTR pszwDuplicateName, INetConnection **ppCon) override
STDMETHOD()	GetProperties (NETCON_PROPERTIES **ppProps) override
STDMETHOD()	GetUiObjectClassId (CLSID *pclsid) override
STDMETHOD()	Rename (LPCWSTR pszwDuplicateName) override
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 ()
HRESULT	Connect ()
HRESULT	Disconnect ()
HRESULT	Delete ()
HRESULT	Duplicate ([in, string] LPCWSTR pszwDuplicateName, [out] INetConnection **ppCon)
HRESULT	GetProperties ([out] NETCON_PROPERTIES **ppProps)
HRESULT	GetUiObjectClassId ([out, ref] CLSID *pclsid)
HRESULT	Rename ([in, string] LPCWSTR pszwNewName)
Public Member Functions inherited from IUnknown
HRESULT	QueryInterface ([in] REFIID riid, [out, iid_is(riid)] void **ppvObject)
ULONG	AddRef ()
ULONG	Release ()

Private Attributes
NETCON_PROPERTIES	m_Props
DWORD	m_dwAdapterIndex

Additional Inherited Members
Public Types inherited from INetConnection
enum	tagNETCON_CHARACTERISTIC_FLAGS {   NCCF_NONE = 0x0000 , NCCF_ALL_USERS = 0x0001 , NCCF_ALLOW_DUPLICATION = 0x0002 , NCCF_ALLOW_REMOVAL = 0x0004 ,   NCCF_ALLOW_RENAME = 0x0008 , NCCF_INCOMING_ONLY = 0x0020 , NCCF_OUTGOING_ONLY = 0x0040 , NCCF_BRANDED = 0x0080 ,   NCCF_SHARED = 0x0100 , NCCF_BRIDGED = 0x0200 , NCCF_FIREWALLED = 0x0400 , NCCF_DEFAULT = 0x0800 ,   NCCF_HOMENET_CAPABLE = 0x1000 , NCCF_SHARED_PRIVATE = 0x2000 , NCCF_QUARANTINED = 0x4000 , NCCF_RESERVED = 0x8000 ,   NCCF_BLUETOOTH_MASK = 0x000F0000 , NCCF_LAN_MASK = 0x00F00000 }
enum	tagNETCON_STATUS {   NCS_DISCONNECTED , NCS_CONNECTING , NCS_CONNECTED , NCS_DISCONNECTING ,   NCS_HARDWARE_NOT_PRESENT , NCS_HARDWARE_DISABLED , NCS_HARDWARE_MALFUNCTION , NCS_MEDIA_DISCONNECTED ,   NCS_AUTHENTICATING , NCS_AUTHENTICATION_SUCCEEDED , NCS_AUTHENTICATION_FAILED , NCS_INVALID_ADDRESS ,   NCS_CREDENTIALS_REQUIRED }
enum	tagNETCON_TYPE {   NCT_DIRECT_CONNECT , NCT_INBOUND , NCT_INTERNET , NCT_LAN ,   NCT_PHONE , NCT_TUNNEL , NCT_BRIDGE }
enum	tagNETCON_MEDIATYPE {   NCM_NONE , NCM_DIRECT , NCM_ISDN , NCM_LAN ,   NCM_PHONE , NCM_TUNNEL , NCM_PPPOE , NCM_BRIDGE ,   NCM_SHAREDACCESSHOST_LAN , NCM_SHAREDACCESSHOST_RAS }
typedef enum INetConnection::tagNETCON_CHARACTERISTIC_FLAGS	NETCON_CHARACTERISTIC_FLAGS
typedef enum INetConnection::tagNETCON_STATUS	NETCON_STATUS
typedef enum INetConnection::tagNETCON_TYPE	NETCON_TYPE
typedef enum INetConnection::tagNETCON_MEDIATYPE	NETCON_MEDIATYPE
typedef struct INetConnection::tagNETCON_PROPERTIES	NETCON_PROPERTIES
Public Types inherited from IUnknown
typedef IUnknown *	LPUNKNOWN
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 44 of file connectmanager.h.

Constructor & Destructor Documentation

~CNetConnection()

CNetConnection::~CNetConnection

(

)

Definition at line 40 of file connectmanager.cpp.

{
    CoTaskMemFree(m_Props.pszwName);
    CoTaskMemFree(m_Props.pszwDeviceName);
}

Member Function Documentation

Connect()

HRESULT WINAPI CNetConnection::Connect ( )

override

Implements INetConnection.

Definition at line 48 of file connectmanager.cpp.

{
    return E_NOTIMPL;
}

Delete()

HRESULT WINAPI CNetConnection::Delete ( )

override

Implements INetConnection.

Definition at line 192 of file connectmanager.cpp.

{
    return E_NOTIMPL;
}

Disconnect()

HRESULT WINAPI CNetConnection::Disconnect ( )

override

Implements INetConnection.

Definition at line 98 of file connectmanager.cpp.

{
    HKEY hKey;
    NETCON_PROPERTIES * pProperties;
    LPOLESTR pDisplayName;
    WCHAR szPath[200];
    DWORD dwSize, dwType;
    LPWSTR pPnp;
    HDEVINFO hInfo;
    SP_DEVINFO_DATA DevInfo;
    SP_PROPCHANGE_PARAMS PropChangeParams;
    HRESULT hr;
 
    hr = GetProperties(&pProperties);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hInfo = SetupDiGetClassDevsW(&GUID_DEVCLASS_NET, NULL, NULL, DIGCF_PRESENT );
    if (!hInfo)
    {
        NcFreeNetconProperties(pProperties);
        return E_FAIL;
    }
 
    if (FAILED(StringFromCLSID((CLSID)pProperties->guidId, &pDisplayName)))
    {
        NcFreeNetconProperties(pProperties);
        SetupDiDestroyDeviceInfoList(hInfo);
        return E_FAIL;
    }
    NcFreeNetconProperties(pProperties);
 
    if (FindNetworkAdapter(hInfo, &DevInfo, pDisplayName))
    {
        PropChangeParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER);
        PropChangeParams.ClassInstallHeader.InstallFunction = DIF_PROPERTYCHANGE; //;
        PropChangeParams.StateChange = DICS_DISABLE;
        PropChangeParams.Scope = DICS_FLAG_CONFIGSPECIFIC;
        PropChangeParams.HwProfile = 0;
 
        if (!SetupDiSetClassInstallParams(hInfo, &DevInfo, &PropChangeParams.ClassInstallHeader, sizeof(SP_PROPCHANGE_PARAMS)) ||
            !SetupDiCallClassInstaller(DIF_PROPERTYCHANGE, hInfo, &DevInfo))
        {
            hr = HRESULT_FROM_WIN32(GetLastError());
        }
    }
    SetupDiDestroyDeviceInfoList(hInfo);
 
    swprintf(szPath, L"SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\%s\\Connection", pDisplayName);
    CoTaskMemFree(pDisplayName);
 
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szPath, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
        return E_FAIL;
 
    dwSize = 0;
    if (RegQueryValueExW(hKey, L"PnpInstanceID", NULL, &dwType, NULL, &dwSize) != ERROR_SUCCESS || dwType != REG_SZ)
    {
        RegCloseKey(hKey);
        return E_FAIL;
    }
 
    pPnp = static_cast<PWSTR>(CoTaskMemAlloc(dwSize));
    if (!pPnp)
    {
        RegCloseKey(hKey);
        return E_FAIL;
    }
 
    if (RegQueryValueExW(hKey, L"PnpInstanceID", NULL, &dwType, (LPBYTE)pPnp, &dwSize) != ERROR_SUCCESS)
    {
        CoTaskMemFree(pPnp);
        RegCloseKey(hKey);
        return E_FAIL;
    }
    RegCloseKey(hKey);
 
    swprintf(szPath, L"System\\CurrentControlSet\\Hardware Profiles\\Current\\System\\CurrentControlSet\\Enum\\%s", pPnp);
    CoTaskMemFree(pPnp);
 
    if (RegCreateKeyExW(HKEY_LOCAL_MACHINE, szPath, 0, NULL, 0, KEY_WRITE, NULL, &hKey, NULL) != ERROR_SUCCESS)
        return E_FAIL;
 
    dwSize = 1; /* enable = 0, disable = 1 */
    RegSetValueExW(hKey, L"CSConfigFlags", 0, REG_DWORD, (LPBYTE)&dwSize, sizeof(DWORD));
    RegCloseKey(hKey);
 
    return S_OK;
}

Duplicate()

HRESULT WINAPI CNetConnection::Duplicate ( LPCWSTR  pszwDuplicateName, INetConnection **  ppCon  )

override

Implements INetConnection.

Definition at line 199 of file connectmanager.cpp.

{
    return E_NOTIMPL;
}

GetProperties()

HRESULT WINAPI CNetConnection::GetProperties ( NETCON_PROPERTIES **  ppProps )

override

Implements INetConnection.

Definition at line 208 of file connectmanager.cpp.

{
    MIB_IFROW IfEntry;
    HKEY hKey;
    LPOLESTR pStr;
    WCHAR szName[140];
    DWORD dwShowIcon, dwNotifyDisconnect, dwType, dwSize;
    NETCON_PROPERTIES * pProperties;
    HRESULT hr;
 
    if (!ppProps)
        return E_POINTER;
 
    pProperties = static_cast<NETCON_PROPERTIES*>(CoTaskMemAlloc(sizeof(NETCON_PROPERTIES)));
    if (!pProperties)
        return E_OUTOFMEMORY;
 
    CopyMemory(pProperties, &m_Props, sizeof(NETCON_PROPERTIES));
    pProperties->pszwName = NULL;
 
    if (m_Props.pszwDeviceName)
    {
        pProperties->pszwDeviceName = static_cast<LPWSTR>(CoTaskMemAlloc((wcslen(m_Props.pszwDeviceName)+1)*sizeof(WCHAR)));
        if (pProperties->pszwDeviceName)
            wcscpy(pProperties->pszwDeviceName, m_Props.pszwDeviceName);
    }
 
    *ppProps = pProperties;
 
    /* get updated adapter characteristics */
    ZeroMemory(&IfEntry, sizeof(IfEntry));
    IfEntry.dwIndex = m_dwAdapterIndex;
    if (GetIfEntry(&IfEntry) != NO_ERROR)
        return NOERROR;
 
    NormalizeOperStatus(&IfEntry, pProperties);
 
 
    hr = StringFromCLSID((CLSID)m_Props.guidId, &pStr);
    if (SUCCEEDED(hr))
    {
        wcscpy(szName, L"SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\");
        wcscat(szName, pStr);
        wcscat(szName, L"\\Connection");
 
        if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szName, 0, KEY_READ, &hKey) == ERROR_SUCCESS)
        {
            dwSize = sizeof(dwShowIcon);
            if (RegQueryValueExW(hKey, L"ShowIcon", NULL, &dwType, (LPBYTE)&dwShowIcon, &dwSize) == ERROR_SUCCESS && dwType == REG_DWORD)
            {
                if (dwShowIcon)
                    pProperties->dwCharacter |= NCCF_SHOW_ICON;
                else
                    pProperties->dwCharacter &= ~NCCF_SHOW_ICON;
            }
 
            dwSize = sizeof(dwNotifyDisconnect);
            if (RegQueryValueExW(hKey, L"IpCheckingEnabled", NULL, &dwType, (LPBYTE)&dwNotifyDisconnect, &dwSize) == ERROR_SUCCESS && dwType == REG_DWORD)
            {
                if (dwNotifyDisconnect)
                    pProperties->dwCharacter |= NCCF_NOTIFY_DISCONNECTED;
                else
                    pProperties->dwCharacter &= ~NCCF_NOTIFY_DISCONNECTED;
            }
 
            dwSize = sizeof(szName);
            if (RegQueryValueExW(hKey, L"Name", NULL, &dwType, (LPBYTE)szName, &dwSize) == ERROR_SUCCESS)
            {
                /* use updated name */
                dwSize = wcslen(szName) + 1;
                pProperties->pszwName = static_cast<PWSTR>(CoTaskMemAlloc(dwSize * sizeof(WCHAR)));
                if (pProperties->pszwName)
                    CopyMemory(pProperties->pszwName, szName, dwSize * sizeof(WCHAR));
            }
            else
            {
                /* use cached name */
                if (m_Props.pszwName)
                {
                    pProperties->pszwName = static_cast<PWSTR>(CoTaskMemAlloc((wcslen(m_Props.pszwName)+1)*sizeof(WCHAR)));
                    if (pProperties->pszwName)
                        wcscpy(pProperties->pszwName, m_Props.pszwName);
                }
            }
            RegCloseKey(hKey);
        }
        CoTaskMemFree(pStr);
    }
 
    /* Enable 'Rename' and 'Delete' for Adminstrators only */
    if (IsUserAdmin())
    {
        pProperties->dwCharacter |= NCCF_ALLOW_RENAME;
 
        /* Virtual network interfaces can be deleted */
        if (IfEntry.dwType == IF_TYPE_TUNNEL)
        {
            pProperties->dwCharacter |= NCCF_ALLOW_REMOVAL;
        }
    }
    else
    {
        pProperties->dwCharacter &= ~(NCCF_ALLOW_RENAME | NCCF_ALLOW_REMOVAL);
    }
 
    return S_OK;
}

Referenced by Disconnect().

GetUiObjectClassId()

HRESULT WINAPI CNetConnection::GetUiObjectClassId ( CLSID *  pclsid )

override

Implements INetConnection.

Definition at line 318 of file connectmanager.cpp.

{
    if (m_Props.MediaType == NCM_LAN)
    {
        CopyMemory(pclsid, &CLSID_LanConnectionUi, sizeof(CLSID));
        return S_OK;
    }
 
    return E_NOTIMPL;
}

Initialize()

HRESULT WINAPI CNetConnection::Initialize

(

PINetConnectionItem

pItem

)

Definition at line 18 of file connectmanager.cpp.

{
    m_Props = pItem->Props;
    m_dwAdapterIndex = pItem->dwAdapterIndex;
 
    if (pItem->Props.pszwName)
    {
        m_Props.pszwName = static_cast<PWSTR>(CoTaskMemAlloc((wcslen(pItem->Props.pszwName)+1)*sizeof(WCHAR)));
        if (m_Props.pszwName)
            wcscpy(m_Props.pszwName, pItem->Props.pszwName);
    }
 
    if (pItem->Props.pszwDeviceName)
    {
        m_Props.pszwDeviceName = static_cast<PWSTR>(CoTaskMemAlloc((wcslen(pItem->Props.pszwDeviceName)+1)*sizeof(WCHAR)));
        if (m_Props.pszwDeviceName)
            wcscpy(m_Props.pszwDeviceName, pItem->Props.pszwDeviceName);
    }
 
    return S_OK;
}

Rename()

HRESULT WINAPI CNetConnection::Rename ( LPCWSTR  pszwDuplicateName )

override

Implements INetConnection.

Definition at line 331 of file connectmanager.cpp.

{
    WCHAR szName[140];
    LPOLESTR pStr;
    DWORD dwSize;
    HKEY hKey;
    HRESULT hr;
 
    if (pszwDuplicateName == NULL || wcslen(pszwDuplicateName) == 0)
        return S_OK;
 
    if (m_Props.pszwName)
    {
        CoTaskMemFree(m_Props.pszwName);
        m_Props.pszwName = NULL;
    }
 
    dwSize = (wcslen(pszwDuplicateName) + 1) * sizeof(WCHAR);
    m_Props.pszwName = static_cast<PWSTR>(CoTaskMemAlloc(dwSize));
    if (m_Props.pszwName == NULL)
        return E_OUTOFMEMORY;
 
    wcscpy(m_Props.pszwName, pszwDuplicateName);
 
    hr = StringFromCLSID((CLSID)m_Props.guidId, &pStr);
    if (SUCCEEDED(hr))
    {
        wcscpy(szName, L"SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}\\");
        wcscat(szName, pStr);
        wcscat(szName, L"\\Connection");
 
        if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, szName, 0, KEY_WRITE, &hKey) == ERROR_SUCCESS)
        {
            RegSetValueExW(hKey, L"Name", NULL, REG_SZ, (LPBYTE)m_Props.pszwName, dwSize);
            RegCloseKey(hKey);
        }
 
        CoTaskMemFree(pStr);
    }
 
    return hr;
}

Member Data Documentation

m_dwAdapterIndex

DWORD CNetConnection::m_dwAdapterIndex

private

Definition at line 50 of file connectmanager.h.

Referenced by GetProperties(), and Initialize().

m_Props

NETCON_PROPERTIES CNetConnection::m_Props

private

Definition at line 49 of file connectmanager.h.

Referenced by GetProperties(), GetUiObjectClassId(), Initialize(), Rename(), and ~CNetConnection().

---

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

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