fixes related to new changes

radkesvat · May 14, 2024 · af1ce1a · af1ce1a
1 parent 3421a89
commit af1ce1a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tunnels/adapters/connector/tcp/tcp_connector.c b/tunnels/adapters/connector/tcp/tcp_connector.c
@@ -171,14 +171,15 @@ static void upStream(tunnel_t *self, context_t *c)
             unsigned int bytes  = bufLen(c->payload);
             int          nwrite = hio_write(cstate->io, c->payload);
             c->payload          = NULL;
-            destroyContext(c);
 
             if (nwrite >= 0 && nwrite < bytes)
             {
                 pauseLineDownSide(c->line);
                 cstate->write_paused = true;
                 hio_setcb_write(cstate->io, onWriteComplete);
             }
+            destroyContext(c);
+
         }
     }
     else

diff --git a/tunnels/adapters/listener/tcp/tcp_listener.c b/tunnels/adapters/listener/tcp/tcp_listener.c
@@ -172,14 +172,15 @@ static void downStream(tunnel_t *self, context_t *c)
             unsigned int bytes  = bufLen(c->payload);
             int          nwrite = hio_write(cstate->io, c->payload);
             c->payload          = NULL;
-            destroyContext(c);
 
             if (nwrite >= 0 && nwrite < bytes)
             {
                 pauseLineUpSide(c->line);
                 cstate->write_paused = true;
                 hio_setcb_write(cstate->io, onWriteComplete);
             }
+            destroyContext(c);
+
         }
     }
     else