// Copyright (C) 1999 AT&T Laboratories Cambridge. All Rights Reserved.
//
// This file is part of the VNC system.
//
// The VNC system 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 program 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 program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
// USA.
//
// If the source code for the VNC system is not available from the place
// whence you received this file, check http://www.uk.research.att.com/vnc or contact
// the authors on vnc@uk.research.att.com for information on obtaining it.
#include "stdhdrs.h"
#include "vncviewer.h"
#include "VNCviewerApp.h"
#include "Exception.h"
// For WinCE Palm, you might want to use this for debugging, since it
// seems impossible to give the command some arguments.
// #define PALM_LOG 1
VNCviewerApp *pApp;
VNCviewerApp::VNCviewerApp(HINSTANCE hInstance, LPTSTR szCmdLine) {
pApp = this;
m_instance = hInstance;
// Read the command line
m_options.SetFromCommandLine(szCmdLine);
// Logging info
log.SetLevel(m_options.m_logLevel);
if (m_options.m_logToConsole) {
log.SetMode(Log::ToConsole | Log::ToDebug);
}
if (m_options.m_logToFile) {
log.SetFile(m_options.m_logFilename);
}
#ifdef PALM_LOG
// Hack override for WinCE Palm developers who can't give
// options to commands, not even via shortcuts.
log.SetLevel(20);
log.SetFile(_T("\\Temp\\log"));
#endif
// Clear connection list
for (int i = 0; i < MAX_CONNECTIONS; i++)
m_clilist[i] = NULL;
// Initialise winsock
WORD wVersionRequested = MAKEWORD(2, 0);
WSADATA wsaData;
if (WSAStartup(wVersionRequested, &wsaData) != 0) {
MessageBox(NULL, _T("Error initialising sockets library"), _T("VNC info"), MB_OK | MB_ICONSTOP);
PostQuitMessage(1);
}
log.Print(3, _T("Started and Winsock (v %d) initialised\n"), wsaData.wVersion);
}
// The list of clients should fill up from the start and have NULLs
// afterwards. If the first entry is a NULL, the list is empty.
void VNCviewerApp::RegisterConnection(ClientConnection *pConn) {
omni_mutex_lock l(m_clilistMutex);
int i;
for (i = 0; i < MAX_CONNECTIONS; i++) {
if (m_clilist[i] == NULL) {
m_clilist[i] = pConn;
log.Print(4,_T("Registered connection with app\n"));
return;
}
}
// If we've got here, something is wrong.
log.Print(-1, _T("Client list overflow!\n"));
MessageBox(NULL, _T("Client list overflow!"), _T("VNC error"),
MB_OK | MB_ICONSTOP);
PostQuitMessage(1);
}
void VNCviewerApp::DeregisterConnection(ClientConnection *pConn) {
omni_mutex_lock l(m_clilistMutex);
int i;
for (i = 0; i < MAX_CONNECTIONS; i++) {
if (m_clilist[i] == pConn) {
// shuffle everything above downwards
for (int j = i; m_clilist[j] && j < MAX_CONNECTIONS-1 ; j++)
m_clilist[j] = m_clilist[j+1];
m_clilist[MAX_CONNECTIONS-1] = NULL;
log.Print(4,_T("Deregistered connection from app\n"));
// No clients left? then we should finish, unless we're in
// listening mode.
if ((m_clilist[0] == NULL) && (!pApp->m_options.m_listening))
PostQuitMessage(0);
return;
}
}
// If we've got here, something is wrong.
log.Print(-1, _T("Client not found for deregistering!\n"));
PostQuitMessage(1);
}
// ----------------------------------------------
VNCviewerApp::~VNCviewerApp() {
// Clean up winsock
WSACleanup();
log.Print(2, _T("VNC viewer closing down\n"));
}