[reactos] 01/01: [BROWSEUI] auto-completion: Reduce failures in ACListISF (#3589)

[Katayama Hirofumi MZ]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=2eaf0afcd2e0c91876fa5…

commit 2eaf0afcd2e0c91876fa534dd5f6e74740eb0a75
Author:     Katayama Hirofumi MZ <katayama.hirofumi.mz(a)gmail.com>
AuthorDate: Tue Apr 6 21:56:40 2021 +0900
Commit:     GitHub <noreply(a)github.com>
CommitDate: Tue Apr 6 21:56:40 2021 +0900

    [BROWSEUI] auto-completion: Reduce failures in ACListISF (#3589)
    
    Reduce failures in ACListISF testcase. Skip left space in auto-completion. Accept the
pair of drive letter and colon. CORE-9281
---
 dll/win32/browseui/aclistisf.cpp | 122 ++++++++++++++++++---------------------
 dll/win32/browseui/aclistisf.h   |   6 +-
 2 files changed, 59 insertions(+), 69 deletions(-)

diff --git a/dll/win32/browseui/aclistisf.cpp b/dll/win32/browseui/aclistisf.cpp
index cd087ec2f3f..458ab046b5b 100644
--- a/dll/win32/browseui/aclistisf.cpp
+++ b/dll/win32/browseui/aclistisf.cpp
@@ -174,7 +174,9 @@ HRESULT CACListISF::GetDisplayName(LPCITEMIDLIST pidlChild,
CComHeapPtr<WCHAR>&
     return hr;
 }
 
-HRESULT CACListISF::GetPathName(LPCITEMIDLIST pidlChild, CComHeapPtr<WCHAR>&
pszPath)
+HRESULT
+CACListISF::GetPaths(LPCITEMIDLIST pidlChild, CComHeapPtr<WCHAR>& pszRaw,
+                     CComHeapPtr<WCHAR>& pszExpanded)
 {
     TRACE("(%p, %p)\n", this, pidlChild);
 
@@ -183,29 +185,24 @@ HRESULT CACListISF::GetPathName(LPCITEMIDLIST pidlChild,
CComHeapPtr<WCHAR>& psz
     if (FAILED_UNEXPECTEDLY(hr))
         return hr;
 
-    CStringW szPath;
-    if (m_szExpand.GetLength() && m_iNextLocation == LT_DIRECTORY)
+    CStringW szRawPath, szExpanded;
+    if (m_szRawPath.GetLength() && m_iNextLocation == LT_DIRECTORY)
     {
-        INT cchExpand = m_szExpand.GetLength();
-        if (StrCmpNIW(pszChild, m_szExpand, cchExpand) != 0 ||
+        INT cchExpand = m_szRawPath.GetLength();
+        if (StrCmpNIW(pszChild, m_szRawPath, cchExpand) != 0 ||
             pszChild[0] != L'\\' || pszChild[1] != L'\\')
         {
-            szPath = m_szExpand;
+            szRawPath = m_szRawPath;
+            szExpanded = m_szExpanded;
         }
     }
-    szPath += pszChild;
+    szRawPath += pszChild;
+    szExpanded += pszChild;
 
-    INT cchMax = szPath.GetLength() + 1;
-    CComHeapPtr<WCHAR> pszFullPath;
-    if (!pszFullPath.Allocate(cchMax))
-    {
-        ERR("Out of memory\n");
-        return E_OUTOFMEMORY;
-    }
-
-    StringCchCopyW(pszFullPath, cchMax, szPath);
-    pszPath.Attach(pszFullPath.Detach());
-    TRACE("pszPath: '%S'\n", static_cast<LPCWSTR>(pszPath));
+    SHStrDupW(szRawPath, &pszRaw);
+    SHStrDupW(szExpanded, &pszExpanded);
+    TRACE("pszRaw: '%S'\n", static_cast<LPCWSTR>(pszRaw));
+    TRACE("pszExpanded: '%S'\n",
static_cast<LPCWSTR>(pszExpanded));
     return S_OK;
 }
 
@@ -232,7 +229,7 @@ STDMETHODIMP CACListISF::Next(ULONG celt, LPOLESTR *rgelt, ULONG
*pceltFetched)
 
     HRESULT hr;
     CComHeapPtr<ITEMIDLIST> pidlChild;
-    CComHeapPtr<WCHAR> pszPathName;
+    CComHeapPtr<WCHAR> pszRawPath, pszExpanded;
 
     do
     {
@@ -243,38 +240,16 @@ STDMETHODIMP CACListISF::Next(ULONG celt, LPOLESTR *rgelt, ULONG
*pceltFetched)
             if (hr != S_OK)
                 break;
 
-            pszPathName.Free();
-            GetPathName(pidlChild, pszPathName);
-            if (!pszPathName)
+            pszRawPath.Free();
+            pszExpanded.Free();
+            GetPaths(pidlChild, pszRawPath, pszExpanded);
+            if (!pszRawPath || !pszExpanded)
                 continue;
 
-            if (m_dwOptions & (ACLO_FILESYSONLY | ACLO_FILESYSDIRS))
-            {
-                DWORD attrs = SFGAO_FILESYSANCESTOR | SFGAO_FILESYSTEM;
-                LPCITEMIDLIST pidlRef = pidlChild;
-                hr = m_pShellFolder->GetAttributesOf(1, &pidlRef, &attrs);
-                if (SUCCEEDED(hr))
-                {
-                   if (!(attrs & (SFGAO_FILESYSTEM | SFGAO_FILESYSANCESTOR)))
-                        continue;
-                }
-            }
-
-            if (m_dwOptions & ACLO_FILESYSDIRS)
-            {
-                if (wcschr(pszPathName, L'%') != NULL)
-                {
-                    WCHAR szPath[MAX_PATH];
-                    ExpandEnvironmentStringsW(pszPathName, szPath, _countof(szPath));
-                    if (!PathIsDirectoryW(szPath))
-                        continue;
-                }
-                else
-                {
-                    if (!PathIsDirectoryW(pszPathName))
-                        continue;
-                }
-            }
+            if ((m_dwOptions & ACLO_FILESYSDIRS) &&
!PathIsDirectoryW(pszExpanded))
+                continue;
+            else if ((m_dwOptions & ACLO_FILESYSONLY) &&
!PathFileExistsW(pszExpanded))
+                continue;
 
             hr = S_OK;
             break;
@@ -283,7 +258,7 @@ STDMETHODIMP CACListISF::Next(ULONG celt, LPOLESTR *rgelt, ULONG
*pceltFetched)
 
     if (hr == S_OK)
     {
-        *rgelt = pszPathName.Detach();
+        *rgelt = pszRawPath.Detach();
         if (pceltFetched)
             *pceltFetched = 1;
     }
@@ -301,7 +276,7 @@ STDMETHODIMP CACListISF::Reset()
     TRACE("(%p)\n", this);
 
     m_iNextLocation = LT_DIRECTORY;
-    m_szExpand = L"";
+    m_szRawPath = L"";
 
     SHELLSTATE ss = { 0 };
     SHGetSetSettings(&ss, SSF_SHOWALLOBJECTS, FALSE);
@@ -341,34 +316,48 @@ STDMETHODIMP CACListISF::Expand(LPCOLESTR pszExpand)
 {
     TRACE("(%p, %ls)\n", this, pszExpand);
 
-    m_szExpand = pszExpand;
+    m_szRawPath = pszExpand;
     m_iNextLocation = LT_DIRECTORY;
 
+    // skip left space
+    while (*pszExpand == L' ')
+        ++pszExpand;
+
     // expand environment variables (%WINDIR% etc.)
-    WCHAR szExpanded[MAX_PATH];
-    if (wcschr(pszExpand, L'%') != NULL &&
-        ExpandEnvironmentStringsW(pszExpand, szExpanded, _countof(szExpanded)))
-    {
-        pszExpand = szExpanded;
-    }
+    WCHAR szExpanded[MAX_PATH], szPath1[MAX_PATH], szPath2[MAX_PATH];
+    ExpandEnvironmentStringsW(pszExpand, szExpanded, _countof(szExpanded));
+    pszExpand = szExpanded;
 
     // get full path
-    WCHAR szPath1[MAX_PATH], szPath2[MAX_PATH];
-    if (PathIsRelativeW(pszExpand) &&
-        SHGetPathFromIDListW(m_pidlCurDir, szPath1) &&
-        PathCombineW(szPath2, szPath1, pszExpand))
+    if (szExpanded[0] && szExpanded[1] == L':' && szExpanded[2]
== 0)
     {
-        pszExpand = szPath2;
+        // 'C:' --> 'C:\'
+        szExpanded[2] = L'\\';
+        szExpanded[3] = 0;
+    }
+    else
+    {
+        if (PathIsRelativeW(pszExpand) &&
+            SHGetPathFromIDListW(m_pidlCurDir, szPath1) &&
+            PathCombineW(szPath2, szPath1, pszExpand))
+        {
+            pszExpand = szPath2;
+        }
+        GetFullPathNameW(pszExpand, _countof(szPath1), szPath1, NULL);
+        pszExpand = szPath1;
     }
-    GetFullPathNameW(pszExpand, _countof(szPath1), szPath1, NULL);
 
     CComHeapPtr<ITEMIDLIST> pidl;
-    HRESULT hr = SHParseDisplayName(szPath1, NULL, &pidl, NULL, NULL);
+    m_szExpanded = pszExpand;
+    HRESULT hr = SHParseDisplayName(m_szExpanded, NULL, &pidl, NULL, NULL);
     if (SUCCEEDED(hr))
     {
         hr = SetLocation(pidl.Detach());
         if (FAILED_UNEXPECTEDLY(hr))
-            m_szExpand = L"";
+        {
+            m_szRawPath = L"";
+            m_szExpanded = L"";
+        }
     }
     return hr;
 }
@@ -448,4 +437,3 @@ STDMETHODIMP CACListISF::SetDirectory(LPCWSTR pwzPath)
     m_pidlCurDir.Attach(pidl);
     return S_OK;
 }
-
diff --git a/dll/win32/browseui/aclistisf.h b/dll/win32/browseui/aclistisf.h
index abc63efd45c..d09b47c3c8f 100644
--- a/dll/win32/browseui/aclistisf.h
+++ b/dll/win32/browseui/aclistisf.h
@@ -43,7 +43,8 @@ private:
     DWORD m_dwOptions;
     LOCATION_TYPE m_iNextLocation;
     BOOL m_fShowHidden;
-    CStringW m_szExpand;
+    CStringW m_szRawPath;
+    CStringW m_szExpanded;
     CComHeapPtr<ITEMIDLIST> m_pidlLocation;
     CComHeapPtr<ITEMIDLIST> m_pidlCurDir;
     CComPtr<IEnumIDList> m_pEnumIDList;
@@ -57,7 +58,8 @@ public:
     HRESULT NextLocation();
     HRESULT SetLocation(LPITEMIDLIST pidl);
     HRESULT GetDisplayName(LPCITEMIDLIST pidlChild, CComHeapPtr<WCHAR>&
pszChild);
-    HRESULT GetPathName(LPCITEMIDLIST pidlChild, CComHeapPtr<WCHAR>& pszPath);
+    HRESULT GetPaths(LPCITEMIDLIST pidlChild, CComHeapPtr<WCHAR>& pszRaw,
+                     CComHeapPtr<WCHAR>& pszExpanded);
 
     // *** IEnumString methods ***
     STDMETHODIMP Next(ULONG celt, LPOLESTR *rgelt, ULONG *pceltFetched) override;