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_WINSOCK_SOCKET
24
25 #include "absl/log/check.h"
26 #include "src/core/lib/experiments/experiments.h"
27 #include "src/core/lib/iomgr/iocp_windows.h"
28 #include "src/core/lib/iomgr/iomgr.h"
29 #include "src/core/lib/iomgr/pollset_windows.h"
30 #include "src/core/lib/iomgr/resolve_address.h"
31 #include "src/core/lib/iomgr/resolve_address_windows.h"
32 #include "src/core/lib/iomgr/sockaddr_windows.h"
33 #include "src/core/lib/iomgr/socket_windows.h"
34 #include "src/core/lib/iomgr/tcp_client.h"
35 #include "src/core/lib/iomgr/tcp_server.h"
36 #include "src/core/lib/iomgr/timer.h"
37 #include "src/core/util/crash.h"
38
39 extern grpc_tcp_server_vtable grpc_windows_tcp_server_vtable;
40 extern grpc_tcp_server_vtable grpc_windows_event_engine_tcp_server_vtable;
41 extern grpc_tcp_client_vtable grpc_windows_tcp_client_vtable;
42 extern grpc_timer_vtable grpc_generic_timer_vtable;
43 extern grpc_pollset_vtable grpc_windows_pollset_vtable;
44 extern grpc_pollset_set_vtable grpc_windows_pollset_set_vtable;
45
46 // Windows' io manager is going to be fully designed using IO completion
47 // ports. All of what we're doing here is basically make sure that
48 // Windows sockets are initialized in and out.
49
winsock_init(void)50 static void winsock_init(void) {
51 WSADATA wsaData;
52 int status = WSAStartup(MAKEWORD(2, 0), &wsaData);
53 CHECK_EQ(status, 0);
54 }
55
winsock_shutdown(void)56 static void winsock_shutdown(void) {
57 int status = WSACleanup();
58 CHECK_EQ(status, 0);
59 }
60
iomgr_platform_init(void)61 static void iomgr_platform_init(void) {
62 winsock_init();
63 grpc_iocp_init();
64 grpc_pollset_global_init();
65 grpc_wsa_socket_flags_init();
66 grpc_core::ResetDNSResolver(std::make_unique<grpc_core::NativeDNSResolver>());
67 }
68
iomgr_platform_flush(void)69 static void iomgr_platform_flush(void) { grpc_iocp_flush(); }
70
iomgr_platform_shutdown(void)71 static void iomgr_platform_shutdown(void) {
72 grpc_pollset_global_shutdown();
73 grpc_iocp_shutdown();
74 winsock_shutdown();
75 grpc_core::ResetDNSResolver(nullptr); // delete the resolver
76 }
77
iomgr_platform_shutdown_background_closure(void)78 static void iomgr_platform_shutdown_background_closure(void) {}
79
iomgr_platform_is_any_background_poller_thread(void)80 static bool iomgr_platform_is_any_background_poller_thread(void) {
81 return false;
82 }
83
iomgr_platform_add_closure_to_background_poller(grpc_closure *,grpc_error_handle)84 static bool iomgr_platform_add_closure_to_background_poller(
85 grpc_closure* /* closure */, grpc_error_handle /* error */) {
86 return false;
87 }
88
89 static grpc_iomgr_platform_vtable vtable = {
90 iomgr_platform_init,
91 iomgr_platform_flush,
92 iomgr_platform_shutdown,
93 iomgr_platform_shutdown_background_closure,
94 iomgr_platform_is_any_background_poller_thread,
95 iomgr_platform_add_closure_to_background_poller};
96
grpc_set_default_iomgr_platform()97 void grpc_set_default_iomgr_platform() {
98 grpc_set_tcp_client_impl(&grpc_windows_tcp_client_vtable);
99 if (grpc_core::IsEventEngineListenerEnabled()) {
100 grpc_set_tcp_server_impl(&grpc_windows_event_engine_tcp_server_vtable);
101 } else {
102 grpc_set_tcp_server_impl(&grpc_windows_tcp_server_vtable);
103 }
104 grpc_set_timer_impl(&grpc_generic_timer_vtable);
105 grpc_set_pollset_vtable(&grpc_windows_pollset_vtable);
106 grpc_set_pollset_set_vtable(&grpc_windows_pollset_set_vtable);
107 grpc_set_iomgr_platform_vtable(&vtable);
108 }
109
grpc_iomgr_run_in_background()110 bool grpc_iomgr_run_in_background() {
111 return grpc_core::IsEventEngineCallbackCqEnabled();
112 }
113
114 #endif // GRPC_WINSOCK_SOCKET
115