The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
263 lines
10 KiB
263 lines
10 KiB
/* |
|
* |
|
* Copyright 2015, Google Inc. |
|
* All rights reserved. |
|
* |
|
* Redistribution and use in source and binary forms, with or without |
|
* modification, are permitted provided that the following conditions are |
|
* met: |
|
* |
|
* * Redistributions of source code must retain the above copyright |
|
* notice, this list of conditions and the following disclaimer. |
|
* * Redistributions in binary form must reproduce the above |
|
* copyright notice, this list of conditions and the following disclaimer |
|
* in the documentation and/or other materials provided with the |
|
* distribution. |
|
* * Neither the name of Google Inc. nor the names of its |
|
* contributors may be used to endorse or promote products derived from |
|
* this software without specific prior written permission. |
|
* |
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
* |
|
*/ |
|
|
|
#include <grpc/support/port_platform.h> |
|
|
|
#include <limits.h> |
|
#include <memory.h> |
|
|
|
#include <grpc/grpc.h> |
|
#include <grpc/support/alloc.h> |
|
#include <grpc/support/log.h> |
|
#include <grpc/support/time.h> |
|
#include "src/core/lib/channel/channel_stack.h" |
|
#include "src/core/lib/channel/compress_filter.h" |
|
#include "src/core/lib/channel/connected_channel.h" |
|
#include "src/core/lib/channel/deadline_filter.h" |
|
#include "src/core/lib/channel/handshaker_registry.h" |
|
#include "src/core/lib/channel/http_client_filter.h" |
|
#include "src/core/lib/channel/http_server_filter.h" |
|
#include "src/core/lib/channel/message_size_filter.h" |
|
#include "src/core/lib/debug/trace.h" |
|
#include "src/core/lib/http/parser.h" |
|
#include "src/core/lib/iomgr/combiner.h" |
|
#include "src/core/lib/iomgr/executor.h" |
|
#include "src/core/lib/iomgr/iomgr.h" |
|
#include "src/core/lib/iomgr/resource_quota.h" |
|
#include "src/core/lib/profiling/timers.h" |
|
#include "src/core/lib/slice/slice_internal.h" |
|
#include "src/core/lib/surface/api_trace.h" |
|
#include "src/core/lib/surface/call.h" |
|
#include "src/core/lib/surface/channel_init.h" |
|
#include "src/core/lib/surface/completion_queue.h" |
|
#include "src/core/lib/surface/init.h" |
|
#include "src/core/lib/surface/lame_client.h" |
|
#include "src/core/lib/surface/server.h" |
|
#include "src/core/lib/transport/bdp_estimator.h" |
|
#include "src/core/lib/transport/connectivity_state.h" |
|
#include "src/core/lib/transport/transport_impl.h" |
|
|
|
/* (generated) built in registry of plugins */ |
|
extern void grpc_register_built_in_plugins(void); |
|
|
|
#define MAX_PLUGINS 128 |
|
|
|
static gpr_once g_basic_init = GPR_ONCE_INIT; |
|
static gpr_mu g_init_mu; |
|
static int g_initializations; |
|
|
|
static void do_basic_init(void) { |
|
gpr_log_verbosity_init(); |
|
gpr_mu_init(&g_init_mu); |
|
grpc_register_built_in_plugins(); |
|
g_initializations = 0; |
|
} |
|
|
|
static bool append_filter(grpc_exec_ctx *exec_ctx, |
|
grpc_channel_stack_builder *builder, void *arg) { |
|
return grpc_channel_stack_builder_append_filter( |
|
builder, (const grpc_channel_filter *)arg, NULL, NULL); |
|
} |
|
|
|
static bool prepend_filter(grpc_exec_ctx *exec_ctx, |
|
grpc_channel_stack_builder *builder, void *arg) { |
|
return grpc_channel_stack_builder_prepend_filter( |
|
builder, (const grpc_channel_filter *)arg, NULL, NULL); |
|
} |
|
|
|
static bool maybe_add_http_filter(grpc_exec_ctx *exec_ctx, |
|
grpc_channel_stack_builder *builder, |
|
void *arg) { |
|
grpc_transport *t = grpc_channel_stack_builder_get_transport(builder); |
|
if (t && strstr(t->vtable->name, "http")) { |
|
return grpc_channel_stack_builder_prepend_filter( |
|
builder, (const grpc_channel_filter *)arg, NULL, NULL); |
|
} |
|
return true; |
|
} |
|
|
|
static void register_builtin_channel_init() { |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_DIRECT_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
prepend_filter, (void *)&grpc_client_deadline_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_SERVER_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, prepend_filter, |
|
(void *)&grpc_server_deadline_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_SUBCHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
prepend_filter, (void *)&grpc_message_size_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_DIRECT_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
prepend_filter, (void *)&grpc_message_size_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_SERVER_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, prepend_filter, |
|
(void *)&grpc_message_size_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, prepend_filter, |
|
(void *)&grpc_compress_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_DIRECT_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
prepend_filter, (void *)&grpc_compress_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_SERVER_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, prepend_filter, |
|
(void *)&grpc_compress_filter); |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_SUBCHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
maybe_add_http_filter, (void *)&grpc_http_client_filter); |
|
grpc_channel_init_register_stage(GRPC_CLIENT_SUBCHANNEL, |
|
GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
grpc_add_connected_filter, NULL); |
|
grpc_channel_init_register_stage( |
|
GRPC_CLIENT_DIRECT_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
maybe_add_http_filter, (void *)&grpc_http_client_filter); |
|
grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, |
|
GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
grpc_add_connected_filter, NULL); |
|
grpc_channel_init_register_stage( |
|
GRPC_SERVER_CHANNEL, GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
maybe_add_http_filter, (void *)&grpc_http_server_filter); |
|
grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, |
|
GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
grpc_add_connected_filter, NULL); |
|
grpc_channel_init_register_stage(GRPC_CLIENT_LAME_CHANNEL, |
|
GRPC_CHANNEL_INIT_BUILTIN_PRIORITY, |
|
append_filter, (void *)&grpc_lame_filter); |
|
grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX, prepend_filter, |
|
(void *)&grpc_server_top_filter); |
|
} |
|
|
|
typedef struct grpc_plugin { |
|
void (*init)(); |
|
void (*destroy)(); |
|
} grpc_plugin; |
|
|
|
static grpc_plugin g_all_of_the_plugins[MAX_PLUGINS]; |
|
static int g_number_of_plugins = 0; |
|
|
|
void grpc_register_plugin(void (*init)(void), void (*destroy)(void)) { |
|
GRPC_API_TRACE("grpc_register_plugin(init=%p, destroy=%p)", 2, |
|
((void *)(intptr_t)init, (void *)(intptr_t)destroy)); |
|
GPR_ASSERT(g_number_of_plugins != MAX_PLUGINS); |
|
g_all_of_the_plugins[g_number_of_plugins].init = init; |
|
g_all_of_the_plugins[g_number_of_plugins].destroy = destroy; |
|
g_number_of_plugins++; |
|
} |
|
|
|
void grpc_init(void) { |
|
int i; |
|
gpr_once_init(&g_basic_init, do_basic_init); |
|
|
|
gpr_mu_lock(&g_init_mu); |
|
if (++g_initializations == 1) { |
|
gpr_time_init(); |
|
grpc_slice_intern_init(); |
|
grpc_mdctx_global_init(); |
|
grpc_channel_init_init(); |
|
grpc_register_tracer("api", &grpc_api_trace); |
|
grpc_register_tracer("channel", &grpc_trace_channel); |
|
grpc_register_tracer("connectivity_state", &grpc_connectivity_state_trace); |
|
grpc_register_tracer("channel_stack_builder", |
|
&grpc_trace_channel_stack_builder); |
|
grpc_register_tracer("http1", &grpc_http1_trace); |
|
grpc_register_tracer("compression", &grpc_compression_trace); |
|
grpc_register_tracer("queue_pluck", &grpc_cq_pluck_trace); |
|
grpc_register_tracer("combiner", &grpc_combiner_trace); |
|
grpc_register_tracer("server_channel", &grpc_server_channel_trace); |
|
grpc_register_tracer("bdp_estimator", &grpc_bdp_estimator_trace); |
|
// Default pluck trace to 1 |
|
grpc_cq_pluck_trace = 1; |
|
grpc_register_tracer("queue_timeout", &grpc_cq_event_timeout_trace); |
|
// Default timeout trace to 1 |
|
grpc_cq_event_timeout_trace = 1; |
|
grpc_register_tracer("op_failure", &grpc_trace_operation_failures); |
|
grpc_register_tracer("resource_quota", &grpc_resource_quota_trace); |
|
grpc_register_tracer("call_error", &grpc_call_error_trace); |
|
#ifndef NDEBUG |
|
grpc_register_tracer("pending_tags", &grpc_trace_pending_tags); |
|
#endif |
|
grpc_security_pre_init(); |
|
grpc_iomgr_init(); |
|
grpc_executor_init(); |
|
gpr_timers_global_init(); |
|
grpc_cq_global_init(); |
|
grpc_handshaker_factory_registry_init(); |
|
grpc_security_init(); |
|
for (i = 0; i < g_number_of_plugins; i++) { |
|
if (g_all_of_the_plugins[i].init != NULL) { |
|
g_all_of_the_plugins[i].init(); |
|
} |
|
} |
|
/* register channel finalization AFTER all plugins, to ensure that it's run |
|
* at the appropriate time */ |
|
grpc_register_security_filters(); |
|
register_builtin_channel_init(); |
|
grpc_tracer_init("GRPC_TRACE"); |
|
/* no more changes to channel init pipelines */ |
|
grpc_channel_init_finalize(); |
|
} |
|
gpr_mu_unlock(&g_init_mu); |
|
GRPC_API_TRACE("grpc_init(void)", 0, ()); |
|
} |
|
|
|
void grpc_shutdown(void) { |
|
int i; |
|
GRPC_API_TRACE("grpc_shutdown(void)", 0, ()); |
|
grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
|
gpr_mu_lock(&g_init_mu); |
|
if (--g_initializations == 0) { |
|
grpc_executor_shutdown(&exec_ctx); |
|
grpc_cq_global_shutdown(); |
|
grpc_iomgr_shutdown(&exec_ctx); |
|
gpr_timers_global_destroy(); |
|
grpc_tracer_shutdown(); |
|
for (i = g_number_of_plugins; i >= 0; i--) { |
|
if (g_all_of_the_plugins[i].destroy != NULL) { |
|
g_all_of_the_plugins[i].destroy(); |
|
} |
|
} |
|
grpc_mdctx_global_shutdown(&exec_ctx); |
|
grpc_handshaker_factory_registry_shutdown(&exec_ctx); |
|
grpc_slice_intern_shutdown(); |
|
} |
|
gpr_mu_unlock(&g_init_mu); |
|
grpc_exec_ctx_finish(&exec_ctx); |
|
} |
|
|
|
int grpc_is_initialized(void) { |
|
int r; |
|
gpr_once_init(&g_basic_init, do_basic_init); |
|
gpr_mu_lock(&g_init_mu); |
|
r = g_initializations > 0; |
|
gpr_mu_unlock(&g_init_mu); |
|
return r; |
|
}
|
|
|