prepared for kinsole 1.1 release1.1

git-svn-id: http://svn.klever.net/kin/kinsole/trunk@7 fe716a7a-6dde-0310-88d9-d003556173a8

1 files changed, 3 insertions, 3 deletions

diff --git a/kinsole.cpp b/kinsole.cpp
index 463db1c..01286f3 100644
--- a/kinsole.cpp
+++ b/kinsole.cpp
@@ -1,211 +1,211 @@
 #include <stdio.h>

 

         #ifdef_DEBUG

 #include <afxwin.h>

 #else

 #define ASSERT(f)          ((void)0)

 #define VERIFY(f)          ((void)(f))

 #define TRACE0(sz)

 #define TRACE1(sz, p1)

 #define TRACE2(sz, p1, p2)

 #define TRACE3(sz, p1, p2, p3)

 #endif

 

 #include <winsock.h>

 #include "resource.h"

 #include "windowsx.h"

 

         #define         DAMN_KIN_NAME   "KINSole"

-        #define         DAMN_KIN_VERSION"v1.00 Beta 3"

+        #define         DAMN_KIN_VERSION"1.1"

 

 #ifdef _DEBUG

 #define new DEBUG_NEW

 #undef THIS_FILE

 static char THIS_FILE[] = __FILE__;

 #endif

 

 enum {

         WM_USERSOCKET = WM_USER+16,

         WM_USERKEY,

         WM_USERNOP

 };

 enum {

         tnIAC = 255,

         tnSE = 240,

         tnNOP = 241,

         tnDM = 242,

         tnBreak = 243,

         tnIP = 244,

         tnAO = 245,

         tnAYT = 246,

         tnEC = 247,

         tnEL = 248,

         tnGA = 249,

         tnSB = 250,

         tnWILL = 251,

         tnWONT = 252,

         tnDO = 253,

         tnDONT = 254

 };

 

 WSADATA wsaData;

 ATOM wsaWC = NULL;

 HWND wsaW = NULL;

 

 CHAR remoteHost[256];

 CHAR remoteProt[256];

 sockaddr_in remoteSIN;

 

 

 SOCKET telnetSocket;

 HANDLE hConsoleInput;

 HANDLE hConsoleOutput;

 HWND hConsoleWindow;

 HANDLE hConsoleThread;

 DWORD consoleThreadID;

 HANDLE hDispatchThread;

 DWORD dispatchThreadID;

 BOOL bTelnet,bTermPulled;

 

         enum_cState {

         cstateNone = 0, cstateIAC, cstateDO, cstateSB, cstateSBData,

         cstateSBDataIAC, cstateWILL, cstateDONT,cstateWONT,

         }connState = cstateNone;

 BYTE negOption = 0;

 

 

 BOOL SelectSocket() {

         return WSAAsyncSelect(telnetSocket,wsaW,WM_USERSOCKET,FD_READ|FD_OOB|FD_CLOSE)!=SOCKET_ERROR;

 }

 

 BOOL ShowWill(BYTE o)

 {

         TRACE1("We're WILLing to %d\n",(WORD)o);

 static

 BYTE d[3] = {tnIAC,tnWILL,0};

         d[2] = o;

 BOOL rv = send(telnetSocket,(char*)d,sizeof(d),0)==sizeof(d);

         SelectSocket();

         return rv;

 }

 BOOL ShowUnwill(BYTE o)

 {

         TRACE1("We're NOT WILLing to %d\n",(WORD)o);

 static

 BYTE d[3] = {tnIAC,tnWONT,0};

         d[2] = o;

 BOOL rv = send(telnetSocket,(char*)d,sizeof(d),0)==sizeof(d);

         SelectSocket();

         return rv;

 }

 BOOL BegDo(BYTE o)

 {

         TRACE1("We beg to DO %d\n",(WORD)o);

 static

 BYTE d[3] = {tnIAC,tnDO,0};

         d[2] = o;

 BOOL rv = send(telnetSocket,(char*)d,sizeof(d),0)==sizeof(d);

         SelectSocket();

         return rv;

 }

 BOOL BegDont(BYTE o)

 {

         TRACE1("We beg DONT'T %d\n",(WORD)o);

 static

 BYTE d[3] = {tnIAC,tnDONT,0};

         d[2] = o;

 BOOL rv = send(telnetSocket,(char*)d,sizeof(d),0)==sizeof(d);

         SelectSocket();

         return rv;

 }

 BOOL SubNegotiate(BYTE o,LPBYTE data,UINT size)

 {

 LPBYTE d = new BYTE[3+size*2+2];

 int ds = 0;

         d[ds++]=tnIAC; d[ds++]=tnSB; d[ds++]=o;

         for(UINT tmp=0;tmp<size;tmp++)

                 if(data[tmp]!=tnIAC)

                         d[ds++]=data[tmp];

                 else{

                         d[ds++]=tnIAC; d[ds++]=tnIAC;

                 }

         d[ds++]=tnIAC;d[ds++]=tnSE;

 BOOL rv = send(telnetSocket,(char*)d,ds,0)==ds;

         delete d;

         SelectSocket();

         return rv;

 }

 BOOL SendLiteral(CHAR c)

 {

 BYTE d[2] = {tnIAC,0};

 BOOL rv = FALSE;

         if(c==tnIAC){

                 d[1]=c;

                 rv = send(telnetSocket,(char*)d,2,0)==2;

         }else

                 rv = send(telnetSocket,&c,1,0)==1;

         return rv;

 }

 BOOL SendLiteral(LPCTSTR c,UINT size)

 {

         for(UINT tmp=0;tmp<size;tmp++)

                 SendLiteral(c[tmp]);

         return TRUE;

 }

 BOOL SendLiteral(LPCTSTR c)

 {

         return SendLiteral(c,strlen(c));

 }

 BOOL SendCommand(BYTE c)

 {

         TRACE1("Issuing %d command\n",(WORD)c);

 static

 BYTE d[2] = {tnIAC,0};

         d[1] = c;

 BOOL rv = send(telnetSocket,(char*)d,sizeof(d),0)==sizeof(d);

         SelectSocket();

         return rv;

 

 }

 

 BOOL CALLBACK consoleCtrlHandler(DWORD dwCtrlType)

 {

         switch(dwCtrlType){

         case CTRL_BREAK_EVENT:

                 SendCommand(tnIP);

                 return TRUE;

         case CTRL_CLOSE_EVENT:

         case CTRL_LOGOFF_EVENT:

         case CTRL_SHUTDOWN_EVENT:

                 PostMessage(wsaW,WM_QUIT,0,0);

                 return TRUE;

         }

         return FALSE;

 }

 

 #include "options.cpp"

 

 BOOL ProcessConsoleInput()

 {

 INPUT_RECORD ir[512];

 DWORD got;

         while(GetNumberOfConsoleInputEvents(hConsoleInput,&got) && got){

                 VERIFY(ReadConsoleInput(hConsoleInput,ir,(sizeof(ir)/sizeof(*ir)),&got));

                 for(DWORD tmp=0;tmp<got;tmp++){

                         if(ir[tmp].EventType==KEY_EVENT && ir[tmp].Event.KeyEvent.bKeyDown){

                                 if((

                                         ir[tmp].Event.KeyEvent.wVirtualKeyCode=='X' ||

                                         ir[tmp].Event.KeyEvent.wVirtualKeyCode=='Q'

                                         ) && ir[tmp].Event.KeyEvent.dwControlKeyState&(LEFT_ALT_PRESSED|RIGHT_ALT_PRESSED)

                                 ){

                                         PostMessage(wsaW,WM_QUIT,0,0);

                                         return FALSE;

                                 }else if(

                                         ir[tmp].Event.KeyEvent.wVirtualKeyCode=='O'

                                         && ir[tmp].Event.KeyEvent.dwControlKeyState&(LEFT_ALT_PRESSED|RIGHT_ALT_PRESSED)

                                 ){

                                         SendCommand(tnAO);

                                 }else if(

                                         ir[tmp].Event.KeyEvent.wVirtualKeyCode=='Y'

                                         && ir[tmp].Event.KeyEvent.dwControlKeyState&(LEFT_ALT_PRESSED|RIGHT_ALT_PRESSED)

                                 ){

@@ -361,376 +361,376 @@ void ProcessIACByte(BYTE c)
         case tnSE:

                 TRACE0("SE\n");break;

         case tnNOP:

                 TRACE0("NOP\n");break;

         case tnDM:

                 TRACE0("DM\n");break;

         case tnBreak:

                 TRACE0("Break\n");break;

         case tnIP:

                 TRACE0("IP\n");break;

         case tnAO:

                 TRACE0("AO\n");break;

         case tnAYT:

                 TRACE0("AYT\n");break;

         case tnEC:

                 TRACE0("EC\n");break;

         case tnEL:

                 TRACE0("EL\n");break;

         case tnGA:

                 TRACE0("GA\n");break;

         case tnSB:

                 connState = cstateSB;

                 break;

         case tnWILL:

                 connState = cstateWILL;

                 break;

         case tnWONT:

                 connState = cstateWONT;

                 break;

         case tnDO:

                 connState = cstateDO;

                 break;

         case tnDONT:

                 connState = cstateDONT;

                 break;

         default:

                 TRACE1("Unknown OpCode = %d\n",(WORD)c);

                 break;

         }

 }

 

 void ProcessNetByte(BYTE c)

 {

         //TRACE1("<%d>",connState);

         switch(connState){

         case cstateWONT:

                 ProcessWONT(c);

                 break;

         case cstateDO:

                 ProcessDO(c);

                 break;

         case cstateWILL:

                 ProcessWILL(c);

                 break;

         case cstateDONT:

                 ProcessDONT(c);

                 break;

         case cstateSB:

                 negOption = c;

                 connState = cstateSBData;

                 break;

         case cstateSBData:

         case cstateSBDataIAC:

                 ProcessSBData(c);

                 break;

         case cstateIAC:

                 ProcessIACByte(c);

                 break;

         case cstateNone:

         default:

                 ASSERT(connState==cstateNone);

                 if(c==tnIAC)

                         connState=cstateIAC;

                 else

                         TerminalOut(c);

                 break;

         }

 }

 

 LRESULT WSMessage(WPARAM wP,LPARAM lP)

 {

         if(WSAGETSELECTERROR(lP)){

                 TRACE0("SelectError\n");

                 PostMessage(wsaW,WM_QUIT,0,0);

                 return 0;

         }

         if(WSAGETSELECTEVENT(lP)&FD_READ){

         //??    TRACE0("FD_READ\n");

         BYTE input[80*12];

         int got;

                 TerminalPreO();

         //??            TRACE0("rv\n");

                         got=recv(telnetSocket,(CHAR*)input,sizeof(input),0);

         //??            TRACE1("/rv %d\n",got);

                         for(int tmp=0;tmp<got;tmp++)

                                 ProcessNetByte(input[tmp]);

                 TerminalPostO();

         //??    TRACE0("/FD_READ\n");

                 return 0;

         }

         if(WSAGETSELECTEVENT(lP)&FD_OOB){

                 TRACE0("OOB\n");

         }

         if(WSAGETSELECTEVENT(lP)&FD_CLOSE){

                 TRACE0("CLOSE\n");

                 PostMessage(wsaW,WM_QUIT,0,0);

                 return 0;

         }

         VERIFY(SelectSocket());

         return 0;

 }

 

 LRESULT CALLBACK WSWndProc(HWND hWnd,UINT uMsg,WPARAM wParam,LPARAM lParam)

 {

         switch(uMsg){

         case WM_USERSOCKET:

                 return WSMessage(wParam,lParam);

         case WM_USERKEY:

                 return ProcessConsoleInput();

         default:

                 TRACE0("DEFWINDOWPROC\n");

                 return ::DefWindowProc(hWnd,uMsg,wParam,lParam);

         }

         return 0;

 }

 

 

 BOOL InitializeWinsock()

 {

         if(WSAStartup(0x101,&wsaData)){

                 printf("Failed to initialize winsock services\n");

                 return FALSE;

         }

 WNDCLASS wc;

         memset(&wc,0,sizeof(wc));

         wc.lpfnWndProc=WSWndProc; wc.hInstance=::GetModuleHandle(NULL);

         wc.lpszClassName = "_WSTFWC_";

         wsaWC = RegisterClass(&wc);

         if(!wsaWC){

                 printf("Failed to initialize winsock services - 1\n");

                 return FALSE;

         }

         wsaW = ::CreateWindow("_WSTFWC_","KIN Sole Mio",0,0,0,0,0,NULL,NULL,::GetModuleHandle(NULL),NULL);

         if(!wsaW){

                 printf("Failed to initialize winsock services\n");

                 return FALSE;

         }

         return TRUE;

 }

 void DeinitializeWinsock()

 {

         if(wsaW)

                 ::DestroyWindow(wsaW);

         wsaW=NULL;

         if(wsaWC)

                 ::UnregisterClass("_WSTFWC_",::GetModuleHandle(NULL));

         wsaWC=NULL;

         WSACleanup();

 }

 

 HWND GetThisConsoleWnd()

 {

 DWORD pid = GetCurrentProcessId();

 CHAR title[512];

 CHAR* t = title;

         if(!GetConsoleTitle(title,sizeof(title)))

                 t = NULL;

 HWND hrv = FindWindowEx(NULL,NULL,"tty",t);

 HWND nopro = NULL;

 UINT nopros=0;

         do{

         DWORD wpid;

                 if(!GetWindowThreadProcessId(hrv,&wpid))

                         continue;

                 if(wpid==pid)

                         return hrv;

                 nopro=hrv;

                 nopros++;

                 hrv = FindWindowEx(NULL,hrv,"tty",t);

         }while(hrv);

         if(nopros==1){

                 ASSERT(nopro);

                 return nopro;

         }

         return NULL;

 }

 

 main(int argc,char*argv[])

 {

         if(argc<2){

 usagebye:

                 printf(

-                        DAMN_KIN_NAME " " DAMN_KIN_VERSION ", Copyright (c) 1998, 2002 Klever Group (http://www.klever.net/)\n\n"

+                        DAMN_KIN_NAME " " DAMN_KIN_VERSION ", Copyright (c) 1998-2004 Klever Group (http://www.klever.net/)\n\n"

                         "Usage:\t" DAMN_KIN_NAME " [<options> ]<host-name/ip-address>[ <port>]\n\n"

                         "Options are:\n"

                         "-r## or -##\tSet number of rows in console screenbuffer\n"

                         "-c##\t\tSet number of columns in console screenbuffer\n"

                                 "\tnote: changing console screenbuffer size may not work properly\n"

                                 "\twhen in full-screen mode\n"

                         "-l<user>\tPass username to remote server in environment\n"

                         "-e<var>=<val>\tPass environment variable to remote server\n"

                         "-v<var>=<val>\tPass user environment variable to remote server\n"

                         "-t<termtype>\tChange preferred terminal type\n"

                                 "\tnote: there are only two different terminal emulations in this\n"

-                                "\tbeta release - one for dumb terminal and one for vt terminal\n"

+                                "\trelease - one for dumb terminal and one for vt terminal\n"

                 );

                 CleanEnvars();

                 return 1;

         }

         if(!InitializeWinsock()){

                 DeinitializeWinsock();

                 return 2;

         }

 CONSOLE_SCREEN_BUFFER_INFO csbi;

 int ac = 0;

 CHAR *ho = NULL, *po = NULL;

 HANDLE hConsole = CreateFile("CONOUT$",GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,

                                                 NULL,OPEN_EXISTING,0,NULL);

         ASSERT(hConsole);

         // *** GetStdHandle(STD_OUTPUT_HANDLE);// *?*?* Do something about redirections and not only here.

         VERIFY(GetConsoleScreenBufferInfo(hConsole,&csbi));

 BOOL bSized=FALSE;

         bTermPulled=FALSE;

         for(int tmp=1;tmp<argc;tmp++){

         char* v = argv[tmp];

                 ASSERT(v && *v);

                 if((*v)=='/' || (*v)=='-'){

                 int lines = atoi(&v[1]);

                         if(lines){

                                 if(lines<csbi.dwSize.Y){

                                 SMALL_RECT wi = {0,0,csbi.dwSize.X-1,lines-1};

                                         VERIFY(SetConsoleWindowInfo(hConsole,TRUE,&wi));

                                 }

                         COORD ns = {csbi.dwSize.X,lines};

                                 if(SetConsoleScreenBufferSize(hConsole,ns))

                                         bSized=TRUE;

                         }else if(v[1]=='r'){

                         int lines = atoi(&v[2]);

                                 if(lines){

                                         if(lines<csbi.dwSize.Y){

                                         SMALL_RECT wi = {0,0,csbi.dwSize.X-1,lines-1};

                                                 VERIFY(SetConsoleWindowInfo(hConsole,TRUE,&wi));

                                         }

                                 COORD ns = {csbi.dwSize.X,lines};

                                         if(SetConsoleScreenBufferSize(hConsole,ns))

                                                 bSized=TRUE;

                                 }else

                                         goto usagebye;

                         }else if(v[1]=='c'){

                         int rows = atoi(&v[2]);

                                 if(rows){

                                         if(rows<csbi.dwSize.X){

                                         SMALL_RECT wi = {0,0,rows-1,csbi.dwSize.Y-1};

                                                 VERIFY(SetConsoleWindowInfo(hConsole,TRUE,&wi));

                                         }

                                 COORD ns = {rows,csbi.dwSize.Y};

                                         if(SetConsoleScreenBufferSize(hConsole,ns))

                                                 bSized=TRUE;

                                 }else

                                         goto usagebye;

                         }else if(v[1]=='l'){

                         CHAR* vv = &v[2];

                                 VERIFY(AddEnvar(nesbVar,"USER",vv));

                                 }else if(v[1]=='e'){    // -e<name>=<value>VAR

                         CHAR* n = &v[2];

                         CHAR* vv = strchr(&v[2],'=');

                                 if(!vv)

                                         goto usagebye;

                                 *(vv++)=0;

                                 VERIFY(AddEnvar(nesbVar,n,vv));

                                 }else if(v[1]=='v'){    // -v<name>=<value>USERVAR

                         CHAR* n = &v[2];

                         CHAR* vv = strchr(n,'=');

                                 if(!vv)

                                         goto usagebye;

                                 *(vv++)=0;

                                 VERIFY(AddEnvar(nesbUserVar,n,vv));

                                 }else if(v[1]=='t'){// -t<ttype>  -t<tname>=<ttype>

                         CHAR* n = &v[2];

                         CHAR* nn = strchr(n,'=');

                                 if(nn){

                                         *(nn++)=0;

                                         if(!*nn)

                                                 nn=NULL;

                                 }

                                 if(!TerminalPullType(nn?nn:n,nn?n:NULL)){

                                         printf("Available terminal types are:");

                                         TerminalPrintTypes();

                                         printf("\n");

                                         goto usagebye;

                                 }

                                 bTermPulled=TRUE;

                         }else if(v[1]=='#'){

                         int cp = atoi(&v[2]);

         #ifdef_DEBUG

                                 TRACE2("SetCP(%d)=%d\n",cp,SetConsoleCP(cp));

                                 TRACE2("SetOutCP(%d)=%d\n",cp,SetConsoleOutputCP(cp));

 #else

                                 SetConsoleCP(cp);

                                 SetConsoleOutputCP(cp);

 #endif

                                 TRACE2("CP,OCP=%d,%d\n",GetConsoleCP(),GetConsoleOutputCP());

                         }else

                                 goto usagebye;

                 }else{

                         if(ac==0){

                                 ho = v;

                                 ac++;

                         }else if(ac==1){

                                 po = v;

                                 ac++;

                         }else

                                 goto usagebye;

                 }

         }

         if(!ho)

                 goto usagebye;

 servent* se = getservbyname("telnet","tcp");

 UINT port = po?atol(po):(se?ntohs(se->s_port):23);

         if(port==23 || (se && port==ntohs(se->s_port)))

                 bTelnet = TRUE;

         if(po && !port){

                 se = getservbyname(po,"tcp");

                 if(!se){

                         printf("Failed to resolve tcp-service port name\n");

                         DeinitializeWinsock();

                         return 2;

                 }

                 port = ntohs(se->s_port);

                 if(!stricmp(po,"telnet"))

                         bTelnet = TRUE;

                 else

                         bTelnet = FALSE;

         }else{

                 se = getservbyport(htons(port),"tcp");

                 if(se){

                         po = se->s_name;

                         if(!stricmp(po,"telnet"))

                                 bTelnet=TRUE;

                 }else{

                         VERIFY(_itoa(port,remoteProt,10));

                         po = NULL;

                         bTelnet=FALSE;

                 }

         }

         if(po)

                 strcpy(remoteProt,po);

 HICON hIcon = LoadIcon(GetModuleHandle(NULL),MAKEINTRESOURCE(IDI_IKON));

 LONG oldBIcon = NULL, oldSIcon = NULL;

         hConsoleWindow = GetThisConsoleWnd();

         if(hConsoleWindow){

                 oldBIcon = SendMessage(hConsoleWindow,WM_SETICON,ICON_BIG,(LPARAM)hIcon);

                 oldSIcon = SendMessage(hConsoleWindow,WM_SETICON,ICON_SMALL,(LPARAM)hIcon);

         }

         Telnet(ho,port);

         CleanEnvars();

         if(hConsoleWindow){

                 SendMessage(hConsoleWindow,WM_SETICON,ICON_BIG,(LPARAM)oldBIcon);

                 SendMessage(hConsoleWindow,WM_SETICON,ICON_SMALL,(LPARAM)oldSIcon);

         }

         Sleep(150);

         if(bSized){

         CONSOLE_SCREEN_BUFFER_INFO CSBI;

                 VERIFY(GetConsoleScreenBufferInfo(hConsole,&CSBI));

                 if(CSBI.dwSize.Y>csbi.dwSize.Y || CSBI.dwSize.X>csbi.dwSize.X){

                 SMALL_RECT wi = {0,0,csbi.dwSize.X-1,csbi.dwSize.Y-1};

                         VERIFY(SetConsoleWindowInfo(hConsole,TRUE,&wi));

                 }

         COORD ns = {csbi.dwSize.X,csbi.dwSize.Y};

                 VERIFY(SetConsoleScreenBufferSize(hConsole,ns));

         }

         Sleep(100);

         DeinitializeWinsock();

         Sleep(100);

         return 0;

 }