#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
-#include <winsock.h>
#include "putty.h"
#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 */
&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;
- } else {
- term_out();
- if( inbuf_head == inbuf_reap ) len++; else break;
- }
- }
+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) {
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);
}
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();
*/
deactivate_option (o->option==TELOPT_NEW_ENVIRON ? &o_oenv : &o_nenv);
}
- if (o->option == TELOPT_ECHO)
- {
- cfg.ldisc_term = FALSE;
- ldisc = &ldisc_simple;
- }
+ option_side_effects(o, 1);
}
static void refused_option (struct Opt *o) {
send_opt (WILL, TELOPT_OLD_ENVIRON);
o_oenv.state = REQUESTED;
}
- if (o->option == TELOPT_ECHO)
- {
- cfg.ldisc_term = TRUE;
- ldisc = &ldisc_term;
- }
+ option_side_effects(o, 0);
}
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:
strcpy(b+4, cfg.termspeed);
n = 4 + strlen(cfg.termspeed);
b[n] = IAC; b[n+1] = SE;
- s_write (b, n+2);
+ sk_write(s, b, n+2);
logevent("server:\tSB TSPEED SEND");
sprintf(logbuf, "client:\tSB TSPEED IS %s", cfg.termspeed);
logevent (logbuf);
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;
logevent("server:\tSB TTYPE SEND");
sprintf(logbuf, "client:\tSB TTYPE IS %s", b+4);
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>");
logevent (logbuf);
} telnet_state = TOPLEVEL;
static void do_telnet_read (char *buf, int len) {
- unsigned char b[10];
while (len--) {
int c = (unsigned char) *buf++;
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
else if (c == WILL) telnet_state = SEENWILL;
else if (c == WONT) telnet_state = SEENWONT;
else if (c == SB) telnet_state = SEENSB;
+ else if (c == DM) {
+ in_synch = 0;
+ telnet_state = TOPLEVEL;
+ }
else {
/* ignore everything else; print it if it's IAC */
if (c == IAC) {
- b[0] = c;
- c_write(b,1);
+ c_write1(c);
}
telnet_state = TOPLEVEL;
}
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
}
}
+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;
+ }
+ if(urgent) in_synch = TRUE;
+ 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 */
/*
* 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, 1, 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.
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];
-
- /*
- * Because reading less than the whole of the available pending
- * data can generate an FD_READ event, we need to allow for the
- * possibility that FD_READ may arrive with FD_CLOSE already in
- * the queue; so it's possible that we can get here even with s
- * invalid. If so, we return 1 and don't worry about it.
- */
- if (s == INVALID_SOCKET)
- return 1;
-
- if (WSAGETSELECTERROR(lParam) != 0)
- return -WSAGETSELECTERROR(lParam);
-
- switch (WSAGETSELECTEVENT(lParam)) {
- case FD_READ:
- case FD_CLOSE:
- 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;
- }
-#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;
- }
- return 1; /* shouldn't happen, but WTF */
-}
-
/*
* Called to send data down the Telnet connection.
*/
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;
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++;
}
}
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]);
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[1] = DM;
- send (s, b, 2, MSG_OOB);
- break;
+ 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.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
};