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.
- checksums.yaml +4 -4
- data/Makefile +403 -153
- data/include/grpc/grpc.h +0 -8
- data/include/grpc/grpc_security.h +59 -2
- data/include/grpc/impl/codegen/grpc_types.h +8 -2
- data/include/grpc/impl/codegen/log.h +112 -0
- data/include/grpc/module.modulemap +2 -0
- data/include/grpc/support/log.h +2 -88
- data/include/grpc/support/string_util.h +2 -0
- data/src/boringssl/err_data.c +597 -593
- data/src/core/ext/filters/client_channel/client_channel.cc +715 -770
- data/src/core/ext/filters/client_channel/client_channel.h +5 -0
- data/src/core/ext/filters/client_channel/client_channel_channelz.cc +111 -0
- data/src/core/ext/filters/client_channel/client_channel_channelz.h +69 -0
- data/src/core/ext/filters/client_channel/client_channel_plugin.cc +9 -0
- data/src/core/ext/filters/client_channel/http_proxy.cc +22 -5
- data/src/core/ext/filters/client_channel/lb_policy.h +15 -0
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/grpclb.cc +3 -0
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/load_balancer_api.cc +3 -3
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/load_balancer_api.h +3 -1
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/duration.pb.c +19 -0
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/duration.pb.h +54 -0
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/timestamp.pb.c +19 -0
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/google/protobuf/timestamp.pb.h +54 -0
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/load_balancer.pb.c +4 -17
- data/src/core/ext/filters/client_channel/lb_policy/grpclb/proto/grpc/lb/v1/load_balancer.pb.h +37 -63
- data/src/core/ext/filters/client_channel/lb_policy/pick_first/pick_first.cc +79 -0
- data/src/core/ext/filters/client_channel/lb_policy/round_robin/round_robin.cc +5 -2
- data/src/core/ext/filters/client_channel/lb_policy_factory.cc +8 -0
- data/src/core/ext/filters/client_channel/lb_policy_factory.h +4 -0
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/dns_resolver_ares.cc +2 -2
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.cc +317 -0
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.h +48 -9
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver_posix.cc +40 -293
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.cc +106 -84
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h +6 -2
- data/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper_fallback.cc +6 -5
- data/src/core/ext/filters/client_channel/subchannel.cc +36 -6
- data/src/core/ext/filters/client_channel/subchannel.h +4 -0
- data/src/core/ext/filters/deadline/deadline_filter.cc +18 -15
- data/src/core/ext/filters/deadline/deadline_filter.h +5 -5
- data/src/core/ext/filters/http/client/http_client_filter.cc +10 -9
- data/src/core/ext/filters/http/server/http_server_filter.h +1 -1
- data/src/core/ext/transport/chttp2/client/insecure/channel_create_posix.cc +1 -1
- data/src/core/ext/transport/chttp2/server/insecure/server_chttp2_posix.cc +3 -2
- data/src/core/ext/transport/chttp2/transport/chttp2_transport.cc +33 -22
- data/src/core/ext/transport/chttp2/transport/hpack_parser.cc +1 -1
- data/src/core/ext/transport/chttp2/transport/internal.h +10 -3
- data/src/core/ext/transport/chttp2/transport/stream_lists.cc +17 -0
- data/src/core/ext/transport/chttp2/transport/writing.cc +21 -16
- data/src/core/ext/transport/inproc/inproc_transport.cc +46 -6
- data/src/core/lib/channel/channel_stack.cc +22 -24
- data/src/core/lib/channel/channel_trace.cc +28 -63
- data/src/core/lib/channel/channel_trace.h +13 -17
- data/src/core/lib/channel/channelz.cc +143 -0
- data/src/core/lib/channel/channelz.h +124 -0
- data/src/core/lib/channel/channelz_registry.cc +7 -24
- data/src/core/lib/channel/channelz_registry.h +12 -8
- data/src/core/lib/channel/connected_channel.cc +8 -1
- data/src/core/{ext/filters/load_reporting/server_load_reporting_filter.h → lib/gpr/alloc.h} +7 -9
- data/src/core/lib/gpr/arena.cc +8 -8
- data/src/core/lib/gpr/string.cc +28 -0
- data/src/core/lib/gpr/string.h +10 -0
- data/src/core/lib/gprpp/abstract.h +5 -2
- data/src/core/lib/gprpp/inlined_vector.h +57 -3
- data/src/core/lib/gprpp/memory.h +2 -2
- data/src/core/lib/gprpp/ref_counted_ptr.h +5 -0
- data/src/core/lib/gprpp/thd_posix.cc +1 -1
- data/src/core/lib/iomgr/call_combiner.h +80 -0
- data/src/core/lib/iomgr/closure.h +3 -2
- data/src/core/lib/iomgr/endpoint_pair_posix.cc +2 -2
- data/src/core/lib/iomgr/error.cc +12 -0
- data/src/core/lib/iomgr/error.h +5 -0
- data/src/core/lib/iomgr/ev_epoll1_linux.cc +36 -9
- data/src/core/lib/iomgr/ev_epollex_linux.cc +172 -46
- data/src/core/lib/iomgr/ev_epollsig_linux.cc +47 -21
- data/src/core/lib/iomgr/ev_poll_posix.cc +10 -4
- data/src/core/lib/iomgr/ev_posix.cc +17 -9
- data/src/core/lib/iomgr/ev_posix.h +20 -4
- data/src/core/lib/iomgr/executor.cc +196 -140
- data/src/core/lib/iomgr/executor.h +47 -14
- data/src/core/lib/iomgr/iomgr.cc +2 -0
- data/src/core/lib/iomgr/iomgr.h +5 -0
- data/src/core/lib/iomgr/is_epollexclusive_available.cc +1 -0
- data/src/core/lib/iomgr/socket_utils.h +9 -0
- data/src/core/lib/iomgr/socket_utils_common_posix.cc +4 -0
- data/src/core/lib/iomgr/socket_utils_uv.cc +4 -0
- data/src/core/lib/iomgr/socket_utils_windows.cc +4 -0
- data/src/core/lib/iomgr/tcp_client_posix.cc +3 -5
- data/src/core/lib/iomgr/tcp_posix.cc +6 -1
- data/src/core/lib/iomgr/tcp_server_posix.cc +3 -3
- data/src/core/lib/iomgr/tcp_server_utils_posix_common.cc +1 -1
- data/src/core/lib/iomgr/timer_manager.cc +0 -1
- data/src/core/lib/iomgr/udp_server.cc +2 -3
- data/src/core/lib/json/json.cc +10 -0
- data/src/core/lib/json/json.h +5 -0
- data/src/core/lib/security/context/security_context.cc +8 -8
- data/src/core/lib/security/context/security_context.h +6 -2
- data/src/core/lib/security/credentials/google_default/google_default_credentials.cc +2 -1
- data/src/core/lib/security/credentials/local/local_credentials.cc +77 -0
- data/src/core/lib/security/credentials/local/local_credentials.h +40 -0
- data/src/core/lib/security/credentials/ssl/ssl_credentials.cc +17 -3
- data/src/core/lib/security/security_connector/local_security_connector.cc +245 -0
- data/src/core/lib/security/security_connector/local_security_connector.h +58 -0
- data/src/core/lib/security/security_connector/security_connector.cc +30 -5
- data/src/core/lib/security/security_connector/security_connector.h +1 -0
- data/src/core/lib/security/transport/client_auth_filter.cc +5 -1
- data/src/core/lib/security/transport/server_auth_filter.cc +4 -5
- data/src/core/lib/surface/call.cc +75 -32
- data/src/core/lib/surface/call.h +2 -0
- data/src/core/lib/surface/channel.cc +32 -13
- data/src/core/lib/surface/channel.h +4 -0
- data/src/core/lib/surface/version.cc +1 -1
- data/src/core/lib/transport/transport.cc +20 -9
- data/src/core/lib/transport/transport.h +12 -10
- data/src/core/lib/transport/transport_op_string.cc +0 -7
- data/src/core/plugin_registry/grpc_plugin_registry.cc +0 -4
- data/src/core/tsi/alts/handshaker/alts_handshaker_service_api_util.h +2 -2
- data/src/core/tsi/alts/handshaker/alts_tsi_handshaker.cc +2 -1
- data/src/core/tsi/alts/handshaker/altscontext.pb.c +0 -1
- data/src/core/tsi/alts/handshaker/altscontext.pb.h +1 -2
- data/src/core/tsi/alts/handshaker/handshaker.pb.c +0 -1
- data/src/core/tsi/alts/handshaker/handshaker.pb.h +1 -2
- data/src/core/tsi/alts/handshaker/transport_security_common.pb.c +0 -1
- data/src/core/tsi/alts/handshaker/transport_security_common.pb.h +1 -1
- data/src/core/tsi/alts/handshaker/transport_security_common_api.h +2 -2
- data/src/core/tsi/alts/zero_copy_frame_protector/alts_grpc_integrity_only_record_protocol.cc +47 -1
- data/src/core/tsi/alts/zero_copy_frame_protector/alts_grpc_integrity_only_record_protocol.h +3 -1
- data/src/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector.cc +12 -11
- data/src/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector.h +7 -2
- data/src/core/tsi/local_transport_security.cc +209 -0
- data/src/core/tsi/local_transport_security.h +51 -0
- data/src/core/tsi/ssl_transport_security.cc +2 -3
- data/src/{core/ext → cpp/ext/filters}/census/grpc_context.cc +0 -0
- data/src/ruby/ext/grpc/rb_channel_credentials.c +3 -3
- data/src/ruby/ext/grpc/rb_grpc_imports.generated.c +18 -18
- data/src/ruby/ext/grpc/rb_grpc_imports.generated.h +29 -29
- data/src/ruby/lib/grpc/generic/active_call.rb +19 -23
- data/src/ruby/lib/grpc/version.rb +1 -1
- data/src/ruby/spec/call_credentials_spec.rb +1 -1
- data/src/ruby/spec/call_spec.rb +1 -1
- data/src/ruby/spec/channel_credentials_spec.rb +1 -1
- data/src/ruby/spec/channel_spec.rb +1 -1
- data/src/ruby/spec/client_auth_spec.rb +1 -12
- data/src/ruby/spec/client_server_spec.rb +1 -1
- data/src/ruby/spec/compression_options_spec.rb +1 -1
- data/src/ruby/spec/error_sanity_spec.rb +1 -1
- data/src/ruby/spec/generic/client_stub_spec.rb +13 -1
- data/src/ruby/spec/generic/rpc_desc_spec.rb +1 -1
- data/src/ruby/spec/generic/rpc_server_pool_spec.rb +1 -1
- data/src/ruby/spec/generic/service_spec.rb +1 -1
- data/src/ruby/spec/google_rpc_status_utils_spec.rb +1 -12
- data/src/ruby/spec/pb/duplicate/codegen_spec.rb +1 -0
- data/src/ruby/spec/pb/health/checker_spec.rb +1 -1
- data/src/ruby/spec/server_credentials_spec.rb +1 -1
- data/src/ruby/spec/server_spec.rb +1 -1
- data/src/ruby/spec/spec_helper.rb +1 -0
- data/src/ruby/spec/support/services.rb +1 -1
- data/src/ruby/spec/time_consts_spec.rb +1 -1
- data/third_party/boringssl/crypto/asn1/tasn_dec.c +40 -19
- data/third_party/boringssl/crypto/bytestring/cbs.c +1 -0
- data/third_party/boringssl/crypto/cipher_extra/e_aesccm.c +47 -15
- data/third_party/boringssl/crypto/ec_extra/ec_asn1.c +9 -10
- data/third_party/boringssl/crypto/ecdh/ecdh.c +4 -3
- data/third_party/boringssl/crypto/fipsmodule/bn/add.c +30 -54
- data/third_party/boringssl/crypto/fipsmodule/bn/bn.c +7 -1
- data/third_party/boringssl/crypto/fipsmodule/bn/cmp.c +8 -8
- data/third_party/boringssl/crypto/fipsmodule/bn/div.c +97 -11
- data/third_party/boringssl/crypto/fipsmodule/bn/gcd.c +274 -218
- data/third_party/boringssl/crypto/fipsmodule/bn/internal.h +111 -34
- data/third_party/boringssl/crypto/fipsmodule/bn/montgomery.c +2 -2
- data/third_party/boringssl/crypto/fipsmodule/bn/montgomery_inv.c +1 -1
- data/third_party/boringssl/crypto/fipsmodule/bn/mul.c +24 -6
- data/third_party/boringssl/crypto/fipsmodule/bn/prime.c +324 -63
- data/third_party/boringssl/crypto/fipsmodule/bn/random.c +74 -21
- data/third_party/boringssl/crypto/fipsmodule/bn/shift.c +128 -86
- data/third_party/boringssl/crypto/fipsmodule/bn/sqrt.c +1 -1
- data/third_party/boringssl/crypto/fipsmodule/ec/ec_key.c +67 -112
- data/third_party/boringssl/crypto/fipsmodule/ec/internal.h +8 -1
- data/third_party/boringssl/crypto/fipsmodule/ec/oct.c +5 -5
- data/third_party/boringssl/crypto/fipsmodule/ec/p224-64.c +9 -17
- data/third_party/boringssl/crypto/fipsmodule/ec/p256-x86_64-table.h +5378 -5418
- data/third_party/boringssl/crypto/fipsmodule/ec/simple.c +32 -32
- data/third_party/boringssl/crypto/fipsmodule/ecdsa/ecdsa.c +5 -11
- data/third_party/boringssl/crypto/fipsmodule/rsa/blinding.c +16 -40
- data/third_party/boringssl/crypto/fipsmodule/rsa/internal.h +1 -6
- data/third_party/boringssl/crypto/fipsmodule/rsa/rsa.c +41 -29
- data/third_party/boringssl/crypto/fipsmodule/rsa/rsa_impl.c +63 -49
- data/third_party/boringssl/crypto/x509/vpm_int.h +1 -0
- data/third_party/boringssl/crypto/x509/x509_vfy.c +4 -0
- data/third_party/boringssl/crypto/x509/x509_vpm.c +44 -22
- data/third_party/boringssl/include/openssl/aead.h +8 -2
- data/third_party/boringssl/include/openssl/asn1.h +1 -0
- data/third_party/boringssl/include/openssl/base.h +4 -0
- data/third_party/boringssl/include/openssl/bn.h +13 -3
- data/third_party/boringssl/include/openssl/bytestring.h +4 -4
- data/third_party/boringssl/include/openssl/ec.h +10 -4
- data/third_party/boringssl/include/openssl/ec_key.h +0 -3
- data/third_party/boringssl/include/openssl/rsa.h +1 -0
- data/third_party/boringssl/include/openssl/ssl.h +8 -3
- data/third_party/boringssl/include/openssl/ssl3.h +0 -1
- data/third_party/boringssl/include/openssl/x509.h +1 -0
- data/third_party/boringssl/include/openssl/x509v3.h +1 -0
- data/third_party/boringssl/ssl/handshake_client.cc +36 -64
- data/third_party/boringssl/ssl/ssl_cipher.cc +4 -0
- data/third_party/boringssl/ssl/ssl_lib.cc +1 -1
- metadata +45 -38
- data/src/core/ext/filters/load_reporting/server_load_reporting_filter.cc +0 -222
- data/src/core/ext/filters/load_reporting/server_load_reporting_plugin.cc +0 -71
- data/src/core/ext/filters/load_reporting/server_load_reporting_plugin.h +0 -61
- data/src/ruby/spec/pb/package_with_underscore/checker_spec.rb +0 -51
- data/src/ruby/spec/pb/package_with_underscore/data.proto +0 -23
- 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_ =
|
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
|
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*
|
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*
|
42
|
-
|
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
|
-
/*
|
45
|
-
|
46
|
-
|
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
|
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
|
-
|
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
|
-
|
51
|
-
|
52
|
-
|
53
|
-
|
54
|
-
|
55
|
-
|
56
|
-
|
57
|
-
|
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
|
-
|
101
|
-
|
102
|
-
|
103
|
-
|
104
|
-
|
105
|
-
|
106
|
-
|
107
|
-
|
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
|
-
|
127
|
-
|
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
|
151
|
-
|
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
|
-
|
175
|
-
|
176
|
-
|
177
|
-
|
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
|
-
|
200
|
-
|
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
|
-
|
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
|
-
|
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
|
-
|
248
|
-
|
249
|
-
|
250
|
-
|
251
|
-
|
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
|
-
|
266
|
-
|
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
|
-
|
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
|
-
|
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) */
|