X-Git-Url: https://git.distorted.org.uk/u/mdw/putty/blobdiff_plain/37508af4ab231b1fde58345f5237f3ee82803829..b47b21989c2dd9c7ce0e973b3730965f9d99b990:/telnet.c diff --git a/telnet.c b/telnet.c index b6e4802d..8cb02a03 100644 --- a/telnet.c +++ b/telnet.c @@ -1,7 +1,6 @@ #include #include #include -#include #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,50 +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; - int c = (unsigned char) *buf; - 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) { @@ -185,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); } @@ -202,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(); @@ -214,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) { @@ -222,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) { @@ -256,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: @@ -285,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 "); + logevent ("server:\tSB TSPEED "); break; case TELOPT_TTYPE: if (sb_len == 1 && sb_buf[0] == TELQUAL_SEND) { @@ -301,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 \r\n"); + logevent("server:\tSB TTYPE \r\n"); break; case TELOPT_OLD_ENVIRON: case TELOPT_NEW_ENVIRON: @@ -317,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; @@ -366,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 ? "" : ""); - lognegot (logbuf); + logevent (logbuf); } break; } @@ -381,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++; @@ -394,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 @@ -411,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); @@ -443,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 @@ -467,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. */ + sk_close(s); + s = NULL; + connection_fatal(data); + return 0; + } else 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 */ @@ -505,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"; - } - -#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"; - } + s = sk_new(addr, port, 0, telnet_receive); + if ( (err = sk_socket_error(s)) ) + return err; - 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. @@ -564,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; @@ -651,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++; } } @@ -667,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); } /* @@ -686,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 };