Steve Grubb writes:
[oweals/busybox.git] / networking / telnet.c
index 92ddfaebfbb6baed526946ce99c8f31f80aa253c..6c5f3d15b64bf5f8cc7f6eb787acbc52ab0d9693 100644 (file)
@@ -28,6 +28,8 @@
  * Modified 2000/06/13 for inclusion into BusyBox by Erik Andersen <andersen@codepoet.org>
  * Modified 2001/05/07 to add ability to pass TTYPE to remote host by Jim McQuillan
  * <jam@ltsp.org>
+ * Modified 2004/02/11 to add ability to pass the USER variable to remote host
+ * by Fernando Silveira <swrh@gmx.net>
  *
  */
 
@@ -52,7 +54,7 @@ static const int DOTRACE = 1;
 #include <arpa/inet.h> /* for inet_ntoa()... */
 #define TRACE(x, y) do { if (x) printf y; } while (0)
 #else
-#define TRACE(x, y) 
+#define TRACE(x, y)
 #endif
 
 #if 0
@@ -97,8 +99,8 @@ static struct Globalvars {
        /* buffer to handle telnet negotiations */
        char    iacbuf[IACBUFSIZE];
        short   iaclen; /* could even use byte */
-       struct termios termios_def;     
-       struct termios termios_raw;     
+       struct termios termios_def;
+       struct termios termios_raw;
 } G;
 
 #define xUSE_GLOBALVAR_PTR /* xUSE... -> don't use :D (makes smaller code) */
@@ -129,6 +131,10 @@ static int one = 1;
 static char *ttype;
 #endif
 
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+static char *autologin;
+#endif
+
 #ifdef CONFIG_FEATURE_AUTOWIDTH
 static int win_width, win_height;
 #endif
@@ -137,7 +143,7 @@ static void doexit(int ev)
 {
        cookmode();
        exit(ev);
-}      
+}
 
 static void conescape(void)
 {
@@ -184,10 +190,10 @@ static void conescape(void)
 
        if (G.gotsig)
                cookmode();
-       
+
  rrturn:
        G.gotsig = 0;
-       
+
 }
 static void handlenetoutput(int len)
 {
@@ -355,6 +361,34 @@ static void putiac_subopt(byte c, char *str)
 }
 #endif
 
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+static void putiac_subopt_autologin(void)
+{
+       int len = strlen(autologin) + 6;        // (2 + 1 + 1 + strlen + 2)
+       char *user = "USER";
+
+       if (G.iaclen + len > IACBUFSIZE)
+               iacflush();
+
+       putiac(IAC);
+       putiac(SB);
+       putiac(TELOPT_NEW_ENVIRON);
+       putiac(TELQUAL_IS);
+       putiac(NEW_ENV_VAR);
+
+       while(*user)
+               putiac(*user++);
+
+       putiac(NEW_ENV_VALUE);
+
+       while(*autologin)
+               putiac(*autologin++);
+
+       putiac(IAC);
+       putiac(SE);
+}
+#endif
+
 #ifdef CONFIG_FEATURE_AUTOWIDTH
 static void putiac_naws(byte c, int x, int y)
 {
@@ -408,7 +442,7 @@ static void will_charmode(void)
        G.charmode = CHM_TRY;
        G.telflags |= (UF_ECHO | UF_SGA);
        setConMode();
-  
+
        putiac2(DO, TELOPT_ECHO);
        putiac2(DO, TELOPT_SGA);
        iacflush();
@@ -438,7 +472,7 @@ static inline void to_echo(void)
        /* if server requests ECHO, don't agree */
        if      (G.telwish == DO) {     putiac2(WONT, TELOPT_ECHO);     return; }
        else if (G.telwish == DONT)     return;
-  
+
        if (G.telflags & UF_ECHO)
        {
                if (G.telwish == WILL)
@@ -472,7 +506,7 @@ static inline void to_sga(void)
        else
                if (G.telwish == WONT)
                        return;
-  
+
        if ((G.telflags ^= UF_SGA) & UF_SGA) /* toggle */
                putiac2(DO, TELOPT_SGA);
        else
@@ -495,13 +529,27 @@ static inline void to_ttype(void)
 }
 #endif
 
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+static inline void to_new_environ(void)
+{
+       /* Tell server we will (or will not) do AUTOLOGIN */
+
+       if (autologin)
+               putiac2(WILL, TELOPT_NEW_ENVIRON);
+       else
+               putiac2(WONT, TELOPT_NEW_ENVIRON);
+
+       return;
+}
+#endif
+
 #ifdef CONFIG_FEATURE_AUTOWIDTH
 static inline void to_naws(void)
-{ 
+{
        /* Tell server we will do NAWS */
        putiac2(WILL, TELOPT_NAWS);
        return;
-}         
+}
 #endif
 
 static void telopt(byte c)
@@ -513,6 +561,9 @@ static void telopt(byte c)
 #ifdef CONFIG_FEATURE_TELNET_TTYPE
                case TELOPT_TTYPE:              to_ttype();break;
 #endif
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+               case TELOPT_NEW_ENVIRON:        to_new_environ();       break;
+#endif
 #ifdef CONFIG_FEATURE_AUTOWIDTH
                case TELOPT_NAWS:               to_naws();
                                                                putiac_naws(c, win_width, win_height);
@@ -539,6 +590,11 @@ static int subneg(byte c)
                else
                if (c == TELOPT_TTYPE)
                        putiac_subopt(TELOPT_TTYPE,ttype);
+#endif
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+               else
+               if (c == TELOPT_NEW_ENVIRON)
+                       putiac_subopt_autologin();
 #endif
                break;
        case TS_SUB2:
@@ -561,7 +617,7 @@ static void fgotsig(int sig)
 static void rawmode(void)
 {
        tcsetattr(0, TCSADRAIN, &G.termios_raw);
-}      
+}
 
 static void cookmode(void)
 {
@@ -570,16 +626,18 @@ static void cookmode(void)
 
 extern int telnet_main(int argc, char** argv)
 {
-       char *host;
-       char *port;
        int len;
        struct sockaddr_in s_in;
 #ifdef USE_POLL
        struct pollfd ufds[2];
-#else  
+#else
        fd_set readfds;
        int maxfd;
-#endif 
+#endif
+
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+       int opt;
+#endif
 
 #ifdef CONFIG_FEATURE_AUTOWIDTH
        get_terminal_width_height(0, &win_width, &win_height);
@@ -593,16 +651,41 @@ extern int telnet_main(int argc, char** argv)
 
        if (tcgetattr(0, &G.termios_def) < 0)
                exit(1);
-       
+
        G.termios_raw = G.termios_def;
        cfmakeraw(&G.termios_raw);
-       
-       if (argc < 2)   bb_show_usage();
-       port = (argc > 2)? argv[2] : "23";
-       
-       host = argv[1];
-       
-       bb_lookup_host(&s_in, host, port);
+
+       if (argc < 2)
+               bb_show_usage();
+
+#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN
+       autologin = NULL;
+       while ((opt = getopt(argc, argv, "al:")) != EOF) {
+               switch (opt) {
+                       case 'l':
+                               autologin = bb_xstrdup(optarg);
+                               break;
+                       case 'a':
+                               autologin = getenv("USER");
+                               break;
+                       case '?':
+                               bb_show_usage();
+                               break;
+               }
+       }
+       if (optind < argc) {
+               bb_lookup_host(&s_in, argv[optind++]);
+               s_in.sin_port = bb_lookup_port((optind < argc) ? argv[optind++] :
+                               "telnet", "tcp", 23);
+               if (optind < argc)
+                       bb_show_usage();
+       } else
+               bb_show_usage();
+#else
+       bb_lookup_host(&s_in, argv[1]);
+       s_in.sin_port = bb_lookup_port((argc == 3) ? argv[2] : "telnet", "tcp", 23);
+#endif
+
        G.netfd = xconnect(&s_in);
 
        setsockopt(G.netfd, SOL_SOCKET, SO_KEEPALIVE, &one, sizeof one);
@@ -612,22 +695,22 @@ extern int telnet_main(int argc, char** argv)
 #ifdef USE_POLL
        ufds[0].fd = 0; ufds[1].fd = G.netfd;
        ufds[0].events = ufds[1].events = POLLIN;
-#else  
+#else
        FD_ZERO(&readfds);
        FD_SET(0, &readfds);
        FD_SET(G.netfd, &readfds);
        maxfd = G.netfd + 1;
 #endif
-       
+
        while (1)
        {
 #ifndef USE_POLL
                fd_set rfds = readfds;
-               
+
                switch (select(maxfd, &rfds, NULL, NULL, NULL))
 #else
                switch (poll(ufds, 2, -1))
-#endif                 
+#endif
                {
                case 0:
                        /* timeout */
@@ -642,9 +725,9 @@ extern int telnet_main(int argc, char** argv)
 
 #ifdef USE_POLL
                        if (ufds[0].revents) /* well, should check POLLIN, but ... */
-#else                          
+#else
                        if (FD_ISSET(0, &rfds))
-#endif                         
+#endif
                        {
                                len = read(0, G.buf, DATABUFSIZE);
 
@@ -652,15 +735,15 @@ extern int telnet_main(int argc, char** argv)
                                        doexit(0);
 
                                TRACE(0, ("Read con: %d\n", len));
-                               
+
                                handlenetoutput(len);
                        }
 
 #ifdef USE_POLL
                        if (ufds[1].revents) /* well, should check POLLIN, but ... */
-#else                          
+#else
                        if (FD_ISSET(G.netfd, &rfds))
-#endif                         
+#endif
                        {
                                len = read(G.netfd, G.buf, DATABUFSIZE);