233 lines
5.5 KiB
C++
233 lines
5.5 KiB
C++
/*
|
|
* Copyright (c) 1995, 1996, 1997 Kungliga Tekniska Högskolan
|
|
* (Royal Institute of Technology, Stockholm, Sweden).
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* 3. Neither the name of the Institute nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
/* $Id: TelnetApp.cpp,v 1.1.1.1 2000/06/16 18:46:22 thorpej Exp $ */
|
|
|
|
// TelnetApp.cpp
|
|
// Author: Jörgen Karlsson - d93-jka@nada.kth.se
|
|
|
|
#include <Windows.h>
|
|
#include <winsock.h>
|
|
#include "TelnetApp.h"
|
|
#include "TelnetEngine.h"
|
|
#include "TerminalEngine.h"
|
|
#include "TelnetSession.h"
|
|
#include "Resource.h"
|
|
|
|
#include "KClient.h"
|
|
|
|
TelnetApp *theApp;
|
|
TelnetSession *theSession;
|
|
HANDLE theEvent[1];
|
|
int EventCount;
|
|
|
|
const int hostnameSZ = 64;
|
|
char hostname[hostnameSZ] = "";
|
|
char username[64] = "";
|
|
int telnet_port;
|
|
char telnet_port_str[64] = "";
|
|
|
|
BOOL CALLBACK
|
|
host_dialog_proc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
switch(uMsg) {
|
|
case WM_INITDIALOG:
|
|
SetDlgItemText(hwndDlg, IDC_EDIT1, hostname);
|
|
SetDlgItemText(hwndDlg, IDC_EDIT2, username);
|
|
SetDlgItemText(hwndDlg, IDC_EDIT3, "telnet");
|
|
return TRUE;
|
|
break;
|
|
case WM_COMMAND:
|
|
switch(wParam) {
|
|
case IDOK:
|
|
if(!GetDlgItemText(hwndDlg,IDC_EDIT1,
|
|
hostname ,hostnameSZ))
|
|
EndDialog(hwndDlg, IDCANCEL);
|
|
GetDlgItemText(hwndDlg, IDC_EDIT2,
|
|
username, sizeof(username));
|
|
GetDlgItemText(hwndDlg, IDC_EDIT3,
|
|
telnet_port_str, sizeof(telnet_port_str));
|
|
case IDCANCEL:
|
|
EndDialog(hwndDlg, wParam);
|
|
return TRUE;
|
|
break;
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
int WINAPI
|
|
WinMain(HINSTANCE hInst, HINSTANCE hPrevInst,
|
|
LPSTR CmdLine, int nShowCmd)
|
|
{
|
|
MSG msg;
|
|
TelnetApp app(hInst, CmdLine);
|
|
|
|
while(TRUE) {
|
|
GetMessage(&msg, 0,0,0);
|
|
switch(msg.message) {
|
|
case WM_QUIT:
|
|
return msg.wParam;
|
|
|
|
case WM_ENDSESSION:
|
|
theApp->CloseAllSessions();
|
|
return 0;
|
|
break;
|
|
|
|
default:
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
TelnetApp::TelnetApp(HINSTANCE hInst, char *hostname)
|
|
{
|
|
WORD version;
|
|
WSADATA data;
|
|
TelnetSession *session;
|
|
|
|
AppInstance = hInst;
|
|
theApp = this;
|
|
version = MAKEWORD(1, 1);
|
|
if(WSAStartup(version, &data) != 0)
|
|
exit(1);
|
|
|
|
session = new TelnetSession();
|
|
|
|
KClientGetUserName(username);
|
|
|
|
HKEY key;
|
|
|
|
RegCreateKeyEx(HKEY_CURRENT_USER,
|
|
"voodoo",
|
|
0,
|
|
"hostname",
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS,
|
|
NULL,
|
|
&key,
|
|
NULL);
|
|
|
|
char rhost[64];
|
|
if(*hostname == '\0') {
|
|
DWORD pcbData = sizeof(rhost);
|
|
|
|
if(RegQueryValueEx(key,
|
|
"hostname",
|
|
0,
|
|
NULL,
|
|
(unsigned char*)rhost,
|
|
&pcbData) == 0) {
|
|
hostname = rhost;
|
|
}
|
|
hostname = GetHostName(hostname);
|
|
}
|
|
|
|
if(hostname != NULL) {
|
|
strncpy(rhost, hostname, sizeof(rhost));
|
|
rhost[sizeof(rhost) - 1] = 0;
|
|
hostname = rhost;
|
|
RegSetValueEx(key,
|
|
"hostname",
|
|
0,
|
|
REG_SZ,
|
|
(unsigned char *)hostname,
|
|
strlen(hostname) + 1);
|
|
|
|
session = new TelnetSession();
|
|
if(session->Connect(hostname, username))
|
|
theSession = session;
|
|
} else
|
|
SessionClosed(0);
|
|
RegCloseKey(key);
|
|
}
|
|
|
|
TelnetApp::~TelnetApp(void)
|
|
{
|
|
if(WSACleanup() == SOCKET_ERROR)
|
|
exit(1);
|
|
}
|
|
|
|
void TelnetApp::Error(int ErrorCode, char *ErrorString)
|
|
{
|
|
Message(ErrorString);
|
|
exit(ErrorCode);
|
|
}
|
|
|
|
void TelnetApp::SessionClosed(TelnetSession *thisTelnetSession)
|
|
{
|
|
PostQuitMessage(0);
|
|
}
|
|
|
|
char* TelnetApp::GetHostName(char* defaultHost)
|
|
{
|
|
PHOSTENT host;
|
|
|
|
strcpy(hostname, defaultHost);
|
|
askforhost:
|
|
HWND wnd = GetActiveWindow();
|
|
HANDLE hInst = theApp->AppInstance;
|
|
switch(DialogBox(hInst,MAKEINTRESOURCE(IDD_DIALOG1),wnd,(DLGPROC)host_dialog_proc)) {
|
|
case IDOK:
|
|
if(!(host = gethostbyname(hostname))) {
|
|
TelnetApp::Message("Unknown host.");
|
|
goto askforhost;
|
|
}
|
|
return host->h_name;
|
|
case IDCANCEL:
|
|
default:
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
|
|
void TelnetApp::RegisterRecEvent(HANDLE RecEvent)
|
|
{
|
|
theEvent[0] = RecEvent;
|
|
EventCount = 1;
|
|
}
|
|
|
|
void TelnetApp::CloseAllSessions(void)
|
|
{
|
|
theSession->Close();
|
|
}
|
|
|
|
void TelnetApp::Message(char *text)
|
|
{
|
|
HWND wnd = GetActiveWindow();
|
|
MessageBox(wnd, text, "Voodoo message", MB_ICONERROR|MB_OK|MB_APPLMODAL);
|
|
}
|