1 files changed, 18 insertions, 7 deletions

diff --git a/socket.c b/socket.c
index 4151eb8..b346e1c 100644
--- a/socket.c
+++ b/socket.c
@@ -401,7 +401,15 @@ int sowrite(struct socket *so)
     DEBUG_ARG("so = %p", so);
 
     if (so->so_urgc) {
-        sosendoob(so);
+        uint32_t expected = so->so_urgc;
+        if (sosendoob(so) < expected) {
+            /* Treat a short write as a fatal error too,
+             * rather than continuing on and sending the urgent
+             * data as if it were non-urgent and leaving the
+             * so_urgc count wrong.
+             */
+            goto err_disconnected;
+        }
         if (sb->sb_cc == 0)
             return 0;
     }
@@ -448,12 +456,7 @@ int sowrite(struct socket *so)
         return 0;
 
     if (nn <= 0) {
-        DEBUG_MISC(
-            (dfd, " --- sowrite disconnected, so->so_state = %x, errno = %d\n",
-             so->so_state, errno));
-        sofcantsendmore(so);
-        tcp_sockclosed(sototcpcb(so));
-        return -1;
+        goto err_disconnected;
     }
 
 #ifndef HAVE_READV
@@ -480,6 +483,14 @@ int sowrite(struct socket *so)
         sofcantsendmore(so);
 
     return nn;
+
+err_disconnected:
+    DEBUG_MISC((dfd,
+                " --- sowrite disconnected, so->so_state = %x, errno = %d\n",
+                so->so_state, errno));
+    sofcantsendmore(so);
+    tcp_sockclosed(sototcpcb(so));
+    return -1;
 }
 
 /*