• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  *
3  * Copyright 2015 gRPC authors.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  */
18 
19 #include <grpc/support/port_platform.h>
20 
21 #include "src/core/lib/iomgr/port.h"
22 
23 #ifdef GRPC_CFSTREAM_IOMGR
24 
25 #include "src/core/lib/debug/trace.h"
26 #include "src/core/lib/iomgr/ev_posix.h"
27 #include "src/core/lib/iomgr/iomgr_internal.h"
28 #include "src/core/lib/iomgr/iomgr_posix.h"
29 #include "src/core/lib/iomgr/resolve_address.h"
30 #include "src/core/lib/iomgr/tcp_client.h"
31 #include "src/core/lib/iomgr/tcp_posix.h"
32 #include "src/core/lib/iomgr/tcp_server.h"
33 #include "src/core/lib/iomgr/timer.h"
34 
35 static const char* grpc_cfstream_env_var = "grpc_cfstream";
36 
37 extern grpc_tcp_server_vtable grpc_posix_tcp_server_vtable;
38 extern grpc_tcp_client_vtable grpc_posix_tcp_client_vtable;
39 extern grpc_tcp_client_vtable grpc_cfstream_client_vtable;
40 extern grpc_timer_vtable grpc_generic_timer_vtable;
41 extern grpc_pollset_vtable grpc_posix_pollset_vtable;
42 extern grpc_pollset_set_vtable grpc_posix_pollset_set_vtable;
43 extern grpc_address_resolver_vtable grpc_posix_resolver_vtable;
44 
iomgr_platform_init(void)45 static void iomgr_platform_init(void) {
46   grpc_wakeup_fd_global_init();
47   grpc_event_engine_init();
48 }
49 
iomgr_platform_flush(void)50 static void iomgr_platform_flush(void) {}
51 
iomgr_platform_shutdown(void)52 static void iomgr_platform_shutdown(void) {
53   grpc_event_engine_shutdown();
54   grpc_wakeup_fd_global_destroy();
55 }
56 
57 static grpc_iomgr_platform_vtable vtable = {
58     iomgr_platform_init, iomgr_platform_flush, iomgr_platform_shutdown};
59 
grpc_set_default_iomgr_platform()60 void grpc_set_default_iomgr_platform() {
61   char* enable_cfstream = getenv(grpc_cfstream_env_var);
62   grpc_tcp_client_vtable* client_vtable = &grpc_posix_tcp_client_vtable;
63   if (enable_cfstream != nullptr && enable_cfstream[0] == '1') {
64     client_vtable = &grpc_cfstream_client_vtable;
65   }
66   grpc_set_tcp_client_impl(client_vtable);
67   grpc_set_tcp_server_impl(&grpc_posix_tcp_server_vtable);
68   grpc_set_timer_impl(&grpc_generic_timer_vtable);
69   grpc_set_pollset_vtable(&grpc_posix_pollset_vtable);
70   grpc_set_pollset_set_vtable(&grpc_posix_pollset_set_vtable);
71   grpc_set_resolver_impl(&grpc_posix_resolver_vtable);
72   grpc_set_iomgr_platform_vtable(&vtable);
73 }
74 
75 #endif /* GRPC_CFSTREAM_IOMGR */
76