Ensure all backends _remember_ the connection has closed after
[u/mdw/putty] / telnet.c
index 00b081b..cd70d9b 100644 (file)
--- a/telnet.c
+++ b/telnet.c
@@ -1,7 +1,6 @@
 #include <windows.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <winsock.h>
 
 #include "putty.h"
 
@@ -12,7 +11,7 @@
 #define TRUE 1
 #endif
 
-static SOCKET s = INVALID_SOCKET;
+static Socket s = NULL;
 
 #define        IAC     255             /* interpret as command: */
 #define        DONT    254             /* you are not to use option */
@@ -133,49 +132,17 @@ static struct Opt *opts[] = {
     &o_we_sga, &o_they_sga, NULL
 };
 
-#if 0
-static int in_synch;
-#endif
+static int echoing = TRUE, editing = TRUE;
 
+static int in_synch;
 static int sb_opt, sb_len;
 static char *sb_buf = NULL;
 static int sb_size = 0;
 #define SB_DELTA 1024
 
-static void try_write (void) {
-    while (outbuf_head != outbuf_reap) {
-       int end = (outbuf_reap < outbuf_head ? outbuf_head : OUTBUF_SIZE);
-       int len = end - outbuf_reap;
-       int ret;
-
-       ret = send (s, outbuf+outbuf_reap, len, 0);
-       if (ret > 0)
-           outbuf_reap = (outbuf_reap + ret) & OUTBUF_MASK;
-       if (ret < len)
-           return;
-    }
-}
-
-static void s_write (void *buf, int len) {
-    unsigned char *p = buf;
-    while (len--) {
-       int new_head = (outbuf_head + 1) & OUTBUF_MASK;
-       if (new_head != outbuf_reap) {
-           outbuf[outbuf_head] = *p++;
-           outbuf_head = new_head;
-       }
-    }
-    try_write();
-}
-
-static void c_write (char *buf, int len) {
-    while (len--) {
-       int new_head = (inbuf_head + 1) & INBUF_MASK;
-       if (new_head != inbuf_reap) {
-           inbuf[inbuf_head] = *buf++;
-           inbuf_head = new_head;
-       }
-    }
+static void c_write1(int c) {
+    char cc = (char)c;
+    from_backend(0, &cc, 1);
 }
 
 static void log_option (char *sender, int cmd, int option) {
@@ -184,14 +151,14 @@ static void log_option (char *sender, int cmd, int option) {
            (cmd == WILL ? "WILL" : cmd == WONT ? "WONT" :
             cmd == DO ? "DO" : cmd == DONT ? "DONT" : "<??>"),
            telopt(option));
-    lognegot(buf);
+    logevent(buf);
 }
 
 static void send_opt (int cmd, int option) {
     unsigned char b[3];
 
     b[0] = IAC; b[1] = cmd; b[2] = option;
-    s_write (b, 3);
+    sk_write(s, b, 3);
     log_option("client", cmd, option);
 }
 
@@ -201,6 +168,17 @@ static void deactivate_option (struct Opt *o) {
     o->state = REALLY_INACTIVE;
 }
 
+/*
+ * Generate side effects of enabling or disabling an option.
+ */
+static void option_side_effects(struct Opt *o, int enabled) {
+    if (o->option == TELOPT_ECHO && o->send == DO)
+        echoing = !enabled;
+    else if (o->option = TELOPT_SGA && o->send == DO)
+        editing = !enabled;
+    ldisc_send(NULL, 0);               /* cause ldisc to notice the change */
+}
+
 static void activate_option (struct Opt *o) {
     if (o->send == WILL && o->option == TELOPT_NAWS)
        telnet_size();
@@ -213,6 +191,7 @@ static void activate_option (struct Opt *o) {
         */
        deactivate_option (o->option==TELOPT_NEW_ENVIRON ? &o_oenv : &o_nenv);
     }
+    option_side_effects(o, 1);
 }
 
 static void refused_option (struct Opt *o) {
@@ -221,6 +200,7 @@ static void refused_option (struct Opt *o) {
        send_opt (WILL, TELOPT_OLD_ENVIRON);
        o_oenv.state = REQUESTED;
     }
+    option_side_effects(o, 0);
 }
 
 static void proc_rec_opt (int cmd, int option) {
@@ -255,6 +235,7 @@ static void proc_rec_opt (int cmd, int option) {
              case ACTIVE:
                (*o)->state = INACTIVE;
                send_opt ((*o)->nsend, option);
+                option_side_effects(*o, 0);
                break;
              case INACTIVE:
              case REALLY_INACTIVE:
@@ -284,12 +265,12 @@ static void process_subneg (void) {
            strcpy(b+4, cfg.termspeed);
            n = 4 + strlen(cfg.termspeed);
            b[n] = IAC; b[n+1] = SE;
-           s_write (b, n+2);
-           lognegot("server:\tSB TSPEED SEND");
+           sk_write(s, b, n+2);
+           logevent("server:\tSB TSPEED SEND");
            sprintf(logbuf, "client:\tSB TSPEED IS %s", cfg.termspeed);
-           lognegot (logbuf);
+           logevent (logbuf);
        } else
-           lognegot ("server:\tSB TSPEED <something weird>");
+           logevent ("server:\tSB TSPEED <something weird>");
        break;
       case TELOPT_TTYPE:
        if (sb_len == 1 && sb_buf[0] == TELQUAL_SEND) {
@@ -300,13 +281,13 @@ static void process_subneg (void) {
                b[n+4] = (cfg.termtype[n] >= 'a' && cfg.termtype[n] <= 'z' ?
                          cfg.termtype[n] + 'A'-'a' : cfg.termtype[n]);
            b[n+4] = IAC; b[n+5] = SE;
-           s_write (b, n+6);
+           sk_write(s, b, n+6);
            b[n+4] = 0;
-           lognegot("server:\tSB TTYPE SEND");
+           logevent("server:\tSB TTYPE SEND");
            sprintf(logbuf, "client:\tSB TTYPE IS %s", b+4);
-           lognegot(logbuf);
+           logevent(logbuf);
        } else
-           lognegot("server:\tSB TTYPE <something weird>\r\n");
+           logevent("server:\tSB TTYPE <something weird>\r\n");
        break;
       case TELOPT_OLD_ENVIRON:
       case TELOPT_NEW_ENVIRON: 
@@ -316,7 +297,7 @@ static void process_subneg (void) {
            char logbuf[50];
            p++;
            sprintf (logbuf, "server:\tSB %s SEND", telopt(sb_opt));
-           lognegot (logbuf);
+           logevent (logbuf);
            if (sb_opt == TELOPT_OLD_ENVIRON) {
                if (cfg.rfc_environ) {
                    value = RFC_VALUE;
@@ -365,10 +346,10 @@ static void process_subneg (void) {
                while (*e) b[n++] = *e++;
            }
            b[n++] = IAC; b[n++] = SE;
-           s_write (b, n);
+           sk_write(s, b, n);
            sprintf(logbuf, "client:\tSB %s IS %s", telopt(sb_opt),
                    n==6 ? "<nothing>" : "<stuff>");
-           lognegot (logbuf);
+           logevent (logbuf);
        }
        break;
     }
@@ -380,7 +361,6 @@ static enum {
 } telnet_state = TOPLEVEL;
 
 static void do_telnet_read (char *buf, int len) {
-    unsigned char b[10];
 
     while (len--) {
        int c = (unsigned char) *buf++;
@@ -393,11 +373,20 @@ static void do_telnet_read (char *buf, int len) {
            else if (c == IAC)
                telnet_state = SEENIAC;
            else {
-               b[0] = c;
-#if 0
                if (!in_synch)
+                   c_write1(c);
+
+#if 1
+               /* I can't get the F***ing winsock to insert the urgent IAC
+                * into the right position! Even with SO_OOBINLINE it gives
+                * it to recv too soon. And of course the DM byte (that
+                * arrives in the same packet!) appears several K later!!
+                *
+                * Oh well, we do get the DM in the right place so I'll
+                * just stop hiding on the next 0xf2 and hope for the best.
+                */
+               else if (c == DM) in_synch = 0;
 #endif
-                   c_write (b, 1);
                if (c == CR)
                    telnet_state = SEENCR;
                else
@@ -410,7 +399,17 @@ static void do_telnet_read (char *buf, int len) {
            else if (c == WILL) telnet_state = SEENWILL;
            else if (c == WONT) telnet_state = SEENWONT;
            else if (c == SB) telnet_state = SEENSB;
-           else telnet_state = TOPLEVEL;/* ignore _everything_ else! */
+           else if (c == DM) {
+               in_synch = 0;
+               telnet_state = TOPLEVEL;
+           } 
+           else {
+               /* ignore everything else; print it if it's IAC */
+               if (c == IAC) {
+                   c_write1(c);
+               }
+               telnet_state = TOPLEVEL;
+           }
            break;
          case SEENWILL:
            proc_rec_opt (WILL, c);
@@ -442,8 +441,8 @@ static void do_telnet_read (char *buf, int len) {
                    char *newbuf;
                    sb_size += SB_DELTA;
                    newbuf = (sb_buf ?
-                             realloc(sb_buf, sb_size) :
-                             malloc(sb_size));
+                             srealloc(sb_buf, sb_size) :
+                             smalloc(sb_size));
                    if (newbuf)
                        sb_buf = newbuf;
                    else
@@ -466,37 +465,40 @@ static void do_telnet_read (char *buf, int len) {
     }
 }
 
+static int telnet_receive(Socket skt, int urgent, char *data, int len) {
+    if (urgent==3) {
+        /* A socket error has occurred. */
+        s = NULL;
+        connection_fatal(data);
+        len = 0;
+    }
+    if (!len) {
+       /* Connection has closed. */
+       sk_close(s);
+       s = NULL;
+       return 0;
+    }
+    do_telnet_read (data, len);
+    return 1;
+}
+
 /*
- * Called to set up the Telnet connection. Will arrange for
- * WM_NETEVENT messages to be passed to the specified window, whose
- * window procedure should then call telnet_msg().
+ * Called to set up the Telnet connection.
  *
  * Returns an error message, or NULL on success.
  *
  * Also places the canonical host name into `realhost'.
  */
-static char *telnet_init (HWND hwnd, char *host, int port, char **realhost) {
-    SOCKADDR_IN addr;
-    struct hostent *h;
-    unsigned long a;
+static char *telnet_init (char *host, int port, char **realhost) {
+    SockAddr addr;
+    char *err;
 
     /*
      * Try to find host.
      */
-    if ( (a = inet_addr(host)) == (unsigned long) INADDR_NONE) {
-       if ( (h = gethostbyname(host)) == NULL)
-           switch (WSAGetLastError()) {
-             case WSAENETDOWN: return "Network is down";
-             case WSAHOST_NOT_FOUND: case WSANO_DATA:
-               return "Host does not exist";
-             case WSATRY_AGAIN: return "Host not found";
-             default: return "gethostbyname: unknown error";
-           }
-       memcpy (&a, h->h_addr, sizeof(a));
-       *realhost = h->h_name;
-    } else
-       *realhost = host;
-    a = ntohl(a);
+    addr = sk_namelookup(host, realhost);
+    if ( (err = sk_addr_error(addr)) )
+       return err;
 
     if (port < 0)
        port = 23;                     /* default telnet port */
@@ -504,53 +506,11 @@ static char *telnet_init (HWND hwnd, char *host, int port, char **realhost) {
     /*
      * Open socket.
      */
-    s = socket(AF_INET, SOCK_STREAM, 0);
-    if (s == INVALID_SOCKET)
-       switch (WSAGetLastError()) {
-         case WSAENETDOWN: return "Network is down";
-         case WSAEAFNOSUPPORT: return "TCP/IP support not present";
-         default: return "socket(): unknown error";
-       }
+    s = sk_new(addr, port, 0, telnet_receive);
+    if ( (err = sk_socket_error(s)) )
+       return err;
 
-#if 0
-    {
-       BOOL b = TRUE;
-       setsockopt (s, SOL_SOCKET, SO_OOBINLINE, (void *)&b, sizeof(b));
-    }
-#endif
-
-    /*
-     * Bind to local address.
-     */
-    addr.sin_family = AF_INET;
-    addr.sin_addr.s_addr = htonl(INADDR_ANY);
-    addr.sin_port = htons(0);
-    if (bind (s, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)
-       switch (WSAGetLastError()) {
-         case WSAENETDOWN: return "Network is down";
-         default: return "bind(): unknown error";
-       }
-
-    /*
-     * Connect to remote address.
-     */
-    addr.sin_addr.s_addr = htonl(a);
-    addr.sin_port = htons((short)port);
-    if (connect (s, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)
-       switch (WSAGetLastError()) {
-         case WSAENETDOWN: return "Network is down";
-         case WSAECONNREFUSED: return "Connection refused";
-         case WSAENETUNREACH: return "Network is unreachable";
-         case WSAEHOSTUNREACH: return "No route to host";
-         default: return "connect(): unknown error";
-       }
-
-    if (WSAAsyncSelect (s, hwnd, WM_NETEVENT, FD_READ |
-                       FD_WRITE | FD_OOB | FD_CLOSE) == SOCKET_ERROR)
-       switch (WSAGetLastError()) {
-         case WSAENETDOWN: return "Network is down";
-         default: return "WSAAsyncSelect(): unknown error";
-       }
+    sk_addr_free(addr);
 
     /*
      * Initialise option states.
@@ -563,86 +523,24 @@ static char *telnet_init (HWND hwnd, char *host, int port, char **realhost) {
                send_opt ((*o)->send, (*o)->option);
     }
 
-#if 0
     /*
      * Set up SYNCH state.
      */
     in_synch = FALSE;
-#endif
 
     return NULL;
 }
 
 /*
- * Process a WM_NETEVENT message. Will return 0 if the connection
- * has closed, or <0 for a socket error.
- */
-static int telnet_msg (WPARAM wParam, LPARAM lParam) {
-    int ret;
-    char buf[256];
-
-    if (s == INVALID_SOCKET)          /* how the hell did we get here?! */
-       return -5000;
-
-    if (WSAGETSELECTERROR(lParam) != 0)
-       return -WSAGETSELECTERROR(lParam);
-
-    switch (WSAGETSELECTEVENT(lParam)) {
-      case FD_READ:
-       ret = recv(s, buf, sizeof(buf), 0);
-       if (ret < 0 && WSAGetLastError() == WSAEWOULDBLOCK)
-           return 1;
-       if (ret < 0)                   /* any _other_ error */
-           return -10000-WSAGetLastError();
-       if (ret == 0) {
-           s = INVALID_SOCKET;
-           return 0;                  /* can't happen, in theory */
-       }
-#if 0
-       if (in_synch) {
-           BOOL i;
-           if (ioctlsocket (s, SIOCATMARK, &i) < 0) {
-               return -20000-WSAGetLastError();
-           }
-           if (i)
-               in_synch = FALSE;
-       }
-#endif
-       do_telnet_read (buf, ret);
-       return 1;
-      case FD_OOB:
-       do {
-           ret = recv(s, buf, sizeof(buf), 0);
-       } while (ret > 0);
-       telnet_state = TOPLEVEL;
-       do {
-           ret = recv(s, buf, 1, MSG_OOB);
-           if (ret > 0)
-               do_telnet_read (buf, ret);
-       } while (ret > 0);
-       if (ret < 0 && WSAGetLastError() != WSAEWOULDBLOCK)
-           return -30000-WSAGetLastError();
-       return 1;
-      case FD_WRITE:
-       if (outbuf_head != outbuf_reap)
-           try_write();
-       return 1;
-      case FD_CLOSE:
-       s = INVALID_SOCKET;
-       return 0;
-    }
-    return 1;                         /* shouldn't happen, but WTF */
-}
-
-/*
  * Called to send data down the Telnet connection.
  */
 static void telnet_send (char *buf, int len) {
     char *p;
     static unsigned char iac[2] = { IAC, IAC };
     static unsigned char cr[2] = { CR, NUL };
+    static unsigned char nl[2] = { CR, LF };
 
-    if (s == INVALID_SOCKET)
+    if (s == NULL)
        return;
 
     p = buf;
@@ -650,10 +548,10 @@ static void telnet_send (char *buf, int len) {
        char *q = p;
 
        while (iswritable((unsigned char)*p) && p < buf+len) p++;
-       s_write (q, p-q);
+       sk_write(s, q, p-q);
 
        while (p < buf+len && !iswritable((unsigned char)*p)) {
-           s_write ((unsigned char)*p == IAC ? iac : cr, 2);
+           sk_write(s, (unsigned char)*p == IAC ? iac : nl, 2);
            p++;
        }
     }
@@ -666,17 +564,17 @@ static void telnet_size(void) {
     unsigned char b[16];
     char logbuf[50];
 
-    if (s == INVALID_SOCKET || o_naws.state != ACTIVE)
+    if (s == NULL || o_naws.state != ACTIVE)
        return;
     b[0] = IAC; b[1] = SB; b[2] = TELOPT_NAWS;
     b[3] = cols >> 8; b[4] = cols & 0xFF;
     b[5] = rows >> 8; b[6] = rows & 0xFF;
     b[7] = IAC; b[8] = SE;
-    s_write (b, 9);
+    sk_write(s, b, 9);
     sprintf(logbuf, "client:\tSB NAWS %d,%d",
            ((unsigned char)b[3] << 8) + (unsigned char)b[4],
            ((unsigned char)b[5] << 8) + (unsigned char)b[6]);
-    lognegot (logbuf);
+    logevent (logbuf);
 }
 
 /*
@@ -685,35 +583,66 @@ static void telnet_size(void) {
 static void telnet_special (Telnet_Special code) {
     unsigned char b[2];
 
-    if (s == INVALID_SOCKET)
+    if (s == NULL)
        return;
 
     b[0] = IAC;
     switch (code) {
-      case TS_AYT: b[1] = AYT; s_write (b, 2); break;
-      case TS_BRK: b[1] = BREAK; s_write (b, 2); break;
-      case TS_EC: b[1] = EC; s_write (b, 2); break;
-      case TS_EL: b[1] = EL; s_write (b, 2); break;
-      case TS_GA: b[1] = GA; s_write (b, 2); break;
-      case TS_NOP: b[1] = NOP; s_write (b, 2); break;
-      case TS_ABORT: b[1] = ABORT; s_write (b, 2); break;
-      case TS_AO: b[1] = AO; s_write (b, 2); break;
-      case TS_IP: b[1] = IP; s_write (b, 2); break;
-      case TS_SUSP: b[1] = SUSP; s_write (b, 2); break;
-      case TS_EOR: b[1] = EOR; s_write (b, 2); break;
-      case TS_EOF: b[1] = xEOF; s_write (b, 2); break;
+      case TS_AYT: b[1] = AYT; sk_write(s, b, 2); break;
+      case TS_BRK: b[1] = BREAK; sk_write(s, b, 2); break;
+      case TS_EC: b[1] = EC; sk_write(s, b, 2); break;
+      case TS_EL: b[1] = EL; sk_write(s, b, 2); break;
+      case TS_GA: b[1] = GA; sk_write(s, b, 2); break;
+      case TS_NOP: b[1] = NOP; sk_write(s, b, 2); break;
+      case TS_ABORT: b[1] = ABORT; sk_write(s, b, 2); break;
+      case TS_AO: b[1] = AO; sk_write(s, b, 2); break;
+      case TS_IP: b[1] = IP; sk_write(s, b, 2); break;
+      case TS_SUSP: b[1] = SUSP; sk_write(s, b, 2); break;
+      case TS_EOR: b[1] = EOR; sk_write(s, b, 2); break;
+      case TS_EOF: b[1] = xEOF; sk_write(s, b, 2); break;
       case TS_SYNCH:
-       outbuf_head = outbuf_reap = 0;
-       b[0] = DM;
-       send (s, b, 1, MSG_OOB);
+        b[1] = DM;
+        sk_write (s, b, 1);
+        sk_write_oob (s, b+1, 1);
+        break;
+      case TS_RECHO:
+       if (o_echo.state == INACTIVE || o_echo.state == REALLY_INACTIVE) {
+           o_echo.state = REQUESTED;
+           send_opt (o_echo.send, o_echo.option);
+       }
+       break;
+      case TS_LECHO:
+       if (o_echo.state == ACTIVE) {
+           o_echo.state = REQUESTED;
+           send_opt (o_echo.nsend, o_echo.option);
+       }
        break;
+      case TS_PING: 
+       if (o_they_sga.state == ACTIVE) {
+           b[1] = NOP; 
+           sk_write(s, b, 2); 
+       }
+        break;
     }
 }
 
+static Socket telnet_socket(void) { return s; }
+
+static int telnet_sendok(void) { return 1; }
+
+static int telnet_ldisc(int option) {
+    if (option == LD_ECHO) return echoing;
+    if (option == LD_EDIT) return editing;
+    return FALSE;
+}
+
 Backend telnet_backend = {
     telnet_init,
-    telnet_msg,
     telnet_send,
     telnet_size,
-    telnet_special
+    telnet_special,
+    telnet_socket,
+    telnet_sendok,
+    telnet_ldisc,
+    23
 };