ミラー元
https://github.com/TigerVNC/tigervnc.git
前回の同期 2024-08-21 05:58:01 +02:00
9fb4e0e6b9
* Changed description in README files. * Added our copyright. * The HTTP server presents itself as TightVNC, rather than RealVNC. * Using registry path Software\TightVNC\WinVNC4 * Updated descriptions in resource files. git-svn-id: svn://svn.code.sf.net/p/tigervnc/code/trunk@67 3789f03b-4d11-0410-bbf8-ca57d06f2519
371 行
10 KiB
C++
371 行
10 KiB
C++
/* Copyright (C) 2002-2004 RealVNC Ltd. 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.
|
|
*/
|
|
|
|
// -=- VNC Viewer for Win32
|
|
|
|
#include <string.h>
|
|
#ifdef WIN32
|
|
#define strcasecmp _stricmp
|
|
#endif
|
|
#include <list>
|
|
|
|
#include <vncviewer/resource.h>
|
|
#include <vncviewer/CViewManager.h>
|
|
#include <vncviewer/CView.h>
|
|
#include <vncviewer/OptionsDialog.h>
|
|
|
|
#include <rfb/Logger_stdio.h>
|
|
#include <rfb/Logger_file.h>
|
|
#include <rfb/LogWriter.h>
|
|
#include <rfb/Exception.h>
|
|
|
|
#include <rfb_win32/RegConfig.h>
|
|
#include <rfb_win32/TrayIcon.h>
|
|
#include <rfb_win32/Win32Util.h>
|
|
#include <rfb_win32/AboutDialog.h>
|
|
|
|
#include <network/TcpSocket.h>
|
|
|
|
#ifdef _DIALOG_CAPTURE
|
|
#include <extra/LoadBMP.h>
|
|
#endif
|
|
|
|
using namespace rfb;
|
|
using namespace rfb::win32;
|
|
using namespace rdr;
|
|
using namespace network;
|
|
|
|
static LogWriter vlog("main");
|
|
|
|
TStr rfb::win32::AppName("VNC Viewer");
|
|
|
|
|
|
#ifdef _DIALOG_CAPTURE
|
|
BoolParameter captureDialogs("CaptureDialogs", "", false);
|
|
#endif
|
|
|
|
//
|
|
// -=- Listener
|
|
// Class to handle listening on a particular port for incoming connections
|
|
// from servers, and spawning of clients
|
|
//
|
|
|
|
static BoolParameter acceptIncoming("Listen", "Accept incoming connections from VNC servers.", false);
|
|
|
|
|
|
//
|
|
// -=- AboutDialog global values
|
|
//
|
|
|
|
const WORD rfb::win32::AboutDialog::DialogId = IDD_ABOUT;
|
|
const WORD rfb::win32::AboutDialog::Copyright = IDC_COPYRIGHT;
|
|
const WORD rfb::win32::AboutDialog::Version = IDC_VERSION;
|
|
const WORD rfb::win32::AboutDialog::BuildTime = IDC_BUILDTIME;
|
|
const WORD rfb::win32::AboutDialog::Description = IDC_DESCRIPTION;
|
|
|
|
|
|
//
|
|
// -=- VNCviewer Tray Icon
|
|
//
|
|
|
|
class CViewTrayIcon : public TrayIcon {
|
|
public:
|
|
CViewTrayIcon(CViewManager& mgr) : manager(mgr) {
|
|
setIcon(IDI_ICON);
|
|
setToolTip(_T("VNC Viewer"));
|
|
}
|
|
virtual LRESULT processMessage(UINT msg, WPARAM wParam, LPARAM lParam) {
|
|
switch(msg) {
|
|
|
|
case WM_USER:
|
|
switch (lParam) {
|
|
case WM_LBUTTONDBLCLK:
|
|
SendMessage(getHandle(), WM_COMMAND, ID_NEW_CONNECTION, 0);
|
|
break;
|
|
case WM_RBUTTONUP:
|
|
HMENU menu = LoadMenu(GetModuleHandle(0), MAKEINTRESOURCE(IDR_TRAY));
|
|
HMENU trayMenu = GetSubMenu(menu, 0);
|
|
|
|
// First item is New Connection, the default
|
|
SetMenuDefaultItem(trayMenu, ID_NEW_CONNECTION, FALSE);
|
|
|
|
// SetForegroundWindow is required, otherwise Windows ignores the
|
|
// TrackPopupMenu because the window isn't the foreground one, on
|
|
// some older Windows versions...
|
|
SetForegroundWindow(getHandle());
|
|
|
|
// Display the menu
|
|
POINT pos;
|
|
GetCursorPos(&pos);
|
|
TrackPopupMenu(trayMenu, 0, pos.x, pos.y, 0, getHandle(), 0);
|
|
break;
|
|
}
|
|
return 0;
|
|
|
|
case WM_COMMAND:
|
|
switch (LOWORD(wParam)) {
|
|
case ID_NEW_CONNECTION:
|
|
manager.addClient(0);
|
|
break;
|
|
case ID_OPTIONS:
|
|
OptionsDialog::global.showDialog(0);
|
|
break;
|
|
case ID_ABOUT:
|
|
AboutDialog::instance.showDialog();
|
|
break;
|
|
case ID_CLOSE:
|
|
SendMessage(getHandle(), WM_CLOSE, 0, 0);
|
|
break;
|
|
}
|
|
return 0;
|
|
|
|
case WM_CLOSE:
|
|
PostQuitMessage(0);
|
|
return 0;
|
|
}
|
|
|
|
return TrayIcon::processMessage(msg, wParam, lParam);
|
|
}
|
|
protected:
|
|
CViewManager& manager;
|
|
};
|
|
|
|
//
|
|
// -=- processParams
|
|
// Read in the command-line parameters and interpret them.
|
|
//
|
|
|
|
void
|
|
programInfo() {
|
|
win32::FileVersionInfo inf;
|
|
_tprintf(_T("%s - %s, Version %s\n"),
|
|
inf.getVerString(_T("ProductName")),
|
|
inf.getVerString(_T("FileDescription")),
|
|
inf.getVerString(_T("FileVersion")));
|
|
printf("%s\n", buildTime);
|
|
_tprintf(_T("%s\n\n"), inf.getVerString(_T("LegalCopyright")));
|
|
}
|
|
|
|
void
|
|
programUsage() {
|
|
printf("usage: vncviewer <options> <hostname>[:<display>]\n");
|
|
printf("Command-line options:\n");
|
|
printf(" -help - Provide usage information.\n");
|
|
printf(" -config <file> - Load connection settings from VNCViewer 3.3 settings file\n");
|
|
printf(" -console - Run with a console window visible.\n");
|
|
printf(" <setting>=<value> - Set the named configuration parameter.\n");
|
|
printf(" (Parameter values specified on the command-line override those specified by other configuration methods.)\n");
|
|
printf("\nLog names:\n");
|
|
LogWriter::listLogWriters();
|
|
printf("\nLog destinations:\n");
|
|
Logger::listLoggers();
|
|
printf("\nParameters:\n");
|
|
Configuration::listParams();
|
|
}
|
|
|
|
|
|
bool print_usage = false;
|
|
bool close_console = true;
|
|
std::list<char*> hosts;
|
|
std::list<char*> configFiles;
|
|
|
|
void
|
|
processParams(int argc, char* argv[]) {
|
|
for (int i=1; i<argc; i++) {
|
|
try {
|
|
|
|
if (strcasecmp(argv[i], "-console") == 0) {
|
|
close_console = false;
|
|
|
|
} else if (((strcasecmp(argv[i], "-config") == 0) ||
|
|
(strcasecmp(argv[i], "/config") == 0)) && (i < argc-1)) {
|
|
configFiles.push_back(strDup(argv[i+1]));
|
|
i++;
|
|
|
|
} else if ((strcasecmp(argv[i], "-help") == 0) ||
|
|
(strcasecmp(argv[i], "--help") == 0) ||
|
|
(strcasecmp(argv[i], "-h") == 0) ||
|
|
(strcasecmp(argv[i], "/?") == 0)) {
|
|
print_usage = true;
|
|
close_console = false;
|
|
break;
|
|
|
|
} else {
|
|
// Try to process <option>=<value>, or -<bool>
|
|
if (Configuration::setParam(argv[i], true))
|
|
continue;
|
|
// Try to process -<option> <value>
|
|
if ((argv[i][0] == '-') && (i+1 < argc)) {
|
|
if (Configuration::setParam(&argv[i][1], argv[i+1], true)) {
|
|
i++;
|
|
continue;
|
|
}
|
|
}
|
|
// If it's -<option> then it's not recognised - error
|
|
// If it's <host> then add it to the list to connect to.
|
|
if ((argv[i][0] == '-') || (argv[i][0] == '/')) {
|
|
const char* fmt = "The option %s was not recognized. Use -help to see VNC Viewer usage";
|
|
CharArray tmp(strlen(argv[i])+strlen(fmt)+1);
|
|
sprintf(tmp.buf, fmt, argv[i]);
|
|
MsgBox(0, TStr(tmp.buf), MB_ICONSTOP | MB_OK);
|
|
exit(1);
|
|
} else {
|
|
hosts.push_back(strDup(argv[i]));
|
|
}
|
|
}
|
|
|
|
} catch (rdr::Exception& e) {
|
|
vlog.error(e.str());
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
//
|
|
// -=- main
|
|
//
|
|
|
|
int WINAPI WinMain(HINSTANCE inst, HINSTANCE prevInst, char* cmdLine, int cmdShow) {
|
|
|
|
try {
|
|
|
|
// - Initialise the available loggers
|
|
initStdIOLoggers();
|
|
initFileLogger("C:\\temp\\vncviewer4.log");
|
|
|
|
// - By default, just log errors to stderr
|
|
logParams.setDefault("*:stderr:0");
|
|
|
|
// - Process the command-line
|
|
int argc = __argc;
|
|
char** argv = __argv;
|
|
processParams(argc, argv);
|
|
|
|
// - By default the console will be closed
|
|
if (close_console) {
|
|
if (!FreeConsole())
|
|
vlog.info("unable to close console:%u", GetLastError());
|
|
} else {
|
|
AllocConsole();
|
|
freopen("CONIN$","rb",stdin);
|
|
freopen("CONOUT$","wb",stdout);
|
|
freopen("CONOUT$","wb",stderr);
|
|
setbuf(stderr, 0);
|
|
}
|
|
|
|
#ifdef _DIALOG_CAPTURE
|
|
if (captureDialogs) {
|
|
CView::userConfigKey.openKey(HKEY_CURRENT_USER, _T("Software\\TightVNC\\VNCViewer4"));
|
|
OptionsDialog::global.showDialog(0, true);
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
// - If no clients are specified, bring up a connection dialog
|
|
if (configFiles.empty() && hosts.empty() && !acceptIncoming && !print_usage)
|
|
hosts.push_back(0);
|
|
|
|
programInfo();
|
|
|
|
// - Connect to the clients
|
|
if (!configFiles.empty() || !hosts.empty() || acceptIncoming) {
|
|
// - Configure the registry configuration reader
|
|
win32::RegistryReader reg_reader;
|
|
reg_reader.setKey(HKEY_CURRENT_USER, _T("Software\\TightVNC\\VNCViewer4"));
|
|
|
|
// - Tell the rest of VNC Viewer where to write config data to
|
|
CView::userConfigKey.openKey(HKEY_CURRENT_USER, _T("Software\\TightVNC\\VNCViewer4"));
|
|
|
|
// - Start the Socket subsystem for TCP
|
|
TcpSocket::initTcpSockets();
|
|
|
|
// Create the client connection manager
|
|
CViewManager view_manager;
|
|
|
|
if (acceptIncoming) {
|
|
int port = 5500;
|
|
|
|
// Listening viewer
|
|
if (hosts.size() > 1) {
|
|
programUsage();
|
|
exit(2);
|
|
}
|
|
if (!hosts.empty()) {
|
|
port = atoi(hosts.front());
|
|
}
|
|
|
|
vlog.debug("opening listener");
|
|
|
|
CViewTrayIcon tray(view_manager);
|
|
|
|
view_manager.addDefaultTCPListener(port);
|
|
|
|
// Run the view manager
|
|
// Also processes the tray icon if necessary
|
|
MSG msg;
|
|
while (GetMessage(&msg, NULL, 0, 0) > 0) {
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
|
|
vlog.debug("quitting viewer");
|
|
} else {
|
|
// Read each config file in turn
|
|
while (!configFiles.empty()) {
|
|
char* filename = configFiles.front();
|
|
view_manager.addClient(filename, true);
|
|
strFree(filename);
|
|
configFiles.pop_front();
|
|
}
|
|
|
|
// Connect to each client in turn
|
|
while (!hosts.empty()) {
|
|
char* hostinfo = hosts.front();
|
|
view_manager.addClient(hostinfo);
|
|
strFree(hostinfo);
|
|
hosts.pop_front();
|
|
}
|
|
|
|
// Run the view manager
|
|
MSG msg;
|
|
while (GetMessage(&msg, NULL, 0, 0) > 0) {
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
|
|
vlog.debug("quitting viewer");
|
|
}
|
|
|
|
}
|
|
|
|
// - If necessary, print the program's usage info
|
|
if (print_usage)
|
|
programUsage();
|
|
|
|
if (!close_console) {
|
|
printf("Press Enter/Return key to continue\n");
|
|
char c = getchar();
|
|
}
|
|
|
|
} catch (rdr::Exception& e) {
|
|
MsgBox(0, TStr(e.str()), MB_ICONSTOP | MB_OK);
|
|
}
|
|
|
|
return 0;
|
|
}
|