正确关闭libUV句柄

时间:2014-09-02 03:18:50

标签: c libuv

我试图找出如何解决我在使用Valgrind运行此程序时遇到的内存泄漏问题。 nShell_client_main中的两个分配发生泄漏。但我不是 确定如何正确释放它们。

我已尝试在nShell_Connect上释放它们,但它导致libUV中止该程序。我已经尝试在nShell_client_main结束时释放它们,但是在关闭循环时我得到了读/写错误。有谁知道我应该如何关闭这些手柄?我读了this,这让我开始了。但是,由于uv_ip4_addr在最新版本中有不同的原型,因此它接近过时。

nShell_main是"条目"点)

#include "nPort.h"
#include "nShell-main.h"

void nShell_Close(
    uv_handle_t * term_handle
){
}

void nShell_Connect(uv_connect_t * term_handle, int status){
    uv_close((uv_handle_t *) term_handle, 0);
}

nError * nShell_client_main(nShell * n_shell, uv_loop_t * n_shell_loop){

    int uv_error = 0;

    nError * n_error = 0;

    uv_tcp_t * n_shell_socket = 0;
    uv_connect_t * n_shell_connect = 0;

    struct sockaddr_in dest_addr;

    n_shell_socket = malloc(sizeof(uv_tcp_t));

    if (!n_shell_socket){
        // handle error
    }

    uv_error = uv_tcp_init(n_shell_loop, n_shell_socket);

    if (uv_error){
        // handle error
    }

    uv_error = uv_ip4_addr("127.0.0.1", NPORT, &dest_addr);

    if (uv_error){
        // handle error
    }

    n_shell_connect = malloc(sizeof(uv_connect_t));

    if (!n_shell_connect){
        // handle error
    }

    uv_error = uv_tcp_connect(n_shell_connect, n_shell_socket, (struct sockaddr *) &dest_addr, nShell_Connect);

    if (uv_error){
        // handle error
    }

    uv_error = uv_run(n_shell_loop, UV_RUN_DEFAULT);

    if (uv_error){
        // handle error
    }

    return 0;
}

nError * nShell_loop_main(nShell * n_shell){

    int uv_error = 0;

    nError * n_error = 0;

    uv_loop_t * n_shell_loop = 0;

    n_shell_loop = malloc(sizeof(uv_loop_t));

    if (!n_shell_loop){
        // handle error
    }

    uv_error = uv_loop_init(n_shell_loop);

    if (uv_error){
        // handle error
    }

    n_error = nShell_client_main(n_shell, n_shell_loop);

    if (n_error){
        // handle error
    }

    uv_loop_close(n_shell_loop);
    free(n_shell_loop);

    return 0;
}

断言发生在这段代码摘要中的switch语句的末尾(摘自Joyith在Github上的libUV页面):

void uv_close(uv_handle_t* handle, uv_close_cb close_cb) {
  assert(!(handle->flags & (UV_CLOSING | UV_CLOSED)));

  handle->flags |= UV_CLOSING;
  handle->close_cb = close_cb;

  switch (handle->type) {
  case UV_NAMED_PIPE:
    uv__pipe_close((uv_pipe_t*)handle);
    break;

  case UV_TTY:
    uv__stream_close((uv_stream_t*)handle);
    break;

  case UV_TCP:
    uv__tcp_close((uv_tcp_t*)handle);
    break;

  case UV_UDP:
    uv__udp_close((uv_udp_t*)handle);
    break;

  case UV_PREPARE:
    uv__prepare_close((uv_prepare_t*)handle);
    break;

  case UV_CHECK:
    uv__check_close((uv_check_t*)handle);
    break;

  case UV_IDLE:
    uv__idle_close((uv_idle_t*)handle);
    break;

  case UV_ASYNC:
    uv__async_close((uv_async_t*)handle);
    break;

  case UV_TIMER:
    uv__timer_close((uv_timer_t*)handle);
    break;

  case UV_PROCESS:
    uv__process_close((uv_process_t*)handle);
    break;

  case UV_FS_EVENT:
    uv__fs_event_close((uv_fs_event_t*)handle);
    break;

  case UV_POLL:
    uv__poll_close((uv_poll_t*)handle);
    break;

  case UV_FS_POLL:
    uv__fs_poll_close((uv_fs_poll_t*)handle);
    break;

  case UV_SIGNAL:
    uv__signal_close((uv_signal_t*) handle);
    /* Signal handles may not be closed immediately. The signal code will */
    /* itself close uv__make_close_pending whenever appropriate. */
    return;

  default:
    assert(0); // assertion is happening here
  }

  uv__make_close_pending(handle);
}

我可以手动调用uv__tcp_close,但它不在公共标题中(并且可能不是正确的解决方案)。

2 个答案:

答案 0 :(得分:20)

在调用close回调之前,libuv不会使用句柄。这是你可以释放手柄的确切时刻。

我看到你拨打uv_loop_close,但是你没有检查返回值。如果仍有待处理的句柄,它将返回UV_EBUSY,因此您应该检查它。

如果要关闭循环并关闭所有句柄,则需要执行以下操作:

  • 使用uv_stop停止循环
  • 使用uv_walk并在未关闭的所有句柄上调用uv_close
  • 使用uv_run再次运行循环,以便调用所有关闭的回调,并且可以释放回调中的内存
  • 致电uv_loop_close,现在应该返回0

答案 1 :(得分:5)

我终于想出了如何停止循环并清理所有句柄。 我创建了一堆句柄和 SIGINT 信号句柄:

uv_signal_t *sigint = new uv_signal_t;
uv_signal_init(uv_default_loop(), sigint);
uv_signal_start(sigint, on_sigint_received, SIGINT);

当收到 SIGINT (按下控制台中的Ctrl + C)时,将调用on_sigint_received回调。 on_sigint_received看起来像是:

void on_sigint_received(uv_signal_t *handle, int signum)
{
    int result = uv_loop_close(handle->loop);
    if (result == UV_EBUSY)
    {
        uv_walk(handle->loop, on_uv_walk, NULL);
    }
}

它会触发回调函数on_uv_walk

void on_uv_walk(uv_handle_t* handle, void* arg)
{
    uv_close(handle, on_uv_close);
}

它尝试关闭每个打开的libuv句柄。 注意:我不会在uv_stop之前致电uv_walk,而是mentioned saghul。 调用on_sigint_received函数后,libuv循环连续执行,并在下一次迭代时为每个打开的句柄调用on_uv_close。如果您调用uv_stop函数,则不会调用on_uv_close回调。

void on_uv_close(uv_handle_t* handle)
{
    if (handle != NULL)
    {
        delete handle;
    }
}

之后libuv没有打开句柄并完成循环(退出uv_run):

uv_run(uv_default_loop(), UV_RUN_DEFAULT);
int result = uv_loop_close(uv_default_loop());
if (result)
{
    cerr << "failed to close libuv loop: " << uv_err_name(result) << endl;
}
else
{
    cout << "libuv loop is closed successfully!\n";
}