Updated for "Windows 10 Anniversary Update" (Version 1607)

1 files changed, 434 insertions, 0 deletions

diff --git a/network/wlan/ihvsampleui/IHVSampleExtUISec.cpp b/network/wlan/ihvsampleui/IHVSampleExtUISec.cpp
new file mode 100644
index 00000000..1cfa3714
--- /dev/null
+++ b/network/wlan/ihvsampleui/IHVSampleExtUISec.cpp
@@ -0,0 +1,434 @@
+//
+// Copyright (C) Microsoft Corporation 2005
+// IHV UI Extension sample
+//
+
+#include "precomp.h"
+
+extern HINSTANCE g_hInst;
+
+// used by the con prop extensions
+CIhvSecurityProfile* pIhvSecProfile;
+
+CDot11SampleExtUISecProperty::CDot11SampleExtUISecProperty(): 
+    m_crefCount(0), m_fInitialized(false), 
+    m_ExtType(DOT11_EXT_UI_SECURITY), m_fModified(FALSE), 
+    m_IhvSecurityType(IHVSecurityInvalid)
+{
+    m_bstrFN = NULL;
+    InterlockedIncrement(&g_objRefCount);
+}
+
+CDot11SampleExtUISecProperty::~CDot11SampleExtUISecProperty()
+{
+    SysFreeString(m_bstrFN);
+    InterlockedDecrement(&g_objRefCount);        
+}
+
+
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::GetDot11ExtUIPropertyFriendlyName(BSTR* bstrPropertyName)
+{
+    HRESULT hr = E_INVALIDARG; 
+    if (false == m_fInitialized)
+    {
+        return hr;
+    }
+
+    if (NULL != bstrPropertyName)
+    {
+        *bstrPropertyName = SysAllocString(m_bstrFN);
+        hr = S_OK;
+    }
+
+    return hr;
+}
+
+//Used to extend property
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::DisplayDot11ExtUIProperty(
+    HWND hParent, // Parent Window Handle 
+    BSTR bstrIHVProfile, // IHV data from the profile 
+    PDOT11EXT_IHV_PARAMS pIHVParams, // Select profile MS security settings
+    BSTR* bstrModifiedIHVProfile, // modified IHV data to be stored in the profile
+    BOOL* pbIsModified // flag to denote if profile was modified
+    )
+{
+    HRESULT hr = S_OK; 
+
+    UNREFERENCED_PARAMETER(pIHVParams);
+    
+    if (!m_fInitialized)
+    {
+        hr = E_INVALIDARG;
+        goto error;
+    }
+
+    // Store the passed-in string in a member variable so dialog box can display it
+    pIhvSecProfile = new(std::nothrow) CIhvSecurityProfile();
+    if (pIhvSecProfile == NULL)
+    {
+        hr = E_OUTOFMEMORY;
+        goto error;
+    }
+    pIhvSecProfile->LoadXml(bstrIHVProfile);
+    m_fModified = FALSE;
+
+    // Dialog will store the string in a member variable
+    DialogBoxParam(
+        g_hInst, 
+        MAKEINTRESOURCE(IDD_PROPPAGE_SMALL), 
+        hParent, 
+        SimpleDialogProcSec,
+        (LPARAM)pIhvSecProfile
+        );
+
+    m_fModified = pIhvSecProfile->GetModified();
+
+    if (NULL != bstrModifiedIHVProfile)
+    {
+        if (m_fModified)
+        {
+            pIhvSecProfile->EmitXml(bstrModifiedIHVProfile); 
+        }
+    }
+
+    if (NULL != pbIsModified)
+    {
+        *pbIsModified = m_fModified;
+    }
+
+error:
+    if(pIhvSecProfile)
+    {
+        delete pIhvSecProfile;
+        pIhvSecProfile = NULL;
+    }
+
+    return hr;
+}
+
+
+//Used to get the currently chosen entry to display as selected in the dropdown list
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::Dot11ExtUIPropertyGetSelected(
+    BSTR bstrIHVProfile, // IHV data from the profile 
+    PDOT11EXT_IHV_PARAMS pIHVParams, // Select profile MS security settings
+    BOOL* pfIsSelected // flag denoting if this is the selected profile
+    )
+{
+    HRESULT hr = S_OK;
+    IHV_SECURITY_TYPE currentSecurityType = IHVSecurityInvalid;
+
+    UNREFERENCED_PARAMETER(pIHVParams);
+
+    if(pfIsSelected == NULL)
+    {
+        hr = E_INVALIDARG;
+        goto error;
+    }
+
+    *pfIsSelected = FALSE;
+
+    pIhvSecProfile = new(std::nothrow) CIhvSecurityProfile();
+    if (pIhvSecProfile == NULL)
+    {
+        hr = E_OUTOFMEMORY;
+        goto error;
+    }
+    
+    pIhvSecProfile->LoadXml(bstrIHVProfile);
+
+    hr = pIhvSecProfile->GetSecurityType(&currentSecurityType);
+    if(FAILED(hr))
+    {
+        hr = S_OK;
+        if(IHVSecurityV1 == m_IhvSecurityType)
+        {
+            *pfIsSelected = TRUE;
+        }
+    }
+    else if(currentSecurityType == m_IhvSecurityType)
+    {
+        *pfIsSelected = TRUE;
+    }
+    
+error:
+    if(pIhvSecProfile != NULL)
+    {
+        delete pIhvSecProfile;
+        pIhvSecProfile = NULL;
+    }    
+    return hr;
+}
+
+//Used to set the current entry as chosen from the dropdown list
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::Dot11ExtUIPropertySetSelected(
+    BSTR bstrIHVProfile, // IHV data from the profile 
+    PDOT11EXT_IHV_PARAMS pIHVParams, // Select profile MS security settings
+    BSTR* bstrModifiedIHVProfile, // modified IHV data to be stored in the profile
+    BOOL* pbIsModified // flag to denote if profile was modified
+    )
+{
+    HRESULT hr = S_OK;
+
+    UNREFERENCED_PARAMETER(pIHVParams);
+
+    if(bstrModifiedIHVProfile == NULL || pbIsModified == NULL)
+    {
+        hr = E_INVALIDARG;
+        goto error;
+    }
+
+    pIhvSecProfile = new(std::nothrow) CIhvSecurityProfile();
+    if (pIhvSecProfile == NULL)
+    {
+        hr = E_OUTOFMEMORY;
+        goto error;
+    }
+    
+    pIhvSecProfile->LoadXml(bstrIHVProfile);
+
+    pIhvSecProfile->SetSecurityType(m_IhvSecurityType);
+    pIhvSecProfile->SetFullSecurityFlag(TRUE);
+
+    pIhvSecProfile->EmitXml(bstrModifiedIHVProfile );
+    *pbIsModified = pIhvSecProfile->GetModified();
+    
+error:
+    if(pIhvSecProfile != NULL)
+    {
+        delete pIhvSecProfile;
+        pIhvSecProfile = NULL;
+    }
+    return hr;    
+}
+
+
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::Dot11ExtUIPropertyHasConfigurationUI(
+    BOOL *fHasConfigurationUI)
+{
+    // this page always wants to show a config UI
+    *fHasConfigurationUI = TRUE;
+    return S_OK;
+}
+
+//Used to get additional display data (ciphers for auth types)
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::Dot11ExtUIPropertyGetDisplayInfo(
+    DOT11_EXT_UI_DISPLAY_INFO_TYPE dot11ExtUIDisplayInfoType, // the diapaly type to be described
+    BSTR bstrIHVProfile, // IHV data from the profile 
+    PDOT11EXT_IHV_PARAMS pIHVParams, // Select profile MS security settings
+    ULONG *pcEntries, // number of dependent strings
+    ULONG *puDefaultSelection, // the entry in the array to be selected by default
+    DOT11_EXT_UI_PROPERTY_DISPLAY_INFO **ppDot11ExtUIProperty // array of returned info structure
+    )
+{
+    HRESULT hr = S_OK;
+    DWORD i = 0;
+    CIhvSecurityProfile IhvSecurityProfile;
+    IHV_CIPHER_TYPE cipherType = IHVCipherInvalid;
+
+    UNREFERENCED_PARAMETER(pIHVParams);
+
+    if(dot11ExtUIDisplayInfoType != DOT11_EXT_UI_DISPLAY_INFO_CIPHER)
+    {
+        hr = E_NOTIMPL;
+        goto error;
+    }
+
+    IhvSecurityProfile.LoadXml(bstrIHVProfile);
+    hr = IhvSecurityProfile.GetCipherType(&cipherType);
+    if(FAILED(hr))
+    {
+        goto error;
+    }
+
+    for(i = 0; i < PROP_COUNT_SEC_CIPHERS; ++i)
+    {
+        ciphersInfoArray[m_IhvSecurityType][i].dwDataKey = i;
+        ciphersInfoArray[m_IhvSecurityType][i].dot11ExtUIDisplayInfoType = DOT11_EXT_UI_DISPLAY_INFO_CIPHER;
+        ciphersInfoArray[m_IhvSecurityType][i].bstrDisplayText = SysAllocString(g_IHVCipherFriendlyName[i]);
+    }
+    
+    // for the given auth type we want to return the list of compatible ciphers
+    *ppDot11ExtUIProperty = ciphersInfoArray[m_IhvSecurityType];
+    *pcEntries = PROP_COUNT_SEC_CIPHERS;
+    *puDefaultSelection = cipherType >= IHVCipherInvalid ? 0 : cipherType;
+
+error:
+    return hr;
+}
+
+STDMETHODIMP 
+CDot11SampleExtUISecProperty::Dot11ExtUIPropertySetDisplayInfo(
+    DOT11_EXT_UI_DISPLAY_INFO_TYPE dot11ExtUIDisplayInfoType, // the diapaly type to be modified
+    BSTR bstrIHVProfile, // IHV data from the profile 
+    PDOT11EXT_IHV_PARAMS pIHVParams, // Select profile MS security settings
+    DOT11_EXT_UI_PROPERTY_DISPLAY_INFO *pDot11ExtUIProperty, // selected info structure
+    BSTR* bstrModifiedIHVProfile, // modified IHV data to be stored in the profile
+    BOOL* pbIsModified // flag to denote if profile was modified
+    )
+{
+    HRESULT hr = S_OK;
+
+    UNREFERENCED_PARAMETER(pIHVParams);
+
+    CIhvSecurityProfile IhvSecurityProfile;
+
+    if(dot11ExtUIDisplayInfoType != DOT11_EXT_UI_DISPLAY_INFO_CIPHER)
+    {
+        hr = E_NOTIMPL;
+        goto error;
+    }
+
+    if(pDot11ExtUIProperty == NULL)
+    {
+        hr = E_INVALIDARG;
+        goto error;
+    }
+
+    hr = IhvSecurityProfile.LoadXml(bstrIHVProfile);
+    if(FAILED(hr))
+    {
+        goto error;
+    }
+    hr = IhvSecurityProfile.SetCipherType((IHV_CIPHER_TYPE)pDot11ExtUIProperty->dwDataKey);
+    if(FAILED(hr))
+    {
+        goto error;
+    }
+
+    hr = IhvSecurityProfile.EmitXml(bstrModifiedIHVProfile);
+    *pbIsModified = IhvSecurityProfile.GetModified();
+    
+error:
+    return hr;
+}
+
+STDMETHODIMP
+CDot11SampleExtUISecProperty::Dot11ExtUIPropertyIsStandardSecurity(
+    BOOL *fIsStandardSecurity, // if this interface is a standard auth method
+    DOT11_EXT_UI_SECURITY_TYPE *dot11ExtUISecurityType  // which of the standard auth methods it is
+    )
+{
+    UNREFERENCED_PARAMETER(dot11ExtUISecurityType);
+    *fIsStandardSecurity = FALSE;
+    return S_OK;    
+}
+
+STDMETHODIMP
+CDot11SampleExtUISecProperty::Initialize(BSTR bstrPropertyName, DWORD dwIhvSecurity)
+{
+    HRESULT hr = E_INVALIDARG; 
+    if (false == m_fInitialized)
+    {
+        // Set the FriendlyName
+        m_bstrFN = SysAllocString(bstrPropertyName);
+        m_IhvSecurityType = (IHV_SECURITY_TYPE)dwIhvSecurity;
+        m_fInitialized = true;
+        hr = S_OK;
+    }
+    return hr;
+}
+
+INT_PTR CALLBACK 
+SimpleDialogProcSec(
+    HWND hwndDlg,
+    UINT uMsg,
+    WPARAM wParam,
+    LPARAM lParam
+)
+{
+    BOOL fRetVal = FALSE;
+    WCHAR szBuf[256] = {0};    
+    DWORD dwValue = 0;
+    BSTR bstrText = NULL;
+
+    UNREFERENCED_PARAMETER(lParam);
+
+    if(!pIhvSecProfile)
+    {
+        goto error;
+    }
+    
+    switch(uMsg)
+    {
+    case WM_INITDIALOG:
+        {
+            // Dialog title
+            WCHAR strDialogTitle[MAX_PATH] = {0};
+            (VOID)::LoadString(
+                g_hInst,
+                IDS_IHV_DEFAULT_SEC_TITLE,
+                strDialogTitle,
+                MAX_PATH
+                );
+                    
+            SetWindowText(hwndDlg, strDialogTitle);
+        }
+    
+        // check the checkbox if needed
+        if(FAILED(pIhvSecProfile->GetParamDWORD(&dwValue)))
+        {
+            dwValue = 0;
+        }
+        ::SendMessage(
+            GetDlgItem(hwndDlg, IDC_USE_FASTHANDOFF), 
+            BM_SETCHECK, 
+            (WPARAM)(int)dwValue, 
+            0L
+            );
+        
+        // Set text in the textbox
+        if(FAILED(pIhvSecProfile->GetParamBSTR(&bstrText)))
+        {
+            bstrText = NULL;
+        }
+        SetWindowText(GetDlgItem(hwndDlg, IDC_PARAM_BOX), bstrText);
+        
+        fRetVal = TRUE;
+        break;
+        
+    case WM_COMMAND: 
+        switch (LOWORD(wParam)) 
+        { 
+            case ID_OK: 
+                GetWindowText(GetDlgItem(hwndDlg, IDC_PARAM_BOX), szBuf, 255);
+                if(szBuf)
+                {
+                    DWORD dwNewValue = 0;
+
+                    // get the button state and record it
+                    dwNewValue = (int)::SendMessage(
+                                            GetDlgItem(hwndDlg, IDC_USE_FASTHANDOFF), 
+                                            BM_GETCHECK, 
+                                            0L, 
+                                            0L
+                                            );
+
+                    pIhvSecProfile->SetParamDWORD(dwNewValue);                    
+                    pIhvSecProfile->SetParamBSTR(szBuf);
+                    pIhvSecProfile->SetFullSecurityFlag(TRUE);
+
+                    // Notify the owner window to carry out the task. 
+                    EndDialog(hwndDlg, 1);
+                    fRetVal = TRUE; 
+                }
+                break;
+
+            case ID_CANCEL: 
+                EndDialog(hwndDlg, 0);
+                fRetVal = TRUE;
+                break;
+        }
+        break;
+    }
+
+error:
+    return fRetVal;
+}
+
+