X-Git-Url: https://git.distorted.org.uk/u/mdw/putty/blobdiff_plain/8ddd7f4ec236820df4656a2c00c53ae0b9597bd2..7e78000d8966255e919e3d8753fd784f4559ee55:/telnet.c?ds=sidebyside

diff --git a/telnet.c b/telnet.c
index 65953421..29daf0f1 100644
--- a/telnet.c
+++ b/telnet.c
@@ -465,19 +465,17 @@ static void do_telnet_read (char *buf, int len) {
     }
 }
 
-static int telnet_receive(Socket skt, int urgent, char *data, int len) {
-    if (urgent==3) {
+static int telnet_closing (Plug plug, char *error_msg, int error_code, int calling_back) {
+    sk_close(s);
+    s = NULL;
+    if (error_msg) {
         /* 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;
-    }
+        connection_fatal (error_msg);
+    } /* Otherwise, the remote side closed the connection normally. */
+    return 0;
+}
+
+static int telnet_receive(Plug plug, int urgent, char *data, int len) {
     if(urgent) in_synch = TRUE;
     do_telnet_read (data, len);
     return 1;
@@ -491,6 +489,11 @@ static int telnet_receive(Socket skt, int urgent, char *data, int len) {
  * Also places the canonical host name into `realhost'.
  */
 static char *telnet_init (char *host, int port, char **realhost) {
+    static struct plug_function_table fn_table = {
+	telnet_closing,
+	telnet_receive
+    }, *fn_table_ptr = &fn_table;
+
     SockAddr addr;
     char *err;
 
@@ -507,7 +510,7 @@ static char *telnet_init (char *host, int port, char **realhost) {
     /*
      * Open socket.
      */
-    s = sk_new(addr, port, 0, 1, telnet_receive);
+    s = sk_new(addr, port, 0, 1, &fn_table_ptr);
     if ( (err = sk_socket_error(s)) )
 	return err;