dlls/browseui/progressdlg.c

   1 /*
   2  *      Progress dialog
   3  *
   4  *      Copyright 2007  Mikolaj Zalewski
   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 #include "config.h"
  22
  23 #include <stdarg.h>
  24
  25 #define COBJMACROS
  26
  27 #include "wine/debug.h"
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winreg.h"
  31 #include "winuser.h"
  32 #include "shlwapi.h"
  33 #include "winerror.h"
  34 #include "objbase.h"
  35
  36 #include "shlguid.h"
  37 #include "shlobj.h"
  38
  39 #include "wine/unicode.h"
  40
  41 #include "browseui.h"
  42 #include "resids.h"
  43
  44 WINE_DEFAULT_DEBUG_CHANNEL(browseui);
  45
  46 #define CANCEL_MSG_LINE 2
  47
  48 /* Note: to avoid a deadlock we don't want to send messages to the dialog
  49  * with the critical section held. Instead we only mark what fields shoud be
  50  * updated and the dialog proc does the update */
  51 #define UPDATE_PROGRESS         0x1
  52 #define UPDATE_TITLE            0x2
  53 #define UPDATE_LINE1            0x4
  54 #define UPDATE_LINE2            (UPDATE_LINE1<<1)
  55 #define UPDATE_LINE3            (UPDATE_LINE2<<2)
  56
  57
  58 #define WM_DLG_UPDATE   (WM_APP+1)  /* set to the dialog when it should update */
  59 #define WM_DLG_DESTROY  (WM_APP+2)  /* DestroyWindow must be called from the owning thread */
  60
  61 typedef struct tagProgressDialog {
  62     const IProgressDialogVtbl *vtbl;
  63     LONG refCount;
  64     CRITICAL_SECTION cs;
  65     HWND hwnd;
  66     DWORD dwFlags;
  67     DWORD dwUpdate;
  68     LPWSTR lines[3];
  69     LPWSTR cancelMsg;
  70     LPWSTR title;
  71     BOOL isCancelled;
  72     ULONGLONG ullCompleted;
  73     ULONGLONG ullTotal;
  74     HWND hwndDisabledParent;    /* For modal dialog: the parent that need to be re-enabled when the dialog ends */
  75 } ProgressDialog;
  76
  77 static const IProgressDialogVtbl ProgressDialogVtbl;
  78
  79 static void set_buffer(LPWSTR *buffer, LPCWSTR string)
  80 {
  81     const WCHAR empty_string = {0};
  82     IMalloc *malloc;
  83     int cb;
  84
  85     if (string == NULL)
  86         string = empty_string;
  87     CoGetMalloc(1, &malloc);
  88
  89     cb = (strlenW(string) + 1)*sizeof(WCHAR);
  90     if (*buffer == NULL || cb > IMalloc_GetSize(malloc, *buffer))
  91         *buffer = IMalloc_Realloc(malloc, *buffer, cb);
  92     memcpy(*buffer, string, cb);
  93 }
  94
  95 struct create_params
  96 {
  97     ProgressDialog *This;
  98     HANDLE hEvent;
  99     HWND hwndParent;
 100 };
 101
 102 static LPWSTR load_string(HINSTANCE hInstance, UINT uiResourceId)
 103 {
 104     WCHAR string[256];
 105     LPWSTR ret;
 106
 107     LoadStringW(hInstance, uiResourceId, string, sizeof(string)/sizeof(string[0]));
 108     ret = HeapAlloc(GetProcessHeap(), 0, (strlenW(string) + 1) * sizeof(WCHAR));
 109     strcpyW(ret, string);
 110     return ret;
 111 }
 112
 113 static void set_progress_marquee(ProgressDialog *This)
 114 {
 115     HWND hProgress = GetDlgItem(This->hwnd, IDC_PROGRESS_BAR);
 116     SetWindowLongW(hProgress, GWL_STYLE,
 117         GetWindowLongW(hProgress, GWL_STYLE)|PBS_MARQUEE);
 118 }
 119
 120 void update_dialog(ProgressDialog *This, DWORD dwUpdate)
 121 {
 122     WCHAR empty[] = {0};
 123
 124     if (dwUpdate & UPDATE_TITLE)
 125         SetWindowTextW(This->hwnd, This->title);
 126
 127     if (dwUpdate & UPDATE_LINE1)
 128         SetDlgItemTextW(This->hwnd, IDC_TEXT_LINE, (This->isCancelled ? empty : This->lines[0]));
 129     if (dwUpdate & UPDATE_LINE2)
 130         SetDlgItemTextW(This->hwnd, IDC_TEXT_LINE+1, (This->isCancelled ? empty : This->lines[1]));
 131     if (dwUpdate & UPDATE_LINE3)
 132         SetDlgItemTextW(This->hwnd, IDC_TEXT_LINE+2, (This->isCancelled ? This->cancelMsg : This->lines[2]));
 133
 134     if (dwUpdate & UPDATE_PROGRESS)
 135     {
 136         ULONGLONG ullTotal = This->ullTotal;
 137         ULONGLONG ullCompleted = This->ullCompleted;
 138
 139         /* progress bar requires 32-bit coordinates */
 140         while (ullTotal >> 32)
 141         {
 142             ullTotal >>= 1;
 143             ullCompleted >>= 1;
 144         }
 145
 146         SendDlgItemMessageW(This->hwnd, IDC_PROGRESS_BAR, PBM_SETRANGE32, 0, (DWORD)ullTotal);
 147         SendDlgItemMessageW(This->hwnd, IDC_PROGRESS_BAR, PBM_SETPOS, (DWORD)ullCompleted, 0);
 148     }
 149 }
 150
 151 static void end_dialog(ProgressDialog *This)
 152 {
 153     SendMessageW(This->hwnd, WM_DLG_DESTROY, 0, 0);
 154     /* native doesn't reenable the window? */
 155     if (This->hwndDisabledParent)
 156         EnableWindow(This->hwndDisabledParent, TRUE);
 157     This->hwnd = NULL;
 158 }
 159
 160 static INT_PTR CALLBACK dialog_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
 161 {
 162     ProgressDialog *This = (ProgressDialog *)GetWindowLongPtrW(hwnd, DWLP_USER);
 163
 164     switch (msg)
 165     {
 166         case WM_INITDIALOG:
 167         {
 168             struct create_params *params = (struct create_params *)lParam;
 169
 170             /* Note: until we set the hEvent, the object is protected by
 171              * the critical section held by StartProgress */
 172             SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)params->This);
 173             This = params->This;
 174             This->hwnd = hwnd;
 175
 176             if (This->dwFlags & PROGDLG_NOPROGRESSBAR)
 177                 ShowWindow(GetDlgItem(hwnd, IDC_PROGRESS_BAR), SW_HIDE);
 178             if (This->dwFlags & PROGDLG_NOCANCEL)
 179                 ShowWindow(GetDlgItem(hwnd, IDCANCEL), SW_HIDE);
 180             if (This->dwFlags & PROGDLG_MARQUEEPROGRESS)
 181                 set_progress_marquee(This);
 182
 183             update_dialog(This, 0xffffffff);
 184             This->dwUpdate = 0;
 185             This->isCancelled = FALSE;
 186             SetEvent(params->hEvent);
 187             return TRUE;
 188         }
 189
 190         case WM_DLG_UPDATE:
 191             EnterCriticalSection(&This->cs);
 192             update_dialog(This, This->dwUpdate);
 193             This->dwUpdate = 0;
 194             LeaveCriticalSection(&This->cs);
 195             return TRUE;
 196
 197         case WM_DLG_DESTROY:
 198             DestroyWindow(hwnd);
 199             PostThreadMessageW(GetCurrentThreadId(), WM_NULL, 0, 0); /* wake up the GetMessage */
 200             return TRUE;
 201
 202         case WM_CLOSE:
 203         case WM_COMMAND:
 204             if (msg == WM_CLOSE || wParam == IDCANCEL)
 205             {
 206                 EnterCriticalSection(&This->cs);
 207                 This->isCancelled = TRUE;
 208
 209                 if (!This->cancelMsg)
 210                     This->cancelMsg = load_string(BROWSEUI_hinstance, IDS_CANCELLING);
 211
 212                 set_progress_marquee(This);
 213                 EnableWindow(GetDlgItem(This->hwnd, IDCANCEL), FALSE);
 214                 update_dialog(This, UPDATE_LINE1|UPDATE_LINE2|UPDATE_LINE3);
 215                 LeaveCriticalSection(&This->cs);
 216             }
 217             return TRUE;
 218     }
 219     return FALSE;
 220 }
 221
 222 static DWORD WINAPI dialog_thread(LPVOID lpParameter)
 223 {
 224     /* Note: until we set the hEvent in WM_INITDIALOG, the ProgressDialog object
 225      * is protected by the critical section held by StartProgress */
 226     struct create_params *params = (struct create_params *)lpParameter;
 227     HWND hwnd;
 228     MSG msg;
 229
 230     hwnd = CreateDialogParamW(BROWSEUI_hinstance, MAKEINTRESOURCEW(IDD_PROGRESS_DLG),
 231         params->hwndParent, dialog_proc, (LPARAM)params);
 232
 233     while (GetMessageW(&msg, NULL, 0, 0) > 0)
 234     {
 235         if (!IsWindow(hwnd))
 236             break;
 237         if(!IsDialogMessageW(hwnd, &msg))
 238         {
 239             TranslateMessage(&msg);
 240             DispatchMessageW(&msg);
 241         }
 242     }
 243
 244     return 0;
 245 }
 246
 247 HRESULT WINAPI ProgressDialog_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
 248 {
 249     ProgressDialog *This;
 250     if (pUnkOuter)
 251         return CLASS_E_NOAGGREGATION;
 252
 253     This = CoTaskMemAlloc(sizeof(ProgressDialog));
 254     if (This == NULL)
 255         return E_OUTOFMEMORY;
 256     ZeroMemory(This, sizeof(*This));
 257     This->vtbl = &ProgressDialogVtbl;
 258     This->refCount = 1;
 259     InitializeCriticalSection(&This->cs);
 260
 261     TRACE("returning %p\n", This);
 262     *ppOut = (IUnknown *)This;
 263     BROWSEUI_refCount++;
 264     return S_OK;
 265 }
 266
 267 static void WINAPI ProgressDialog_Destructor(ProgressDialog *This)
 268 {
 269     TRACE("destroying %p\n", This);
 270     if (This->hwnd)
 271         end_dialog(This);
 272     CoTaskMemFree(This->lines[0]);
 273     CoTaskMemFree(This->lines[1]);
 274     CoTaskMemFree(This->lines[2]);
 275     CoTaskMemFree(This->cancelMsg);
 276     CoTaskMemFree(This->title);
 277     CoTaskMemFree(This);
 278     BROWSEUI_refCount--;
 279 }
 280
 281 static HRESULT WINAPI ProgressDialog_QueryInterface(IProgressDialog *iface, REFIID iid, LPVOID *ppvOut)
 282 {
 283     ProgressDialog *This = (ProgressDialog *)iface;
 284     *ppvOut = NULL;
 285
 286     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_IProgressDialog))
 287     {
 288         *ppvOut = This;
 289     }
 290
 291     if (*ppvOut)
 292     {
 293         IUnknown_AddRef(iface);
 294         return S_OK;
 295     }
 296
 297     WARN("unsupported interface: %s\n", debugstr_guid(iid));
 298     return E_NOINTERFACE;
 299 }
 300
 301 static ULONG WINAPI ProgressDialog_AddRef(IProgressDialog *iface)
 302 {
 303     ProgressDialog *This = (ProgressDialog *)iface;
 304     return InterlockedIncrement(&This->refCount);
 305 }
 306
 307 static ULONG WINAPI ProgressDialog_Release(IProgressDialog *iface)
 308 {
 309     ProgressDialog *This = (ProgressDialog *)iface;
 310     ULONG ret;
 311
 312     ret = InterlockedDecrement(&This->refCount);
 313     if (ret == 0)
 314         ProgressDialog_Destructor(This);
 315     return ret;
 316 }
 317
 318 static HRESULT WINAPI ProgressDialog_StartProgressDialog(IProgressDialog *iface, HWND hwndParent, IUnknown *punkEnableModeless, DWORD dwFlags, LPCVOID reserved)
 319 {
 320     ProgressDialog *This = (ProgressDialog *)iface;
 321     struct create_params params;
 322     HANDLE hThread;
 323
 324     TRACE("(%p, %p, %x, %p)\n", iface, punkEnableModeless, dwFlags, reserved);
 325     if (punkEnableModeless || reserved)
 326         FIXME("Reserved parameters not null (%p, %p)\n", punkEnableModeless, reserved);
 327     if (dwFlags & PROGDLG_AUTOTIME)
 328         FIXME("Flags PROGDLG_AUTOTIME not supported\n");
 329     if (dwFlags & PROGDLG_NOTIME)
 330         FIXME("Flags PROGDLG_NOTIME not supported\n");
 331     if (dwFlags & PROGDLG_NOMINIMIZE)
 332         FIXME("Flags PROGDLG_NOMINIMIZE not supported\n");
 333
 334     EnterCriticalSection(&This->cs);
 335
 336     if (This->hwnd)
 337     {
 338         LeaveCriticalSection(&This->cs);
 339         return S_OK;  /* as on XP */
 340     }
 341     This->dwFlags = dwFlags;
 342     params.This = This;
 343     params.hwndParent = hwndParent;
 344     params.hEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
 345
 346     hThread = CreateThread(NULL, 0, dialog_thread, &params, 0, NULL);
 347     WaitForSingleObject(params.hEvent, INFINITE);
 348
 349     This->hwndDisabledParent = NULL;
 350     if (hwndParent && (dwFlags & PROGDLG_MODAL))
 351     {
 352         HWND hwndDisable = GetAncestor(hwndParent, GA_ROOT);
 353         if (EnableWindow(hwndDisable, FALSE))
 354             This->hwndDisabledParent = hwndDisable;
 355     }
 356
 357     LeaveCriticalSection(&This->cs);
 358
 359     return S_OK;
 360 }
 361
 362 static HRESULT WINAPI ProgressDialog_StopProgressDialog(IProgressDialog *iface)
 363 {
 364     ProgressDialog *This = (ProgressDialog *)iface;
 365
 366     EnterCriticalSection(&This->cs);
 367     if (This->hwnd)
 368         end_dialog(This);
 369     LeaveCriticalSection(&This->cs);
 370
 371     return S_OK;
 372 }
 373
 374 static HRESULT WINAPI ProgressDialog_SetTitle(IProgressDialog *iface, LPCWSTR pwzTitle)
 375 {
 376     ProgressDialog *This = (ProgressDialog *)iface;
 377     HWND hwnd;
 378
 379     TRACE("(%p, %s)\n", This, wine_dbgstr_w(pwzTitle));
 380
 381     EnterCriticalSection(&This->cs);
 382     set_buffer(&This->title, pwzTitle);
 383     This->dwUpdate |= UPDATE_TITLE;
 384     hwnd = This->hwnd;
 385     LeaveCriticalSection(&This->cs);
 386
 387     if (hwnd)
 388         SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 389
 390     return S_OK;
 391 }
 392
 393 static HRESULT WINAPI ProgressDialog_SetAnimation(IProgressDialog *iface, HINSTANCE hInstance, UINT uiResourceId)
 394 {
 395     FIXME("(%p, %p, %d) - stub\n", iface, hInstance, uiResourceId);
 396     return S_OK;
 397 }
 398
 399 static BOOL WINAPI ProgressDialog_HasUserCancelled(IProgressDialog *iface)
 400 {
 401     ProgressDialog *This = (ProgressDialog *)iface;
 402     return This->isCancelled;
 403 }
 404
 405 static HRESULT WINAPI ProgressDialog_SetProgress64(IProgressDialog *iface, ULONGLONG ullCompleted, ULONGLONG ullTotal)
 406 {
 407     ProgressDialog *This = (ProgressDialog *)iface;
 408     HWND hwnd;
 409
 410     TRACE("(%p, 0x%s, 0x%s)\n", This, wine_dbgstr_longlong(ullCompleted), wine_dbgstr_longlong(ullTotal));
 411
 412     EnterCriticalSection(&This->cs);
 413     This->ullTotal = ullTotal;
 414     This->ullCompleted = ullCompleted;
 415     This->dwUpdate |= UPDATE_PROGRESS;
 416     hwnd = This->hwnd;
 417     LeaveCriticalSection(&This->cs);
 418
 419     if (hwnd)
 420         SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 421
 422     return S_OK;  /* Windows sometimes returns S_FALSE */
 423 }
 424
 425 static HRESULT WINAPI ProgressDialog_SetProgress(IProgressDialog *iface, DWORD dwCompleted, DWORD dwTotal)
 426 {
 427     return IProgressDialog_SetProgress64(iface, dwCompleted, dwTotal);
 428 }
 429
 430 static HRESULT WINAPI ProgressDialog_SetLine(IProgressDialog *iface, DWORD dwLineNum, LPCWSTR pwzLine, BOOL bPath, LPCVOID reserved)
 431 {
 432     ProgressDialog *This = (ProgressDialog *)iface;
 433     HWND hwnd;
 434
 435     TRACE("(%p, %d, %s, %d)\n", This, dwLineNum, wine_dbgstr_w(pwzLine), bPath);
 436
 437     if (reserved)
 438         FIXME("reserved pointer not null (%p)\n", reserved);
 439
 440     dwLineNum--;
 441     if (dwLineNum >= 3)  /* Windows seems to do something like that */
 442         dwLineNum = 0;
 443
 444     EnterCriticalSection(&This->cs);
 445     set_buffer(&This->lines[dwLineNum], pwzLine);
 446     This->dwUpdate |= UPDATE_LINE1 << dwLineNum;
 447     hwnd = (This->isCancelled ? NULL : This->hwnd); /* no sense to send the message if window cancelled */
 448     LeaveCriticalSection(&This->cs);
 449
 450     if (hwnd)
 451         SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 452
 453     return S_OK;
 454 }
 455
 456 static HRESULT WINAPI ProgressDialog_SetCancelMsg(IProgressDialog *iface, LPCWSTR pwzMsg, LPCVOID reserved)
 457 {
 458     ProgressDialog *This = (ProgressDialog *)iface;
 459     HWND hwnd;
 460
 461     TRACE("(%p, %s)\n", This, wine_dbgstr_w(pwzMsg));
 462
 463     if (reserved)
 464         FIXME("reserved pointer not null (%p)\n", reserved);
 465
 466     EnterCriticalSection(&This->cs);
 467     set_buffer(&This->cancelMsg, pwzMsg);
 468     This->dwUpdate |= UPDATE_LINE1 << CANCEL_MSG_LINE;
 469     hwnd = (This->isCancelled ? This->hwnd : NULL); /* no sense to send the message if window not cancelled */
 470     LeaveCriticalSection(&This->cs);
 471
 472     if (hwnd)
 473         SendMessageW(hwnd, WM_DLG_UPDATE, 0, 0);
 474
 475     return S_OK;
 476 }
 477
 478 static HRESULT WINAPI ProgressDialog_Timer(IProgressDialog *iface, DWORD dwTimerAction, LPCVOID reserved)
 479 {
 480     ProgressDialog *This = (ProgressDialog *)iface;
 481
 482     FIXME("(%p, %d, %p) - stub\n", This, dwTimerAction, reserved);
 483
 484     if (reserved)
 485         FIXME("Reserved field not NULL but %p\n", reserved);
 486
 487     return S_OK;
 488 }
 489
 490 static const IProgressDialogVtbl ProgressDialogVtbl =
 491 {
 492     ProgressDialog_QueryInterface,
 493     ProgressDialog_AddRef,
 494     ProgressDialog_Release,
 495
 496     ProgressDialog_StartProgressDialog,
 497     ProgressDialog_StopProgressDialog,
 498     ProgressDialog_SetTitle,
 499     ProgressDialog_SetAnimation,
 500     ProgressDialog_HasUserCancelled,
 501     ProgressDialog_SetProgress,
 502     ProgressDialog_SetProgress64,
 503     ProgressDialog_SetLine,
 504     ProgressDialog_SetCancelMsg,
 505     ProgressDialog_Timer
 506 };