From afa2d6342fc7c608141f058a7d4bb0778ae24f53 Mon Sep 17 00:00:00 2001
From: Craig Tiller <craig.tiller@gmail.com>
Date: Tue, 26 May 2015 16:39:13 -0700
Subject: [PATCH] Fix C core tests
---
src/core/surface/server.c | 41 +++++++++++++++++++++++++++++++++++++++
1 file changed, 41 insertions(+)
diff --git a/src/core/surface/server.c b/src/core/surface/server.c
index 69f531fe1d..37ceca43ba 100644
--- a/src/core/surface/server.c
+++ b/src/core/surface/server.c
@@ -902,6 +902,47 @@ void grpc_server_listener_destroy_done(void *s) {
gpr_mu_unlock(&server->mu);
}
+void grpc_server_cancel_all_calls(grpc_server *server) {
+ call_data *calld;
+ grpc_call **calls;
+ size_t call_count;
+ size_t call_capacity;
+ int is_first = 1;
+ size_t i;
+
+ gpr_mu_lock(&server->mu);
+
+ GPR_ASSERT(server->shutdown);
+
+ if (!server->lists[ALL_CALLS]) {
+ gpr_mu_unlock(&server->mu);
+ return;
+ }
+
+ call_capacity = 8;
+ call_count = 0;
+ calls = gpr_malloc(sizeof(grpc_call *) * call_capacity);
+
+ for (calld = server->lists[ALL_CALLS]; calld != server->lists[ALL_CALLS] || is_first; calld = calld->links[ALL_CALLS].next) {
+ if (call_count == call_capacity) {
+ call_capacity *= 2;
+ calls = gpr_realloc(calls, sizeof(grpc_call *) * call_capacity);
+ }
+ calls[call_count++] = calld->call;
+ GRPC_CALL_INTERNAL_REF(calld->call, "cancel_all");
+ is_first = 0;
+ }
+
+ gpr_mu_unlock(&server->mu);
+
+ for (i = 0; i < call_count; i++) {
+ grpc_call_cancel_with_status(calls[i], GRPC_STATUS_UNAVAILABLE, "Unavailable");
+ GRPC_CALL_INTERNAL_UNREF(calls[i], "cancel_all", 1);
+ }
+
+ gpr_free(calls);
+}
+
void grpc_server_destroy(grpc_server *server) {
channel_data *c;
listener *l;
--
GitLab