put the buffer back to the pool when userdata is null

radkesvat · Apr 24, 2024 · 890e6f9 · 890e6f9
1 parent faa413a
commit 890e6f9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/tunnels/adapters/connector/tcp/tcp_connector.c b/tunnels/adapters/connector/tcp/tcp_connector.c
@@ -128,6 +128,7 @@ static void onRecv(hio_t *restrict io, shift_buffer_t *buf)
     tcp_connector_con_state_t *cstate = (tcp_connector_con_state_t *) (hevent_userdata(io));
     if (cstate == NULL)
     {
+        reuseBuffer(hloop_bufferpool(hevent_loop(io)),buf);
         return;
     }
     shift_buffer_t *payload = buf;

diff --git a/tunnels/adapters/connector/udp/udp_connector.c b/tunnels/adapters/connector/udp/udp_connector.c
@@ -11,7 +11,11 @@ static void cleanup(udp_connector_con_state_t *cstate)
 static void onRecv(hio_t *io, shift_buffer_t *buf)
 {
     udp_connector_con_state_t *cstate = (udp_connector_con_state_t *) (hevent_userdata(io));
-
+    if (cstate == NULL)
+    {
+        reuseBuffer(hloop_bufferpool(hevent_loop(io)),buf);
+        return;
+    }
     shift_buffer_t *payload = buf;
     tunnel_t *      self    = (cstate)->tunnel;
     line_t *        line    = (cstate)->line;

diff --git a/tunnels/adapters/listener/tcp/tcp_listener.c b/tunnels/adapters/listener/tcp/tcp_listener.c
@@ -253,6 +253,7 @@ static void onRecv(hio_t *io, shift_buffer_t *buf)
     tcp_listener_con_state_t *cstate = (tcp_listener_con_state_t *) (hevent_userdata(io));
     if (cstate == NULL)
     {
+        reuseBuffer(hloop_bufferpool(hevent_loop(io)),buf);
         return;
     }
     shift_buffer_t *payload           = buf;