Prevent duplicate sk_close() calls on the same socket when the

connection dies unexpectedly (CONNABORTED / CONNRESET)

git-svn-id: svn://svn.tartarus.org/sgt/putty@910 cda61777-01e9-0310-a592-d414129be87e

diff --git a/raw.c b/raw.c
index 27af2c4..9cbb606 100644 (file)
--- a/raw.c
+++ b/raw.c
@@ -27,11 +27,12 @@ static void c_write (char *buf, int len) {
 static int raw_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);
         len = 0;
-    }
-    if (!len) {
+        return 0;
+    } else if (!len) {
        /* Connection has closed. */
        sk_close(s);
        s = NULL;

diff --git a/rlogin.c b/rlogin.c
index e0552f2..850d906 100644 (file)
--- a/rlogin.c
+++ b/rlogin.c
@@ -27,11 +27,11 @@ static void c_write (char *buf, int len) {
 static int rlogin_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);
-        len = 0;
-    }
-    if (!len) {
+        return 0;
+    } else if (!len) {
        /* Connection has closed. */
        sk_close(s);
        s = NULL;

diff --git a/ssh.c b/ssh.c
index 4849d91..5ad1ea4 100644 (file)
--- a/ssh.c
+++ b/ssh.c
@@ -1153,11 +1153,11 @@ static int ssh_receive(Socket skt, int urgent, char *data, int len) {
     if (urgent==3) {
         /* A socket error has occurred. */
         ssh_state = SSH_STATE_CLOSED;
+        sk_close(s);
         s = NULL;
         connection_fatal(data);
-        len = 0;
-    }
-    if (!len) {
+        return 0;
+    } else if (!len) {
        /* Connection has closed. */
        ssh_state = SSH_STATE_CLOSED;
        sk_close(s);

diff --git a/telnet.c b/telnet.c
index cd70d9b..8cb02a0 100644 (file)
--- a/telnet.c
+++ b/telnet.c
@@ -468,11 +468,11 @@ 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);
-        len = 0;
-    }
-    if (!len) {
+        return 0;
+    } else if (!len) {
        /* Connection has closed. */
        sk_close(s);
        s = NULL;