[WLANWIZ] Early implementation of connection mode/state

Author: SigmaTel71

dll/shellext/wlanwiz/CMakeLists.txt

@@ -26,6 +26,6 @@ add_library(wlanwiz MODULE
 set_module_type(wlanwiz win32dll UNICODE)
 set_target_cpp_properties(wlanwiz WITH_EXCEPTIONS)
 target_link_libraries(wlanwiz cpprt cppstl uuid atl_classes)
-add_importlibs(wlanwiz version gdi32 user32 advapi32 comctl32 ole32 msvcrt msimg32 kernel32 shell32 shlwapi ntdll uxtheme wlanapi)
+add_importlibs(wlanwiz version gdi32 user32 advapi32 comctl32 ole32 msvcrt msimg32 kernel32 shell32 shlwapi ntdll uxtheme wlanapi xmllite)
 add_cd_file(TARGET wlanwiz DESTINATION reactos/system32 FOR all)
 set_property(TARGET wlanwiz PROPERTY CXX_STANDARD 20)

dll/shellext/wlanwiz/draw.cpp

@@ -5,6 +5,7 @@
  * COPYRIGHT:   Copyright 2024-2025 Vitaly Orekhov <[email protected]>
  */
 #include "main.h"
+#include <xmllite.h>
 
 LRESULT CWlanWizard::OnMeasureItem(UINT nMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
 {
@@ -97,7 +98,7 @@ LRESULT CWlanWizard::OnDrawItem(UINT nMsg, WPARAM wParam, LPARAM lParam, BOOL& b
                     HICON hSidebarIcon = NULL;
 
                     wParam == IDC_WLANWIZ_PREFERRED_APS
-                        ? hSidebarIcon = LoadIconW(GetModuleHandleW(L"shell32.dll"), MAKEINTRESOURCE(44)) /* 'Favorites' icon */
+                        ? hSidebarIcon = LoadIconW(GetModuleHandleW(L"shell32.dll"), MAKEINTRESOURCE(IDI_SHELL32_FAVORITES))
                         : hSidebarIcon = LoadIconW(wlanwiz_hInstance, MAKEINTRESOURCEW(wParam + 20));
 
                     DrawIconEx(this->hThemeEB ? hDCBtn : pdis->hDC,
@@ -374,6 +375,82 @@ LRESULT CWlanWizard::OnDrawItem(UINT nMsg, WPARAM wParam, LPARAM lParam, BOOL& b
                     DrawTextW(pdis->hDC, cswExpandedText, cswExpandedText.GetLength(), &rcExpandedText, DT_WORDBREAK | DT_LEFT);
                 }
 
+                /* Connection state or preference as set in saved profile */
+                ATL::CStringW cswConnState;
+                GUID interfaceGuid;
+                LPWSTR pstrProfileXml = NULL;
+                DWORD dwFlags = 0, dwGrantedAccess = WLAN_READ_ACCESS;
+                IIDFromString(this->m_sGUID, &interfaceGuid);
+
+                if (   wcslen(pWlanNetwork->strProfileName)
+                    && SUCCEEDED(WlanGetProfile(this->hWlanClient, &interfaceGuid, pWlanNetwork->strProfileName, NULL, &pstrProfileXml, &dwFlags, &dwGrantedAccess))
+                    && pstrProfileXml != NULL)
+                {
+                    ATL::CComPtr<IStream> xmlStream = CreateDataStream(pstrProfileXml, wcslen(pstrProfileXml));                    
+                    ATL::CComPtr<IXmlReader> xmlReader;
+                    WlanFreeMemory(pstrProfileXml);
+                    
+                    CreateXmlReader(IID_IXmlReader, reinterpret_cast<LPVOID*>(&xmlReader), 0);
+                    xmlReader->SetInput(xmlStream);
+
+                    bool bNodeFound = false;
+                    XmlNodeType xnType = XmlNodeType_None;
+                    while (!bNodeFound)
+                    {
+                        LPCWSTR pwszLocalName;
+                        xmlReader->Read(&xnType);
+
+                        switch(xnType)
+                        {
+                        case XmlNodeType_Element:
+                        {
+                            xmlReader->GetLocalName(&pwszLocalName, NULL);
+
+                            if (pwszLocalName && wcsicmp(pwszLocalName, L"connectionMode") == 0)
+                            {
+                                LPCWSTR pwszConnectionMode;
+                                xmlReader->GetValue(&pwszConnectionMode, NULL);
+
+                                cswConnState.LoadStringW(wcsicmp(pwszConnectionMode, L"auto") == 0
+                                    ? IDS_WLANWIZ_CONNECTED_AUTO
+                                    : IDS_WLANWIZ_CONNECTED_MANU);
+
+                                RECT rcConnMode =
+                                {
+                                    .left = pdis->rcItem.right - 70,
+                                    .top = pdis->rcItem.top + 3,
+                                    .right = pdis->rcItem.right - 25,
+                                    .bottom = pdis->rcItem.top + 22
+                                };
+
+                                DrawTextW(pdis->hDC, cswConnState, cswConnState.GetLength(), &rcConnMode, DT_RIGHT);
+
+                                HICON hFav = LoadIconW(GetModuleHandleW(L"shell32.dll"), MAKEINTRESOURCE(IDI_SHELL32_FAVORITES));
+
+                                DrawIconEx(pdis->hDC,
+                                    pdis->rcItem.right - 20, pdis->rcItem.top + 3,
+                                    hFav,
+                                    16, 16,
+                                    0,
+                                    0,
+                                    DI_NORMAL
+                                );
+
+                                DestroyIcon(hFav);
+                                bNodeFound = true;
+                            }
+
+                            break;
+                        }
+                        default:
+                            break;
+                        }
+                    }
+
+                    xmlReader.Release();
+                    xmlStream.Release();
+                }
+
                 SetTextColor(pdis->hDC, crItemText);
 
                 /* Shrink focus rectangle for listbox items */

dll/shellext/wlanwiz/helpers.cpp

@@ -82,4 +82,33 @@ HWND CWlanWizard::CreateToolTip(_In_ int nID)
     hWndTip.SendMessageW(TTM_ADDTOOL, NULL, reinterpret_cast<LPARAM>(&toolInfo));
 
     return hWndTip;
+}
+
+ATL::CComPtr<IStream> CWlanWizard::CreateDataStream(const PVOID pvData, size_t size)
+{
+    ATL::CComPtr<IStream> stream;
+    HRESULT hr;
+
+    HGLOBAL hGlobal = GlobalAlloc(GHND, size);
+
+    if (FAILED(HRESULT_FROM_WIN32(GetLastError())))
+    {
+        DPRINT1("GlobalAlloc failed: 0x%lx\n", HRESULT_FROM_WIN32(GetLastError()));
+        return nullptr;
+    }
+
+    PVOID ptr = GlobalLock(hGlobal);
+    memcpy(ptr, pvData, size);
+
+    hr = CreateStreamOnHGlobal(hGlobal, TRUE, &stream);
+
+    if (FAILED(hr))
+    {
+        DPRINT1("CreateStreamOnHGlobal failed: 0x%lx\n", hr);
+        return nullptr;
+    }
+
+    GlobalUnlock(hGlobal);
+
+    return stream;
 }

dll/shellext/wlanwiz/main.h

@@ -33,6 +33,7 @@
 #include "resource.h"
 
 #define IDT_SCANNING_NETWORKS 700
+#define IDI_SHELL32_FAVORITES 44
 
 enum WLAN_SCAN_STATES
 {
@@ -200,6 +201,7 @@ class CWlanWizard : public CDialogImpl<CWlanWizard>
 	HWND CreateToolTip(_In_ int toolID);
 	static DWORD WINAPI ScanNetworksThread(_In_ LPVOID lpParameter);
 	static ATL::CStringW APNameToUnicode(_In_ PDOT11_SSID dot11Ssid);
+	static ATL::CComPtr<IStream> CreateDataStream(const PVOID pvData, size_t size);
 	void TryInsertToAdHoc(_Inout_ std::set<DWORD>& setAdHoc, _In_ DWORD dwIndex);
 	void TryInsertToKnown(_Inout_ std::set<DWORD>& setProfiles, _In_ DWORD dwIndex);
 	DWORD TryFindConnected(_In_ DWORD dwIndex);