grpc 1.13.0 → 1.14.0

Sign up to get free protection for your applications and to get access to all the features.

Potentially problematic release.


This version of grpc might be problematic. Click here for more details.

Files changed (213) hide show
  1. checksums.yaml +4 -4
  2. data/Makefile +403 -153
  3. data/include/grpc/grpc.h +0 -8
  4. data/include/grpc/grpc_security.h +59 -2
  5. data/include/grpc/impl/codegen/grpc_types.h +8 -2
  6. data/include/grpc/impl/codegen/log.h +112 -0
  7. data/include/grpc/module.modulemap +2 -0
  8. data/include/grpc/support/log.h +2 -88
  9. data/include/grpc/support/string_util.h +2 -0
  10. data/src/boringssl/err_data.c +597 -593
  11. data/src/core/ext/filters/client_channel/client_channel.cc +715 -770
  12. data/src/core/ext/filters/client_channel/client_channel.h +5 -0
  13. data/src/core/ext/filters/client_channel/client_channel_channelz.cc +111 -0
  14. data/src/core/ext/filters/client_channel/client_channel_channelz.h +69 -0
  15. data/src/core/ext/filters/client_channel/client_channel_plugin.cc +9 -0
  16. data/src/core/ext/filters/client_channel/http_proxy.cc +22 -5
  17. data/src/core/ext/filters/client_channel/lb_policy.h +15 -0
  18. data/src/core/ext/filters/client_channel/lb_policy/grpclb/grpclb.cc +3 -0
  19. data/src/core/ext/filters/client_channel/lb_policy/grpclb/load_balancer_api.cc +3 -3
  20. data/src/core/ext/filters/client_channel/lb_policy/grpclb/load_balancer_api.h +3 -1
  21. data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/duration.pb.c +19 -0
  22. data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/duration.pb.h +54 -0
  23. data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/timestamp.pb.c +19 -0
  24. data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/timestamp.pb.h +54 -0
  25. data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/load_balancer.pb.c +4 -17
  26. data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/load_balancer.pb.h +37 -63
  27. data/src/core/ext/filters/client_channel/lb_policy/pick_first/pick_first.cc +79 -0
  28. data/src/core/ext/filters/client_channel/lb_policy/round_robin/round_robin.cc +5 -2
  29. data/src/core/ext/filters/client_channel/lb_policy_factory.cc +8 -0
  30. data/src/core/ext/filters/client_channel/lb_policy_factory.h +4 -0
  31. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/dns_resolver_ares.cc +2 -2
  32. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.cc +317 -0
  33. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.h +48 -9
  34. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver_posix.cc +40 -293
  35. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.cc +106 -84
  36. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h +6 -2
  37. data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper_fallback.cc +6 -5
  38. data/src/core/ext/filters/client_channel/subchannel.cc +36 -6
  39. data/src/core/ext/filters/client_channel/subchannel.h +4 -0
  40. data/src/core/ext/filters/deadline/deadline_filter.cc +18 -15
  41. data/src/core/ext/filters/deadline/deadline_filter.h +5 -5
  42. data/src/core/ext/filters/http/client/http_client_filter.cc +10 -9
  43. data/src/core/ext/filters/http/server/http_server_filter.h +1 -1
  44. data/src/core/ext/transport/chttp2/client/insecure/channel_create_posix.cc +1 -1
  45. data/src/core/ext/transport/chttp2/server/insecure/server_chttp2_posix.cc +3 -2
  46. data/src/core/ext/transport/chttp2/transport/chttp2_transport.cc +33 -22
  47. data/src/core/ext/transport/chttp2/transport/hpack_parser.cc +1 -1
  48. data/src/core/ext/transport/chttp2/transport/internal.h +10 -3
  49. data/src/core/ext/transport/chttp2/transport/stream_lists.cc +17 -0
  50. data/src/core/ext/transport/chttp2/transport/writing.cc +21 -16
  51. data/src/core/ext/transport/inproc/inproc_transport.cc +46 -6
  52. data/src/core/lib/channel/channel_stack.cc +22 -24
  53. data/src/core/lib/channel/channel_trace.cc +28 -63
  54. data/src/core/lib/channel/channel_trace.h +13 -17
  55. data/src/core/lib/channel/channelz.cc +143 -0
  56. data/src/core/lib/channel/channelz.h +124 -0
  57. data/src/core/lib/channel/channelz_registry.cc +7 -24
  58. data/src/core/lib/channel/channelz_registry.h +12 -8
  59. data/src/core/lib/channel/connected_channel.cc +8 -1
  60. data/src/core/{ext/filters/load_reporting/server_load_reporting_filter.h → lib/gpr/alloc.h} +7 -9
  61. data/src/core/lib/gpr/arena.cc +8 -8
  62. data/src/core/lib/gpr/string.cc +28 -0
  63. data/src/core/lib/gpr/string.h +10 -0
  64. data/src/core/lib/gprpp/abstract.h +5 -2
  65. data/src/core/lib/gprpp/inlined_vector.h +57 -3
  66. data/src/core/lib/gprpp/memory.h +2 -2
  67. data/src/core/lib/gprpp/ref_counted_ptr.h +5 -0
  68. data/src/core/lib/gprpp/thd_posix.cc +1 -1
  69. data/src/core/lib/iomgr/call_combiner.h +80 -0
  70. data/src/core/lib/iomgr/closure.h +3 -2
  71. data/src/core/lib/iomgr/endpoint_pair_posix.cc +2 -2
  72. data/src/core/lib/iomgr/error.cc +12 -0
  73. data/src/core/lib/iomgr/error.h +5 -0
  74. data/src/core/lib/iomgr/ev_epoll1_linux.cc +36 -9
  75. data/src/core/lib/iomgr/ev_epollex_linux.cc +172 -46
  76. data/src/core/lib/iomgr/ev_epollsig_linux.cc +47 -21
  77. data/src/core/lib/iomgr/ev_poll_posix.cc +10 -4
  78. data/src/core/lib/iomgr/ev_posix.cc +17 -9
  79. data/src/core/lib/iomgr/ev_posix.h +20 -4
  80. data/src/core/lib/iomgr/executor.cc +196 -140
  81. data/src/core/lib/iomgr/executor.h +47 -14
  82. data/src/core/lib/iomgr/iomgr.cc +2 -0
  83. data/src/core/lib/iomgr/iomgr.h +5 -0
  84. data/src/core/lib/iomgr/is_epollexclusive_available.cc +1 -0
  85. data/src/core/lib/iomgr/socket_utils.h +9 -0
  86. data/src/core/lib/iomgr/socket_utils_common_posix.cc +4 -0
  87. data/src/core/lib/iomgr/socket_utils_uv.cc +4 -0
  88. data/src/core/lib/iomgr/socket_utils_windows.cc +4 -0
  89. data/src/core/lib/iomgr/tcp_client_posix.cc +3 -5
  90. data/src/core/lib/iomgr/tcp_posix.cc +6 -1
  91. data/src/core/lib/iomgr/tcp_server_posix.cc +3 -3
  92. data/src/core/lib/iomgr/tcp_server_utils_posix_common.cc +1 -1
  93. data/src/core/lib/iomgr/timer_manager.cc +0 -1
  94. data/src/core/lib/iomgr/udp_server.cc +2 -3
  95. data/src/core/lib/json/json.cc +10 -0
  96. data/src/core/lib/json/json.h +5 -0
  97. data/src/core/lib/security/context/security_context.cc +8 -8
  98. data/src/core/lib/security/context/security_context.h +6 -2
  99. data/src/core/lib/security/credentials/google_default/google_default_credentials.cc +2 -1
  100. data/src/core/lib/security/credentials/local/local_credentials.cc +77 -0
  101. data/src/core/lib/security/credentials/local/local_credentials.h +40 -0
  102. data/src/core/lib/security/credentials/ssl/ssl_credentials.cc +17 -3
  103. data/src/core/lib/security/security_connector/local_security_connector.cc +245 -0
  104. data/src/core/lib/security/security_connector/local_security_connector.h +58 -0
  105. data/src/core/lib/security/security_connector/security_connector.cc +30 -5
  106. data/src/core/lib/security/security_connector/security_connector.h +1 -0
  107. data/src/core/lib/security/transport/client_auth_filter.cc +5 -1
  108. data/src/core/lib/security/transport/server_auth_filter.cc +4 -5
  109. data/src/core/lib/surface/call.cc +75 -32
  110. data/src/core/lib/surface/call.h +2 -0
  111. data/src/core/lib/surface/channel.cc +32 -13
  112. data/src/core/lib/surface/channel.h +4 -0
  113. data/src/core/lib/surface/version.cc +1 -1
  114. data/src/core/lib/transport/transport.cc +20 -9
  115. data/src/core/lib/transport/transport.h +12 -10
  116. data/src/core/lib/transport/transport_op_string.cc +0 -7
  117. data/src/core/plugin_registry/grpc_plugin_registry.cc +0 -4
  118. data/src/core/tsi/alts/handshaker/alts_handshaker_service_api_util.h +2 -2
  119. data/src/core/tsi/alts/handshaker/alts_tsi_handshaker.cc +2 -1
  120. data/src/core/tsi/alts/handshaker/altscontext.pb.c +0 -1
  121. data/src/core/tsi/alts/handshaker/altscontext.pb.h +1 -2
  122. data/src/core/tsi/alts/handshaker/handshaker.pb.c +0 -1
  123. data/src/core/tsi/alts/handshaker/handshaker.pb.h +1 -2
  124. data/src/core/tsi/alts/handshaker/transport_security_common.pb.c +0 -1
  125. data/src/core/tsi/alts/handshaker/transport_security_common.pb.h +1 -1
  126. data/src/core/tsi/alts/handshaker/transport_security_common_api.h +2 -2
  127. data/src/core/tsi/alts/zero_copy_frame_protector/alts_grpc_integrity_only_record_protocol.cc +47 -1
  128. data/src/core/tsi/alts/zero_copy_frame_protector/alts_grpc_integrity_only_record_protocol.h +3 -1
  129. data/src/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector.cc +12 -11
  130. data/src/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector.h +7 -2
  131. data/src/core/tsi/local_transport_security.cc +209 -0
  132. data/src/core/tsi/local_transport_security.h +51 -0
  133. data/src/core/tsi/ssl_transport_security.cc +2 -3
  134. data/src/{core/ext → cpp/ext/filters}/census/grpc_context.cc +0 -0
  135. data/src/ruby/ext/grpc/rb_channel_credentials.c +3 -3
  136. data/src/ruby/ext/grpc/rb_grpc_imports.generated.c +18 -18
  137. data/src/ruby/ext/grpc/rb_grpc_imports.generated.h +29 -29
  138. data/src/ruby/lib/grpc/generic/active_call.rb +19 -23
  139. data/src/ruby/lib/grpc/version.rb +1 -1
  140. data/src/ruby/spec/call_credentials_spec.rb +1 -1
  141. data/src/ruby/spec/call_spec.rb +1 -1
  142. data/src/ruby/spec/channel_credentials_spec.rb +1 -1
  143. data/src/ruby/spec/channel_spec.rb +1 -1
  144. data/src/ruby/spec/client_auth_spec.rb +1 -12
  145. data/src/ruby/spec/client_server_spec.rb +1 -1
  146. data/src/ruby/spec/compression_options_spec.rb +1 -1
  147. data/src/ruby/spec/error_sanity_spec.rb +1 -1
  148. data/src/ruby/spec/generic/client_stub_spec.rb +13 -1
  149. data/src/ruby/spec/generic/rpc_desc_spec.rb +1 -1
  150. data/src/ruby/spec/generic/rpc_server_pool_spec.rb +1 -1
  151. data/src/ruby/spec/generic/service_spec.rb +1 -1
  152. data/src/ruby/spec/google_rpc_status_utils_spec.rb +1 -12
  153. data/src/ruby/spec/pb/duplicate/codegen_spec.rb +1 -0
  154. data/src/ruby/spec/pb/health/checker_spec.rb +1 -1
  155. data/src/ruby/spec/server_credentials_spec.rb +1 -1
  156. data/src/ruby/spec/server_spec.rb +1 -1
  157. data/src/ruby/spec/spec_helper.rb +1 -0
  158. data/src/ruby/spec/support/services.rb +1 -1
  159. data/src/ruby/spec/time_consts_spec.rb +1 -1
  160. data/third_party/boringssl/crypto/asn1/tasn_dec.c +40 -19
  161. data/third_party/boringssl/crypto/bytestring/cbs.c +1 -0
  162. data/third_party/boringssl/crypto/cipher_extra/e_aesccm.c +47 -15
  163. data/third_party/boringssl/crypto/ec_extra/ec_asn1.c +9 -10
  164. data/third_party/boringssl/crypto/ecdh/ecdh.c +4 -3
  165. data/third_party/boringssl/crypto/fipsmodule/bn/add.c +30 -54
  166. data/third_party/boringssl/crypto/fipsmodule/bn/bn.c +7 -1
  167. data/third_party/boringssl/crypto/fipsmodule/bn/cmp.c +8 -8
  168. data/third_party/boringssl/crypto/fipsmodule/bn/div.c +97 -11
  169. data/third_party/boringssl/crypto/fipsmodule/bn/gcd.c +274 -218
  170. data/third_party/boringssl/crypto/fipsmodule/bn/internal.h +111 -34
  171. data/third_party/boringssl/crypto/fipsmodule/bn/montgomery.c +2 -2
  172. data/third_party/boringssl/crypto/fipsmodule/bn/montgomery_inv.c +1 -1
  173. data/third_party/boringssl/crypto/fipsmodule/bn/mul.c +24 -6
  174. data/third_party/boringssl/crypto/fipsmodule/bn/prime.c +324 -63
  175. data/third_party/boringssl/crypto/fipsmodule/bn/random.c +74 -21
  176. data/third_party/boringssl/crypto/fipsmodule/bn/shift.c +128 -86
  177. data/third_party/boringssl/crypto/fipsmodule/bn/sqrt.c +1 -1
  178. data/third_party/boringssl/crypto/fipsmodule/ec/ec_key.c +67 -112
  179. data/third_party/boringssl/crypto/fipsmodule/ec/internal.h +8 -1
  180. data/third_party/boringssl/crypto/fipsmodule/ec/oct.c +5 -5
  181. data/third_party/boringssl/crypto/fipsmodule/ec/p224-64.c +9 -17
  182. data/third_party/boringssl/crypto/fipsmodule/ec/p256-x86_64-table.h +5378 -5418
  183. data/third_party/boringssl/crypto/fipsmodule/ec/simple.c +32 -32
  184. data/third_party/boringssl/crypto/fipsmodule/ecdsa/ecdsa.c +5 -11
  185. data/third_party/boringssl/crypto/fipsmodule/rsa/blinding.c +16 -40
  186. data/third_party/boringssl/crypto/fipsmodule/rsa/internal.h +1 -6
  187. data/third_party/boringssl/crypto/fipsmodule/rsa/rsa.c +41 -29
  188. data/third_party/boringssl/crypto/fipsmodule/rsa/rsa_impl.c +63 -49
  189. data/third_party/boringssl/crypto/x509/vpm_int.h +1 -0
  190. data/third_party/boringssl/crypto/x509/x509_vfy.c +4 -0
  191. data/third_party/boringssl/crypto/x509/x509_vpm.c +44 -22
  192. data/third_party/boringssl/include/openssl/aead.h +8 -2
  193. data/third_party/boringssl/include/openssl/asn1.h +1 -0
  194. data/third_party/boringssl/include/openssl/base.h +4 -0
  195. data/third_party/boringssl/include/openssl/bn.h +13 -3
  196. data/third_party/boringssl/include/openssl/bytestring.h +4 -4
  197. data/third_party/boringssl/include/openssl/ec.h +10 -4
  198. data/third_party/boringssl/include/openssl/ec_key.h +0 -3
  199. data/third_party/boringssl/include/openssl/rsa.h +1 -0
  200. data/third_party/boringssl/include/openssl/ssl.h +8 -3
  201. data/third_party/boringssl/include/openssl/ssl3.h +0 -1
  202. data/third_party/boringssl/include/openssl/x509.h +1 -0
  203. data/third_party/boringssl/include/openssl/x509v3.h +1 -0
  204. data/third_party/boringssl/ssl/handshake_client.cc +36 -64
  205. data/third_party/boringssl/ssl/ssl_cipher.cc +4 -0
  206. data/third_party/boringssl/ssl/ssl_lib.cc +1 -1
  207. metadata +45 -38
  208. data/src/core/ext/filters/load_reporting/server_load_reporting_filter.cc +0 -222
  209. data/src/core/ext/filters/load_reporting/server_load_reporting_plugin.cc +0 -71
  210. data/src/core/ext/filters/load_reporting/server_load_reporting_plugin.h +0 -61
  211. data/src/ruby/spec/pb/package_with_underscore/checker_spec.rb +0 -51
  212. data/src/ruby/spec/pb/package_with_underscore/data.proto +0 -23
  213. data/src/ruby/spec/pb/package_with_underscore/service.proto +0 -23
@@ -101,6 +101,10 @@ grpc_arg grpc_lb_addresses_create_channel_arg(
101
101
  grpc_lb_addresses* grpc_lb_addresses_find_channel_arg(
102
102
  const grpc_channel_args* channel_args);
103
103
 
104
+ // Returns true if addresses contains at least one balancer address.
105
+ bool grpc_lb_addresses_contains_balancer_address(
106
+ const grpc_lb_addresses& addresses);
107
+
104
108
  //
105
109
  // LB policy factory
106
110
  //
@@ -414,10 +414,10 @@ void AresDnsResolver::StartResolvingLocked() {
414
414
  resolving_ = true;
415
415
  lb_addresses_ = nullptr;
416
416
  service_config_json_ = nullptr;
417
- pending_request_ = grpc_dns_lookup_ares(
417
+ pending_request_ = grpc_dns_lookup_ares_locked(
418
418
  dns_server_, name_to_resolve_, kDefaultPort, interested_parties_,
419
419
  &on_resolved_, &lb_addresses_, true /* check_grpclb */,
420
- request_service_config_ ? &service_config_json_ : nullptr);
420
+ request_service_config_ ? &service_config_json_ : nullptr, combiner());
421
421
  last_resolution_timestamp_ = grpc_core::ExecCtx::Get()->Now();
422
422
  }
423
423
 
@@ -0,0 +1,317 @@
1
+ /*
2
+ *
3
+ * Copyright 2016 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
+ #include <grpc/support/port_platform.h>
19
+
20
+ #include "src/core/lib/iomgr/port.h"
21
+ #if GRPC_ARES == 1 && defined(GRPC_POSIX_SOCKET_ARES_EV_DRIVER)
22
+
23
+ #include <ares.h>
24
+ #include <string.h>
25
+ #include <sys/ioctl.h>
26
+
27
+ #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.h"
28
+
29
+ #include <grpc/support/alloc.h>
30
+ #include <grpc/support/log.h>
31
+ #include <grpc/support/string_util.h>
32
+ #include <grpc/support/time.h>
33
+ #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h"
34
+ #include "src/core/lib/gpr/string.h"
35
+ #include "src/core/lib/iomgr/ev_posix.h"
36
+ #include "src/core/lib/iomgr/iomgr_internal.h"
37
+ #include "src/core/lib/iomgr/sockaddr_utils.h"
38
+
39
+ typedef struct fd_node {
40
+ /** the owner of this fd node */
41
+ grpc_ares_ev_driver* ev_driver;
42
+ /** a closure wrapping on_readable_locked, which should be
43
+ invoked when the grpc_fd in this node becomes readable. */
44
+ grpc_closure read_closure;
45
+ /** a closure wrapping on_writable_locked, which should be
46
+ invoked when the grpc_fd in this node becomes writable. */
47
+ grpc_closure write_closure;
48
+ /** next fd node in the list */
49
+ struct fd_node* next;
50
+
51
+ /** wrapped fd that's polled by grpc's poller for the current platform */
52
+ grpc_core::GrpcPolledFd* grpc_polled_fd;
53
+ /** if the readable closure has been registered */
54
+ bool readable_registered;
55
+ /** if the writable closure has been registered */
56
+ bool writable_registered;
57
+ /** if the fd has been shutdown yet from grpc iomgr perspective */
58
+ bool already_shutdown;
59
+ } fd_node;
60
+
61
+ struct grpc_ares_ev_driver {
62
+ /** the ares_channel owned by this event driver */
63
+ ares_channel channel;
64
+ /** pollset set for driving the IO events of the channel */
65
+ grpc_pollset_set* pollset_set;
66
+ /** refcount of the event driver */
67
+ gpr_refcount refs;
68
+
69
+ /** combiner to synchronize c-ares and I/O callbacks on */
70
+ grpc_combiner* combiner;
71
+ /** a list of grpc_fd that this event driver is currently using. */
72
+ fd_node* fds;
73
+ /** is this event driver currently working? */
74
+ bool working;
75
+ /** is this event driver being shut down */
76
+ bool shutting_down;
77
+ /** request object that's using this ev driver */
78
+ grpc_ares_request* request;
79
+ };
80
+
81
+ static void grpc_ares_notify_on_event_locked(grpc_ares_ev_driver* ev_driver);
82
+
83
+ static grpc_ares_ev_driver* grpc_ares_ev_driver_ref(
84
+ grpc_ares_ev_driver* ev_driver) {
85
+ gpr_log(GPR_DEBUG, "Ref ev_driver %" PRIuPTR, (uintptr_t)ev_driver);
86
+ gpr_ref(&ev_driver->refs);
87
+ return ev_driver;
88
+ }
89
+
90
+ static void grpc_ares_ev_driver_unref(grpc_ares_ev_driver* ev_driver) {
91
+ gpr_log(GPR_DEBUG, "Unref ev_driver %" PRIuPTR, (uintptr_t)ev_driver);
92
+ if (gpr_unref(&ev_driver->refs)) {
93
+ gpr_log(GPR_DEBUG, "destroy ev_driver %" PRIuPTR, (uintptr_t)ev_driver);
94
+ GPR_ASSERT(ev_driver->fds == nullptr);
95
+ GRPC_COMBINER_UNREF(ev_driver->combiner, "free ares event driver");
96
+ ares_destroy(ev_driver->channel);
97
+ grpc_ares_complete_request_locked(ev_driver->request);
98
+ gpr_free(ev_driver);
99
+ }
100
+ }
101
+
102
+ static void fd_node_destroy_locked(fd_node* fdn) {
103
+ gpr_log(GPR_DEBUG, "delete fd: %s", fdn->grpc_polled_fd->GetName());
104
+ GPR_ASSERT(!fdn->readable_registered);
105
+ GPR_ASSERT(!fdn->writable_registered);
106
+ GPR_ASSERT(fdn->already_shutdown);
107
+ grpc_core::Delete(fdn->grpc_polled_fd);
108
+ gpr_free(fdn);
109
+ }
110
+
111
+ static void fd_node_shutdown_locked(fd_node* fdn, const char* reason) {
112
+ if (!fdn->already_shutdown) {
113
+ fdn->already_shutdown = true;
114
+ fdn->grpc_polled_fd->ShutdownLocked(
115
+ GRPC_ERROR_CREATE_FROM_STATIC_STRING(reason));
116
+ }
117
+ }
118
+
119
+ grpc_error* grpc_ares_ev_driver_create_locked(grpc_ares_ev_driver** ev_driver,
120
+ grpc_pollset_set* pollset_set,
121
+ grpc_combiner* combiner,
122
+ grpc_ares_request* request) {
123
+ *ev_driver = static_cast<grpc_ares_ev_driver*>(
124
+ gpr_malloc(sizeof(grpc_ares_ev_driver)));
125
+ ares_options opts;
126
+ memset(&opts, 0, sizeof(opts));
127
+ opts.flags |= ARES_FLAG_STAYOPEN;
128
+ int status = ares_init_options(&(*ev_driver)->channel, &opts, ARES_OPT_FLAGS);
129
+ grpc_core::ConfigureAresChannelLocked(&(*ev_driver)->channel);
130
+ gpr_log(GPR_DEBUG, "grpc_ares_ev_driver_create_locked");
131
+ if (status != ARES_SUCCESS) {
132
+ char* err_msg;
133
+ gpr_asprintf(&err_msg, "Failed to init ares channel. C-ares error: %s",
134
+ ares_strerror(status));
135
+ grpc_error* err = GRPC_ERROR_CREATE_FROM_COPIED_STRING(err_msg);
136
+ gpr_free(err_msg);
137
+ gpr_free(*ev_driver);
138
+ return err;
139
+ }
140
+ (*ev_driver)->combiner = GRPC_COMBINER_REF(combiner, "ares event driver");
141
+ gpr_ref_init(&(*ev_driver)->refs, 1);
142
+ (*ev_driver)->pollset_set = pollset_set;
143
+ (*ev_driver)->fds = nullptr;
144
+ (*ev_driver)->working = false;
145
+ (*ev_driver)->shutting_down = false;
146
+ (*ev_driver)->request = request;
147
+ return GRPC_ERROR_NONE;
148
+ }
149
+
150
+ void grpc_ares_ev_driver_on_queries_complete_locked(
151
+ grpc_ares_ev_driver* ev_driver) {
152
+ // We mark the event driver as being shut down. If the event driver
153
+ // is working, grpc_ares_notify_on_event_locked will shut down the
154
+ // fds; if it's not working, there are no fds to shut down.
155
+ ev_driver->shutting_down = true;
156
+ grpc_ares_ev_driver_unref(ev_driver);
157
+ }
158
+
159
+ void grpc_ares_ev_driver_shutdown_locked(grpc_ares_ev_driver* ev_driver) {
160
+ ev_driver->shutting_down = true;
161
+ fd_node* fn = ev_driver->fds;
162
+ while (fn != nullptr) {
163
+ fd_node_shutdown_locked(fn, "grpc_ares_ev_driver_shutdown");
164
+ fn = fn->next;
165
+ }
166
+ }
167
+
168
+ // Search fd in the fd_node list head. This is an O(n) search, the max possible
169
+ // value of n is ARES_GETSOCK_MAXNUM (16). n is typically 1 - 2 in our tests.
170
+ static fd_node* pop_fd_node_locked(fd_node** head, ares_socket_t as) {
171
+ fd_node dummy_head;
172
+ dummy_head.next = *head;
173
+ fd_node* node = &dummy_head;
174
+ while (node->next != nullptr) {
175
+ if (node->next->grpc_polled_fd->GetWrappedAresSocketLocked() == as) {
176
+ fd_node* ret = node->next;
177
+ node->next = node->next->next;
178
+ *head = dummy_head.next;
179
+ return ret;
180
+ }
181
+ node = node->next;
182
+ }
183
+ return nullptr;
184
+ }
185
+
186
+ static void on_readable_locked(void* arg, grpc_error* error) {
187
+ fd_node* fdn = static_cast<fd_node*>(arg);
188
+ grpc_ares_ev_driver* ev_driver = fdn->ev_driver;
189
+ const ares_socket_t as = fdn->grpc_polled_fd->GetWrappedAresSocketLocked();
190
+ fdn->readable_registered = false;
191
+ gpr_log(GPR_DEBUG, "readable on %s", fdn->grpc_polled_fd->GetName());
192
+ if (error == GRPC_ERROR_NONE) {
193
+ do {
194
+ ares_process_fd(ev_driver->channel, as, ARES_SOCKET_BAD);
195
+ } while (fdn->grpc_polled_fd->IsFdStillReadableLocked());
196
+ } else {
197
+ // If error is not GRPC_ERROR_NONE, it means the fd has been shutdown or
198
+ // timed out. The pending lookups made on this ev_driver will be cancelled
199
+ // by the following ares_cancel() and the on_done callbacks will be invoked
200
+ // with a status of ARES_ECANCELLED. The remaining file descriptors in this
201
+ // ev_driver will be cleaned up in the follwing
202
+ // grpc_ares_notify_on_event_locked().
203
+ ares_cancel(ev_driver->channel);
204
+ }
205
+ grpc_ares_notify_on_event_locked(ev_driver);
206
+ grpc_ares_ev_driver_unref(ev_driver);
207
+ }
208
+
209
+ static void on_writable_locked(void* arg, grpc_error* error) {
210
+ fd_node* fdn = static_cast<fd_node*>(arg);
211
+ grpc_ares_ev_driver* ev_driver = fdn->ev_driver;
212
+ const ares_socket_t as = fdn->grpc_polled_fd->GetWrappedAresSocketLocked();
213
+ fdn->writable_registered = false;
214
+ gpr_log(GPR_DEBUG, "writable on %s", fdn->grpc_polled_fd->GetName());
215
+ if (error == GRPC_ERROR_NONE) {
216
+ ares_process_fd(ev_driver->channel, ARES_SOCKET_BAD, as);
217
+ } else {
218
+ // If error is not GRPC_ERROR_NONE, it means the fd has been shutdown or
219
+ // timed out. The pending lookups made on this ev_driver will be cancelled
220
+ // by the following ares_cancel() and the on_done callbacks will be invoked
221
+ // with a status of ARES_ECANCELLED. The remaining file descriptors in this
222
+ // ev_driver will be cleaned up in the follwing
223
+ // grpc_ares_notify_on_event_locked().
224
+ ares_cancel(ev_driver->channel);
225
+ }
226
+ grpc_ares_notify_on_event_locked(ev_driver);
227
+ grpc_ares_ev_driver_unref(ev_driver);
228
+ }
229
+
230
+ ares_channel* grpc_ares_ev_driver_get_channel_locked(
231
+ grpc_ares_ev_driver* ev_driver) {
232
+ return &ev_driver->channel;
233
+ }
234
+
235
+ // Get the file descriptors used by the ev_driver's ares channel, register
236
+ // driver_closure with these filedescriptors.
237
+ static void grpc_ares_notify_on_event_locked(grpc_ares_ev_driver* ev_driver) {
238
+ fd_node* new_list = nullptr;
239
+ if (!ev_driver->shutting_down) {
240
+ ares_socket_t socks[ARES_GETSOCK_MAXNUM];
241
+ int socks_bitmask =
242
+ ares_getsock(ev_driver->channel, socks, ARES_GETSOCK_MAXNUM);
243
+ for (size_t i = 0; i < ARES_GETSOCK_MAXNUM; i++) {
244
+ if (ARES_GETSOCK_READABLE(socks_bitmask, i) ||
245
+ ARES_GETSOCK_WRITABLE(socks_bitmask, i)) {
246
+ fd_node* fdn = pop_fd_node_locked(&ev_driver->fds, socks[i]);
247
+ // Create a new fd_node if sock[i] is not in the fd_node list.
248
+ if (fdn == nullptr) {
249
+ fdn = static_cast<fd_node*>(gpr_malloc(sizeof(fd_node)));
250
+ fdn->grpc_polled_fd = grpc_core::NewGrpcPolledFdLocked(
251
+ socks[i], ev_driver->pollset_set);
252
+ gpr_log(GPR_DEBUG, "new fd: %s", fdn->grpc_polled_fd->GetName());
253
+ fdn->ev_driver = ev_driver;
254
+ fdn->readable_registered = false;
255
+ fdn->writable_registered = false;
256
+ fdn->already_shutdown = false;
257
+ GRPC_CLOSURE_INIT(&fdn->read_closure, on_readable_locked, fdn,
258
+ grpc_combiner_scheduler(ev_driver->combiner));
259
+ GRPC_CLOSURE_INIT(&fdn->write_closure, on_writable_locked, fdn,
260
+ grpc_combiner_scheduler(ev_driver->combiner));
261
+ }
262
+ fdn->next = new_list;
263
+ new_list = fdn;
264
+ // Register read_closure if the socket is readable and read_closure has
265
+ // not been registered with this socket.
266
+ if (ARES_GETSOCK_READABLE(socks_bitmask, i) &&
267
+ !fdn->readable_registered) {
268
+ grpc_ares_ev_driver_ref(ev_driver);
269
+ gpr_log(GPR_DEBUG, "notify read on: %s",
270
+ fdn->grpc_polled_fd->GetName());
271
+ fdn->grpc_polled_fd->RegisterForOnReadableLocked(&fdn->read_closure);
272
+ fdn->readable_registered = true;
273
+ }
274
+ // Register write_closure if the socket is writable and write_closure
275
+ // has not been registered with this socket.
276
+ if (ARES_GETSOCK_WRITABLE(socks_bitmask, i) &&
277
+ !fdn->writable_registered) {
278
+ gpr_log(GPR_DEBUG, "notify write on: %s",
279
+ fdn->grpc_polled_fd->GetName());
280
+ grpc_ares_ev_driver_ref(ev_driver);
281
+ fdn->grpc_polled_fd->RegisterForOnWriteableLocked(
282
+ &fdn->write_closure);
283
+ fdn->writable_registered = true;
284
+ }
285
+ }
286
+ }
287
+ }
288
+ // Any remaining fds in ev_driver->fds were not returned by ares_getsock() and
289
+ // are therefore no longer in use, so they can be shut down and removed from
290
+ // the list.
291
+ while (ev_driver->fds != nullptr) {
292
+ fd_node* cur = ev_driver->fds;
293
+ ev_driver->fds = ev_driver->fds->next;
294
+ fd_node_shutdown_locked(cur, "c-ares fd shutdown");
295
+ if (!cur->readable_registered && !cur->writable_registered) {
296
+ fd_node_destroy_locked(cur);
297
+ } else {
298
+ cur->next = new_list;
299
+ new_list = cur;
300
+ }
301
+ }
302
+ ev_driver->fds = new_list;
303
+ // If the ev driver has no working fd, all the tasks are done.
304
+ if (new_list == nullptr) {
305
+ ev_driver->working = false;
306
+ gpr_log(GPR_DEBUG, "ev driver stop working");
307
+ }
308
+ }
309
+
310
+ void grpc_ares_ev_driver_start_locked(grpc_ares_ev_driver* ev_driver) {
311
+ if (!ev_driver->working) {
312
+ ev_driver->working = true;
313
+ grpc_ares_notify_on_event_locked(ev_driver);
314
+ }
315
+ }
316
+
317
+ #endif /* GRPC_ARES == 1 && defined(GRPC_POSIX_SOCKET_ARES_EV_DRIVER) */
@@ -22,6 +22,8 @@
22
22
  #include <grpc/support/port_platform.h>
23
23
 
24
24
  #include <ares.h>
25
+ #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h"
26
+ #include "src/core/lib/gprpp/abstract.h"
25
27
  #include "src/core/lib/iomgr/pollset_set.h"
26
28
 
27
29
  typedef struct grpc_ares_ev_driver grpc_ares_ev_driver;
@@ -29,25 +31,62 @@ typedef struct grpc_ares_ev_driver grpc_ares_ev_driver;
29
31
  /* Start \a ev_driver. It will keep working until all IO on its ares_channel is
30
32
  done, or grpc_ares_ev_driver_destroy() is called. It may notify the callbacks
31
33
  bound to its ares_channel when necessary. */
32
- void grpc_ares_ev_driver_start(grpc_ares_ev_driver* ev_driver);
34
+ void grpc_ares_ev_driver_start_locked(grpc_ares_ev_driver* ev_driver);
33
35
 
34
36
  /* Returns the ares_channel owned by \a ev_driver. To bind a c-ares query to
35
37
  \a ev_driver, use the ares_channel owned by \a ev_driver as the arg of the
36
38
  query. */
37
- ares_channel* grpc_ares_ev_driver_get_channel(grpc_ares_ev_driver* ev_driver);
39
+ ares_channel* grpc_ares_ev_driver_get_channel_locked(
40
+ grpc_ares_ev_driver* ev_driver);
38
41
 
39
42
  /* Creates a new grpc_ares_ev_driver. Returns GRPC_ERROR_NONE if \a ev_driver is
40
43
  created successfully. */
41
- grpc_error* grpc_ares_ev_driver_create(grpc_ares_ev_driver** ev_driver,
42
- grpc_pollset_set* pollset_set);
44
+ grpc_error* grpc_ares_ev_driver_create_locked(grpc_ares_ev_driver** ev_driver,
45
+ grpc_pollset_set* pollset_set,
46
+ grpc_combiner* combiner,
47
+ grpc_ares_request* request);
43
48
 
44
- /* Destroys \a ev_driver asynchronously. Pending lookups made on \a ev_driver
45
- will be cancelled and their on_done callbacks will be invoked with a status
46
- of ARES_ECANCELLED. */
47
- void grpc_ares_ev_driver_destroy(grpc_ares_ev_driver* ev_driver);
49
+ /* Called back when all DNS lookups have completed. */
50
+ void grpc_ares_ev_driver_on_queries_complete_locked(
51
+ grpc_ares_ev_driver* ev_driver);
48
52
 
49
53
  /* Shutdown all the grpc_fds used by \a ev_driver */
50
- void grpc_ares_ev_driver_shutdown(grpc_ares_ev_driver* ev_driver);
54
+ void grpc_ares_ev_driver_shutdown_locked(grpc_ares_ev_driver* ev_driver);
55
+
56
+ namespace grpc_core {
57
+
58
+ /* A wrapped fd that integrates with the grpc iomgr of the current platform.
59
+ * A GrpcPolledFd knows how to create grpc platform-specific iomgr endpoints
60
+ * from "ares_socket_t" sockets, and then sign up for readability/writeability
61
+ * with that poller, and do shutdown and destruction. */
62
+ class GrpcPolledFd {
63
+ public:
64
+ virtual ~GrpcPolledFd() {}
65
+ /* Called when c-ares library is interested and there's no pending callback */
66
+ virtual void RegisterForOnReadableLocked(grpc_closure* read_closure)
67
+ GRPC_ABSTRACT;
68
+ /* Called when c-ares library is interested and there's no pending callback */
69
+ virtual void RegisterForOnWriteableLocked(grpc_closure* write_closure)
70
+ GRPC_ABSTRACT;
71
+ /* Indicates if there is data left even after just being read from */
72
+ virtual bool IsFdStillReadableLocked() GRPC_ABSTRACT;
73
+ /* Called once and only once. Must cause cancellation of any pending
74
+ * read/write callbacks. */
75
+ virtual void ShutdownLocked(grpc_error* error) GRPC_ABSTRACT;
76
+ /* Get the underlying ares_socket_t that this was created from */
77
+ virtual ares_socket_t GetWrappedAresSocketLocked() GRPC_ABSTRACT;
78
+ /* A unique name, for logging */
79
+ virtual const char* GetName() GRPC_ABSTRACT;
80
+
81
+ GRPC_ABSTRACT_BASE_CLASS
82
+ };
83
+
84
+ /* Creates a new wrapped fd for the current platform */
85
+ GrpcPolledFd* NewGrpcPolledFdLocked(ares_socket_t as,
86
+ grpc_pollset_set* driver_pollset_set);
87
+ void ConfigureAresChannelLocked(ares_channel* channel);
88
+
89
+ } // namespace grpc_core
51
90
 
52
91
  #endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_RESOLVER_DNS_C_ARES_GRPC_ARES_EV_DRIVER_H \
53
92
  */
@@ -21,6 +21,7 @@
21
21
  #if GRPC_ARES == 1 && defined(GRPC_POSIX_SOCKET_ARES_EV_DRIVER)
22
22
 
23
23
  #include <ares.h>
24
+ #include <string.h>
24
25
  #include <sys/ioctl.h>
25
26
 
26
27
  #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.h"
@@ -35,317 +36,63 @@
35
36
  #include "src/core/lib/iomgr/iomgr_internal.h"
36
37
  #include "src/core/lib/iomgr/sockaddr_utils.h"
37
38
 
38
- typedef struct fd_node {
39
- /** the owner of this fd node */
40
- grpc_ares_ev_driver* ev_driver;
41
- /** a closure wrapping on_readable_cb, which should be invoked when the
42
- grpc_fd in this node becomes readable. */
43
- grpc_closure read_closure;
44
- /** a closure wrapping on_writable_cb, which should be invoked when the
45
- grpc_fd in this node becomes writable. */
46
- grpc_closure write_closure;
47
- /** next fd node in the list */
48
- struct fd_node* next;
39
+ namespace grpc_core {
49
40
 
50
- /** mutex guarding the rest of the state */
51
- gpr_mu mu;
52
- /** the grpc_fd owned by this fd node */
53
- grpc_fd* fd;
54
- /** if the readable closure has been registered */
55
- bool readable_registered;
56
- /** if the writable closure has been registered */
57
- bool writable_registered;
58
- /** if the fd is being shut down */
59
- bool shutting_down;
60
- } fd_node;
61
-
62
- struct grpc_ares_ev_driver {
63
- /** the ares_channel owned by this event driver */
64
- ares_channel channel;
65
- /** pollset set for driving the IO events of the channel */
66
- grpc_pollset_set* pollset_set;
67
- /** refcount of the event driver */
68
- gpr_refcount refs;
69
-
70
- /** mutex guarding the rest of the state */
71
- gpr_mu mu;
72
- /** a list of grpc_fd that this event driver is currently using. */
73
- fd_node* fds;
74
- /** is this event driver currently working? */
75
- bool working;
76
- /** is this event driver being shut down */
77
- bool shutting_down;
78
- };
79
-
80
- static void grpc_ares_notify_on_event_locked(grpc_ares_ev_driver* ev_driver);
81
-
82
- static grpc_ares_ev_driver* grpc_ares_ev_driver_ref(
83
- grpc_ares_ev_driver* ev_driver) {
84
- gpr_log(GPR_DEBUG, "Ref ev_driver %" PRIuPTR, (uintptr_t)ev_driver);
85
- gpr_ref(&ev_driver->refs);
86
- return ev_driver;
87
- }
88
-
89
- static void grpc_ares_ev_driver_unref(grpc_ares_ev_driver* ev_driver) {
90
- gpr_log(GPR_DEBUG, "Unref ev_driver %" PRIuPTR, (uintptr_t)ev_driver);
91
- if (gpr_unref(&ev_driver->refs)) {
92
- gpr_log(GPR_DEBUG, "destroy ev_driver %" PRIuPTR, (uintptr_t)ev_driver);
93
- GPR_ASSERT(ev_driver->fds == nullptr);
94
- gpr_mu_destroy(&ev_driver->mu);
95
- ares_destroy(ev_driver->channel);
96
- gpr_free(ev_driver);
41
+ class GrpcPolledFdPosix : public GrpcPolledFd {
42
+ public:
43
+ GrpcPolledFdPosix(ares_socket_t as, grpc_pollset_set* driver_pollset_set)
44
+ : as_(as) {
45
+ gpr_asprintf(&name_, "c-ares fd: %d", (int)as);
46
+ fd_ = grpc_fd_create((int)as, name_, false);
47
+ driver_pollset_set_ = driver_pollset_set;
48
+ grpc_pollset_set_add_fd(driver_pollset_set_, fd_);
97
49
  }
98
- }
99
50
 
100
- static void fd_node_destroy(fd_node* fdn) {
101
- gpr_log(GPR_DEBUG, "delete fd: %d", grpc_fd_wrapped_fd(fdn->fd));
102
- GPR_ASSERT(!fdn->readable_registered);
103
- GPR_ASSERT(!fdn->writable_registered);
104
- gpr_mu_destroy(&fdn->mu);
105
- /* c-ares library has closed the fd inside grpc_fd. This fd may be picked up
106
- immediately by another thread, and should not be closed by the following
107
- grpc_fd_orphan. */
108
- grpc_fd_orphan(fdn->fd, nullptr, nullptr, true /* already_closed */,
109
- "c-ares query finished");
110
- gpr_free(fdn);
111
- }
112
-
113
- static void fd_node_shutdown(fd_node* fdn) {
114
- gpr_mu_lock(&fdn->mu);
115
- fdn->shutting_down = true;
116
- if (!fdn->readable_registered && !fdn->writable_registered) {
117
- gpr_mu_unlock(&fdn->mu);
118
- fd_node_destroy(fdn);
119
- } else {
120
- grpc_fd_shutdown(
121
- fdn->fd, GRPC_ERROR_CREATE_FROM_STATIC_STRING("c-ares fd shutdown"));
122
- gpr_mu_unlock(&fdn->mu);
51
+ ~GrpcPolledFdPosix() {
52
+ gpr_free(name_);
53
+ grpc_pollset_set_del_fd(driver_pollset_set_, fd_);
54
+ /* c-ares library will close the fd inside grpc_fd. This fd may be picked up
55
+ immediately by another thread, and should not be closed by the following
56
+ grpc_fd_orphan. */
57
+ int dummy_release_fd;
58
+ grpc_fd_orphan(fd_, nullptr, &dummy_release_fd, "c-ares query finished");
123
59
  }
124
- }
125
60
 
126
- grpc_error* grpc_ares_ev_driver_create(grpc_ares_ev_driver** ev_driver,
127
- grpc_pollset_set* pollset_set) {
128
- *ev_driver = static_cast<grpc_ares_ev_driver*>(
129
- gpr_malloc(sizeof(grpc_ares_ev_driver)));
130
- int status = ares_init(&(*ev_driver)->channel);
131
- gpr_log(GPR_DEBUG, "grpc_ares_ev_driver_create");
132
- if (status != ARES_SUCCESS) {
133
- char* err_msg;
134
- gpr_asprintf(&err_msg, "Failed to init ares channel. C-ares error: %s",
135
- ares_strerror(status));
136
- grpc_error* err = GRPC_ERROR_CREATE_FROM_COPIED_STRING(err_msg);
137
- gpr_free(err_msg);
138
- gpr_free(*ev_driver);
139
- return err;
61
+ void RegisterForOnReadableLocked(grpc_closure* read_closure) override {
62
+ grpc_fd_notify_on_read(fd_, read_closure);
140
63
  }
141
- gpr_mu_init(&(*ev_driver)->mu);
142
- gpr_ref_init(&(*ev_driver)->refs, 1);
143
- (*ev_driver)->pollset_set = pollset_set;
144
- (*ev_driver)->fds = nullptr;
145
- (*ev_driver)->working = false;
146
- (*ev_driver)->shutting_down = false;
147
- return GRPC_ERROR_NONE;
148
- }
149
64
 
150
- void grpc_ares_ev_driver_destroy(grpc_ares_ev_driver* ev_driver) {
151
- // It's not safe to shut down remaining fds here directly, becauses
152
- // ares_host_callback does not provide an exec_ctx. We mark the event driver
153
- // as being shut down. If the event driver is working,
154
- // grpc_ares_notify_on_event_locked will shut down the fds; if it's not
155
- // working, there are no fds to shut down.
156
- gpr_mu_lock(&ev_driver->mu);
157
- ev_driver->shutting_down = true;
158
- gpr_mu_unlock(&ev_driver->mu);
159
- grpc_ares_ev_driver_unref(ev_driver);
160
- }
161
-
162
- void grpc_ares_ev_driver_shutdown(grpc_ares_ev_driver* ev_driver) {
163
- gpr_mu_lock(&ev_driver->mu);
164
- ev_driver->shutting_down = true;
165
- fd_node* fn = ev_driver->fds;
166
- while (fn != nullptr) {
167
- grpc_fd_shutdown(fn->fd, GRPC_ERROR_CREATE_FROM_STATIC_STRING(
168
- "grpc_ares_ev_driver_shutdown"));
169
- fn = fn->next;
65
+ void RegisterForOnWriteableLocked(grpc_closure* write_closure) override {
66
+ grpc_fd_notify_on_write(fd_, write_closure);
170
67
  }
171
- gpr_mu_unlock(&ev_driver->mu);
172
- }
173
68
 
174
- // Search fd in the fd_node list head. This is an O(n) search, the max possible
175
- // value of n is ARES_GETSOCK_MAXNUM (16). n is typically 1 - 2 in our tests.
176
- static fd_node* pop_fd_node(fd_node** head, int fd) {
177
- fd_node dummy_head;
178
- dummy_head.next = *head;
179
- fd_node* node = &dummy_head;
180
- while (node->next != nullptr) {
181
- if (grpc_fd_wrapped_fd(node->next->fd) == fd) {
182
- fd_node* ret = node->next;
183
- node->next = node->next->next;
184
- *head = dummy_head.next;
185
- return ret;
186
- }
187
- node = node->next;
69
+ bool IsFdStillReadableLocked() override {
70
+ size_t bytes_available = 0;
71
+ return ioctl(grpc_fd_wrapped_fd(fd_), FIONREAD, &bytes_available) == 0 &&
72
+ bytes_available > 0;
188
73
  }
189
- return nullptr;
190
- }
191
-
192
- /* Check if \a fd is still readable */
193
- static bool grpc_ares_is_fd_still_readable(grpc_ares_ev_driver* ev_driver,
194
- int fd) {
195
- size_t bytes_available = 0;
196
- return ioctl(fd, FIONREAD, &bytes_available) == 0 && bytes_available > 0;
197
- }
198
74
 
199
- static void on_readable_cb(void* arg, grpc_error* error) {
200
- fd_node* fdn = static_cast<fd_node*>(arg);
201
- grpc_ares_ev_driver* ev_driver = fdn->ev_driver;
202
- gpr_mu_lock(&fdn->mu);
203
- const int fd = grpc_fd_wrapped_fd(fdn->fd);
204
- fdn->readable_registered = false;
205
- if (fdn->shutting_down && !fdn->writable_registered) {
206
- gpr_mu_unlock(&fdn->mu);
207
- fd_node_destroy(fdn);
208
- grpc_ares_ev_driver_unref(ev_driver);
209
- return;
75
+ void ShutdownLocked(grpc_error* error) override {
76
+ grpc_fd_shutdown(fd_, error);
210
77
  }
211
- gpr_mu_unlock(&fdn->mu);
212
78
 
213
- gpr_log(GPR_DEBUG, "readable on %d", fd);
214
- if (error == GRPC_ERROR_NONE) {
215
- do {
216
- ares_process_fd(ev_driver->channel, fd, ARES_SOCKET_BAD);
217
- } while (grpc_ares_is_fd_still_readable(ev_driver, fd));
218
- } else {
219
- // If error is not GRPC_ERROR_NONE, it means the fd has been shutdown or
220
- // timed out. The pending lookups made on this ev_driver will be cancelled
221
- // by the following ares_cancel() and the on_done callbacks will be invoked
222
- // with a status of ARES_ECANCELLED. The remaining file descriptors in this
223
- // ev_driver will be cleaned up in the follwing
224
- // grpc_ares_notify_on_event_locked().
225
- ares_cancel(ev_driver->channel);
226
- }
227
- gpr_mu_lock(&ev_driver->mu);
228
- grpc_ares_notify_on_event_locked(ev_driver);
229
- gpr_mu_unlock(&ev_driver->mu);
230
- grpc_ares_ev_driver_unref(ev_driver);
231
- }
79
+ ares_socket_t GetWrappedAresSocketLocked() override { return as_; }
232
80
 
233
- static void on_writable_cb(void* arg, grpc_error* error) {
234
- fd_node* fdn = static_cast<fd_node*>(arg);
235
- grpc_ares_ev_driver* ev_driver = fdn->ev_driver;
236
- gpr_mu_lock(&fdn->mu);
237
- const int fd = grpc_fd_wrapped_fd(fdn->fd);
238
- fdn->writable_registered = false;
239
- if (fdn->shutting_down && !fdn->readable_registered) {
240
- gpr_mu_unlock(&fdn->mu);
241
- fd_node_destroy(fdn);
242
- grpc_ares_ev_driver_unref(ev_driver);
243
- return;
244
- }
245
- gpr_mu_unlock(&fdn->mu);
81
+ const char* GetName() override { return name_; }
246
82
 
247
- gpr_log(GPR_DEBUG, "writable on %d", fd);
248
- if (error == GRPC_ERROR_NONE) {
249
- ares_process_fd(ev_driver->channel, ARES_SOCKET_BAD, fd);
250
- } else {
251
- // If error is not GRPC_ERROR_NONE, it means the fd has been shutdown or
252
- // timed out. The pending lookups made on this ev_driver will be cancelled
253
- // by the following ares_cancel() and the on_done callbacks will be invoked
254
- // with a status of ARES_ECANCELLED. The remaining file descriptors in this
255
- // ev_driver will be cleaned up in the follwing
256
- // grpc_ares_notify_on_event_locked().
257
- ares_cancel(ev_driver->channel);
258
- }
259
- gpr_mu_lock(&ev_driver->mu);
260
- grpc_ares_notify_on_event_locked(ev_driver);
261
- gpr_mu_unlock(&ev_driver->mu);
262
- grpc_ares_ev_driver_unref(ev_driver);
263
- }
83
+ char* name_;
84
+ ares_socket_t as_;
85
+ grpc_fd* fd_;
86
+ grpc_pollset_set* driver_pollset_set_;
87
+ };
264
88
 
265
- ares_channel* grpc_ares_ev_driver_get_channel(grpc_ares_ev_driver* ev_driver) {
266
- return &ev_driver->channel;
89
+ GrpcPolledFd* NewGrpcPolledFdLocked(ares_socket_t as,
90
+ grpc_pollset_set* driver_pollset_set) {
91
+ return grpc_core::New<GrpcPolledFdPosix>(as, driver_pollset_set);
267
92
  }
268
93
 
269
- // Get the file descriptors used by the ev_driver's ares channel, register
270
- // driver_closure with these filedescriptors.
271
- static void grpc_ares_notify_on_event_locked(grpc_ares_ev_driver* ev_driver) {
272
- fd_node* new_list = nullptr;
273
- if (!ev_driver->shutting_down) {
274
- ares_socket_t socks[ARES_GETSOCK_MAXNUM];
275
- int socks_bitmask =
276
- ares_getsock(ev_driver->channel, socks, ARES_GETSOCK_MAXNUM);
277
- for (size_t i = 0; i < ARES_GETSOCK_MAXNUM; i++) {
278
- if (ARES_GETSOCK_READABLE(socks_bitmask, i) ||
279
- ARES_GETSOCK_WRITABLE(socks_bitmask, i)) {
280
- fd_node* fdn = pop_fd_node(&ev_driver->fds, socks[i]);
281
- // Create a new fd_node if sock[i] is not in the fd_node list.
282
- if (fdn == nullptr) {
283
- char* fd_name;
284
- gpr_asprintf(&fd_name, "ares_ev_driver-%" PRIuPTR, i);
285
- fdn = static_cast<fd_node*>(gpr_malloc(sizeof(fd_node)));
286
- gpr_log(GPR_DEBUG, "new fd: %d", socks[i]);
287
- fdn->fd = grpc_fd_create(socks[i], fd_name);
288
- fdn->ev_driver = ev_driver;
289
- fdn->readable_registered = false;
290
- fdn->writable_registered = false;
291
- fdn->shutting_down = false;
292
- gpr_mu_init(&fdn->mu);
293
- GRPC_CLOSURE_INIT(&fdn->read_closure, on_readable_cb, fdn,
294
- grpc_schedule_on_exec_ctx);
295
- GRPC_CLOSURE_INIT(&fdn->write_closure, on_writable_cb, fdn,
296
- grpc_schedule_on_exec_ctx);
297
- grpc_pollset_set_add_fd(ev_driver->pollset_set, fdn->fd);
298
- gpr_free(fd_name);
299
- }
300
- fdn->next = new_list;
301
- new_list = fdn;
302
- gpr_mu_lock(&fdn->mu);
303
- // Register read_closure if the socket is readable and read_closure has
304
- // not been registered with this socket.
305
- if (ARES_GETSOCK_READABLE(socks_bitmask, i) &&
306
- !fdn->readable_registered) {
307
- grpc_ares_ev_driver_ref(ev_driver);
308
- gpr_log(GPR_DEBUG, "notify read on: %d", grpc_fd_wrapped_fd(fdn->fd));
309
- grpc_fd_notify_on_read(fdn->fd, &fdn->read_closure);
310
- fdn->readable_registered = true;
311
- }
312
- // Register write_closure if the socket is writable and write_closure
313
- // has not been registered with this socket.
314
- if (ARES_GETSOCK_WRITABLE(socks_bitmask, i) &&
315
- !fdn->writable_registered) {
316
- gpr_log(GPR_DEBUG, "notify write on: %d",
317
- grpc_fd_wrapped_fd(fdn->fd));
318
- grpc_ares_ev_driver_ref(ev_driver);
319
- grpc_fd_notify_on_write(fdn->fd, &fdn->write_closure);
320
- fdn->writable_registered = true;
321
- }
322
- gpr_mu_unlock(&fdn->mu);
323
- }
324
- }
325
- }
326
- // Any remaining fds in ev_driver->fds were not returned by ares_getsock() and
327
- // are therefore no longer in use, so they can be shut down and removed from
328
- // the list.
329
- while (ev_driver->fds != nullptr) {
330
- fd_node* cur = ev_driver->fds;
331
- ev_driver->fds = ev_driver->fds->next;
332
- fd_node_shutdown(cur);
333
- }
334
- ev_driver->fds = new_list;
335
- // If the ev driver has no working fd, all the tasks are done.
336
- if (new_list == nullptr) {
337
- ev_driver->working = false;
338
- gpr_log(GPR_DEBUG, "ev driver stop working");
339
- }
340
- }
94
+ void ConfigureAresChannelLocked(ares_channel* channel) {}
341
95
 
342
- void grpc_ares_ev_driver_start(grpc_ares_ev_driver* ev_driver) {
343
- gpr_mu_lock(&ev_driver->mu);
344
- if (!ev_driver->working) {
345
- ev_driver->working = true;
346
- grpc_ares_notify_on_event_locked(ev_driver);
347
- }
348
- gpr_mu_unlock(&ev_driver->mu);
349
- }
96
+ } // namespace grpc_core
350
97
 
351
98
  #endif /* GRPC_ARES == 1 && defined(GRPC_POSIX_SOCKET_ARES_EV_DRIVER) */