The Tor Browser / file revision

 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

 /* vim:set ts=2 sw=2 sts=2 et cindent: */

 /* This Source Code Form is subject to the terms of the Mozilla Public

  * License, v. 2.0. If a copy of the MPL was not distributed with this

  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

 #include <stdio.h>

 #include <windows.h>

 #include <commctrl.h>

 #include <process.h>

 #include <io.h>

 #include "resource.h"

 #include "progressui.h"

 #include "readstrings.h"

 #include "errors.h"

 #define TIMER_ID 1

 #define TIMER_INTERVAL 100

 #define RESIZE_WINDOW(hwnd, extrax, extray) \

   { \

     RECT windowSize; \

     GetWindowRect(hwnd, &windowSize); \

     SetWindowPos(hwnd, 0, 0, 0, windowSize.right - windowSize.left + extrax, \

                  windowSize.bottom - windowSize.top + extray, \

                  SWP_NOMOVE | SWP_NOZORDER); \

   }

 #define MOVE_WINDOW(hwnd, dx, dy) \

   { \

     RECT rc; \

     POINT pt; \

     GetWindowRect(hwnd, &rc); \

     pt.x = rc.left; \

     pt.y = rc.top; \

     ScreenToClient(GetParent(hwnd), &pt); \

     SetWindowPos(hwnd, 0, pt.x + dx, pt.y + dy, 0, 0, \

                  SWP_NOSIZE | SWP_NOZORDER); \

   }

 static float sProgress;  // between 0 and 100

 static BOOL  sQuit = FALSE;

 static BOOL sIndeterminate = FALSE;

 static StringTable sUIStrings;

 static BOOL

 GetStringsFile(WCHAR filename[MAX_PATH])

 {

   if (!GetModuleFileNameW(nullptr, filename, MAX_PATH))

     return FALSE;

   WCHAR *dot = wcsrchr(filename, '.');

   if (!dot || wcsicmp(dot + 1, L"exe"))

     return FALSE;

   wcscpy(dot + 1, L"ini");

   return TRUE;

 }

 static void

 UpdateDialog(HWND hDlg)

 {

   int pos = int(sProgress + 0.5f);

   HWND hWndPro = GetDlgItem(hDlg, IDC_PROGRESS);

   SendMessage(hWndPro, PBM_SETPOS, pos, 0L);

 }

 // The code in this function is from MSDN:

 // http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winui/winui/windowsuserinterface/windowing/dialogboxes/usingdialogboxes.asp

 static void

 CenterDialog(HWND hDlg)

 {

   RECT rc, rcOwner, rcDlg;

   // Get the owner window and dialog box rectangles. 

   HWND desktop = GetDesktopWindow();

   GetWindowRect(desktop, &rcOwner); 

   GetWindowRect(hDlg, &rcDlg); 

   CopyRect(&rc, &rcOwner); 

   // Offset the owner and dialog box rectangles so that 

   // right and bottom values represent the width and 

   // height, and then offset the owner again to discard 

   // space taken up by the dialog box. 

   OffsetRect(&rcDlg, -rcDlg.left, -rcDlg.top); 

   OffsetRect(&rc, -rc.left, -rc.top); 

   OffsetRect(&rc, -rcDlg.right, -rcDlg.bottom); 

   // The new position is the sum of half the remaining 

   // space and the owner's original position. 

   SetWindowPos(hDlg, 

                HWND_TOP, 

                rcOwner.left + (rc.right / 2), 

                rcOwner.top + (rc.bottom / 2), 

                0, 0,          // ignores size arguments 

                SWP_NOSIZE); 

 }

 static void

 InitDialog(HWND hDlg)

 {

   WCHAR szwTitle[MAX_TEXT_LEN];

   WCHAR szwInfo[MAX_TEXT_LEN];

   MultiByteToWideChar(CP_UTF8, 0, sUIStrings.title, -1, szwTitle,

                       sizeof(szwTitle)/sizeof(szwTitle[0]));

   MultiByteToWideChar(CP_UTF8, 0, sUIStrings.info, -1, szwInfo,

                       sizeof(szwInfo)/sizeof(szwInfo[0]));

   SetWindowTextW(hDlg, szwTitle);

   SetWindowTextW(GetDlgItem(hDlg, IDC_INFO), szwInfo);

   // Set dialog icon

   HICON hIcon = LoadIcon(GetModuleHandle(nullptr),

                          MAKEINTRESOURCE(IDI_DIALOG));

   if (hIcon)

     SendMessage(hDlg, WM_SETICON, ICON_BIG, (LPARAM) hIcon);

   HWND hWndPro = GetDlgItem(hDlg, IDC_PROGRESS);

   SendMessage(hWndPro, PBM_SETRANGE, 0, MAKELPARAM(0, 100));

   if (sIndeterminate) {

     LONG_PTR val = GetWindowLongPtr(hWndPro, GWL_STYLE);

     SetWindowLongPtr(hWndPro, GWL_STYLE, val|PBS_MARQUEE); 

     SendMessage(hWndPro,(UINT) PBM_SETMARQUEE,(WPARAM) TRUE,(LPARAM)50 );

   }

   // Resize the dialog to fit all of the text if necessary.

   RECT infoSize, textSize;

   HWND hWndInfo = GetDlgItem(hDlg, IDC_INFO);

   // Get the control's font for calculating the new size for the control

   HDC hDCInfo = GetDC(hWndInfo);

   HFONT hInfoFont, hOldFont;

   hInfoFont = (HFONT)SendMessage(hWndInfo, WM_GETFONT, 0, 0);

   if (hInfoFont)

     hOldFont = (HFONT)SelectObject(hDCInfo, hInfoFont);

   // Measure the space needed for the text on a single line. DT_CALCRECT means

   // nothing is drawn.

   if (DrawText(hDCInfo, szwInfo, -1, &textSize,

                DT_CALCRECT | DT_NOCLIP | DT_SINGLELINE)) {

     GetClientRect(hWndInfo, &infoSize);

     SIZE extra;

     // Calculate the additional space needed for the text by subtracting from

     // the rectangle returned by DrawText the existing client rectangle's width

     // and height.

     extra.cx = (textSize.right - textSize.left) - \

                (infoSize.right - infoSize.left);

     extra.cy = (textSize.bottom - textSize.top) - \

                (infoSize.bottom - infoSize.top);

     if (extra.cx < 0)

       extra.cx = 0;

     if (extra.cy < 0)

       extra.cy = 0;

     if ((extra.cx > 0) || (extra.cy > 0)) {

       RESIZE_WINDOW(hDlg, extra.cx, extra.cy);

       RESIZE_WINDOW(hWndInfo, extra.cx, extra.cy);

       RESIZE_WINDOW(hWndPro, extra.cx, 0);

       MOVE_WINDOW(hWndPro, 0, extra.cy);

     }

   }

   if (hOldFont)

     SelectObject(hDCInfo, hOldFont);

   ReleaseDC(hWndInfo, hDCInfo);

   CenterDialog(hDlg);  // make dialog appear in the center of the screen

   SetTimer(hDlg, TIMER_ID, TIMER_INTERVAL, nullptr);

 }

 // Message handler for update dialog.

 static LRESULT CALLBACK

 DialogProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)

 {

   switch (message)

   {

   case WM_INITDIALOG:

     InitDialog(hDlg);

     return TRUE;

   case WM_TIMER:

     if (sQuit) {

       EndDialog(hDlg, 0);

     } else {

       UpdateDialog(hDlg);

     }

     return TRUE;

   case WM_COMMAND:

     return TRUE;

   }

   return FALSE;

 }

 int

 InitProgressUI(int *argc, NS_tchar ***argv)

 {

   return 0;

 }

 /**

  * Initializes the progress UI strings

  * 

  * @return 0 on success, -1 on error

 */

 int

 InitProgressUIStrings() {

   // If we do not have updater.ini, then we should not bother showing UI.

   WCHAR filename[MAX_PATH];

   if (!GetStringsFile(filename)) {

     return -1;

   }

   if (_waccess(filename, 04)) {

     return -1;

   }

   // If the updater.ini doesn't have the required strings, then we should not

   // bother showing UI.

   if (ReadStrings(filename, &sUIStrings) != OK) {

     return -1;

   }

   return 0;

 }

 int

 ShowProgressUI(bool indeterminate, bool initUIStrings)

 {

   sIndeterminate = indeterminate;

   if (!indeterminate) {

     // Only show the Progress UI if the process is taking a significant amount of

     // time where a significant amount of time is defined as .5 seconds after

     // ShowProgressUI is called sProgress is less than 70.

     Sleep(500);

     if (sQuit || sProgress > 70.0f)

       return 0;

   }

   if (initUIStrings && InitProgressUIStrings() == -1) {

     return -1;

   }

   INITCOMMONCONTROLSEX icc = {

     sizeof(INITCOMMONCONTROLSEX),

     ICC_PROGRESS_CLASS

   };

   InitCommonControlsEx(&icc);

   DialogBox(GetModuleHandle(nullptr),

             MAKEINTRESOURCE(IDD_DIALOG), nullptr,

             (DLGPROC) DialogProc);

   return 0;

 }

 void

 QuitProgressUI()

 {

   sQuit = TRUE;

 }

 void

 UpdateProgressUI(float progress)

 {

   sProgress = progress;  // 32-bit writes are atomic

 }