KasmVNC/win/rfb_win32/Dialog.cxx

386 lines
10 KiB
C++
Raw Normal View History

2020-09-20 14:16:44 +02:00
/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
* Copyright (C) 2010 D. R. Commander. All Rights Reserved.
*
* This is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this software; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
* USA.
*/
// -=- Dialog.cxx
// Base-class for any Dialog classes we might require
#include <rfb_win32/Dialog.h>
#include <rfb_win32/TCharArray.h>
#include <rfb/LogWriter.h>
#include <rdr/Exception.h>
#include <rfb_win32/Win32Util.h>
#ifdef _DIALOG_CAPTURE
#ifdef PropSheet_IndexToId
#include <rfb_win32/DeviceFrameBuffer.h>
#include <extra/LoadBMP.cxx>
#else
#undef _DIALOG_CAPTURE
#pragma message(" NOTE: Not building Dialog Capture support.")
#endif
#endif
using namespace rfb;
using namespace rfb::win32;
static LogWriter dlog("Dialog");
static LogWriter plog("PropSheet");
Dialog::Dialog(HINSTANCE inst_)
: inst(inst_), handle(0), alreadyShowing(false)
{
}
Dialog::~Dialog()
{
}
bool Dialog::showDialog(const TCHAR* resource, HWND owner)
{
if (alreadyShowing) return false;
handle = 0;
alreadyShowing = true;
INT_PTR result = DialogBoxParam(inst, resource, owner,
staticDialogProc, (LPARAM)this);
if (result<0)
throw rdr::SystemException("DialogBoxParam failed", GetLastError());
alreadyShowing = false;
return (result == 1);
}
bool Dialog::isItemChecked(int id) {
return SendMessage(GetDlgItem(handle, id), BM_GETCHECK, 0, 0) == BST_CHECKED;
}
int Dialog::getItemInt(int id) {
BOOL trans;
int result = GetDlgItemInt(handle, id, &trans, TRUE);
if (!trans)
throw rdr::Exception("unable to read dialog Int");
return result;
}
TCHAR* Dialog::getItemString(int id) {
TCharArray tmp(256);
if (!GetDlgItemText(handle, id, tmp.buf, 256))
tmp.buf[0] = 0;
return tmp.takeBuf();
}
void Dialog::setItemChecked(int id, bool state) {
SendMessage(GetDlgItem(handle, id), BM_SETCHECK, state ? BST_CHECKED : BST_UNCHECKED, 0);
}
void Dialog::setItemInt(int id, int value) {
SetDlgItemInt(handle, id, value, TRUE);
}
void Dialog::setItemString(int id, const TCHAR* s) {
SetDlgItemText(handle, id, s);
}
void Dialog::enableItem(int id, bool state) {
EnableWindow(GetDlgItem(handle, id), state);
}
INT_PTR CALLBACK Dialog::staticDialogProc(HWND hwnd, UINT msg,
WPARAM wParam, LPARAM lParam)
{
if (msg == WM_INITDIALOG)
SetWindowLongPtr(hwnd, GWLP_USERDATA, lParam);
LONG_PTR self = GetWindowLongPtr(hwnd, GWLP_USERDATA);
if (!self) return FALSE;
return ((Dialog*)self)->dialogProc(hwnd, msg, wParam, lParam);
}
BOOL Dialog::dialogProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
switch (msg) {
case WM_INITDIALOG:
handle = hwnd;
initDialog();
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam)) {
case IDOK:
if (onOk()) {
EndDialog(hwnd, 1);
return TRUE;
}
return FALSE;
case IDCANCEL:
EndDialog(hwnd, 0);
return TRUE;
default:
return onCommand(LOWORD(wParam), HIWORD(wParam));
};
case WM_HELP:
return onHelp(((HELPINFO*)lParam)->iCtrlId);
}
return FALSE;
}
PropSheetPage::PropSheetPage(HINSTANCE inst, const TCHAR* id) : Dialog(inst), propSheet(0) {
page.dwSize = sizeof(page);
page.dwFlags = 0; // PSP_USECALLBACK;
page.hInstance = inst;
page.pszTemplate = id;
page.pfnDlgProc = staticPageProc;
page.lParam = (LPARAM)this;
page.pfnCallback = 0; // staticPageProc;
}
PropSheetPage::~PropSheetPage() {
}
INT_PTR CALLBACK PropSheetPage::staticPageProc(HWND hwnd, UINT msg,
WPARAM wParam, LPARAM lParam)
{
if (msg == WM_INITDIALOG)
SetWindowLongPtr(hwnd, GWLP_USERDATA, ((PROPSHEETPAGE*)lParam)->lParam);
LONG_PTR self = GetWindowLongPtr(hwnd, GWLP_USERDATA);
if (!self) return FALSE;
return ((PropSheetPage*)self)->dialogProc(hwnd, msg, wParam, lParam);
}
BOOL PropSheetPage::dialogProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
switch (msg) {
case WM_INITDIALOG:
handle = hwnd;
initDialog();
return TRUE;
case WM_NOTIFY:
switch (((NMHDR*)lParam)->code) {
case PSN_APPLY:
onOk();
return FALSE;
};
return FALSE;
case WM_COMMAND:
return onCommand(LOWORD(wParam), HIWORD(wParam));
case WM_HELP:
return onHelp(((HELPINFO*)lParam)->iCtrlId);
}
return FALSE;
}
PropSheet::PropSheet(HINSTANCE inst_, const TCHAR* title_, std::list<PropSheetPage*> pages_, HICON icon_)
: icon(icon_), pages(pages_), inst(inst_), title(tstrDup(title_)), handle(0), alreadyShowing(0) {
}
PropSheet::~PropSheet() {
}
// For some reason, DLGTEMPLATEEX isn't defined in the Windows headers - go figure...
struct DLGTEMPLATEEX {
WORD dlgVer;
WORD signature;
DWORD helpID;
DWORD exStyle;
DWORD style;
WORD cDlgItems;
short x;
short y;
short cx;
short cy;
};
static int CALLBACK removeCtxtHelp(HWND hwnd, UINT message, LPARAM lParam) {
if (message == PSCB_PRECREATE) {
// Remove the context-help style, to remove the titlebar ? button
// *** Nasty hack to cope with new & old dialog template formats...
if (((DLGTEMPLATEEX*)lParam)->signature == 0xffff)
((DLGTEMPLATEEX*)lParam)->style &= ~DS_CONTEXTHELP;
else
((LPDLGTEMPLATE)lParam)->style &= ~DS_CONTEXTHELP;
}
return TRUE;
}
bool PropSheet::showPropSheet(HWND owner, bool showApply, bool showCtxtHelp, bool capture) {
if (alreadyShowing) return false;
alreadyShowing = true;
int count = pages.size();
HPROPSHEETPAGE* hpages = new HPROPSHEETPAGE[count];
try {
// Create the PropertSheet page GDI objects.
std::list<PropSheetPage*>::iterator pspi;
int i = 0;
for (pspi=pages.begin(); pspi!=pages.end(); pspi++) {
hpages[i] = CreatePropertySheetPage(&((*pspi)->page));
(*pspi)->setPropSheet(this);
i++;
}
// Initialise and create the PropertySheet itself
PROPSHEETHEADER header;
header.dwSize = sizeof(PROPSHEETHEADER); // Requires comctl32.dll 4.71 or greater, ie IE 4 or later
header.dwFlags = PSH_MODELESS | (showApply ? 0 : PSH_NOAPPLYNOW) | (showCtxtHelp ? 0 : PSH_USECALLBACK);
header.pfnCallback = removeCtxtHelp;
header.hwndParent = owner;
header.hInstance = inst;
header.pszCaption = title.buf;
header.nPages = count;
header.nStartPage = 0;
header.phpage = hpages;
if (icon) {
header.hIcon = icon;
header.dwFlags |= PSH_USEHICON;
}
handle = (HWND)PropertySheet(&header);
if ((handle == 0) || (handle == (HWND)-1))
throw rdr::SystemException("PropertySheet failed", GetLastError());
centerWindow(handle, owner);
plog.info("created %p", handle);
#ifdef _DIALOG_CAPTURE
if (capture) {
plog.info("capturing \"%s\"", (const char*)CStr(title.buf));
char* tmpdir = getenv("TEMP");
HDC dc = GetWindowDC(handle);
DeviceFrameBuffer fb(dc);
int i=0;
while (true) {
int id = PropSheet_IndexToId(handle, i);
if (!id) break;
PropSheet_SetCurSelByID(handle, id);
MSG msg;
while (PeekMessage(&msg, handle, 0, 0, PM_REMOVE)) {
if (!PropSheet_IsDialogMessage(handle, &msg))
DispatchMessage(&msg);
}
fb.grabRect(fb.getRect());
TCHAR title[128];
if (!GetWindowText(PropSheet_GetCurrentPageHwnd(handle), title, sizeof(title)))
_stprintf(title, _T("capture%d"), i);
CharArray pageTitle(strDup(title));
for (int j=0; j<strlen(pageTitle.buf); j++) {
if (pageTitle.buf[j] == '/' || pageTitle.buf[j] == '\\' || pageTitle.buf[j] == ':')
pageTitle.buf[j] = '-';
}
char filename[256];
sprintf(filename, "%s\\%s.bmp", tmpdir, pageTitle.buf);
vlog.debug("writing to %s", filename);
saveBMP(filename, &fb);
i++;
}
ReleaseDC(handle, dc);
} else {
#endif
try {
if (owner)
EnableWindow(owner, FALSE);
// Run the PropertySheet
MSG msg;
while (GetMessage(&msg, 0, 0, 0)) {
if (!PropSheet_IsDialogMessage(handle, &msg))
DispatchMessage(&msg);
if (!PropSheet_GetCurrentPageHwnd(handle))
break;
}
if (owner)
EnableWindow(owner, TRUE);
} catch (...) {
if (owner)
EnableWindow(owner, TRUE);
throw;
}
#ifdef _DIALOG_CAPTURE
}
#endif
plog.info("finished %p", handle);
DestroyWindow(handle);
handle = 0;
alreadyShowing = false;
// Clear up the pages' GDI objects
for (pspi=pages.begin(); pspi!=pages.end(); pspi++)
(*pspi)->setPropSheet(0);
delete [] hpages; hpages = 0;
return true;
} catch (rdr::Exception&) {
alreadyShowing = false;
std::list<PropSheetPage*>::iterator pspi;
for (pspi=pages.begin(); pspi!=pages.end(); pspi++)
(*pspi)->setPropSheet(0);
delete [] hpages; hpages = 0;
throw;
}
}
void PropSheet::reInitPages() {
std::list<PropSheetPage*>::iterator pspi;
for (pspi=pages.begin(); pspi!=pages.end(); pspi++) {
if ((*pspi)->handle)
(*pspi)->initDialog();
}
}
bool PropSheet::commitPages() {
bool result = true;
std::list<PropSheetPage*>::iterator pspi;
for (pspi=pages.begin(); pspi!=pages.end(); pspi++) {
if ((*pspi)->handle)
result = result && (*pspi)->onOk();
}
return result;
}
void PropSheetPage::setChanged(bool changed) {
if (propSheet) {
if (changed)
PropSheet_Changed(propSheet->handle, handle);
else
PropSheet_UnChanged(propSheet->handle, handle);
}
}