diff options
author | Vijay Pai <vpai@google.com> | 2016-02-25 18:10:24 -0800 |
---|---|---|
committer | Vijay Pai <vpai@google.com> | 2016-02-25 18:10:24 -0800 |
commit | e9ef53645150f7a0400a5f9be770eb2b4ba335b5 (patch) | |
tree | 34a8b7094b851589eec500b2ca419b64c9051d89 /test/core/util/test_tcp_server.c | |
parent | e3cd25684051de21e4b1ba93173c06e72fa5ffad (diff) |
Revert "Add an implementation firewall against pollset_set"
Diffstat (limited to 'test/core/util/test_tcp_server.c')
-rw-r--r-- | test/core/util/test_tcp_server.c | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/test/core/util/test_tcp_server.c b/test/core/util/test_tcp_server.c index ab379441d8..e99d5dcffd 100644 --- a/test/core/util/test_tcp_server.c +++ b/test/core/util/test_tcp_server.c @@ -57,8 +57,8 @@ void test_tcp_server_init(test_tcp_server *server, server->tcp_server = NULL; grpc_closure_init(&server->shutdown_complete, on_server_destroyed, server); server->shutdown = 0; - server->pollset = gpr_malloc(grpc_pollset_size()); - grpc_pollset_init(server->pollset, &server->mu); + grpc_pollset_init(&server->pollset); + server->pollsets[0] = &server->pollset; server->on_connect = on_connect; server->cb_data = user_data; } @@ -77,7 +77,7 @@ void test_tcp_server_start(test_tcp_server *server, int port) { grpc_tcp_server_add_port(server->tcp_server, &addr, sizeof(addr)); GPR_ASSERT(port_added == port); - grpc_tcp_server_start(&exec_ctx, server->tcp_server, &server->pollset, 1, + grpc_tcp_server_start(&exec_ctx, server->tcp_server, server->pollsets, 1, server->on_connect, server->cb_data); gpr_log(GPR_INFO, "test tcp server listening on 0.0.0.0:%d", port); @@ -90,10 +90,10 @@ void test_tcp_server_poll(test_tcp_server *server, int seconds) { gpr_time_add(gpr_now(GPR_CLOCK_MONOTONIC), gpr_time_from_seconds(seconds, GPR_TIMESPAN)); grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; - gpr_mu_lock(server->mu); - grpc_pollset_work(&exec_ctx, server->pollset, &worker, + gpr_mu_lock(GRPC_POLLSET_MU(&server->pollset)); + grpc_pollset_work(&exec_ctx, &server->pollset, &worker, gpr_now(GPR_CLOCK_MONOTONIC), deadline); - gpr_mu_unlock(server->mu); + gpr_mu_unlock(GRPC_POLLSET_MU(&server->pollset)); grpc_exec_ctx_finish(&exec_ctx); } @@ -111,9 +111,8 @@ void test_tcp_server_destroy(test_tcp_server *server) { gpr_time_cmp(gpr_now(GPR_CLOCK_MONOTONIC), shutdown_deadline) < 0) { test_tcp_server_poll(server, 1); } - grpc_pollset_shutdown(&exec_ctx, server->pollset, &do_nothing_cb); + grpc_pollset_shutdown(&exec_ctx, &server->pollset, &do_nothing_cb); grpc_exec_ctx_finish(&exec_ctx); - grpc_pollset_destroy(server->pollset); - gpr_free(server->pollset); + grpc_pollset_destroy(&server->pollset); grpc_shutdown(); } |