HRESULT CALLBACK TaskDialogErrorProc(
_In_ HWND hwndDlg,
_In_ UINT uMsg,
_In_ WPARAM wParam,
_In_ LPARAM lParam,
_In_ LONG_PTR dwRefData
)
{
PUPLOAD_CONTEXT context = (PUPLOAD_CONTEXT)dwRefData;
switch (uMsg)
{
case TDN_NAVIGATED:
{
if (context->TaskbarListClass)
{
ITaskbarList3_SetProgressValue(context->TaskbarListClass, PhMainWndHandle, 1, 1);
ITaskbarList3_SetProgressState(context->TaskbarListClass, PhMainWndHandle, TBPF_ERROR);
}
}
break;
}
return S_OK;
}
HRESULT CALLBACK TaskDialogResultFoundProc(
_In_ HWND hwndDlg,
_In_ UINT uMsg,
_In_ WPARAM wParam,
_In_ LPARAM lParam,
_In_ LONG_PTR dwRefData
)
{
PUPLOAD_CONTEXT context = (PUPLOAD_CONTEXT)dwRefData;
switch (uMsg)
{
case TDN_NAVIGATED:
{
if (context->TaskbarListClass)
{
ITaskbarList3_SetProgressState(context->TaskbarListClass, PhMainWndHandle, TBPF_NOPROGRESS);
}
}
break;
case TDN_BUTTON_CLICKED:
{
INT buttonID = (INT)wParam;
if (buttonID == IDOK)
{
ShowVirusTotalProgressDialog(context);
return S_FALSE;
}
else if (buttonID == IDRETRY)
{
if (!PhIsNullOrEmptyString(context->ReAnalyseUrl))
PhShellExecute(hwndDlg, PhGetString(context->ReAnalyseUrl), NULL);
}
else if (buttonID == IDYES)
{
if (!PhIsNullOrEmptyString(context->LaunchCommand))
{
PhShellExecute(hwndDlg, PhGetString(context->LaunchCommand), NULL);
}
}
}
break;
case TDN_VERIFICATION_CLICKED:
{
BOOL verification = (BOOL)wParam;
}
break;
}
return S_OK;
}
static bool win32_set_window_progress(void *data, int progress, bool finished)
{
HWND hwnd = win32_get_window();
dispserv_win32_t *serv = (dispserv_win32_t*)data;
if (serv)
serv->progress = progress;
#ifdef HAS_TASKBAR_EXT
if (!g_taskbarList || !win32_taskbar_is_created())
return false;
if (progress == -1)
{
if (ITaskbarList3_SetProgressState(
g_taskbarList, hwnd, TBPF_INDETERMINATE) != S_OK)
return false;
}
else if (finished)
{
if (ITaskbarList3_SetProgressState(
g_taskbarList, hwnd, TBPF_NOPROGRESS) != S_OK)
return false;
}
else if (progress >= 0)
{
if (ITaskbarList3_SetProgressState(
g_taskbarList, hwnd, TBPF_NORMAL) != S_OK)
return false;
if (ITaskbarList3_SetProgressValue(
g_taskbarList, hwnd, progress, 100) != S_OK)
return false;
}
#endif
return true;
}
HRESULT CALLBACK TaskDialogProgressCallbackProc(
_In_ HWND hwndDlg,
_In_ UINT uMsg,
_In_ WPARAM wParam,
_In_ LPARAM lParam,
_In_ LONG_PTR dwRefData
)
{
PUPLOAD_CONTEXT context = (PUPLOAD_CONTEXT)dwRefData;
switch (uMsg)
{
case TDN_NAVIGATED:
{
SendMessage(hwndDlg, TDM_SET_MARQUEE_PROGRESS_BAR, TRUE, 0);
SendMessage(hwndDlg, TDM_SET_PROGRESS_BAR_MARQUEE, TRUE, 1);
if (context->TaskbarListClass)
ITaskbarList3_SetProgressState(context->TaskbarListClass, PhMainWndHandle, TBPF_INDETERMINATE);
PhReferenceObject(context);
context->UploadThreadHandle = PhCreateThread(0, UploadFileThreadStart, context);
}
break;
case TDN_BUTTON_CLICKED:
{
if ((INT)wParam == IDCANCEL)
{
if (context->UploadThreadHandle)
{
NtClose(context->UploadThreadHandle);
context->UploadThreadHandle = NULL;
}
}
}
break;
}
return S_OK;
}