src/core/lib/surface/init.c - platform/external/grpc-grpc - Gitiles

 /*
  *
  * Copyright 2015-2016, 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/time.h>
 /* TODO(ctiller): find another way? - better not to include census here */
 #include "src/core/lib/census/grpc_plugin.h"
 #include "src/core/lib/channel/channel_stack.h"
 #include "src/core/lib/channel/client_channel.h"
 #include "src/core/lib/channel/compress_filter.h"
 #include "src/core/lib/channel/connected_channel.h"
 #include "src/core/lib/channel/http_client_filter.h"
 #include "src/core/lib/channel/http_server_filter.h"
 #include "src/core/lib/client_config/lb_policies/pick_first.h"
 #include "src/core/lib/client_config/lb_policies/round_robin.h"
 #include "src/core/lib/client_config/lb_policy_registry.h"
 #include "src/core/lib/client_config/resolver_registry.h"
 #include "src/core/lib/client_config/resolvers/dns_resolver.h"
 #include "src/core/lib/client_config/resolvers/sockaddr_resolver.h"
 #include "src/core/lib/client_config/subchannel.h"
 #include "src/core/lib/client_config/subchannel_index.h"
 #include "src/core/lib/debug/trace.h"
 #include "src/core/lib/iomgr/executor.h"
 #include "src/core/lib/iomgr/iomgr.h"
 #include "src/core/lib/profiling/timers.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/surface/surface_trace.h"
 #include "src/core/lib/transport/chttp2_transport.h"
 #include "src/core/lib/transport/connectivity_state.h"
 #include "src/core/lib/transport/transport_impl.h"

 #ifndef GRPC_DEFAULT_NAME_PREFIX
 #define GRPC_DEFAULT_NAME_PREFIX "dns:///"
 #endif

 #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_mu_init(&g_init_mu);
   /* TODO(ctiller): ideally remove this strict linkage */
   grpc_register_plugin(census_grpc_plugin_init, census_grpc_plugin_destroy);
   g_initializations = 0;
 }

 static bool append_filter(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_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_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_CHANNEL, INT_MAX, prepend_filter,
                                    (void *)&grpc_compress_filter);
   grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, INT_MAX,
                                    prepend_filter,
                                    (void *)&grpc_compress_filter);
   grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX, prepend_filter,
                                    (void *)&grpc_compress_filter);
   grpc_channel_init_register_stage(GRPC_CLIENT_SUBCHANNEL, INT_MAX,
                                    maybe_add_http_filter,
                                    (void *)&grpc_http_client_filter);
   grpc_channel_init_register_stage(GRPC_CLIENT_SUBCHANNEL, INT_MAX,
                                    grpc_add_connected_filter, NULL);
   grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, INT_MAX,
                                    maybe_add_http_filter,
                                    (void *)&grpc_http_client_filter);
   grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, INT_MAX,
                                    grpc_add_connected_filter, NULL);
   grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX,
                                    maybe_add_http_filter,
                                    (void *)&grpc_http_server_filter);
   grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX,
                                    grpc_add_connected_filter, NULL);
   grpc_channel_init_register_stage(GRPC_CLIENT_CHANNEL, INT_MAX, append_filter,
                                    (void *)&grpc_client_channel_filter);
   grpc_channel_init_register_stage(GRPC_CLIENT_LAME_CHANNEL, INT_MAX,
                                    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_mdctx_global_init();
     grpc_channel_init_init();
     grpc_lb_policy_registry_init(grpc_pick_first_lb_factory_create());
     grpc_register_lb_policy(grpc_pick_first_lb_factory_create());
     grpc_register_lb_policy(grpc_round_robin_lb_factory_create());
     grpc_resolver_registry_init(GRPC_DEFAULT_NAME_PREFIX);
     grpc_register_resolver_type(grpc_dns_resolver_factory_create());
     grpc_register_resolver_type(grpc_ipv4_resolver_factory_create());
     grpc_register_resolver_type(grpc_ipv6_resolver_factory_create());
 #ifdef GPR_POSIX_SOCKET
     grpc_register_resolver_type(grpc_unix_resolver_factory_create());
 #endif
     grpc_register_tracer("api", &grpc_api_trace);
     grpc_register_tracer("channel", &grpc_trace_channel);
     grpc_register_tracer("http", &grpc_http_trace);
     grpc_register_tracer("flowctl", &grpc_flowctl_trace);
     grpc_register_tracer("connectivity_state", &grpc_connectivity_state_trace);
     grpc_register_tracer("channel_stack_builder",
                          &grpc_trace_channel_stack_builder);
     grpc_security_pre_init();
     grpc_iomgr_init();
     grpc_executor_init();
     grpc_tracer_init("GRPC_TRACE");
     gpr_timers_global_init();
     grpc_cq_global_init();
     grpc_subchannel_index_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();
     /* 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, ());
   gpr_mu_lock(&g_init_mu);
   if (--g_initializations == 0) {
     grpc_executor_shutdown();
     grpc_cq_global_shutdown();
     grpc_iomgr_shutdown();
     grpc_subchannel_index_shutdown();
     gpr_timers_global_destroy();
     grpc_tracer_shutdown();
     grpc_resolver_registry_shutdown();
     grpc_lb_policy_registry_shutdown();
     for (i = 0; i < g_number_of_plugins; i++) {
       if (g_all_of_the_plugins[i].destroy != NULL) {
         g_all_of_the_plugins[i].destroy();
       }
     }
     grpc_channel_init_shutdown();
     grpc_mdctx_global_shutdown();
   }
   gpr_mu_unlock(&g_init_mu);
 }

 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;
 }
	/*
	*
	* Copyright 2015-2016, 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/time.h>
	/* TODO(ctiller): find another way? - better not to include census here */
	#include "src/core/lib/census/grpc_plugin.h"
	#include "src/core/lib/channel/channel_stack.h"
	#include "src/core/lib/channel/client_channel.h"
	#include "src/core/lib/channel/compress_filter.h"
	#include "src/core/lib/channel/connected_channel.h"
	#include "src/core/lib/channel/http_client_filter.h"
	#include "src/core/lib/channel/http_server_filter.h"
	#include "src/core/lib/client_config/lb_policies/pick_first.h"
	#include "src/core/lib/client_config/lb_policies/round_robin.h"
	#include "src/core/lib/client_config/lb_policy_registry.h"
	#include "src/core/lib/client_config/resolver_registry.h"
	#include "src/core/lib/client_config/resolvers/dns_resolver.h"
	#include "src/core/lib/client_config/resolvers/sockaddr_resolver.h"
	#include "src/core/lib/client_config/subchannel.h"
	#include "src/core/lib/client_config/subchannel_index.h"
	#include "src/core/lib/debug/trace.h"
	#include "src/core/lib/iomgr/executor.h"
	#include "src/core/lib/iomgr/iomgr.h"
	#include "src/core/lib/profiling/timers.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/surface/surface_trace.h"
	#include "src/core/lib/transport/chttp2_transport.h"
	#include "src/core/lib/transport/connectivity_state.h"
	#include "src/core/lib/transport/transport_impl.h"

	#ifndef GRPC_DEFAULT_NAME_PREFIX
	#define GRPC_DEFAULT_NAME_PREFIX "dns:///"
	#endif

	#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_mu_init(&g_init_mu);
	/* TODO(ctiller): ideally remove this strict linkage */
	grpc_register_plugin(census_grpc_plugin_init, census_grpc_plugin_destroy);
	g_initializations = 0;
	}

	static bool append_filter(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_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_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_CHANNEL, INT_MAX, prepend_filter,
	(void *)&grpc_compress_filter);
	grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, INT_MAX,
	prepend_filter,
	(void *)&grpc_compress_filter);
	grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX, prepend_filter,
	(void *)&grpc_compress_filter);
	grpc_channel_init_register_stage(GRPC_CLIENT_SUBCHANNEL, INT_MAX,
	maybe_add_http_filter,
	(void *)&grpc_http_client_filter);
	grpc_channel_init_register_stage(GRPC_CLIENT_SUBCHANNEL, INT_MAX,
	grpc_add_connected_filter, NULL);
	grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, INT_MAX,
	maybe_add_http_filter,
	(void *)&grpc_http_client_filter);
	grpc_channel_init_register_stage(GRPC_CLIENT_DIRECT_CHANNEL, INT_MAX,
	grpc_add_connected_filter, NULL);
	grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX,
	maybe_add_http_filter,
	(void *)&grpc_http_server_filter);
	grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX,
	grpc_add_connected_filter, NULL);
	grpc_channel_init_register_stage(GRPC_CLIENT_CHANNEL, INT_MAX, append_filter,
	(void *)&grpc_client_channel_filter);
	grpc_channel_init_register_stage(GRPC_CLIENT_LAME_CHANNEL, INT_MAX,
	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_mdctx_global_init();
	grpc_channel_init_init();
	grpc_lb_policy_registry_init(grpc_pick_first_lb_factory_create());
	grpc_register_lb_policy(grpc_pick_first_lb_factory_create());
	grpc_register_lb_policy(grpc_round_robin_lb_factory_create());
	grpc_resolver_registry_init(GRPC_DEFAULT_NAME_PREFIX);
	grpc_register_resolver_type(grpc_dns_resolver_factory_create());
	grpc_register_resolver_type(grpc_ipv4_resolver_factory_create());
	grpc_register_resolver_type(grpc_ipv6_resolver_factory_create());
	#ifdef GPR_POSIX_SOCKET
	grpc_register_resolver_type(grpc_unix_resolver_factory_create());
	#endif
	grpc_register_tracer("api", &grpc_api_trace);
	grpc_register_tracer("channel", &grpc_trace_channel);
	grpc_register_tracer("http", &grpc_http_trace);
	grpc_register_tracer("flowctl", &grpc_flowctl_trace);
	grpc_register_tracer("connectivity_state", &grpc_connectivity_state_trace);
	grpc_register_tracer("channel_stack_builder",
	&grpc_trace_channel_stack_builder);
	grpc_security_pre_init();
	grpc_iomgr_init();
	grpc_executor_init();
	grpc_tracer_init("GRPC_TRACE");
	gpr_timers_global_init();
	grpc_cq_global_init();
	grpc_subchannel_index_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();
	/* 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, ());
	gpr_mu_lock(&g_init_mu);
	if (--g_initializations == 0) {
	grpc_executor_shutdown();
	grpc_cq_global_shutdown();
	grpc_iomgr_shutdown();
	grpc_subchannel_index_shutdown();
	gpr_timers_global_destroy();
	grpc_tracer_shutdown();
	grpc_resolver_registry_shutdown();
	grpc_lb_policy_registry_shutdown();
	for (i = 0; i < g_number_of_plugins; i++) {
	if (g_all_of_the_plugins[i].destroy != NULL) {
	g_all_of_the_plugins[i].destroy();
	}
	}
	grpc_channel_init_shutdown();
	grpc_mdctx_global_shutdown();
	}
	gpr_mu_unlock(&g_init_mu);
	}

	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;
	}