dlls/shell32/autocomplete.c

   1 /*
   2  *      AutoComplete interfaces implementation.
   3  *
   4  *      Copyright 2004  Maxime Bellengé <maxime.bellenge@laposte.net>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 /*
  22   Implemented:
  23   - ACO_AUTOAPPEND style
  24   - ACO_AUTOSUGGEST style
  25   - ACO_UPDOWNKEYDROPSLIST style
  26
  27   - Handle pwzsRegKeyPath and pwszQuickComplete in Init
  28
  29   TODO:
  30   - implement ACO_SEARCH style
  31   - implement ACO_FILTERPREFIXES style
  32   - implement ACO_USETAB style
  33   - implement ACO_RTLREADING style
  34   - implement ResetEnumerator
  35   - string compares should be case-insensitive, the content of the list should be sorted
  36
  37  */
  38 #include "config.h"
  39
  40 #include <stdarg.h>
  41 #include <stdlib.h>
  42 #include <string.h>
  43
  44 #define COBJMACROS
  45
  46 #include "wine/debug.h"
  47 #include "windef.h"
  48 #include "winbase.h"
  49 #include "winreg.h"
  50 #include "undocshell.h"
  51 #include "shlwapi.h"
  52 #include "winerror.h"
  53 #include "objbase.h"
  54
  55 #include "pidl.h"
  56 #include "shlobj.h"
  57 #include "shldisp.h"
  58 #include "debughlp.h"
  59 #include "shell32_main.h"
  60
  61 #include "wine/unicode.h"
  62
  63 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  64
  65 typedef struct
  66 {
  67     IAutoComplete2 IAutoComplete2_iface;
  68     IAutoCompleteDropDown IAutoCompleteDropDown_iface;
  69     LONG ref;
  70     BOOL initialized;
  71     BOOL enabled;
  72     HWND hwndEdit;
  73     HWND hwndListBox;
  74     WNDPROC wpOrigEditProc;
  75     WNDPROC wpOrigLBoxProc;
  76     WCHAR *txtbackup;
  77     WCHAR *quickComplete;
  78     IEnumString *enumstr;
  79     AUTOCOMPLETEOPTIONS options;
  80 } IAutoCompleteImpl;
  81
  82 enum autoappend_flag
  83 {
  84     autoappend_flag_yes,
  85     autoappend_flag_no,
  86     autoappend_flag_displayempty
  87 };
  88
  89 static const WCHAR autocomplete_propertyW[] = {'W','i','n','e',' ','A','u','t','o',
  90                                                'c','o','m','p','l','e','t','e',' ',
  91                                                'c','o','n','t','r','o','l',0};
  92
  93 static inline IAutoCompleteImpl *impl_from_IAutoComplete2(IAutoComplete2 *iface)
  94 {
  95     return CONTAINING_RECORD(iface, IAutoCompleteImpl, IAutoComplete2_iface);
  96 }
  97
  98 static inline IAutoCompleteImpl *impl_from_IAutoCompleteDropDown(IAutoCompleteDropDown *iface)
  99 {
 100     return CONTAINING_RECORD(iface, IAutoCompleteImpl, IAutoCompleteDropDown_iface);
 101 }
 102
 103 static void set_text_and_selection(IAutoCompleteImpl *ac, HWND hwnd, WCHAR *text, WPARAM start, LPARAM end)
 104 {
 105     /* Send it directly to the edit control to match Windows behavior */
 106     WNDPROC proc = ac->wpOrigEditProc;
 107     if (CallWindowProcW(proc, hwnd, WM_SETTEXT, 0, (LPARAM)text))
 108         CallWindowProcW(proc, hwnd, EM_SETSEL, start, end);
 109 }
 110
 111 static size_t format_quick_complete(WCHAR *dst, const WCHAR *qc, const WCHAR *str, size_t str_len)
 112 {
 113     /* Replace the first %s directly without using snprintf, to avoid
 114        exploits since the format string can be retrieved from the registry */
 115     WCHAR *base = dst;
 116     UINT args = 0;
 117     while (*qc != '\0')
 118     {
 119         if (qc[0] == '%')
 120         {
 121             if (args < 1 && qc[1] == 's')
 122             {
 123                 memcpy(dst, str, str_len * sizeof(WCHAR));
 124                 dst += str_len;
 125                 qc += 2;
 126                 args++;
 127                 continue;
 128             }
 129             qc += (qc[1] == '%');
 130         }
 131         *dst++ = *qc++;
 132     }
 133     *dst = '\0';
 134     return dst - base;
 135 }
 136
 137 static LRESULT change_selection(IAutoCompleteImpl *ac, HWND hwnd, UINT key)
 138 {
 139     INT count = SendMessageW(ac->hwndListBox, LB_GETCOUNT, 0, 0);
 140     INT sel = SendMessageW(ac->hwndListBox, LB_GETCURSEL, 0, 0);
 141     if (key == VK_PRIOR || key == VK_NEXT)
 142     {
 143         if (sel < 0)
 144             sel = (key == VK_PRIOR) ? count - 1 : 0;
 145         else
 146         {
 147             INT base = SendMessageW(ac->hwndListBox, LB_GETTOPINDEX, 0, 0);
 148             INT pgsz = SendMessageW(ac->hwndListBox, LB_GETLISTBOXINFO, 0, 0);
 149             pgsz = max(pgsz - 1, 1);
 150             if (key == VK_PRIOR)
 151             {
 152                 if (sel == 0)
 153                     sel = -1;
 154                 else
 155                 {
 156                     if (sel == base) base -= min(base, pgsz);
 157                     sel = base;
 158                 }
 159             }
 160             else
 161             {
 162                 if (sel == count - 1)
 163                     sel = -1;
 164                 else
 165                 {
 166                     base += pgsz;
 167                     if (sel >= base) base += pgsz;
 168                     sel = min(base, count - 1);
 169                 }
 170             }
 171         }
 172     }
 173     else if (key == VK_UP)
 174         sel = ((sel - 1) < -1) ? count - 1 : sel - 1;
 175     else
 176         sel = ((sel + 1) >= count) ? -1 : sel + 1;
 177
 178     SendMessageW(ac->hwndListBox, LB_SETCURSEL, sel, 0);
 179     if (sel >= 0)
 180     {
 181         WCHAR *msg;
 182         UINT len = SendMessageW(ac->hwndListBox, LB_GETTEXTLEN, sel, 0);
 183         if (!(msg = heap_alloc((len + 1) * sizeof(WCHAR))))
 184             return 0;
 185         len = SendMessageW(ac->hwndListBox, LB_GETTEXT, sel, (LPARAM)msg);
 186         set_text_and_selection(ac, hwnd, msg, len, len);
 187         heap_free(msg);
 188     }
 189     else
 190     {
 191         UINT len = strlenW(ac->txtbackup);
 192         set_text_and_selection(ac, hwnd, ac->txtbackup, len, len);
 193     }
 194     return 0;
 195 }
 196
 197 static void autoappend_str(IAutoCompleteImpl *ac, WCHAR *text, UINT len, WCHAR *str, HWND hwnd)
 198 {
 199     DWORD sel_start;
 200     WCHAR *tmp;
 201     size_t size;
 202
 203     /* Don't auto-append unless the caret is at the end */
 204     SendMessageW(hwnd, EM_GETSEL, (WPARAM)&sel_start, 0);
 205     if (sel_start != len)
 206         return;
 207
 208     /* The character capitalization can be different,
 209        so merge text and str into a new string */
 210     size = len + strlenW(&str[len]) + 1;
 211
 212     if ((tmp = heap_alloc(size * sizeof(*tmp))))
 213     {
 214         memcpy(tmp, text, len * sizeof(*tmp));
 215         memcpy(&tmp[len], &str[len], (size - len) * sizeof(*tmp));
 216     }
 217     else tmp = str;
 218
 219     set_text_and_selection(ac, hwnd, tmp, len, size - 1);
 220     if (tmp != str)
 221         heap_free(tmp);
 222 }
 223
 224 static void autocomplete_text(IAutoCompleteImpl *ac, HWND hwnd, enum autoappend_flag flag)
 225 {
 226     HRESULT hr;
 227     WCHAR *text;
 228     UINT cpt, size, len = SendMessageW(hwnd, WM_GETTEXTLENGTH, 0, 0);
 229
 230     if (flag != autoappend_flag_displayempty && len == 0)
 231     {
 232         if (ac->options & ACO_AUTOSUGGEST)
 233             ShowWindow(ac->hwndListBox, SW_HIDE);
 234         return;
 235     }
 236
 237     size = len + 1;
 238     if (!(text = heap_alloc(size * sizeof(WCHAR))))
 239         return;
 240     len = SendMessageW(hwnd, WM_GETTEXT, size, (LPARAM)text);
 241     if (len + 1 != size)
 242         text = heap_realloc(text, (len + 1) * sizeof(WCHAR));
 243
 244     /* Set txtbackup to point to text itself (which must not be released) */
 245     heap_free(ac->txtbackup);
 246     ac->txtbackup = text;
 247
 248     if (ac->options & ACO_AUTOSUGGEST)
 249     {
 250         SendMessageW(ac->hwndListBox, WM_SETREDRAW, FALSE, 0);
 251         SendMessageW(ac->hwndListBox, LB_RESETCONTENT, 0, 0);
 252     }
 253     IEnumString_Reset(ac->enumstr);
 254     for (cpt = 0;;)
 255     {
 256         LPOLESTR strs = NULL;
 257         ULONG fetched;
 258
 259         hr = IEnumString_Next(ac->enumstr, 1, &strs, &fetched);
 260         if (hr != S_OK)
 261             break;
 262
 263         if (!strncmpiW(text, strs, len))
 264         {
 265             if (cpt == 0 && flag == autoappend_flag_yes)
 266             {
 267                 autoappend_str(ac, text, len, strs, hwnd);
 268                 if (!(ac->options & ACO_AUTOSUGGEST))
 269                 {
 270                     CoTaskMemFree(strs);
 271                     break;
 272                 }
 273             }
 274
 275             if (ac->options & ACO_AUTOSUGGEST)
 276                 SendMessageW(ac->hwndListBox, LB_ADDSTRING, 0, (LPARAM)strs);
 277
 278             cpt++;
 279         }
 280
 281         CoTaskMemFree(strs);
 282     }
 283
 284     if (ac->options & ACO_AUTOSUGGEST)
 285     {
 286         if (cpt)
 287         {
 288             RECT r;
 289             UINT height = SendMessageW(ac->hwndListBox, LB_GETITEMHEIGHT, 0, 0);
 290             SendMessageW(ac->hwndListBox, LB_CARETOFF, 0, 0);
 291             GetWindowRect(hwnd, &r);
 292             /* It seems that Windows XP displays 7 lines at most
 293                and otherwise displays a vertical scroll bar */
 294             SetWindowPos(ac->hwndListBox, HWND_TOP,
 295                          r.left, r.bottom + 1, r.right - r.left, height * min(cpt + 1, 7),
 296                          SWP_SHOWWINDOW );
 297             SendMessageW(ac->hwndListBox, WM_SETREDRAW, TRUE, 0);
 298         }
 299         else
 300             ShowWindow(ac->hwndListBox, SW_HIDE);
 301     }
 302 }
 303
 304 static void destroy_autocomplete_object(IAutoCompleteImpl *ac)
 305 {
 306     ac->hwndEdit = NULL;
 307     if (ac->hwndListBox)
 308         DestroyWindow(ac->hwndListBox);
 309     IAutoComplete2_Release(&ac->IAutoComplete2_iface);
 310 }
 311
 312 /*
 313    Helper for ACEditSubclassProc
 314 */
 315 static LRESULT ACEditSubclassProc_KeyDown(IAutoCompleteImpl *ac, HWND hwnd, UINT uMsg,
 316                                           WPARAM wParam, LPARAM lParam)
 317 {
 318     switch (wParam)
 319     {
 320         case VK_RETURN:
 321             /* If quickComplete is set and control is pressed, replace the string */
 322             if (ac->quickComplete && (GetKeyState(VK_CONTROL) & 0x8000))
 323             {
 324                 WCHAR *text, *buf;
 325                 size_t sz;
 326                 UINT len = SendMessageW(hwnd, WM_GETTEXTLENGTH, 0, 0);
 327                 if (!(text = heap_alloc((len + 1) * sizeof(WCHAR))))
 328                     return 0;
 329                 len = SendMessageW(hwnd, WM_GETTEXT, len + 1, (LPARAM)text);
 330                 sz = strlenW(ac->quickComplete) + 1 + len;
 331
 332                 if ((buf = heap_alloc(sz * sizeof(WCHAR))))
 333                 {
 334                     len = format_quick_complete(buf, ac->quickComplete, text, len);
 335                     set_text_and_selection(ac, hwnd, buf, 0, len);
 336                     heap_free(buf);
 337                 }
 338
 339                 if (ac->options & ACO_AUTOSUGGEST)
 340                     ShowWindow(ac->hwndListBox, SW_HIDE);
 341                 heap_free(text);
 342                 return 0;
 343             }
 344
 345             if (ac->options & ACO_AUTOSUGGEST)
 346                 ShowWindow(ac->hwndListBox, SW_HIDE);
 347             break;
 348         case VK_UP:
 349         case VK_DOWN:
 350         case VK_PRIOR:
 351         case VK_NEXT:
 352             /* Two cases here:
 353                - if the listbox is not visible and ACO_UPDOWNKEYDROPSLIST is
 354                  set, display it with all the entries, without selecting any
 355                - if the listbox is visible, change the selection
 356             */
 357             if (!(ac->options & ACO_AUTOSUGGEST))
 358                 break;
 359
 360             if (!IsWindowVisible(ac->hwndListBox))
 361             {
 362                 if (ac->options & ACO_UPDOWNKEYDROPSLIST)
 363                 {
 364                     autocomplete_text(ac, hwnd, autoappend_flag_displayempty);
 365                     return 0;
 366                 }
 367             }
 368             else
 369                 return change_selection(ac, hwnd, wParam);
 370             break;
 371         case VK_DELETE:
 372         {
 373             LRESULT ret = CallWindowProcW(ac->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 374             autocomplete_text(ac, hwnd, autoappend_flag_no);
 375             return ret;
 376         }
 377     }
 378     return CallWindowProcW(ac->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 379 }
 380
 381 /*
 382   Window procedure for autocompletion
 383  */
 384 static LRESULT APIENTRY ACEditSubclassProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
 385 {
 386     IAutoCompleteImpl *This = GetPropW(hwnd, autocomplete_propertyW);
 387     LRESULT ret;
 388
 389     if (!This->enabled) return CallWindowProcW(This->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 390
 391     switch (uMsg)
 392     {
 393         case CB_SHOWDROPDOWN:
 394             if (This->options & ACO_AUTOSUGGEST)
 395                 ShowWindow(This->hwndListBox, SW_HIDE);
 396             return 0;
 397         case WM_KILLFOCUS:
 398             if ((This->options & ACO_AUTOSUGGEST) && ((HWND)wParam != This->hwndListBox))
 399             {
 400                 ShowWindow(This->hwndListBox, SW_HIDE);
 401             }
 402             break;
 403         case WM_KEYDOWN:
 404             return ACEditSubclassProc_KeyDown(This, hwnd, uMsg, wParam, lParam);
 405         case WM_CHAR:
 406         case WM_UNICHAR:
 407             /* Don't autocomplete at all on most control characters */
 408             if (iscntrlW(wParam) && !(wParam >= '\b' && wParam <= '\r'))
 409                 break;
 410
 411             ret = CallWindowProcW(This->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 412             autocomplete_text(This, hwnd, (This->options & ACO_AUTOAPPEND) && wParam >= ' '
 413                                           ? autoappend_flag_yes : autoappend_flag_no);
 414             return ret;
 415         case WM_SETTEXT:
 416         case WM_CUT:
 417         case WM_CLEAR:
 418         case WM_UNDO:
 419             ret = CallWindowProcW(This->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 420             autocomplete_text(This, hwnd, autoappend_flag_no);
 421             return ret;
 422         case WM_PASTE:
 423             ret = CallWindowProcW(This->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 424             autocomplete_text(This, hwnd, autoappend_flag_yes);
 425             return ret;
 426         case WM_SETFONT:
 427             if (This->hwndListBox)
 428                 SendMessageW(This->hwndListBox, WM_SETFONT, wParam, lParam);
 429             break;
 430         case WM_DESTROY:
 431         {
 432             WNDPROC proc = This->wpOrigEditProc;
 433
 434             SetWindowLongPtrW(hwnd, GWLP_WNDPROC, (LONG_PTR)proc);
 435             RemovePropW(hwnd, autocomplete_propertyW);
 436             destroy_autocomplete_object(This);
 437             return CallWindowProcW(proc, hwnd, uMsg, wParam, lParam);
 438         }
 439     }
 440     return CallWindowProcW(This->wpOrigEditProc, hwnd, uMsg, wParam, lParam);
 441 }
 442
 443 static LRESULT APIENTRY ACLBoxSubclassProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
 444 {
 445     IAutoCompleteImpl *This = (IAutoCompleteImpl *)GetWindowLongPtrW(hwnd, GWLP_USERDATA);
 446     WCHAR *msg;
 447     int sel, len;
 448
 449     switch (uMsg) {
 450         case WM_MOUSEMOVE:
 451             sel = SendMessageW(hwnd, LB_ITEMFROMPOINT, 0, lParam);
 452             SendMessageW(hwnd, LB_SETCURSEL, sel, 0);
 453             break;
 454         case WM_LBUTTONDOWN:
 455             sel = SendMessageW(hwnd, LB_GETCURSEL, 0, 0);
 456             if (sel < 0)
 457                 break;
 458             len = SendMessageW(hwnd, LB_GETTEXTLEN, sel, 0);
 459             if (!(msg = heap_alloc((len + 1) * sizeof(WCHAR))))
 460                 break;
 461             len = SendMessageW(hwnd, LB_GETTEXT, sel, (LPARAM)msg);
 462             set_text_and_selection(This, This->hwndEdit, msg, 0, len);
 463             ShowWindow(hwnd, SW_HIDE);
 464             heap_free(msg);
 465             break;
 466         default:
 467             return CallWindowProcW(This->wpOrigLBoxProc, hwnd, uMsg, wParam, lParam);
 468     }
 469     return 0;
 470 }
 471
 472 static void create_listbox(IAutoCompleteImpl *This)
 473 {
 474     /* FIXME : The listbox should be resizable with the mouse. WS_THICKFRAME looks ugly */
 475     This->hwndListBox = CreateWindowExW(0, WC_LISTBOXW, NULL,
 476                                     WS_BORDER | WS_CHILD | WS_VSCROLL | LBS_HASSTRINGS | LBS_NOTIFY | LBS_NOINTEGRALHEIGHT,
 477                                     0, 0, 0, 0, GetParent(This->hwndEdit), NULL, shell32_hInstance, NULL);
 478
 479     if (This->hwndListBox) {
 480         HFONT edit_font;
 481
 482         This->wpOrigLBoxProc = (WNDPROC) SetWindowLongPtrW( This->hwndListBox, GWLP_WNDPROC, (LONG_PTR) ACLBoxSubclassProc);
 483         SetWindowLongPtrW( This->hwndListBox, GWLP_USERDATA, (LONG_PTR)This);
 484         SetParent(This->hwndListBox, HWND_DESKTOP);
 485
 486         /* Use the same font as the edit control, as it gets destroyed before it anyway */
 487         edit_font = (HFONT)SendMessageW(This->hwndEdit, WM_GETFONT, 0, 0);
 488         if (edit_font)
 489             SendMessageW(This->hwndListBox, WM_SETFONT, (WPARAM)edit_font, FALSE);
 490     }
 491     else
 492         This->options &= ~ACO_AUTOSUGGEST;
 493 }
 494
 495 /**************************************************************************
 496  *  AutoComplete_QueryInterface
 497  */
 498 static HRESULT WINAPI IAutoComplete2_fnQueryInterface(
 499     IAutoComplete2 * iface,
 500     REFIID riid,
 501     LPVOID *ppvObj)
 502 {
 503     IAutoCompleteImpl *This = impl_from_IAutoComplete2(iface);
 504
 505     TRACE("(%p)->(IID:%s,%p)\n", This, shdebugstr_guid(riid), ppvObj);
 506     *ppvObj = NULL;
 507
 508     if (IsEqualIID(riid, &IID_IUnknown) ||
 509         IsEqualIID(riid, &IID_IAutoComplete) ||
 510         IsEqualIID(riid, &IID_IAutoComplete2))
 511     {
 512         *ppvObj = &This->IAutoComplete2_iface;
 513     }
 514     else if (IsEqualIID(riid, &IID_IAutoCompleteDropDown))
 515     {
 516         *ppvObj = &This->IAutoCompleteDropDown_iface;
 517     }
 518
 519     if (*ppvObj)
 520     {
 521         IUnknown_AddRef((IUnknown*)*ppvObj);
 522         TRACE("-- Interface: (%p)->(%p)\n", ppvObj, *ppvObj);
 523         return S_OK;
 524     }
 525     WARN("unsupported interface: %s\n", debugstr_guid(riid));
 526     return E_NOINTERFACE;
 527 }
 528
 529 /******************************************************************************
 530  * IAutoComplete2_fnAddRef
 531  */
 532 static ULONG WINAPI IAutoComplete2_fnAddRef(
 533         IAutoComplete2 * iface)
 534 {
 535     IAutoCompleteImpl *This = impl_from_IAutoComplete2(iface);
 536     ULONG refCount = InterlockedIncrement(&This->ref);
 537
 538     TRACE("(%p)->(%u)\n", This, refCount - 1);
 539
 540     return refCount;
 541 }
 542
 543 /******************************************************************************
 544  * IAutoComplete2_fnRelease
 545  */
 546 static ULONG WINAPI IAutoComplete2_fnRelease(
 547         IAutoComplete2 * iface)
 548 {
 549     IAutoCompleteImpl *This = impl_from_IAutoComplete2(iface);
 550     ULONG refCount = InterlockedDecrement(&This->ref);
 551
 552     TRACE("(%p)->(%u)\n", This, refCount + 1);
 553
 554     if (!refCount) {
 555         TRACE("destroying IAutoComplete(%p)\n", This);
 556         heap_free(This->quickComplete);
 557         heap_free(This->txtbackup);
 558         if (This->enumstr)
 559             IEnumString_Release(This->enumstr);
 560         heap_free(This);
 561     }
 562     return refCount;
 563 }
 564
 565 /******************************************************************************
 566  * IAutoComplete2_fnEnable
 567  */
 568 static HRESULT WINAPI IAutoComplete2_fnEnable(
 569     IAutoComplete2 * iface,
 570     BOOL fEnable)
 571 {
 572     IAutoCompleteImpl *This = impl_from_IAutoComplete2(iface);
 573     HRESULT hr = S_OK;
 574
 575     TRACE("(%p)->(%s)\n", This, (fEnable)?"true":"false");
 576
 577     This->enabled = fEnable;
 578
 579     return hr;
 580 }
 581
 582 /******************************************************************************
 583  * IAutoComplete2_fnInit
 584  */
 585 static HRESULT WINAPI IAutoComplete2_fnInit(
 586     IAutoComplete2 * iface,
 587     HWND hwndEdit,
 588     IUnknown *punkACL,
 589     LPCOLESTR pwzsRegKeyPath,
 590     LPCOLESTR pwszQuickComplete)
 591 {
 592     IAutoCompleteImpl *prev, *This = impl_from_IAutoComplete2(iface);
 593
 594     TRACE("(%p)->(%p, %p, %s, %s)\n",
 595           This, hwndEdit, punkACL, debugstr_w(pwzsRegKeyPath), debugstr_w(pwszQuickComplete));
 596
 597     if (This->options & ACO_SEARCH) FIXME(" ACO_SEARCH not supported\n");
 598     if (This->options & ACO_FILTERPREFIXES) FIXME(" ACO_FILTERPREFIXES not supported\n");
 599     if (This->options & ACO_USETAB) FIXME(" ACO_USETAB not supported\n");
 600     if (This->options & ACO_RTLREADING) FIXME(" ACO_RTLREADING not supported\n");
 601
 602     if (!hwndEdit || !punkACL)
 603         return E_INVALIDARG;
 604
 605     if (This->initialized)
 606     {
 607         WARN("Autocompletion object is already initialized\n");
 608         /* This->hwndEdit is set to NULL when the edit window is destroyed. */
 609         return This->hwndEdit ? E_FAIL : E_UNEXPECTED;
 610     }
 611
 612     if (FAILED (IUnknown_QueryInterface (punkACL, &IID_IEnumString, (LPVOID*)&This->enumstr))) {
 613         WARN("No IEnumString interface\n");
 614         return E_NOINTERFACE;
 615     }
 616
 617     This->initialized = TRUE;
 618     This->hwndEdit = hwndEdit;
 619
 620     /* If another AutoComplete object was previously assigned to this edit control,
 621        release it but keep the same callback on the control, to avoid an infinite
 622        recursive loop in ACEditSubclassProc while the property is set to this object */
 623     prev = GetPropW(hwndEdit, autocomplete_propertyW);
 624     SetPropW(hwndEdit, autocomplete_propertyW, This);
 625
 626     if (prev && prev->initialized) {
 627         This->wpOrigEditProc = prev->wpOrigEditProc;
 628         destroy_autocomplete_object(prev);
 629     }
 630     else
 631         This->wpOrigEditProc = (WNDPROC) SetWindowLongPtrW(hwndEdit, GWLP_WNDPROC, (LONG_PTR) ACEditSubclassProc);
 632
 633     /* Keep at least one reference to the object until the edit window is destroyed */
 634     IAutoComplete2_AddRef(&This->IAutoComplete2_iface);
 635
 636     if (This->options & ACO_AUTOSUGGEST)
 637         create_listbox(This);
 638
 639     if (pwzsRegKeyPath)
 640     {
 641         static const HKEY roots[] = { HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE };
 642         WCHAR *key, *value;
 643         DWORD type, sz;
 644         BYTE *qc;
 645         HKEY hKey;
 646         LSTATUS res;
 647         size_t len;
 648         UINT i;
 649
 650         /* pwszRegKeyPath contains the key as well as the value, so split it */
 651         value = strrchrW(pwzsRegKeyPath, '\\');
 652         len = value - pwzsRegKeyPath;
 653
 654         if (value && (key = heap_alloc((len+1) * sizeof(*key))) != NULL)
 655         {
 656             memcpy(key, pwzsRegKeyPath, len * sizeof(*key));
 657             key[len] = '\0';
 658             value++;
 659
 660             for (i = 0; i < ARRAY_SIZE(roots); i++)
 661             {
 662                 if (RegOpenKeyExW(roots[i], key, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
 663                     continue;
 664                 sz = MAX_PATH * sizeof(WCHAR);
 665
 666                 while ((qc = heap_alloc(sz)) != NULL)
 667                 {
 668                     res = RegQueryValueExW(hKey, value, NULL, &type, qc, &sz);
 669                     if (res == ERROR_SUCCESS && type == REG_SZ)
 670                     {
 671                         This->quickComplete = heap_realloc(qc, sz);
 672                         i = ARRAY_SIZE(roots);
 673                         break;
 674                     }
 675                     heap_free(qc);
 676                     if (res != ERROR_MORE_DATA || type != REG_SZ)
 677                         break;
 678                 }
 679                 RegCloseKey(hKey);
 680             }
 681             heap_free(key);
 682         }
 683     }
 684
 685     if (!This->quickComplete && pwszQuickComplete)
 686     {
 687         size_t len = strlenW(pwszQuickComplete)+1;
 688         if ((This->quickComplete = heap_alloc(len * sizeof(WCHAR))) != NULL)
 689             memcpy(This->quickComplete, pwszQuickComplete, len * sizeof(WCHAR));
 690     }
 691
 692     return S_OK;
 693 }
 694
 695 /**************************************************************************
 696  *  IAutoComplete2_fnGetOptions
 697  */
 698 static HRESULT WINAPI IAutoComplete2_fnGetOptions(
 699     IAutoComplete2 * iface,
 700     DWORD *pdwFlag)
 701 {
 702     IAutoCompleteImpl *This = impl_from_IAutoComplete2(iface);
 703     HRESULT hr = S_OK;
 704
 705     TRACE("(%p) -> (%p)\n", This, pdwFlag);
 706
 707     *pdwFlag = This->options;
 708
 709     return hr;
 710 }
 711
 712 /**************************************************************************
 713  *  IAutoComplete2_fnSetOptions
 714  */
 715 static HRESULT WINAPI IAutoComplete2_fnSetOptions(
 716     IAutoComplete2 * iface,
 717     DWORD dwFlag)
 718 {
 719     IAutoCompleteImpl *This = impl_from_IAutoComplete2(iface);
 720     HRESULT hr = S_OK;
 721
 722     TRACE("(%p) -> (0x%x)\n", This, dwFlag);
 723
 724     This->options = dwFlag;
 725
 726     if ((This->options & ACO_AUTOSUGGEST) && This->hwndEdit && !This->hwndListBox)
 727         create_listbox(This);
 728
 729     return hr;
 730 }
 731
 732 /**************************************************************************
 733  *  IAutoComplete2 VTable
 734  */
 735 static const IAutoComplete2Vtbl acvt =
 736 {
 737     IAutoComplete2_fnQueryInterface,
 738     IAutoComplete2_fnAddRef,
 739     IAutoComplete2_fnRelease,
 740     IAutoComplete2_fnInit,
 741     IAutoComplete2_fnEnable,
 742     /* IAutoComplete2 */
 743     IAutoComplete2_fnSetOptions,
 744     IAutoComplete2_fnGetOptions,
 745 };
 746
 747
 748 static HRESULT WINAPI IAutoCompleteDropDown_fnQueryInterface(IAutoCompleteDropDown *iface,
 749             REFIID riid, LPVOID *ppvObj)
 750 {
 751     IAutoCompleteImpl *This = impl_from_IAutoCompleteDropDown(iface);
 752     return IAutoComplete2_QueryInterface(&This->IAutoComplete2_iface, riid, ppvObj);
 753 }
 754
 755 static ULONG WINAPI IAutoCompleteDropDown_fnAddRef(IAutoCompleteDropDown *iface)
 756 {
 757     IAutoCompleteImpl *This = impl_from_IAutoCompleteDropDown(iface);
 758     return IAutoComplete2_AddRef(&This->IAutoComplete2_iface);
 759 }
 760
 761 static ULONG WINAPI IAutoCompleteDropDown_fnRelease(IAutoCompleteDropDown *iface)
 762 {
 763     IAutoCompleteImpl *This = impl_from_IAutoCompleteDropDown(iface);
 764     return IAutoComplete2_Release(&This->IAutoComplete2_iface);
 765 }
 766
 767 /**************************************************************************
 768  *  IAutoCompleteDropDown_fnGetDropDownStatus
 769  */
 770 static HRESULT WINAPI IAutoCompleteDropDown_fnGetDropDownStatus(
 771     IAutoCompleteDropDown *iface,
 772     DWORD *pdwFlags,
 773     LPWSTR *ppwszString)
 774 {
 775     IAutoCompleteImpl *This = impl_from_IAutoCompleteDropDown(iface);
 776     BOOL dropped;
 777
 778     TRACE("(%p) -> (%p, %p)\n", This, pdwFlags, ppwszString);
 779
 780     dropped = IsWindowVisible(This->hwndListBox);
 781
 782     if (pdwFlags)
 783         *pdwFlags = (dropped ? ACDD_VISIBLE : 0);
 784
 785     if (ppwszString) {
 786         if (dropped) {
 787             int sel;
 788
 789             sel = SendMessageW(This->hwndListBox, LB_GETCURSEL, 0, 0);
 790             if (sel >= 0)
 791             {
 792                 DWORD len;
 793
 794                 len = SendMessageW(This->hwndListBox, LB_GETTEXTLEN, sel, 0);
 795                 *ppwszString = CoTaskMemAlloc((len+1)*sizeof(WCHAR));
 796                 SendMessageW(This->hwndListBox, LB_GETTEXT, sel, (LPARAM)*ppwszString);
 797             }
 798             else
 799                 *ppwszString = NULL;
 800         }
 801         else
 802             *ppwszString = NULL;
 803     }
 804
 805     return S_OK;
 806 }
 807
 808 /**************************************************************************
 809  *  IAutoCompleteDropDown_fnResetEnumarator
 810  */
 811 static HRESULT WINAPI IAutoCompleteDropDown_fnResetEnumerator(
 812     IAutoCompleteDropDown *iface)
 813 {
 814     IAutoCompleteImpl *This = impl_from_IAutoCompleteDropDown(iface);
 815
 816     FIXME("(%p): stub\n", This);
 817
 818     return E_NOTIMPL;
 819 }
 820
 821 /**************************************************************************
 822  *  IAutoCompleteDropDown VTable
 823  */
 824 static const IAutoCompleteDropDownVtbl acdropdownvt =
 825 {
 826     IAutoCompleteDropDown_fnQueryInterface,
 827     IAutoCompleteDropDown_fnAddRef,
 828     IAutoCompleteDropDown_fnRelease,
 829     IAutoCompleteDropDown_fnGetDropDownStatus,
 830     IAutoCompleteDropDown_fnResetEnumerator,
 831 };
 832
 833 /**************************************************************************
 834  *  IAutoComplete_Constructor
 835  */
 836 HRESULT WINAPI IAutoComplete_Constructor(IUnknown * pUnkOuter, REFIID riid, LPVOID * ppv)
 837 {
 838     IAutoCompleteImpl *lpac;
 839     HRESULT hr;
 840
 841     if (pUnkOuter && !IsEqualIID (riid, &IID_IUnknown))
 842         return CLASS_E_NOAGGREGATION;
 843
 844     lpac = heap_alloc_zero(sizeof(*lpac));
 845     if (!lpac)
 846         return E_OUTOFMEMORY;
 847
 848     lpac->ref = 1;
 849     lpac->IAutoComplete2_iface.lpVtbl = &acvt;
 850     lpac->IAutoCompleteDropDown_iface.lpVtbl = &acdropdownvt;
 851     lpac->enabled = TRUE;
 852     lpac->options = ACO_AUTOAPPEND;
 853
 854     hr = IAutoComplete2_QueryInterface(&lpac->IAutoComplete2_iface, riid, ppv);
 855     IAutoComplete2_Release(&lpac->IAutoComplete2_iface);
 856
 857     TRACE("-- (%p)->\n",lpac);
 858
 859     return hr;
 860 }