// WinMain parses the command line and either calls the main App
// routine or, under NT, the main service routine.
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)
{
if (VNCOS.OS_NOTSUPPORTED==true)
{
MessageBoxSecure(NULL, "Error OS not supported","Unsupported OS", MB_ICONERROR);
return true;
}
// make vnc last service to stop
SetProcessShutdownParameters(0x100,false);
// handle dpi on aero
/*HMODULE hUser32 = LoadLibrary(_T("user32.dll"));
typedef BOOL (*SetProcessDPIAwareFunc)();
SetProcessDPIAwareFunc setDPIAware=NULL;
if (hUser32) setDPIAware = (SetProcessDPIAwareFunc)GetProcAddress(hUser32, "SetProcessDPIAware");
if (setDPIAware) setDPIAware();
if (hUser32) FreeLibrary(hUser32);*/
#ifdef IPP
InitIpp();
#endif
#ifdef CRASHRPT
CR_INSTALL_INFO info;
memset(&info, 0, sizeof(CR_INSTALL_INFO));
info.cb = sizeof(CR_INSTALL_INFO);
info.pszAppName = _T("UVNC");
info.pszAppVersion = _T("1.2.0.9");
info.pszEmailSubject = _T("UVNC server 1.2.0.9 Error Report");
info.pszEmailTo = _T("*****@*****.**");
info.uPriorities[CR_SMAPI] = 1; // Third try send report over Simple MAPI
// Install all available exception handlers
info.dwFlags |= CR_INST_ALL_POSSIBLE_HANDLERS;
// Restart the app on crash
info.dwFlags |= CR_INST_APP_RESTART;
info.dwFlags |= CR_INST_SEND_QUEUED_REPORTS;
info.dwFlags |= CR_INST_AUTO_THREAD_HANDLERS;
info.pszRestartCmdLine = _T("/restart");
// Define the Privacy Policy URL
// Install crash reporting
int nResult = crInstall(&info);
if (nResult != 0)
{
// Something goes wrong. Get error message.
TCHAR szErrorMsg[512] = _T("");
crGetLastErrorMsg(szErrorMsg, 512);
_tprintf_s(_T("%s\n"), szErrorMsg);
return 1;
}
#endif
bool Injected_autoreconnect=false;
SPECIAL_SC_EXIT=false;
SPECIAL_SC_PROMPT=false;
setbuf(stderr, 0);
// [v1.0.2-jp1 fix] Load resouce from dll
hInstResDLL = NULL;
//limit the vnclang.dll searchpath to avoid
char szCurrentDir[MAX_PATH];
char szCurrentDir_vnclangdll[MAX_PATH];
if (GetModuleFileName(NULL, szCurrentDir, MAX_PATH))
{
char* p = strrchr(szCurrentDir, '\\');
*p = '\0';
}
strcpy (szCurrentDir_vnclangdll,szCurrentDir);
strcat (szCurrentDir_vnclangdll,"\\");
strcat (szCurrentDir_vnclangdll,"vnclang_server.dll");
hInstResDLL = LoadLibrary(szCurrentDir_vnclangdll);
if (hInstResDLL == NULL)
{
hInstResDLL = hInstance;
}
// RegisterLinkLabel(hInstResDLL);
//Load all messages from ressource file
Load_Localization(hInstResDLL) ;
char WORKDIR[MAX_PATH];
if (GetModuleFileName(NULL, WORKDIR, MAX_PATH))
{
char* p = strrchr(WORKDIR, '\\');
if (p == NULL) return 0;
*p = '\0';
}
char progname[MAX_PATH];
strncpy(progname, WORKDIR, sizeof progname);
progname[MAX_PATH - 1] = 0;
//strcat(WORKDIR,"\\");
//strcat(WORKDIR,"WinVNC.log");
vnclog.SetFile();
//vnclog.SetMode(4);
//vnclog.SetLevel(10);
#ifdef _DEBUG
{
// Get current flag
int tmpFlag = _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG );
// Turn on leak-checking bit
tmpFlag |= _CRTDBG_LEAK_CHECK_DF;
// Set flag to the new value
_CrtSetDbgFlag( tmpFlag );
}
#endif
// Save the application instance and main thread id
hAppInstance = hInstance;
mainthreadId = GetCurrentThreadId();
// Initialise the VSocket system
VSocketSystem socksys;
if (!socksys.Initialised())
{
MessageBoxSecure(NULL, sz_ID_FAILED_INIT, szAppName, MB_OK);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
// look up the current service name in the registry.
GetServiceName(progname, service_name);
// Make the command-line lowercase and parse it
size_t i;
for (i = 0; i < strlen(szCmdLine); i++)
{
szCmdLine[i] = tolower(szCmdLine[i]);
}
BOOL argfound = FALSE;
for (i = 0; i < strlen(szCmdLine); i++)
{
if (szCmdLine[i] <= ' ')
continue;
argfound = TRUE;
if (strncmp(&szCmdLine[i], winvncSettingshelper, strlen(winvncSettingshelper)) == 0)
{
Sleep(3000);
char mycommand[MAX_PATH];
i+=strlen(winvncSettingshelper);
strcpy( mycommand, &(szCmdLine[i+1]));
Set_settings_as_admin(mycommand);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncStopserviceHelper, strlen(winvncStopserviceHelper)) == 0)
{
Sleep(3000);
Set_stop_service_as_admin();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncKill, strlen(winvncKill)) == 0)
{
static HANDLE hShutdownEventTmp;
hShutdownEventTmp = OpenEvent(EVENT_ALL_ACCESS, FALSE, "Global\\SessionEventUltra");
SetEvent(hShutdownEventTmp);
CloseHandle(hShutdownEventTmp);
//adzm 2010-02-10 - Finds the appropriate VNC window for any process. Sends this message to all of them!
// do removed, loops forever with cpu 100
HWND hservwnd = NULL;
hservwnd = FindWinVNCWindow(false);
if (hservwnd!=NULL)
{
PostMessage(hservwnd, WM_COMMAND, 40002, 0);
PostMessage(hservwnd, WM_CLOSE, 0, 0);
}
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncopenhomepage, strlen(winvncopenhomepage)) == 0)
{
Open_homepage();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncopenforum, strlen(winvncopenforum)) == 0)
{
Open_forum();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncStartserviceHelper, strlen(winvncStartserviceHelper)) == 0)
{
Sleep(3000);
Set_start_service_as_admin();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncInstallServiceHelper, strlen(winvncInstallServiceHelper)) == 0)
{
//Sleeps are realy needed, else runas fails...
Sleep(3000);
Set_install_service_as_admin();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncUnInstallServiceHelper, strlen(winvncUnInstallServiceHelper)) == 0)
{
Sleep(3000);
Set_uninstall_service_as_admin();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncSoftwarecadHelper, strlen(winvncSoftwarecadHelper)) == 0)
{
Sleep(3000);
Enable_softwareCAD_elevated();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncdelSoftwarecadHelper, strlen(winvncdelSoftwarecadHelper)) == 0)
{
Sleep(3000);
delete_softwareCAD_elevated();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncRebootSafeHelper, strlen(winvncRebootSafeHelper)) == 0)
{
Sleep(3000);
Reboot_in_safemode_elevated();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncRebootForceHelper, strlen(winvncRebootForceHelper)) == 0)
{
Sleep(3000);
Reboot_with_force_reboot_elevated();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncSecurityEditorHelper, strlen(winvncSecurityEditorHelper)) == 0)
{
Sleep(3000);
winvncSecurityEditorHelper_as_admin();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncSecurityEditor, strlen(winvncSecurityEditor)) == 0)
{
typedef void (*vncEditSecurityFn) (HWND hwnd, HINSTANCE hInstance);
vncEditSecurityFn vncEditSecurity = 0;
char szCurrentDirl[MAX_PATH];
if (GetModuleFileName(NULL, szCurrentDirl, MAX_PATH)) {
char* p = strrchr(szCurrentDirl, '\\');
*p = '\0';
strcat (szCurrentDirl,"\\authSSP.dll");
}
HMODULE hModule = LoadLibrary(szCurrentDirl);
if (hModule) {
vncEditSecurity = (vncEditSecurityFn) GetProcAddress(hModule, "vncEditSecurity");
HRESULT hr = CoInitialize(NULL);
vncEditSecurity(NULL, hAppInstance);
CoUninitialize();
FreeLibrary(hModule);
}
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncSettings, strlen(winvncSettings)) == 0)
{
char mycommand[MAX_PATH];
i+=strlen(winvncSettings);
strcpy( mycommand, &(szCmdLine[i+1]));
Real_settings(mycommand);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], dsmpluginhelper, strlen(dsmpluginhelper)) == 0)
{
char mycommand[MAX_PATH];
i += strlen(dsmpluginhelper);
strcpy(mycommand, &(szCmdLine[i + 1]));
Secure_Plugin_elevated(mycommand);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], dsmplugininstance, strlen(dsmplugininstance)) == 0)
{
char mycommand[MAX_PATH];
i += strlen(dsmplugininstance);
strcpy(mycommand, &(szCmdLine[i + 1]));
Secure_Plugin(mycommand);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncSoftwarecad, strlen(winvncSoftwarecad)) == 0)
{
Enable_softwareCAD();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncdelSoftwarecad, strlen(winvncdelSoftwarecad)) == 0)
{
delete_softwareCAD();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncRebootSafe, strlen(winvncRebootSafe)) == 0)
{
Reboot_in_safemode();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncRebootForce, strlen(winvncRebootForce)) == 0)
{
Reboot_with_force_reboot();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncStopservice, strlen(winvncStopservice)) == 0)
{
Real_stop_service();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncStartservice, strlen(winvncStartservice)) == 0)
{
Real_start_service();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncInstallService, strlen(winvncInstallService)) == 0)
{
// rest of command line service name, if provided.
char *pServiceName = &szCmdLine[i];
// skip over command switch, find next whitepace
while (*pServiceName && !isspace(*(unsigned char*)pServiceName))
++pServiceName;
// skip past whitespace to service name
while (*pServiceName && isspace(*(unsigned char*)pServiceName))
++pServiceName;
// strip off any quotes
if (*pServiceName && *pServiceName == '\"')
++pServiceName;
if (*pServiceName)
{
// look for trailing quote, if found, terminate the string there.
char *pQuote = pServiceName;
pQuote = strrchr(pServiceName, '\"');
if (pQuote)
*pQuote = 0;
}
// if a service name is supplied, and it differs except in case from
// the default, use the supplied service name instead
if (*pServiceName && (_strcmpi(pServiceName, service_name) != 0))
{
strncpy(service_name, pServiceName, 256);
service_name[255] = 0;
}
install_service();
Sleep(2000);
char command[MAX_PATH + 32]; // 29 January 2008 jdp
_snprintf(command, sizeof command, "net start \"%s\"", service_name);
WinExec(command,SW_HIDE);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncUnInstallService, strlen(winvncUnInstallService)) == 0)
{
char command[MAX_PATH + 32]; // 29 January 2008 jdp
// rest of command line service name, if provided.
char *pServiceName = &szCmdLine[i];
// skip over command switch, find next whitepace
while (*pServiceName && !isspace(*(unsigned char*)pServiceName))
++pServiceName;
// skip past whitespace to service name
while (*pServiceName && isspace(*(unsigned char*)pServiceName))
++pServiceName;
// strip off any quotes
if (*pServiceName && *pServiceName == '\"')
++pServiceName;
if (*pServiceName)
{
// look for trailing quote, if found, terminate the string there.
char *pQuote = pServiceName;
pQuote = strrchr(pServiceName, '\"');
if (pQuote)
*pQuote = 0;
}
if (*pServiceName && (_strcmpi(pServiceName, service_name) != 0))
{
strncpy(service_name, pServiceName, 256);
service_name[255] = 0;
}
_snprintf(command, sizeof command, "net stop \"%s\"", service_name);
WinExec(command,SW_HIDE);
uninstall_service();
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncRunService, strlen(winvncRunService)) == 0)
{
//Run as service
if (!Myinit(hInstance)) return 0;
fRunningFromExternalService = true;
vncService::RunningFromExternalService(true);
int returnvalue = WinVNCAppMain();
#ifdef CRASHRPT
crUninstall();
#endif
return returnvalue;
}
if (strncmp(&szCmdLine[i], winvncStartService, strlen(winvncStartService)) == 0)
{
start_service(szCmdLine);
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
if (strncmp(&szCmdLine[i], winvncRunAsUserApp, strlen(winvncRunAsUserApp)) == 0)
{
// WinVNC is being run as a user-level program
if (!Myinit(hInstance)) return 0;
int returnvalue = WinVNCAppMain();
#ifdef CRASHRPT
crUninstall();
#endif
return returnvalue;
}
if (strncmp(&szCmdLine[i], winvncSCexit, strlen(winvncSCexit)) == 0)
{
SPECIAL_SC_EXIT=true;
i+=strlen(winvncSCexit);
continue;
}
if (strncmp(&szCmdLine[i], winvncSCprompt, strlen(winvncSCprompt)) == 0)
{
SPECIAL_SC_PROMPT=true;
i+=strlen(winvncSCprompt);
continue;
}
if (strncmp(&szCmdLine[i], winvncmulti, strlen(winvncmulti)) == 0)
{
multi=true;
i+=strlen(winvncmulti);
continue;
}
if (strncmp(&szCmdLine[i], winvnchttp, strlen(winvnchttp)) == 0)
{
G_HTTP=true;
i+=strlen(winvnchttp);
continue;
}
if (strncmp(&szCmdLine[i], winvncStopReconnect, strlen(winvncStopReconnect)) == 0)
{
i+=strlen(winvncStopReconnect);
vncService::PostAddStopConnectClientAll();
continue;
}
if (strncmp(&szCmdLine[i], winvncAutoReconnect, strlen(winvncAutoReconnect)) == 0)
{
// Note that this "autoreconnect" param MUST be BEFORE the "connect" one
// on the command line !
// wa@2005 -- added support for the AutoReconnectId
i+=strlen(winvncAutoReconnect);
Injected_autoreconnect=true;
int start, end;
char* pszId = NULL;
start = i;
// skip any spaces and grab the parameter
while (szCmdLine[start] <= ' ' && szCmdLine[start] != 0) start++;
if ( strncmp( &szCmdLine[start], winvncAutoReconnectId, strlen(winvncAutoReconnectId) ) == 0 )
{
end = start;
while (szCmdLine[end] > ' ') end++;
if (end - start > 0)
{
pszId = new char[end - start + 1];
strncpy(pszId, &(szCmdLine[start]), end - start);
pszId[end - start] = 0;
pszId = _strupr(pszId);
}
//multiple spaces between autoreconnect and id
i = end;
}// end of condition we found the ID: parameter
// NOTE: id must be NULL or the ID:???? (pointer will get deleted when message is processed)
// We can not contact a runnning service, permissions, so we must store the settings
// and process until the vncmenu has been started
if (!vncService::PostAddAutoConnectClient( pszId ))
{
PostAddAutoConnectClient_bool=true;
if (pszId==NULL)
{
PostAddAutoConnectClient_bool_null=true;
PostAddAutoConnectClient_bool=false;
}
else
{
strcpy(pszId_char,pszId);
//memory leak fix
delete[] pszId; pszId = NULL;
}
}
if (pszId != NULL) delete[] pszId; pszId = NULL;
continue;
}
if ( strncmp( &szCmdLine[i], winvncReconnectId, strlen(winvncReconnectId) ) == 0 )
{
i+=strlen("-");
int start, end;
char* pszId = NULL;
start = i;
end = start;
while (szCmdLine[end] > ' ') end++;
if (end - start > 0)
{
pszId = new char[end - start + 1];
if (pszId != 0)
{
strncpy(pszId, &(szCmdLine[start]), end - start);
pszId[end - start] = 0;
pszId = _strupr(pszId);
}
}
i = end;
if (!vncService::PostAddConnectClient( pszId ))
{
PostAddConnectClient_bool=true;
if (pszId==NULL)
{
PostAddConnectClient_bool_null=true;
PostAddConnectClient_bool=false;
}
else
{
strcpy(pszId_char,pszId);
//memory leak fix
delete[] pszId; pszId = NULL;
}
}
if (pszId != NULL) delete[] pszId; pszId = NULL;
continue;
}
if (strncmp(&szCmdLine[i], winvncConnect, strlen(winvncConnect)) == 0)
{
if (!Injected_autoreconnect)
{
vncService::PostAddStopConnectClient();
}
// Add a new client to an existing copy of winvnc
i+=strlen(winvncConnect);
// First, we have to parse the command line to get the filename to use
int start, end;
start=i;
while (szCmdLine[start] <= ' ' && szCmdLine[start] != 0) start++;
end = start;
while (szCmdLine[end] > ' ') end++;
// Was there a hostname (and optionally a port number) given?
if (end-start > 0)
{
char *name = new char[end-start+1];
if (name != 0) {
strncpy(name, &(szCmdLine[start]), end-start);
name[end-start] = 0;
int port = INCOMING_PORT_OFFSET;
char *portp = strchr(name, ':');
if (portp) {
*portp++ = '\0';
if (*portp == ':') {
port = atoi(++portp); // Port number after "::"
} else {
port = atoi(portp); // Display number after ":"
}
}
vnclog.Print(LL_STATE, VNCLOG("test... %s %d\n"),name,port);
strcpy_s(dnsname,name);
VCard32 address = VSocket::Resolve(name);
delete [] name;
if (address != 0) {
// Post the IP address to the server
// We can not contact a runnning service, permissions, so we must store the settings
// and process until the vncmenu has been started
vnclog.Print(LL_INTERR, VNCLOG("PostAddNewClient III \n"));
if (!vncService::PostAddNewClientInit(address, port))
{
PostAddNewClient_bool=true;
port_int=port;
address_vcard=address;
}
}
else
{
//ask for host,port
PostAddNewClient_bool=true;
port_int=0;
address_vcard=0;
Sleep(2000);
//Beep(200,1000);
return 0;
}
}
i=end;
continue;
}
else
{
// Tell the server to show the Add New Client dialog
// We can not contact a runnning service, permissions, so we must store the settings
// and process until the vncmenu has been started
vnclog.Print(LL_INTERR, VNCLOG("PostAddNewClient IIII\n"));
if (!vncService::PostAddNewClient(0, 0))
{
PostAddNewClient_bool=true;
port_int=0;
address_vcard=0;
}
}
continue;
}
//adzm 2009-06-20
if (strncmp(&szCmdLine[i], winvncRepeater, strlen(winvncRepeater)) == 0)
{
// set the default repeater host
i+=strlen(winvncRepeater);
// First, we have to parse the command line to get the host to use
int start, end;
start=i;
while (szCmdLine[start] <= ' ' && szCmdLine[start] != 0) start++;
end = start;
while (szCmdLine[end] > ' ') end++;
// Was there a hostname (and optionally a port number) given?
if (end-start > 0)
{
if (g_szRepeaterHost) {
delete[] g_szRepeaterHost;
g_szRepeaterHost = NULL;
}
g_szRepeaterHost = new char[end-start+1];
if (g_szRepeaterHost != 0) {
strncpy(g_szRepeaterHost, &(szCmdLine[start]), end-start);
g_szRepeaterHost[end-start] = 0;
// We can not contact a runnning service, permissions, so we must store the settings
// and process until the vncmenu has been started
vnclog.Print(LL_INTERR, VNCLOG("PostAddNewRepeaterClient I\n"));
if (!vncService::PostAddNewRepeaterClient())
{
PostAddNewRepeaterClient_bool=true;
port_int=0;
address_vcard=0;
}
}
i=end;
continue;
}
else
{
/*
// Tell the server to show the Add New Client dialog
// We can not contact a runnning service, permissions, so we must store the settings
// and process until the vncmenu has been started
vnclog.Print(LL_INTERR, VNCLOG("PostAddNewClient IIII\n"));
if (!vncService::PostAddNewClient(0, 0))
{
PostAddNewClient_bool=true;
port_int=0;
address_vcard=0;
}
*/
}
continue;
}
// Either the user gave the -help option or there is something odd on the cmd-line!
// Show the usage dialog
MessageBoxSecure(NULL, winvncUsageText, sz_ID_WINVNC_USAGE, MB_OK | MB_ICONINFORMATION);
break;
};
// If no arguments were given then just run
if (!argfound)
{
if (!Myinit(hInstance))
{
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
int returnvalue= WinVNCAppMain();
#ifdef CRASHRPT
crUninstall();
#endif
return returnvalue;
}
#ifdef CRASHRPT
crUninstall();
#endif
return 0;
}
// Process window messages
LRESULT CALLBACK vncMenu::WndProc(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
{
// This is a static method, so we don't know which instantiation we're
// dealing with. We use Allen Hadden's ([email protected]) suggestion
// from a newsgroup to get the pseudo-this.
vncMenu *_this = helper::SafeGetWindowUserData<vncMenu>(hwnd);
// Beep(100,10);
// vnclog.Print(LL_INTINFO, VNCLOG("iMsg 0x%x \n"),iMsg);
if (iMsg==WM_TASKBARCREATED)
{
if (_this->m_server->RunningFromExternalService())
{
Sleep(1000);
vnclog.Print(LL_INTINFO,
VNCLOG("WM_TASKBARCREATED \n"));
// User has changed!
strcpy(_this->m_username, newuser);
vnclog.Print(LL_INTINFO,
VNCLOG("############## Kill vncMenu thread\n"));
// Order impersonation thread killing
KillTimer(hwnd,1);
PostQuitMessage(0);
}
}
switch (iMsg)
{
// Every five seconds, a timer message causes the icon to update
case WM_TIMER:
// sf@2007 - Can't get the WTS_CONSOLE_CONNECT message work properly for now..
// So use a hack instead
// jdp reread some ini settings
_this->m_properties.ReloadDynamicSettings();
// G_1111==true --> reconnect
if (G_1111==true)
{
if (_this->IsIconSet==true)
{
vnclog.Print(LL_INTERR, VNCLOG("Add client reconnect from timer\n"));
G_1111=false;
PostMessage(hwnd,MENU_ADD_CLIENT_MSG,1111,1111);
}
}
vnclog.Print(LL_INTERR, VNCLOG("########### vncMenu::TIMER TrayIcon 5s hack\n"));
if (_this->m_server->RunningFromExternalService())
{
strcpy(newuser,"");
if (vncService::CurrentUser((char *) &newuser, sizeof(newuser)))
{
// Check whether the user name has changed!
if (_stricmp(newuser, _this->m_username) != 0 || _this->IconFaultCounter>2)
{
Sleep(1000);
vnclog.Print(LL_INTINFO,
VNCLOG("user name has changed\n"));
// User has changed!
strcpy(_this->m_username, newuser);
vnclog.Print(LL_INTINFO,
VNCLOG("############## Kill vncMenu thread\n"));
// Order impersonation thread killing
PostQuitMessage(0);
break;
}
}
}
// *** HACK for running servicified
if (vncService::RunningAsService())
{
vnclog.Print(LL_INTERR, VNCLOG("########### vncMenu::TIMER TrayIcon 5s hack call - Runningasservice\n"));
// Attempt to add the icon if it's not already there
_this->AddTrayIcon();
// Trigger a check of the current user
PostMessage(hwnd, WM_USERCHANGED, 0, 0);
}
// Update the icon
_this->FlashTrayIcon(_this->m_server->AuthClientCount() != 0);
break;
// DEAL WITH NOTIFICATIONS FROM THE SERVER:
case WM_SRV_CLIENT_AUTHENTICATED:
case WM_SRV_CLIENT_DISCONNECT:
// Adjust the icon accordingly
_this->FlashTrayIcon(_this->m_server->AuthClientCount() != 0);
if (_this->m_server->AuthClientCount() != 0) {
if (_this->m_server->RemoveWallpaperEnabled())
KillWallpaper();
if (_this->m_server->RemoveAeroEnabled()) // Moved, redundant if //PGM @ Advantig
DisableAero(); // Moved, redundant if //PGM @ Advantig
} else {
if (_this->m_server->RemoveAeroEnabled()) // Moved, redundant if //PGM @ Advantig
ResetAero(); // Moved, redundant if //PGM @ Advantig
if (_this->m_server->RemoveWallpaperEnabled()) { // Added { //PGM @ Advantig
Sleep(2000); // Added 2 second delay to help wallpaper restore //PGM @ Advantig
RestoreWallpaper();
} //PGM @ Advantig
}
//PGM @ Advantig if (_this->m_server->AuthClientCount() != 0) {
//PGM @ Advantig if (_this->m_server->RemoveAeroEnabled())
//PGM @ Advantig DisableAero();
//PGM @ Advantig } else {
//PGM @ Advantig if (_this->m_server->RemoveAeroEnabled())
//PGM @ Advantig ResetAero();
//PGM @ Advantig }
return 0;
// STANDARD MESSAGE HANDLING
case WM_CREATE:
WM_TASKBARCREATED = RegisterWindowMessage("TaskbarCreated");
return 0;
case WM_COMMAND:
// User has clicked an item on the tray menu
switch (LOWORD(wParam))
{
case ID_DEFAULT_PROPERTIES:
// Show the default properties dialog, unless it is already displayed
vnclog.Print(LL_INTINFO, VNCLOG("show default properties requested\n"));
_this->m_properties.ShowAdmin(TRUE, FALSE);
_this->FlashTrayIcon(_this->m_server->AuthClientCount() != 0);
break;
case ID_PROPERTIES:
// Show the properties dialog, unless it is already displayed
vnclog.Print(LL_INTINFO, VNCLOG("show user properties requested\n"));
_this->m_propertiesPoll.Show(TRUE, TRUE);
_this->FlashTrayIcon(_this->m_server->AuthClientCount() != 0);
break;
case ID_ADMIN_PROPERTIES:
// Show the properties dialog, unless it is already displayed
vnclog.Print(LL_INTINFO, VNCLOG("show user properties requested\n"));
_this->m_properties.ShowAdmin(TRUE, TRUE);
_this->FlashTrayIcon(_this->m_server->AuthClientCount() != 0);
break;
case ID_OUTGOING_CONN:
// Connect out to a listening VNC viewer
{
vncConnDialog *newconn = new vncConnDialog(_this->m_server);
if (newconn)
{
newconn->DoDialog();
// delete newconn; // NO ! Already done in vncConnDialog.
}
}
break;
case ID_KILLCLIENTS:
// Disconnect all currently connected clients
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_KILLCLIENTS \n"));
_this->m_server->KillAuthClients();
break;
// sf@2002
case ID_LISTCLIENTS:
_this->m_ListDlg.Display();
break;
case ID_ABOUT:
// Show the About box
_this->m_about.Show(TRUE);
break;
case ID_VISITUSONLINE_HOMEPAGE:
{
HANDLE hProcess,hPToken;
DWORD id=GetExplorerLogonPid();
if (id!=0)
{
hProcess = OpenProcess(MAXIMUM_ALLOWED,FALSE,id);
if(!OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY
|TOKEN_DUPLICATE|TOKEN_ASSIGN_PRIMARY|TOKEN_ADJUST_SESSIONID
|TOKEN_READ|TOKEN_WRITE,&hPToken)) break;
char dir[MAX_PATH];
char exe_file_name[MAX_PATH];
GetModuleFileName(0, exe_file_name, MAX_PATH);
strcpy(dir, exe_file_name);
strcat(dir, " -openhomepage");
{
STARTUPINFO StartUPInfo;
PROCESS_INFORMATION ProcessInfo;
ZeroMemory(&StartUPInfo,sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo,sizeof(PROCESS_INFORMATION));
StartUPInfo.wShowWindow = SW_SHOW;
StartUPInfo.lpDesktop = "Winsta0\\Default";
StartUPInfo.cb = sizeof(STARTUPINFO);
CreateProcessAsUser(hPToken,NULL,dir,NULL,NULL,FALSE,DETACHED_PROCESS,NULL,NULL,&StartUPInfo,&ProcessInfo);
DWORD error=GetLastError();
if (ProcessInfo.hThread) CloseHandle(ProcessInfo.hThread);
if (ProcessInfo.hProcess) CloseHandle(ProcessInfo.hProcess);
//if (error==1314)
// {
// Open_homepage();
// }
}
}
}
break;
case ID_VISITUSONLINE_FORUM:
{
HANDLE hProcess,hPToken;
DWORD id=GetExplorerLogonPid();
if (id!=0)
{
hProcess = OpenProcess(MAXIMUM_ALLOWED,FALSE,id);
if(!OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY
|TOKEN_DUPLICATE|TOKEN_ASSIGN_PRIMARY|TOKEN_ADJUST_SESSIONID
|TOKEN_READ|TOKEN_WRITE,&hPToken)) break;
char dir[MAX_PATH];
char exe_file_name[MAX_PATH];
GetModuleFileName(0, exe_file_name, MAX_PATH);
strcpy(dir, exe_file_name);
strcat(dir, " -openforum");
{
STARTUPINFO StartUPInfo;
PROCESS_INFORMATION ProcessInfo;
ZeroMemory(&StartUPInfo,sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo,sizeof(PROCESS_INFORMATION));
StartUPInfo.wShowWindow = SW_SHOW;
StartUPInfo.lpDesktop = "Winsta0\\Default";
StartUPInfo.cb = sizeof(STARTUPINFO);
CreateProcessAsUser(hPToken,NULL,dir,NULL,NULL,FALSE,DETACHED_PROCESS,NULL,NULL,&StartUPInfo,&ProcessInfo);
DWORD error=GetLastError();
if (ProcessInfo.hThread) CloseHandle(ProcessInfo.hThread);
if (ProcessInfo.hProcess) CloseHandle(ProcessInfo.hProcess);
//if (error==1314)
// {
// Open_forum();
// }
}
}
}
break;
case ID_CLOSE:
// User selected Close from the tray menu
fShutdownOrdered=TRUE;
Sleep(1000);
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_CLOSE \n"));
_this->m_server->KillAuthClients();
PostMessage(hwnd, WM_CLOSE, 0, 0);
break;
case ID_UNINSTALL_SERVICE:
{
HANDLE hProcess,hPToken;
DWORD id=GetExplorerLogonPid();
if (id!=0)
{
hProcess = OpenProcess(MAXIMUM_ALLOWED,FALSE,id);
if(!OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY
|TOKEN_DUPLICATE|TOKEN_ASSIGN_PRIMARY|TOKEN_ADJUST_SESSIONID
|TOKEN_READ|TOKEN_WRITE,&hPToken)) break;
char dir[MAX_PATH];
char exe_file_name[MAX_PATH];
GetModuleFileName(0, exe_file_name, MAX_PATH);
strcpy(dir, exe_file_name);
strcat(dir, " -uninstallhelper");
{
STARTUPINFO StartUPInfo;
PROCESS_INFORMATION ProcessInfo;
HANDLE Token=NULL;
HANDLE process=NULL;
ZeroMemory(&StartUPInfo,sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo,sizeof(PROCESS_INFORMATION));
StartUPInfo.wShowWindow = SW_SHOW;
StartUPInfo.lpDesktop = "Winsta0\\Default";
StartUPInfo.cb = sizeof(STARTUPINFO);
CreateProcessAsUser(hPToken,NULL,dir,NULL,NULL,FALSE,DETACHED_PROCESS,NULL,NULL,&StartUPInfo,&ProcessInfo);
DWORD errorcode=GetLastError();
if (process) CloseHandle(process);
if (Token) CloseHandle(Token);
if (errorcode==1314)
{
Set_uninstall_service_as_admin();
}
}
fShutdownOrdered=TRUE;
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_CLOSE \n"));
_this->m_server->KillAuthClients();
PostMessage(hwnd, WM_CLOSE, 0, 0);
}
}
break;
case ID_RUNASSERVICE:
{
HANDLE hProcess,hPToken;
DWORD id=GetExplorerLogonPid();
if (id!=0)
{
hProcess = OpenProcess(MAXIMUM_ALLOWED,FALSE,id);
if(!OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY
|TOKEN_DUPLICATE|TOKEN_ASSIGN_PRIMARY|TOKEN_ADJUST_SESSIONID
|TOKEN_READ|TOKEN_WRITE,&hPToken)) break;
char dir[MAX_PATH];
char exe_file_name[MAX_PATH];
GetModuleFileName(0, exe_file_name, MAX_PATH);
strcpy(dir, exe_file_name);
strcat(dir, " -installhelper");
STARTUPINFO StartUPInfo;
PROCESS_INFORMATION ProcessInfo;
HANDLE Token=NULL;
HANDLE process=NULL;
ZeroMemory(&StartUPInfo,sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo,sizeof(PROCESS_INFORMATION));
StartUPInfo.wShowWindow = SW_SHOW;
StartUPInfo.lpDesktop = "Winsta0\\Default";
StartUPInfo.cb = sizeof(STARTUPINFO);
CreateProcessAsUser(hPToken,NULL,dir,NULL,NULL,FALSE,DETACHED_PROCESS,NULL,NULL,&StartUPInfo,&ProcessInfo);
DWORD error=GetLastError();
if (process) CloseHandle(process);
if (Token) CloseHandle(Token);
if (error==1314)
{
Set_install_service_as_admin();
}
}
fShutdownOrdered=TRUE;
Sleep(1000);
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_CLOSE \n"));
_this->m_server->KillAuthClients();
PostMessage(hwnd, WM_CLOSE, 0, 0);
}
break;
case ID_CLOSE_SERVICE:
{
HANDLE hProcess,hPToken;
DWORD id=GetExplorerLogonPid();
if (id!=0)
{
hProcess = OpenProcess(MAXIMUM_ALLOWED,FALSE,id);
if(!OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY
|TOKEN_DUPLICATE|TOKEN_ASSIGN_PRIMARY|TOKEN_ADJUST_SESSIONID
|TOKEN_READ|TOKEN_WRITE,&hPToken))
{
CloseHandle(hProcess);
break;
}
char dir[MAX_PATH];
char exe_file_name[MAX_PATH];
GetModuleFileName(0, exe_file_name, MAX_PATH);
strcpy(dir, exe_file_name);
strcat(dir, " -stopservicehelper");
STARTUPINFO StartUPInfo;
PROCESS_INFORMATION ProcessInfo;
HANDLE Token=NULL;
HANDLE process=NULL;
ZeroMemory(&StartUPInfo,sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo,sizeof(PROCESS_INFORMATION));
StartUPInfo.wShowWindow = SW_SHOW;
StartUPInfo.lpDesktop = "Winsta0\\Default";
StartUPInfo.cb = sizeof(STARTUPINFO);
CreateProcessAsUser(hPToken,NULL,dir,NULL,NULL,FALSE,DETACHED_PROCESS,NULL,NULL,&StartUPInfo,&ProcessInfo);
DWORD error=GetLastError();
if (process) CloseHandle(process);
if (Token) CloseHandle(Token);
if (hProcess) CloseHandle(hProcess);
if (error==1314)
{
Set_stop_service_as_admin();
}
}
}
break;
case ID_START_SERVICE:
{
HANDLE hProcess,hPToken;
DWORD id=GetExplorerLogonPid();
if (id!=0)
{
hProcess = OpenProcess(MAXIMUM_ALLOWED,FALSE,id);
if(!OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY
|TOKEN_DUPLICATE|TOKEN_ASSIGN_PRIMARY|TOKEN_ADJUST_SESSIONID
|TOKEN_READ|TOKEN_WRITE,&hPToken))
{
CloseHandle(hProcess);
break;
}
char dir[MAX_PATH];
char exe_file_name[MAX_PATH];
GetModuleFileName(0, exe_file_name, MAX_PATH);
strcpy(dir, exe_file_name);
strcat(dir, " -startservicehelper");
STARTUPINFO StartUPInfo;
PROCESS_INFORMATION ProcessInfo;
HANDLE Token=NULL;
HANDLE process=NULL;
ZeroMemory(&StartUPInfo,sizeof(STARTUPINFO));
ZeroMemory(&ProcessInfo,sizeof(PROCESS_INFORMATION));
StartUPInfo.wShowWindow = SW_SHOW;
StartUPInfo.lpDesktop = "Winsta0\\Default";
StartUPInfo.cb = sizeof(STARTUPINFO);
CreateProcessAsUser(hPToken,NULL,dir,NULL,NULL,FALSE,DETACHED_PROCESS,NULL,NULL,&StartUPInfo,&ProcessInfo);
DWORD error=GetLastError();
if (hPToken) CloseHandle(hPToken);
if (process) CloseHandle(process);
if (Token) CloseHandle(Token);
if (hProcess) CloseHandle(hProcess);
if (error==1314)
{
Set_start_service_as_admin();
}
fShutdownOrdered=TRUE;
Sleep(1000);
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_CLOSE \n"));
_this->m_server->KillAuthClients();
PostMessage(hwnd, WM_CLOSE, 0, 0);
}
}
break;
}
return 0;
case WM_TRAYNOTIFY:
// User has clicked on the tray icon or the menu
{
// Get the submenu to use as a pop-up menu
HMENU submenu = GetSubMenu(_this->m_hmenu, 0);
// What event are we responding to, RMB click?
if (lParam==WM_RBUTTONUP)
{
if (submenu == NULL)
{
vnclog.Print(LL_INTERR, VNCLOG("no submenu available\n"));
return 0;
}
// Make the first menu item the default (bold font)
SetMenuDefaultItem(submenu, 0, TRUE);
// Get the current cursor position, to display the menu at
POINT mouse;
GetCursorPos(&mouse);
// There's a "bug"
// (Microsoft calls it a feature) in Windows 95 that requires calling
// SetForegroundWindow. To find out more, search for Q135788 in MSDN.
//
SetForegroundWindow(_this->m_nid.hWnd);
// Display the menu at the desired position
TrackPopupMenu(submenu,
0, mouse.x, mouse.y, 0,
_this->m_nid.hWnd, NULL);
PostMessage(hwnd, WM_NULL, 0, 0);
return 0;
}
// Or was there a LMB double click?
if (lParam==WM_LBUTTONDBLCLK)
{
// double click: execute first menu item
SendMessage(_this->m_nid.hWnd,
WM_COMMAND,
GetMenuItemID(submenu, 0),
0);
}
return 0;
}
case WM_CLOSE:
// Only accept WM_CLOSE if the logged on user has AllowShutdown set
if (!_this->m_properties.AllowShutdown())
{
return 0;
}
// tnatsni Wallpaper fix
if (_this->m_server->RemoveWallpaperEnabled())
RestoreWallpaper();
if (_this->m_server->RemoveAeroEnabled())
ResetAero();
vnclog.Print(LL_INTERR, VNCLOG("vncMenu WM_CLOSE call - All cleanup done\n"));
Sleep(2000);
DestroyWindow(hwnd);
break;
case WM_DESTROY:
// The user wants WinVNC to quit cleanly...
vnclog.Print(LL_INTINFO, VNCLOG("quitting from WM_DESTROY\n"));
PostQuitMessage(0);
return 0;
case WM_QUERYENDSESSION:
{
//shutdown or reboot
if((lParam & ENDSESSION_LOGOFF) != ENDSESSION_LOGOFF)
{
fShutdownOrdered=TRUE;
Sleep(1000);
vnclog.Print(LL_INTERR, VNCLOG("SHUTDOWN OS detected\n"));
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_CLOSE \n"));
_this->m_server->KillAuthClients();
PostMessage(hwnd, WM_CLOSE, 0, 0);
break;
}
DWORD SessionID;
SessionID=GetCurrentSessionID();
vnclog.Print(LL_INTERR, VNCLOG("Session ID %i\n"),SessionID);
if (SessionID!=0)
{
fShutdownOrdered=TRUE;
Sleep(1000);
vnclog.Print(LL_INTERR, VNCLOG("WM_QUERYENDSESSION session!=0\n"));
vnclog.Print(LL_INTINFO, VNCLOG("KillAuthClients() ID_CLOSE \n"));
_this->m_server->KillAuthClients();
PostMessage(hwnd, WM_CLOSE, 0, 0);
}
}
break;
case WM_ENDSESSION:
vnclog.Print(LL_INTERR, VNCLOG("WM_ENDSESSION\n"));
break;
case WM_USERCHANGED:
// The current user may have changed.
{
strcpy(newuser,"");
if (vncService::CurrentUser((char *) &newuser, sizeof(newuser)))
{
vnclog.Print(LL_INTINFO,
VNCLOG("############### Usernames change: old=\"%s\", new=\"%s\"\n"),
_this->m_username, newuser);
// Check whether the user name has changed!
if (_stricmp(newuser, _this->m_username) != 0)
{
vnclog.Print(LL_INTINFO,
VNCLOG("user name has changed\n"));
// User has changed!
strcpy(_this->m_username, newuser);
// Redraw the tray icon and set it's state
_this->DelTrayIcon();
_this->AddTrayIcon();
_this->FlashTrayIcon(_this->m_server->AuthClientCount() != 0);
// We should load in the prefs for the new user
if (_this->m_properties.m_fUseRegistry)
{
_this->m_properties.Load(TRUE);
_this->m_propertiesPoll.Load(TRUE);
}
else
{
_this->m_properties.LoadFromIniFile();
_this->m_propertiesPoll.LoadFromIniFile();
}
}
}
}
return 0;
// [v1.0.2-jp1 fix] Don't show IME toolbar on right click menu.
case WM_INITMENU:
case WM_INITMENUPOPUP:
SendMessage(ImmGetDefaultIMEWnd(hwnd), WM_IME_CONTROL, IMC_CLOSESTATUSWINDOW, 0);
return 0;
default:
// Deal with any of our custom message types
// wa@2005 -- added support for the AutoReconnectId
// removed the previous code that used 999,999
if ( iMsg == MENU_AUTO_RECONNECT_MSG )
{
char szId[MAX_PATH] = {0};
UINT ret = 0;
if ( lParam != NULL )
{
ret = GlobalGetAtomName( (ATOM)lParam, szId, sizeof( szId ) );
GlobalDeleteAtom( (ATOM)lParam );
}
_this->m_server->AutoReconnect(true);
if ( ret > 0 )
_this->m_server->AutoReconnectId(szId);
return 0;
}
if ( iMsg == MENU_REPEATER_ID_MSG )
{
char szId[MAX_PATH] = {0};
UINT ret = 0;
if ( lParam != NULL )
{
ret = GlobalGetAtomName( (ATOM)lParam, szId, sizeof( szId ) );
GlobalDeleteAtom( (ATOM)lParam );
}
_this->m_server->IdReconnect(true);
if ( ret > 0 )
_this->m_server->AutoReconnectId(szId);
return 0;
}
if (iMsg == MENU_ADD_CLIENT_MSG)
{
/*
// sf@2005 - FTNoUserImpersonation
// Dirty trick to avoid to add a new MSG... no time
if (lParam == 998)
{
_this->m_server->FTUserImpersonation(false);
return 0;
}
*/
// Add Client message. This message includes an IP address
// of a listening client, to which we should connect.
//adzm 2009-06-20 - Check for special add repeater client message
if (wParam == 0xFFFFFFFF && lParam == 0xFFFFFFFF) {
vncConnDialog *newconn = new vncConnDialog(_this->m_server);
if (newconn)
{
if (IDOK != newconn->DoDialog()) {
if (SPECIAL_SC_PROMPT && _this->m_server->AuthClientCount() == 0 && _this->m_server->UnauthClientCount() == 0) {
PostMessage(hwnd, WM_COMMAND, ID_CLOSE, 0);
}
}
}
return 0;
}
// If there is no IP address then show the connection dialog
if (!lParam) {
vncConnDialog *newconn = new vncConnDialog(_this->m_server);
if (newconn)
{
newconn->DoDialog();
// winvnc -connect fixed
//CHECH memeory leak
// delete newconn;
}
return 0;
}
unsigned short nport = 0;
char *nameDup = 0;
char szAdrName[64];
char szId[MAX_PATH] = {0};
// sf@2003 - Values are already converted
if ((_this->m_server->AutoReconnect()|| _this->m_server->IdReconnect() )&& strlen(_this->m_server->AutoReconnectAdr()) > 0)
{
nport = _this->m_server->AutoReconnectPort();
strcpy(szAdrName, _this->m_server->AutoReconnectAdr());
}
else
{
// Get the IP address stringified
struct in_addr address;
address.S_un.S_addr = lParam;
char *name = inet_ntoa(address);
if (name == 0)
return 0;
nameDup = _strdup(name);
if (nameDup == 0)
return 0;
strcpy(szAdrName, nameDup);
// Free the duplicate name
if (nameDup != 0) free(nameDup);
// Get the port number
nport = (unsigned short)wParam;
if (nport == 0)
nport = INCOMING_PORT_OFFSET;
}
// wa@2005 -- added support for the AutoReconnectId
// (but it's not required)
bool bId = ( strlen(_this->m_server->AutoReconnectId() ) > 0);
if ( bId )
strcpy( szId, _this->m_server->AutoReconnectId() );
// sf@2003
// Stores the client adr/ports the first time we try to connect
// This way we can call this message again later to reconnect with the same values
if ((_this->m_server->AutoReconnect() || _this->m_server->IdReconnect())&& strlen(_this->m_server->AutoReconnectAdr()) == 0)
{
_this->m_server->AutoReconnectAdr(szAdrName);
_this->m_server->AutoReconnectPort(nport);
}
// Attempt to create a new socket
VSocket *tmpsock;
tmpsock = new VSocket;
if (tmpsock) {
// Connect out to the specified host on the VNCviewer listen port
tmpsock->Create();
if (tmpsock->Connect(szAdrName, nport)) {
if ( bId )
{
// wa@2005 -- added support for the AutoReconnectId
// Set the ID for this client -- code taken from vncconndialog.cpp (ln:142)
tmpsock->Send(szId,250);
tmpsock->SetTimeout(0);
// adzm 2009-07-05 - repeater IDs
// Add the new client to this server
// adzm 2009-08-02
_this->m_server->AddClient(tmpsock, TRUE, TRUE, 0, NULL, szId, szAdrName, nport);
} else {
// Add the new client to this server
// adzm 2009-08-02
_this->m_server->AddClient(tmpsock, TRUE, TRUE, 0, NULL, NULL, szAdrName, nport);
}
} else {
delete tmpsock;
_this->m_server->AutoConnectRetry();
}
}
return 0;
}
// Process FileTransfer asynchronous Send Packet Message
if (iMsg == FileTransferSendPacketMessage)
{
vncClient* pClient = (vncClient*) wParam;
if (_this->m_server->IsClient(pClient)) pClient->SendFileChunk();
}
// adzm 2009-07-05 - Tray icon balloon tips
if (iMsg == MENU_TRAYICON_BALLOON_MSG) {
omni_mutex_lock sync(_this->m_mutexTrayIcon);
// adzm 2009-07-05 - Tray icon balloon tips
if (_this->m_BalloonInfo) {
free(_this->m_BalloonInfo);
_this->m_BalloonInfo = NULL;
}
if (_this->m_BalloonTitle) {
free(_this->m_BalloonTitle);
_this->m_BalloonTitle = NULL;
}
char* szInfo = (char*)wParam;
char* szTitle = (char*)lParam;
if (szInfo && (strlen(szInfo) > 0) ) {
_this->m_BalloonInfo = _strdup(szInfo);
}
if (szTitle && (strlen(szTitle) > 0) ) {
_this->m_BalloonTitle = _strdup(szTitle);
}
if (szInfo) {
free(szInfo);
}
if (szTitle) {
free(szTitle);
}
if (_this->IsIconSet) {
_this->SendTrayMsg(NIM_MODIFY, _this->m_nid.hIcon == _this->m_winvnc_icon ? FALSE : TRUE);
}
}
}
// Message not recognised
return DefWindowProc(hwnd, iMsg, wParam, lParam);
}
